tu101 commited on
Commit
a08b649
·
verified ·
1 Parent(s): ec8a648
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ class BasicAgent:
22
  print(f"Agent received question (first 50 chars): {question[:50]}...")
23
  # Wrap question as a human message.
24
  messages = [HumanMessage(content=question)]
25
- messages = self.graph.invoke({"messages": messages})
26
  answer = messages['messages'][-1].content
27
  return answer
28
 
 
22
  print(f"Agent received question (first 50 chars): {question[:50]}...")
23
  # Wrap question as a human message.
24
  messages = [HumanMessage(content=question)]
25
+ messages = self.graph.invoke({"messages": messages, config={"callbacks": [langfuse_handler]}})
26
  answer = messages['messages'][-1].content
27
  return answer
28