WeByT3 commited on
Commit
e39e70f
·
verified ·
1 Parent(s): 55f4a0c

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -9,11 +9,11 @@ from tools import *
9
  tools = [add, substract, multiply, divide, web_search]
10
 
11
  class AgentState(TypedDict):
12
- messages: Annotated[list[AnyMessage], add_messages]
13
 
14
  def assistant(state: AgentState):
15
  return {
16
- "messages": [chat_with_tools.invoke(state["messages"])],
17
  }
18
 
19
  def build_agent():
 
9
  tools = [add, substract, multiply, divide, web_search]
10
 
11
  class AgentState(TypedDict):
12
+ question: str
13
 
14
  def assistant(state: AgentState):
15
  return {
16
+ "answer": [chat_with_tools.invoke(state["question"])],
17
  }
18
 
19
  def build_agent():