Update app.py

app.py

@@ -53,6 +53,10 @@ logging.getLogger("ultralytics").setLevel(logging.ERROR)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+# Agent State Definition
+class AgentState(TypedDict):
+    messages: Annotated[ListType, add_messages]
+
 # System prompt for the agent
 SYSTEM_PROMPT = """You are a precision research assistant for the GAIA benchmark. Your mission is EXTREME ACCURACY.
 
@@ -268,9 +272,6 @@ class GAIAAgent:
 
     def _create_agent_runner(self):
         """Create the LangGraph agent runner"""
-        class AgentState(TypedDict):
-            messages: Annotated[ListType, add_messages]
-        
         model_with_tools = self.llm.bind_tools(self.tools)
         
         def agent_node(state):