zcfrank1st commited on
Commit
4b09d90
·
1 Parent(s): 83a4965

change code

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,8 +31,8 @@ class BasicAgent:
31
  def __call__(self, question: str) -> str:
32
  print(f"Agent received question (first 50 chars): {question[:50]}...")
33
  answer = self.agent.invoke({"messages": [("user", question)]})
34
- print(f"Agent returning answer: {answer}")
35
- return answer
36
 
37
  def run_and_submit_all( profile: gr.OAuthProfile | None):
38
  """
 
31
  def __call__(self, question: str) -> str:
32
  print(f"Agent received question (first 50 chars): {question[:50]}...")
33
  answer = self.agent.invoke({"messages": [("user", question)]})
34
+ print(f"Agent returning answer: {answer['messages'][-1].content}")
35
+ return answer['messages'][-1].content
36
 
37
  def run_and_submit_all( profile: gr.OAuthProfile | None):
38
  """