Update util.py

util.py

@@ -1,7 +1,7 @@
 import os
 import pandas as pd
-from google import genai
-#from openai import OpenAI
+#from google import genai
+from openai import OpenAI
 
 def get_questions(file_path):
     df = pd.read_json(file_path, lines=True)
@@ -62,20 +62,20 @@ def get_final_answer(model, question, answer):
 
         """
 
-    #client = OpenAI()
+    client = OpenAI()
     
-    #completion = client.chat.completions.create(
-    #    model=model,
-    #    messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}]
-    #)
+    completion = client.chat.completions.create(
+        model=model,
+        messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}]
+    )
 
-    client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+    #client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
-    response = client.models.generate_content(
-        model=model, 
-        contents=[prompt_template]
-    )
+    #response = client.models.generate_content(
+    #    model=model, 
+    #    contents=[prompt_template]
+    #)
     
-    #return completion.choices[0].message.content
+    return completion.choices[0].message.content
     
-    return response.text
+    #return response.text