philincloud commited on
Commit
666975c
·
verified ·
1 Parent(s): f33e43c

Update langgraph_agent.py

Browse files
Files changed (1) hide show
  1. langgraph_agent.py +4 -4
langgraph_agent.py CHANGED
@@ -191,10 +191,10 @@ def build_graph(provider: str = "gemini"):
191
  llm_with_tools = llm.bind_tools(tools)
192
 
193
  def assistant(state: MessagesState):
194
- messages_to_send = [sys_msg] + state["messages"]
195
- llm_response = llm_with_tools.invoke(messages_to_send)
196
- print(f"LLM Raw Response: {llm_response}")
197
- return {"messages": [llm_response]}
198
 
199
  builder = StateGraph(MessagesState)
200
  builder.add_node("assistant", assistant)
 
191
  llm_with_tools = llm.bind_tools(tools)
192
 
193
  def assistant(state: MessagesState):
194
+ messages_to_send = [sys_msg] + state["messages"] # This line was moved right
195
+ llm_response = llm_with_tools.invoke(messages_to_send) # This line was moved right
196
+ print(f"LLM Raw Response: {llm_response}") # This line was moved right
197
+ return {"messages": [llm_response]} # This line was moved right
198
 
199
  builder = StateGraph(MessagesState)
200
  builder.add_node("assistant", assistant)