niikun commited on
Commit
2a3d577
·
1 Parent(s): c352afe

update __call__

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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
- return out["messages"][-1]
 
 
 
 
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
  """