philincloud commited on
Commit
17d17f1
·
verified ·
1 Parent(s): 5cce4fd

Update langgraph_agent.py

Browse files
Files changed (1) hide show
  1. langgraph_agent.py +2 -1
langgraph_agent.py CHANGED
@@ -107,7 +107,8 @@ def build_graph(provider: str = "gemini"):
107
  llm_with_tools = llm.bind_tools(tools)
108
 
109
  def assistant(state: MessagesState):
110
- return {"messages": [llm_with_tools.invoke(state["messages"])]}
 
111
 
112
  builder = StateGraph(MessagesState)
113
  builder.add_node("assistant", assistant)
 
107
  llm_with_tools = llm.bind_tools(tools)
108
 
109
  def assistant(state: MessagesState):
110
+ messages_to_send = [sys_msg] + state["messages"]
111
+ return {"messages": [llm_with_tools.invoke(messages_to_send)]}
112
 
113
  builder = StateGraph(MessagesState)
114
  builder.add_node("assistant", assistant)