MariaFilipkowska commited on
Commit
e163e29
·
verified ·
1 Parent(s): cb7ae6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,7 +57,7 @@ def search_web(topic: str) -> str:
57
  """Retrieves information about the topic."""
58
  results = DuckDuckGoSearchRun().invoke(topic)
59
  if results:
60
- return "\n\n".join([doc.text for doc in results[:2]])
61
  else:
62
  return "No matching content found."
63
 
@@ -155,7 +155,7 @@ class BasicAgent:
155
  print(f"Agent received question (first 50 chars): {question[:50]}...")
156
  message = HumanMessage(content=question)
157
  print(message)
158
- answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 3})['messages'][-1].content
159
  answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))
160
  answer = answer.replace('FINAL ANSWER:', '')
161
  answer = answer.replace('FINAL ANSWER', '')
 
57
  """Retrieves information about the topic."""
58
  results = DuckDuckGoSearchRun().invoke(topic)
59
  if results:
60
+ return "\n\n".join([doc.text for doc in results[:6]])
61
  else:
62
  return "No matching content found."
63
 
 
155
  print(f"Agent received question (first 50 chars): {question[:50]}...")
156
  message = HumanMessage(content=question)
157
  print(message)
158
+ answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 5})['messages'][-1].content
159
  answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))
160
  answer = answer.replace('FINAL ANSWER:', '')
161
  answer = answer.replace('FINAL ANSWER', '')