Commit
·
cd7cb86
1
Parent(s):
9882879
fix: prompting and answers
Browse files
agent.py
CHANGED
@@ -184,7 +184,10 @@ def build_graph(provider: str = "openai"):
|
|
184 |
# Node
|
185 |
def assistant(state: MessagesState):
|
186 |
"""Assistant node"""
|
187 |
-
|
|
|
|
|
|
|
188 |
|
189 |
def retriever(state: MessagesState):
|
190 |
"""Retriever node"""
|
|
|
184 |
# Node
|
185 |
def assistant(state: MessagesState):
|
186 |
"""Assistant node"""
|
187 |
+
# Prepend the system prompt
|
188 |
+
messages = [sys_msg] + state["messages"]
|
189 |
+
result = llm_with_tools.invoke(messages)
|
190 |
+
return {"messages": [result]}
|
191 |
|
192 |
def retriever(state: MessagesState):
|
193 |
"""Retriever node"""
|