mirjam-m commited on
Commit
e38c5b6
·
1 Parent(s): a30a657

graph test

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -144,7 +144,7 @@ class BasicAgent:
144
 
145
  def wiki_search(self, state: AnswerState) -> Dict[str, Any]:
146
  print("[wiki_search] Searching for: " + str(state["wiki_request"]))
147
- api_wrapper = WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=10000)
148
  wiki_tool = WikipediaQueryRun(api_wrapper=api_wrapper)
149
  results = wiki_tool.run(str(state["wiki_request"]))
150
  print(f"Wiki results: {results}")
@@ -183,7 +183,7 @@ class BasicAgent:
183
 
184
  if state["is_final_answer"]:
185
  return "FINAL_ANSWER"
186
- elif state["attempt"] > 3:
187
  return "EXCEEDED_ATTEMPTS"
188
  elif state["search_request"]:
189
  return "WEB_SEARCH"
 
144
 
145
  def wiki_search(self, state: AnswerState) -> Dict[str, Any]:
146
  print("[wiki_search] Searching for: " + str(state["wiki_request"]))
147
+ api_wrapper = WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=100000)
148
  wiki_tool = WikipediaQueryRun(api_wrapper=api_wrapper)
149
  results = wiki_tool.run(str(state["wiki_request"]))
150
  print(f"Wiki results: {results}")
 
183
 
184
  if state["is_final_answer"]:
185
  return "FINAL_ANSWER"
186
+ elif state["attempt"] > 5:
187
  return "EXCEEDED_ATTEMPTS"
188
  elif state["search_request"]:
189
  return "WEB_SEARCH"