Prompt

app.py

@@ -31,25 +31,29 @@ class BasicAgent:
         python_tool = PythonInterpreterTool()
         visit_tool = VisitWebpageTool()
 
+        
 
         self.agent = CodeAgent(
             model=self.model,
             tools=[search_tool, wiki_tool, python_tool, visit_tool],
         )
-        self.system_prompt = prompts["system"]
+
+        self.agent.PromptTemplates["final_answer"]["post_messages"] = (
+            code_agent.prompt_templates["final_answer"]["post_messages"] + prompts["system"]
+        )
+      
         print("BasicAgent initialized.")
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        messages = [
-            {"role": "system", "content": self.system_prompt},
-            {"role": "user", "content": question},
-        ]
-        answer = self.agent.run(messages)
-        # Ensure answer is always a string, never a list
-        if isinstance(answer, list):
-            print(f"Warning: Agent returned a list. Converting to string. Value: {answer}")
-            answer = " ".join(str(a) for a in answer)
+
+        answer = self.agent.run(question)
+
+        if isinstance(answer, str) and "Final Answer" in answer:
+            # Extract everything after "Final Answer"
+            idx = answer.find("Final Answer")
+            # Remove "Final Answer" and any colon or dash after it, then strip whitespace
+            answer = answer[idx + len("Final Answer"):].lstrip(": -").strip()
         return answer