diff --git a/internal/tui/components/dialogs/models/models.go b/internal/tui/components/dialogs/models/models.go index e09b040a52ebf911ceefc455b0892c7c9ceba754..d863e1998ad91164822d11253c960d50b8d8199a 100644 --- a/internal/tui/components/dialogs/models/models.go +++ b/internal/tui/components/dialogs/models/models.go @@ -2,6 +2,7 @@ package models import ( "fmt" + "slices" "time" "github.com/charmbracelet/bubbles/v2/help" @@ -95,6 +96,24 @@ func NewModelDialogCmp() ModelDialog { } func (m *modelDialogCmp) Init() tea.Cmd { + providers, err := config.Providers() + if err == nil { + filteredProviders := []catwalk.Provider{} + simpleProviders := []string{ + "anthropic", + "openai", + "gemini", + "xai", + "groq", + "openrouter", + } + for _, p := range providers { + if slices.Contains(simpleProviders, string(p.ID)) { + filteredProviders = append(filteredProviders, p) + } + } + m.modelList.SetProviders(filteredProviders) + } return tea.Batch(m.modelList.Init(), m.apiKeyInput.Init()) }