final_answer

agent.py

@@ -19,7 +19,7 @@ from tools import (
 
 # ─────────────────────────── Configuration ───────────────────────────────
 SYSTEM_PROMPT = """
-You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+You are a general AI assistant. I will ask you a question. Don't report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
 """
 MAX_TOOL_CALLS = 5
 
@@ -36,18 +36,18 @@ MAX_TOOL_CALLS = 5
 # ─────────────────────────── Graph wiring ───────────────────────────────
 
 def build_graph():
-    
-
-    llm = ChatOpenAI(model_name="gpt-4.1-mini", temperature=0.3)
+    """Build and return a create_react_agent."""
+    llm = ChatOpenAI(model_name="gpt-4o-mini", temperature=0.3)
 
     llm_tools = [
         wikipedia_search_tool,
         arxiv_search_tool,
         audio_transcriber_tool,
         excel_tool,
-        analyze_code_tool,
+        analyze_code_tool
     ]
-    # llm = llm.bind_tools(llm_tools)
-    agent = create_react_agent(model=llm, tools=llm_tools, prompt=SYSTEM_PROMPT)
+    
+    # Create the react agent with proper system prompt
+    agent = create_react_agent(model=llm, tools=llm_tools, messages_modifier=SYSTEM_PROMPT)
 
     return agent 

app.py

@@ -26,14 +26,32 @@ class BasicAgent:
         """Run the agent and return whatever FINAL_ANSWER the graph produces."""
         print(f"Agent received question: {question}")
         
-        # The user_question argument for AgentState is the question.
+        # Initialize the state properly with all required fields
+        init_state = {
+            "messages": [
+                SystemMessage(content=SYSTEM_PROMPT),
+                HumanMessage(content=question)
+            ]
+        }
         
-
         # IMPORTANT: invoke() returns a **new** state instance (or an AddableValuesDict),
         # not the object we pass in.  Use the returned value to fetch final_answer.
-        out_state = self.graph.invoke({"messages": [HumanMessage(content=question)]})
+        out_state = self.graph.invoke(init_state)
         print("out_state: ", out_state)
-        return out_state
+        
+        # Extract the final answer from the last message
+        if out_state and "messages" in out_state:
+            last_message = out_state["messages"][-1]
+            if hasattr(last_message, 'content'):
+                content = last_message.content
+                # Look for FINAL ANSWER: pattern
+                if "FINAL ANSWER:" in content:
+                    final_answer = content.split("FINAL ANSWER:")[-1].strip()
+                    return final_answer
+                else:
+                    return content
+        
+        return "No answer generated."
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):