Skip to content

Commit 99f833d

Browse files
authored
Merge pull request #2162 from gtardif/evals_dont_provide_keys_with_gateway
Evals: don't provide all API keys when using models gateway
2 parents ce11fc5 + bba36f7 commit 99f833d

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

pkg/evaluation/eval.go

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -388,13 +388,6 @@ func (r *Runner) runDockerAgentInContainer(ctx context.Context, imageID string,
388388

389389
var env []string
390390

391-
for _, name := range []string{"OPENAI_API_KEY", "ANTHROPIC_API_KEY", "GOOGLE_API_KEY", "MISTRAL_API_KEY", "XAI_API_KEY", "NEBIUS_API_KEY"} {
392-
if val, ok := r.runConfig.EnvProvider().Get(ctx, name); ok && val != "" {
393-
args = append(args, "-e", name)
394-
env = append(env, name+"="+val)
395-
}
396-
}
397-
398391
if r.runConfig.ModelsGateway != "" {
399392
args = append(args, "-e", "DOCKER_AGENT_MODELS_GATEWAY")
400393
env = append(env, "DOCKER_AGENT_MODELS_GATEWAY="+r.runConfig.ModelsGateway)
@@ -403,6 +396,13 @@ func (r *Runner) runDockerAgentInContainer(ctx context.Context, imageID string,
403396
args = append(args, "-e", environment.DockerDesktopTokenEnv)
404397
env = append(env, environment.DockerDesktopTokenEnv+"="+token)
405398
}
399+
} else {
400+
for _, name := range []string{"OPENAI_API_KEY", "ANTHROPIC_API_KEY", "GOOGLE_API_KEY", "MISTRAL_API_KEY", "XAI_API_KEY", "NEBIUS_API_KEY"} {
401+
if val, ok := r.runConfig.EnvProvider().Get(ctx, name); ok && val != "" {
402+
args = append(args, "-e", name)
403+
env = append(env, name+"="+val)
404+
}
405+
}
406406
}
407407

408408
// Pass additional environment variables specified via -e flag

0 commit comments

Comments
 (0)