Update agent.py
Browse files
agent.py
CHANGED
@@ -207,9 +207,7 @@ def build_graph(provider: str = "huggingface"):
|
|
207 |
def assistant(state: MessagesState):
|
208 |
"""Assistant node"""
|
209 |
|
210 |
-
"messages": [sys_msg] + state["messages"]
|
211 |
-
|
212 |
-
return {"messages": [llm_with_tools.invoke(state["messages"])]}
|
213 |
"""
|
214 |
def retriever(state: MessagesState):
|
215 |
#Retriever node
|
|
|
207 |
def assistant(state: MessagesState):
|
208 |
"""Assistant node"""
|
209 |
|
210 |
+
return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])]}
|
|
|
|
|
211 |
"""
|
212 |
def retriever(state: MessagesState):
|
213 |
#Retriever node
|