Update agent.py

agent.py

@@ -5,6 +5,7 @@ from langgraph.prebuilt import tools_condition
 from langgraph.prebuilt import ToolNode
 from langchain_core.messages import SystemMessage, HumanMessage
 from tools import *
+from typing import TypedDict
 
 tools = [add, substract, multiply, divide, web_search]