Spaces:
Sleeping
Sleeping
niikun
commited on
Commit
·
2a3d577
1
Parent(s):
c352afe
update __call__
Browse files
app.py
CHANGED
|
@@ -57,7 +57,11 @@ class BasicAgent:
|
|
| 57 |
def __call__(self, question: str) -> str:
|
| 58 |
state = {"messages": [question]}
|
| 59 |
out = self.graph.invoke(state)
|
| 60 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 61 |
|
| 62 |
def run_and_submit_all( profile: gr.OAuthProfile | None):
|
| 63 |
"""
|
|
|
|
| 57 |
def __call__(self, question: str) -> str:
|
| 58 |
state = {"messages": [question]}
|
| 59 |
out = self.graph.invoke(state)
|
| 60 |
+
last_msg = out["messages"][-1]
|
| 61 |
+
if hasattr(last_msg, "content"):
|
| 62 |
+
return last_msg.content
|
| 63 |
+
else:
|
| 64 |
+
return str(last_msg)
|
| 65 |
|
| 66 |
def run_and_submit_all( profile: gr.OAuthProfile | None):
|
| 67 |
"""
|