luping85 commited on
Commit
54de1bd
·
verified ·
1 Parent(s): b24c4a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ class BasicAgent:
18
  def __call__(self, question: str) -> str:
19
  print(f"Agent received question (first 50 chars): {question[:50]}...")
20
  response = graph.invoke({"messages": question})
21
- answer = response["messages"][-1].content
22
  print(f"Agent returning fixed answer: {answer}")
23
  return answer
24
 
 
18
  def __call__(self, question: str) -> str:
19
  print(f"Agent received question (first 50 chars): {question[:50]}...")
20
  response = graph.invoke({"messages": question})
21
+ answer = response["messages"][-1].content.split(': ')[1]
22
  print(f"Agent returning fixed answer: {answer}")
23
  return answer
24