Update app.py

app.py

@@ -3,9 +3,9 @@ import os
 import inspect
 import gradio as gr
 import requests
-import pandas as pd
-from langchain_core.messages import HumanMessage
-from agent import build_graph
+#import pandas as pd
+#from langchain_core.messages import HumanMessage
+from gaia_agent import HumanMessage, SystemMessage, build_graph
 
 
 
@@ -21,14 +21,35 @@ class BasicAgent:
     """A langgraph agent."""
     def __init__(self):
         print("BasicAgent initialized.")
-        self.graph = build_graph()
+        self.graph = build_graph
+        self.systemMsg = SystemMessage(
+            content="""
+            "You are a helpful assistant tasked with answering questions using a set of tools.\n"
+                "Your final answer must strictly follow this format:\n"
+                "FINAL ANSWER: [ANSWER]\n"
+                "Only write the answer in that exact format. Do not explain anything. Do not include any other text. \n"
+                "If you are provided with a similar question and its final answer, and the current question is **exactly the same**, then simply return the same final answer without using any tools. \n"
+                "Only use tools if the current question is different from the similar one. \n"
+                "Examples: \n"
+                "- FINAL ANSWER: FunkMonk \n"
+                "- FINAL ANSWER: Paris \n"
+                "- FINAL ANSWER: 128 \n"
+                " \n"
+                "Once you output 'FINAL ANSWER', stop reasoning and do not call any tool.\n"
+                "If you do not follow this format exactly, your response will be considered incorrect. \n"
+                "Use the most up-to-date information available to you we are in December 2025.
+            """
+        )
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        messages = [HumanMessage(content=question)]
-        result = self.graph.invoke({"messages": messages})
-        answer = result['messages'][-1].content
-        return answer  # kein [14:] mehr nötig!
+        messages = [self.systemMsg, HumanMessage(content=question)]
+        response = self.graph.invoke({"messages": messages})
+        ai_msg = response["messages"][-1].content
+        if "FINAL ANSWER:" in ai_msg:
+           return ai_msg.split("FINAL ANSWER:")[-1].strip()
+        else:
+            return ai_msg.strip()