Update util.py

util.py

@@ -2,10 +2,9 @@ import pandas as pd
 
 from openai import OpenAI
 
-FILE_PATH = "data/gaia_validation_20.jsonl"
-
-def get_questions():
-    df = pd.read_json(FILE_PATH, lines=True)
+def get_questions(file_path):
+    df = pd.read_json(file_path, lines=True)
+    
     result=[]
     
     for index, row in df.iterrows():
@@ -13,7 +12,7 @@ def get_questions():
 
     return result
 
-def get_final_answer(model, question, initial_answer):
+def get_final_answer(model, question, answer):
     prompt_template = """
         You are an expert question answering assistant. Given a question and an initial answer, your task is to provide the final answer.
         Your final answer must be a number and/or string OR as few words as possible OR a comma-separated list of numbers and/or strings.
@@ -24,7 +23,7 @@ def get_final_answer(model, question, initial_answer):
 
         **Question:** """ + question + """
         
-        **Initial answer:** """ + initial_answer + """
+        **Initial answer:** """ + answer + """
         
         **Example 1:** How many 'r's are in strawberry? 3
         **Example 2:** What is the opposite of black? White
@@ -37,6 +36,7 @@ def get_final_answer(model, question, initial_answer):
         """
 
     client = OpenAI()
+    
     completion = client.chat.completions.create(
         messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}],
         model=model