diff --git a/internal/config/load.go b/internal/config/load.go index 65846925b11547d713b74c7dcb9c032e64097ab0..a2d2155048a8abf634958293056da8a4713e1400 100644 --- a/internal/config/load.go +++ b/internal/config/load.go @@ -168,8 +168,8 @@ func (c *Config) configureProviders(env env.Env, resolver VariableResolver, know } continue } - prepared.ExtraParams["project"] = env.Get("GOOGLE_CLOUD_PROJECT") - prepared.ExtraParams["location"] = env.Get("GOOGLE_CLOUD_LOCATION") + prepared.ExtraParams["project"] = env.Get("VERTEXAI_PROJECT") + prepared.ExtraParams["location"] = env.Get("VERTEXAI_LOCATION") case catwalk.InferenceProviderAzure: endpoint, err := resolver.ResolveValue(p.APIEndpoint) if err != nil || endpoint == "" { @@ -470,10 +470,9 @@ func loadFromReaders(readers []io.Reader) (*Config, error) { } func hasVertexCredentials(env env.Env) bool { - useVertex := env.Get("GOOGLE_GENAI_USE_VERTEXAI") == "true" - hasProject := env.Get("GOOGLE_CLOUD_PROJECT") != "" - hasLocation := env.Get("GOOGLE_CLOUD_LOCATION") != "" - return useVertex && hasProject && hasLocation + hasProject := env.Get("VERTEXAI_PROJECT") != "" + hasLocation := env.Get("VERTEXAI_LOCATION") != "" + return hasProject && hasLocation } func hasAWSCredentials(env env.Env) bool { diff --git a/internal/llm/provider/vertexai.go b/internal/llm/provider/vertexai.go index 1baa08927dcfacd40e3dc3a9909311b7be452826..49a28084ab2096522399c43ea2824c2b12063244 100644 --- a/internal/llm/provider/vertexai.go +++ b/internal/llm/provider/vertexai.go @@ -10,8 +10,8 @@ import ( type VertexAIClient ProviderClient func newVertexAIClient(opts providerClientOptions) VertexAIClient { - project := opts.extraHeaders["project"] - location := opts.extraHeaders["location"] + project := opts.extraParams["project"] + location := opts.extraParams["location"] client, err := genai.NewClient(context.Background(), &genai.ClientConfig{ Project: project, Location: location,