ORromu commited on
Commit
121e0ee
·
verified ·
1 Parent(s): 2448c7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -20,10 +20,9 @@ class BasicAgent:
20
 
21
  def __call__(self, question: str) -> str:
22
  time.sleep(30) # To keep using the free gemini key
23
- print(f"Agent received question (first 50 chars): {question[:50]}...")
24
  messages = [HumanMessage(content=question)]
25
  messages = self.graph.invoke({"messages": messages})
26
- print(messages)
27
  answer = messages['messages'][-1].content
28
  return answer.split("FINAL ANSWER:")[-1].strip()
29
 
 
20
 
21
  def __call__(self, question: str) -> str:
22
  time.sleep(30) # To keep using the free gemini key
23
+ print(f"Agent received question (first 200 chars): {question[:200]}...")
24
  messages = [HumanMessage(content=question)]
25
  messages = self.graph.invoke({"messages": messages})
 
26
  answer = messages['messages'][-1].content
27
  return answer.split("FINAL ANSWER:")[-1].strip()
28