From d4520eb831e904780c9476cc3aaee0201d90a352 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?V=C3=ADctor=20Mayoral=20Vilches?= <v.mayoralv@gmail.com>
Date: Thu, 8 May 2025 19:23:22 +0200
Subject: [PATCH] Retry openai 4
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
---
 ci/benchmarks/.benchmarks.yml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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