add exception handling

openai · Sep 30, 2024 · 2f38c39 · 2f38c39
1 parent 234bcde
commit 2f38c39
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/evals/cli/oaieval.py b/evals/cli/oaieval.py
@@ -224,7 +224,10 @@ def to_number(x: str) -> Union[int, float, str]:
         **extra_eval_params,
     )
     result = eval.run(recorder)
-    add_token_usage_to_result(result, recorder)
+    try:
+        add_token_usage_to_result(result, recorder)
+    except Exception as e:
+        logger.error(f"Failed to add token usage to result: {e}")
     recorder.record_final_report(result)
 
     if not (args.dry_run or args.local_run):