prozorov commited on
Commit
bf7c60c
·
verified ·
1 Parent(s): 4a4ae8f

Update app.py

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