Spaces:
Sleeping
Sleeping
graph test
Browse files
app.py
CHANGED
|
@@ -144,7 +144,11 @@ 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(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 148 |
wiki_tool = WikipediaQueryRun(api_wrapper=api_wrapper)
|
| 149 |
results = wiki_tool.run(str(state["wiki_request"]))
|
| 150 |
print(f"Wiki results: {results}")
|
|
|
|
| 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(
|
| 148 |
+
top_k_results=1,
|
| 149 |
+
doc_content_chars_max=100000,
|
| 150 |
+
load_all_available_meta=True,
|
| 151 |
+
)
|
| 152 |
wiki_tool = WikipediaQueryRun(api_wrapper=api_wrapper)
|
| 153 |
results = wiki_tool.run(str(state["wiki_request"]))
|
| 154 |
print(f"Wiki results: {results}")
|