ORromu commited on
Commit
99407de
·
verified ·
1 Parent(s): 556473a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -27,12 +27,13 @@ class BasicAgent:
27
  self.graph = simple_graph()
28
 
29
  def __call__(self, question: str) -> str:
30
- time.sleep(60) # To keep using the free gemini key
31
  print(f"Agent received question (first 50 chars): {question[:50]}...")
32
  messages = [sys_msg, HumanMessage(content=question)]
33
  print(messages)
34
  messages = self.graph.invoke({"messages": messages})
35
  answer = messages['messages'][-1].content
 
36
  return answer
37
 
38
  def run_and_submit_all( profile: gr.OAuthProfile | None):
 
27
  self.graph = simple_graph()
28
 
29
  def __call__(self, question: str) -> str:
30
+ time.sleep(30) # To keep using the free gemini key
31
  print(f"Agent received question (first 50 chars): {question[:50]}...")
32
  messages = [sys_msg, HumanMessage(content=question)]
33
  print(messages)
34
  messages = self.graph.invoke({"messages": messages})
35
  answer = messages['messages'][-1].content
36
+ answer = answer[answer.index("FINAL_ANSWER: ") + 14:]
37
  return answer
38
 
39
  def run_and_submit_all( profile: gr.OAuthProfile | None):