Update agent.py

agent.py

@@ -167,8 +167,14 @@ def final_answer_formatting(answer, question):
     )
 
     prompt = f"""
-    You are an AI assistant specialized in the GAIA benchmark. For the question provided, generate the answer in the exact format requested by the question. Do not include any other text or creative additions.
+    You are an AI assistant specialized in the GAIA benchmark. For the question provided, generate the answer in the exact format requested by the question. Do not include any other text, formatting or creative additions.
 
+    For example:
+    Question: "what is the capital of France ?"
+    Answer: "The capital of France is Paris. It is the largest city in France and serves as the country's political, cultural, and economic center. Paris has been the capital since its liberation in 1944, although it has held this status intermittently throughout French history"
+    Return: "Paris"
+
+    Your turn:
     Question: {question}
     Answer: {answer}
     """
@@ -217,7 +223,7 @@ audio_agent = CodeAgent(
 manager_agent = CodeAgent(
     name="manager_agent",
     model=LiteLLMModel(
-        model_id="gemini/gemini-2.5-flash-preview-04-17",
+        model_id="gemini/gemini-2.0-flash",
         api_key=os.getenv("GOOGLE_API_KEY"),
     ),
     tools=[GetFileTool(), LoadXlsxFileTool(), LoadTextFileTool(), ImageAnalysisTool(), WebSearchPerplexityTool()],