Killian Mahé commited on
Commit
72f2f08
·
1 Parent(s): c5b142b

Update answer format

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ class BasicAgent:
18
  def __call__(self, question: str) -> str:
19
  print(f"Agent received question (first 50 chars): {question[:50]}...")
20
  messages = gaia_agent.invoke({"messages": [HumanMessage(content=question)]})
21
- answer = messages['messages'][-1].content
22
- print(f"Agent returning fixed answer: {answer}")
23
  return answer
24
 
25
  def run_and_submit_all( profile: gr.OAuthProfile | None):
 
18
  def __call__(self, question: str) -> str:
19
  print(f"Agent received question (first 50 chars): {question[:50]}...")
20
  messages = gaia_agent.invoke({"messages": [HumanMessage(content=question)]})
21
+ answer = messages['messages'][-1].content[14:]
22
+ print(f"Agent returning answer: {answer}")
23
  return answer
24
 
25
  def run_and_submit_all( profile: gr.OAuthProfile | None):