mnab commited on
Commit
503e8d6
·
verified ·
1 Parent(s): 6806cde

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ class BasicAgent:
21
  def __call__(self, question: str) -> str:
22
  print(f"Agent received question (first 50 chars): {question[:50]}...")
23
  messages = [HumanMessage(content=question)]
24
- messages = self.graph.invoke({"messages": messages})
25
  answer = messages["messages"][-1].content
26
  return answer[14:]
27
  # fixed_answer = "This is a default answer."
 
21
  def __call__(self, question: str) -> str:
22
  print(f"Agent received question (first 50 chars): {question[:50]}...")
23
  messages = [HumanMessage(content=question)]
24
+ messages = self.workflow.invoke({"messages": messages})
25
  answer = messages["messages"][-1].content
26
  return answer[14:]
27
  # fixed_answer = "This is a default answer."