Update my_agent.py

my_agent.py

@@ -15,9 +15,8 @@ from my_tools import wiki_search, arxiv_search, web_search, visit_webpage, trans
 
 hf_token = os.getenv("HF_TOKEN")
 
-class AgentResponseState(AgentState):
-    response: str
-    messages: list[HumanMessage | AIMessage]
+class GraphMessagesState(MessagesState):
+    question: str
 
 
 # --- Basic Agent Definition ---