update

examples/api_eval/eval_gemini_google.py

@@ -3,8 +3,12 @@
 import argparse
 import json
 import os
+import sys
 import time
 
+pwd = os.path.abspath(os.path.dirname(__file__))
+sys.path.append(os.path.join(pwd, "../../"))
+
 from google import genai
 from google.genai import types
 
@@ -122,6 +126,7 @@ Your output is:
                 config=generate_content_config,
             )
             time_cost = time.time() - time_begin
+            print(time_cost)
             try:
                 prediction = response.candidates[0].content.parts[0].text
             except TypeError:
@@ -147,7 +152,7 @@ Your output is:
             row_ = json.dumps(row_, ensure_ascii=False)
             fout.write(f"{row_}\n")
 
-            print(f"score: {score}")
+            # print(f"score: {score}")
 
     return
 

examples/api_eval/eval_openai.py

@@ -3,8 +3,12 @@
 import argparse
 import json
 import os
+import sys
 import time
 
+pwd = os.path.abspath(os.path.dirname(__file__))
+sys.path.append(os.path.join(pwd, "../../"))
+
 import openai
 from openai import AzureOpenAI
 
@@ -118,6 +122,7 @@ Your output is:
                     # }
                 )
                 time_cost = time.time() - time_begin
+                print(time_cost)
             except openai.BadRequestError as e:
                 print(f"request failed, error type: {type(e)}, error text: {str(e)}")
                 continue
@@ -145,7 +150,7 @@ Your output is:
             row_ = json.dumps(row_, ensure_ascii=False)
             fout.write(f"{row_}\n")
 
-            print(f"score: {score}")
+            # print(f"score: {score}")
 
     return