Update agent.py

agent.py

@@ -875,6 +875,15 @@ def call_llm(state):
 
 
 from langchain.schema import AIMessage
+from typing import TypedDict, List, Optional
+from langchain_core.messages import BaseMessage
+
+class AgentState(TypedDict):
+    messages: List[BaseMessage]   # Chat history
+    input: str                    # Original input
+    intent: str                   # Derived or predicted intent
+    result: Optional[str]         # Optional result
+
 
 def tool_dispatcher(state: AgentState) -> AgentState:
     last_msg = state["messages"][-1]