Hugging Face's logo

Spaces:
ORromu
/
Final_Assignment_Template
Sleeping

App Files Community
ORromu commited on
Commit
556ab31
·
verified ·
1 Parent(s): 766670a

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -43,7 +43,7 @@ def simple_graph():
43
  return {"messages": [sys_msg] + [chat_with_tools.invoke(state["messages"])]}
44
 
45
  # Build graph / nodes
46
- builder = StateGraph(State)
47
  builder.add_node("assistant", chat) # Assistant
48
  builder.add_node("tools", ToolNode(tools)) # Tools
49
 
 
43
  return {"messages": [sys_msg] + [chat_with_tools.invoke(state["messages"])]}
44
 
45
  # Build graph / nodes
46
+ builder = StateGraph(MessagesState)
47
  builder.add_node("assistant", chat) # Assistant
48
  builder.add_node("tools", ToolNode(tools)) # Tools
49