Update app.py

app.py

@@ -24,15 +24,15 @@ tools = [
 ]
 chat_with_tools = chat.bind_tools(tools)
 
+# Generate the AgentState and Agent graph
+class AgentState(TypedDict):
+    messages: Annotated[list[AnyMessage], add_messages]
+    
 def assistant(state: AgentState):
     return {
         "messages": [chat_with_tools.invoke(state["messages"])],
     }   
 
-# Generate the AgentState and Agent graph
-class AgentState(TypedDict):
-    messages: Annotated[list[AnyMessage], add_messages]
-    
 class BasicAgent:
     
     def __init__(self):