Mohammad Haghir commited on
Commit ·
852d9f3
1
Parent(s): 057b801
agent
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ class BasicAgent:
|
|
| 40 |
|
| 41 |
def __call__(self, question: str) -> str:
|
| 42 |
response = (self.graph).invoke({"messages": question})
|
| 43 |
-
return response
|
| 44 |
|
| 45 |
def agent(self, state: GraphState):
|
| 46 |
|
|
|
|
| 40 |
|
| 41 |
def __call__(self, question: str) -> str:
|
| 42 |
response = (self.graph).invoke({"messages": question})
|
| 43 |
+
return response["messages"]
|
| 44 |
|
| 45 |
def agent(self, state: GraphState):
|
| 46 |
|