fix benchmark for single prediction

valory-xyz · Jun 27, 2024 · c8acbe9 · c8acbe9
1 parent c522f9f
commit c8acbe9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/kongzii/customs/ofv_market_resolver/benchmark.py b/packages/kongzii/customs/ofv_market_resolver/benchmark.py
@@ -140,8 +140,10 @@ def single(
     """
     ofv_run(
         question,
-        openai_api_key=SecretStr(openai_api_key),
-        serper_api_key=SecretStr(serper_api_key),
+        api_keys={
+            "openai": openai_api_key,
+            "serperapi": serper_api_key,
+        },
     )