Update util.py

util.py

@@ -2,12 +2,10 @@ import pandas as pd
 
 from openai import OpenAI
 
-FILE_NAME = "data/gaia_validation_20.jsonl"
-
-FINAL_ANSWER_MODEL = "gpt-4.5-preview"
+FILE_PATH = "data/gaia_validation_20.jsonl"
 
 def get_questions():
-    df = pd.read_json(FILE_NAME, lines=True)
+    df = pd.read_json(FILE_PATH, lines=True)
     result=[]
     
     for index, row in df.iterrows():
@@ -15,7 +13,7 @@ def get_questions():
 
     return result
 
-def get_final_answer(question, initial_answer):
+def get_final_answer(model, question, initial_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.
@@ -41,7 +39,7 @@ def get_final_answer(question, initial_answer):
     client = OpenAI()
     completion = client.chat.completions.create(
         messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}],
-        model=FINAL_ANSWER_MODEL
+        model=model
     )
 
     return completion.choices[0].message.content