Mohammad Haghir commited on
Commit ·
85b75bf
1
Parent(s): a50492d
update
Browse files
app.py
CHANGED
|
@@ -71,7 +71,7 @@ class BasicAgent:
|
|
| 71 |
|
| 72 |
messages = [HumanMessage(content=prompt)]
|
| 73 |
response = (self.graph).invoke({"messages": messages})
|
| 74 |
-
return response["messages"].content
|
| 75 |
|
| 76 |
def agent(self, state: GraphState):
|
| 77 |
|
|
|
|
| 71 |
|
| 72 |
messages = [HumanMessage(content=prompt)]
|
| 73 |
response = (self.graph).invoke({"messages": messages})
|
| 74 |
+
return response["messages"][-1].content
|
| 75 |
|
| 76 |
def agent(self, state: GraphState):
|
| 77 |
|