1db7629 · crush · git.iko.soy

on main help sign in

fix(ui): improve model changed notification (#3015)

1db7629Andrey Nering5 months ago

internal/ui/model/ui.go+9−1

@@ -40,6 +40,7 @@ import (

40 "github.com/charmbracelet/crush/internal/pubsub"

41 "github.com/charmbracelet/crush/internal/session"

42 "github.com/charmbracelet/crush/internal/skills"

43+ "github.com/charmbracelet/crush/internal/stringext"

43 "github.com/charmbracelet/crush/internal/ui/anim"

44 "github.com/charmbracelet/crush/internal/ui/anim"

44 "github.com/charmbracelet/crush/internal/ui/attachments"

45 "github.com/charmbracelet/crush/internal/ui/attachments"

45 "github.com/charmbracelet/crush/internal/ui/chat"

46 "github.com/charmbracelet/crush/internal/ui/chat"

@@ -1701,7 +1702,14 @@ func (m *UI) handleSelectModel(msg dialog.ActionSelectModel) tea.Cmd {

1701 return util.ReportError(err)

1702 return util.ReportError(err)

1702 }

1703 }

1703

1704

1705+ var (

1706+ modelType = stringext.Capitalize(string(msg.ModelType))

1707+ modelName = msg.Model.Model

1708+ )

1709+ if catwalkModel := cfg.GetModel(msg.Model.Provider, msg.Model.Model); catwalkModel != nil && catwalkModel.Name != "" {

1710+ modelName = catwalkModel.Name

1711+ }

1704− modelMsg := fmt.Sprintf("%s model changed to %s", msg.ModelType, msg.Model.Model)

1712+ modelMsg := fmt.Sprintf("%s model changed to %s", modelType, modelName)

1705

1713

1706 return util.NewInfoMsg(modelMsg)

1714 return util.NewInfoMsg(modelMsg)

1707 })

1715 })