Update langgraph_agent.py
Browse files- 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 |
-
|
|
|
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)
|