peerawatchomp commited on
Commit
84f8c5d
·
verified ·
1 Parent(s): 01903c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -68,12 +68,9 @@ class BasicAgent:
68
  print(f"Agent received question (first 50 chars): {question[:50]}...")
69
 
70
  messages = [HumanMessage(content=question)]
71
- print("invoking ......")
72
- print("agent", self.agent)
73
- answer = self.agent.invoke({"messages": messages})
74
- print("answer:", answer)
75
  print(f"Agent returning fixed answer: {answer}")
76
- return fixed_answer
77
 
78
  def run_and_submit_all( profile: gr.OAuthProfile | None):
79
  """
 
68
  print(f"Agent received question (first 50 chars): {question[:50]}...")
69
 
70
  messages = [HumanMessage(content=question)]
71
+ answer = self.agent.invoke({"messages": messages})['messages'][-1]["content"]
 
 
 
72
  print(f"Agent returning fixed answer: {answer}")
73
+ return answer
74
 
75
  def run_and_submit_all( profile: gr.OAuthProfile | None):
76
  """