Update agent.py

agent.py

@@ -35,26 +35,26 @@ def run_agent(level, question, file_name, ground_truth):
 get_final_answer(question, answer):
     client = OpenAI()
 
-    prompt = """
-        You are given a question and a preliminary answer:
-        Question: {question}
-        Preliminary answer: {answer}
-        You must give the precise answer requested. Do not include explanations, steps, reasoning, or additional text. Be direct and specific.
+    prompt_template = """
+        You are given a question and an answer:
+        Question: """ + question + """
+        Answer: """ + answer + """
+        You must ONLY give the precise answer requested. Do not include explanations, steps, reasoning, or additional text. Be direct and specific.
         For example, if asked "What is the capital of France?", respond simply with "Paris".
     """
 
     completion = client.chat.completions.create(
         max_tokens = 1000,
         messages = [{"role": "user",
-                     "content": [{"type": "text", "text": prompt}],
+                     "content": [{"type": "text", "text": prompt_template}],
         model = MODEL_ID,
         temperature = 0.0
     )
 
-    content = completion.choices[0].message.content
+    final_answer = completion.choices[0].message.content
 
     print("###")
-    print(content)
+    print(prompt_template)
     print("###")
 
-    return content
+    return final_answer