Update agent.py
Browse files
agent.py
CHANGED
|
@@ -181,7 +181,7 @@ def build_graph(provider: str = "groq"):
|
|
| 181 |
# Node
|
| 182 |
def assistant(state: MessagesState):
|
| 183 |
"""Assistant node"""
|
| 184 |
-
return {"messages": [llm_with_tools.invoke(state["messages"])]}
|
| 185 |
|
| 186 |
def retriever(state: MessagesState):
|
| 187 |
"""Retriever node"""
|
|
|
|
| 181 |
# Node
|
| 182 |
def assistant(state: MessagesState):
|
| 183 |
"""Assistant node"""
|
| 184 |
+
return {"messages": [sys_msg] + [llm_with_tools.invoke(state["messages"])]}
|
| 185 |
|
| 186 |
def retriever(state: MessagesState):
|
| 187 |
"""Retriever node"""
|