Update agent.py

agent.py

@@ -1,4 +1,4 @@
-import os
+from openai import OpenAI
 
 from smolagents import (
     CodeAgent, 
@@ -33,4 +33,28 @@ def run_agent(level, question, file_name, ground_truth):
     return answer, str(answer == ground_truth)
 
 get_final_answer(question, answer):
-    return "Final 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.
+        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}],
+        model = MODEL_ID,
+        temperature = 0.0
+    )
+
+    content = completion.choices[0].message.content
+
+    print("###")
+    print(content)
+    print("###")
+
+    return content