sampsong commited on
Commit
fe740d6
·
1 Parent(s): 6e43ba6
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -23,10 +23,14 @@ class BasicAgent:
23
  print(f"Agent received question (first 50 chars): {question[:50]}...")
24
  messages = [HumanMessage(content=question)]
25
  print(f"Message = {messages}")
26
- #messages = self.graph.invoke({"messages": messages})
27
- messages = self.graph.invoke({"messages": messages},config={"callbacks": [langfuse_handler]})
 
28
  self.graph.get_graph().draw_mermaid_png()
29
  answer = messages['messages'][-1].content
 
 
 
30
  return answer[14:]
31
 
32
  def run_and_submit_all( profile: gr.OAuthProfile | None):
 
23
  print(f"Agent received question (first 50 chars): {question[:50]}...")
24
  messages = [HumanMessage(content=question)]
25
  print(f"Message = {messages}")
26
+ messages = self.graph.invoke({"messages": messages})
27
+
28
+ #messages = self.graph.invoke({"messages": messages},config={"callbacks": [langfuse_handler]})
29
  self.graph.get_graph().draw_mermaid_png()
30
  answer = messages['messages'][-1].content
31
+ print(f"second last message : {messages['messages'][-2].content}")
32
+ print(f"full answer : {answer}")
33
+
34
  return answer[14:]
35
 
36
  def run_and_submit_all( profile: gr.OAuthProfile | None):