Spaces:
Sleeping
Sleeping
graph test
Browse files
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=
|
| 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"] >
|
| 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"
|