Update agent.py

agent.py

@@ -5,7 +5,7 @@ from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, Too
 from langgraph.graph.message import add_messages
 from langgraph.graph import MessagesState
 from langgraph.graph import StateGraph, START, END
-from typing import Literal
+from typing import TypedDict, Annotated, Literal
 
 from langchain_community.tools import BraveSearch  # web search
 from langchain_experimental.tools.python.tool import PythonAstREPLTool  # for logic/math problems