Spaces:
Sleeping
Sleeping
graph test
Browse files
app.py
CHANGED
|
@@ -144,12 +144,12 @@ class BasicAgent:
|
|
| 144 |
|
| 145 |
def wiki_search(self, state: AnswerState) -> Dict[str, Any]:
|
| 146 |
print("[wiki_search] Searching for: " + str(state["wiki_request"]))
|
| 147 |
-
|
| 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}")
|
| 155 |
state["messages"].append(
|
|
|
|
| 144 |
|
| 145 |
def wiki_search(self, state: AnswerState) -> Dict[str, Any]:
|
| 146 |
print("[wiki_search] Searching for: " + str(state["wiki_request"]))
|
| 147 |
+
wiki_tool = 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}")
|
| 155 |
state["messages"].append(
|