Spaces:
Configuration error
Configuration error
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"""
|