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)
|