chore: small fix

Kujtim Hoxha created 5 months ago

Change summary

internal/fur/provider/provider.go              | 2 ++
internal/tui/components/chat/splash/splash.go  | 1 +
internal/tui/components/dialogs/models/list.go | 7 +++++--
3 files changed, 8 insertions(+), 2 deletions(-)

Detailed changes

internal/fur/provider/provider.go

@@ -27,6 +27,7 @@ const (
 	InferenceProviderBedrock    InferenceProvider = "bedrock"
 	InferenceProviderVertexAI   InferenceProvider = "vertexai"
 	InferenceProviderXAI        InferenceProvider = "xai"
+	InferenceProviderGROQ       InferenceProvider = "groq"
 	InferenceProviderOpenRouter InferenceProvider = "openrouter"
 )
 
@@ -68,6 +69,7 @@ func KnownProviders() []InferenceProvider {
 		InferenceProviderBedrock,
 		InferenceProviderVertexAI,
 		InferenceProviderXAI,
+		InferenceProviderGROQ,
 		InferenceProviderOpenRouter,
 	}
 }

internal/tui/components/chat/splash/splash.go

@@ -105,6 +105,7 @@ func (s *splashCmp) SetOnboarding(onboarding bool) {
 			"openai",
 			"gemini",
 			"xai",
+			"groq",
 			"openrouter",
 		}
 		for _, p := range providers {

internal/tui/components/dialogs/models/list.go

@@ -98,14 +98,17 @@ func (m *ModelListComponent) SetModelType(modelType int) tea.Cmd {
 
 	// First, add any configured providers that are not in the known providers list
 	// These should appear at the top of the list
-	knownProviders := provider.KnownProviders()
+	knownProviders, err := config.Providers()
+	if err != nil {
+		return util.ReportError(err)
+	}
 	for providerID, providerConfig := range cfg.Providers {
 		if providerConfig.Disable {
 			continue
 		}
 
 		// Check if this provider is not in the known providers list
-		if !slices.Contains(knownProviders, provider.InferenceProvider(providerID)) {
+		if !slices.ContainsFunc(knownProviders, func(p provider.Provider) bool { return p.ID == provider.InferenceProvider(providerID) }) {
 			// Convert config provider to provider.Provider format
 			configProvider := provider.Provider{
 				Name:   providerConfig.Name,