KPatelis commited on
Commit
6e33629
·
1 Parent(s): 8df4ab3

changed issue in app

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