diff --git a/ci/benchmarks/.benchmarks.yml b/ci/benchmarks/.benchmarks.yml index 0ac455a3..49c2faa2 100644 --- a/ci/benchmarks/.benchmarks.yml +++ b/ci/benchmarks/.benchmarks.yml @@ -12,7 +12,7 @@ - pip install litellm langchain transformers torch openai tqdm - | for var in $(compgen -e); do - if [[ $var == CTF_* && -n ${!var} ]]; then + if [[ ($var == CTF_* || $var == *_API_KEY || $var == *_API_BASE) && -n ${!var} ]]; then export $var="${!var}" fi done @@ -54,4 +54,5 @@ benchmarks-test-cybermetric-openai: <<: *run_benchmarks variables: MODEL_NAME: "openai/gpt-4o-mini" - BENCHMARK_FILE: "benchmarks/cybermetric/CyberMetric-2-v1.json" \ No newline at end of file + BENCHMARK_FILE: "benchmarks/cybermetric/CyberMetric-2-v1.json" + # GitLab will use the OPENAI_API_KEY defined in CI/CD variables \ No newline at end of file