git.iko.soy/crush
on main help sign in
chore: fix gemini validation
90961a6Kujtim Hoxha
internal/config/config.go+7
compare with main
unifiedsplit
@@ -5,6 +5,7 @@ import (
5 "fmt"
5 "fmt"
6 "log/slog"
6 "log/slog"
7 "net/http"
7 "net/http"
8+ "net/url"
8 "os"
9 "os"
9 "slices"
10 "slices"
10 "strings"
11 "strings"
@@ -471,6 +472,12 @@ func (c *ProviderConfig) TestConnection(resolver VariableResolver) error {
471 testURL = baseURL + "/models"
472 testURL = baseURL + "/models"
472 headers["x-api-key"] = apiKey
473 headers["x-api-key"] = apiKey
473 headers["anthropic-version"] = "2023-06-01"
474 headers["anthropic-version"] = "2023-06-01"
475+ case catwalk.TypeGemini:
476+ baseURL, _ := resolver.ResolveValue(c.BaseURL)
477+ if baseURL == "" {
478+ baseURL = "https://generativelanguage.googleapis.com"
479+ }
480+ testURL = baseURL + "/v1beta/models?key=" + url.QueryEscape(apiKey)
474 }
481 }
475 ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
482 ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
476 defer cancel()
483 defer cancel()