fix
Browse files
agent.py
CHANGED
|
@@ -48,7 +48,7 @@ def answer_message(state: AgentState) -> AgentState:
|
|
| 48 |
"""
|
| 49 |
response = model.invoke(prompt)
|
| 50 |
# Append the model's answer to the messages list
|
| 51 |
-
return {"messages":
|
| 52 |
|
| 53 |
|
| 54 |
def build_graph():
|
|
|
|
| 48 |
"""
|
| 49 |
response = model.invoke(prompt)
|
| 50 |
# Append the model's answer to the messages list
|
| 51 |
+
return {"messages": messages + [response]}
|
| 52 |
|
| 53 |
|
| 54 |
def build_graph():
|