diff --git a/internal/config/config.go b/internal/config/config.go index 0f9fc99b..1307d3b2 100644 --- a/internal/config/config.go +++ b/internal/config/config.go @@ -5,6 +5,7 @@ import ( "fmt" "log/slog" "net/http" + "net/url" "os" "slices" "strings" @@ -471,6 +472,12 @@ func (c *ProviderConfig) TestConnection(resolver VariableResolver) error { testURL = baseURL + "/models" headers["x-api-key"] = apiKey headers["anthropic-version"] = "2023-06-01" + case catwalk.TypeGemini: + baseURL, _ := resolver.ResolveValue(c.BaseURL) + if baseURL == "" { + baseURL = "https://generativelanguage.googleapis.com" + } + testURL = baseURL + "/v1beta/models?key=" + url.QueryEscape(apiKey) } ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel()