Spaces:
Sleeping
Sleeping
Upload tools.py
Browse files- src/tools.py +5 -2
src/tools.py
CHANGED
|
@@ -74,11 +74,14 @@ def wiki_search(query: str) -> str:
|
|
| 74 |
|
| 75 |
Args:
|
| 76 |
query: The search query."""
|
| 77 |
-
search_docs = WikipediaLoader(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 78 |
formatted_search_docs = [
|
| 79 |
{
|
| 80 |
"source": doc.metadata["source"],
|
| 81 |
-
"page": doc.metadata.get("page", ""),
|
| 82 |
"content": doc.page_content,
|
| 83 |
}
|
| 84 |
for doc in search_docs
|
|
|
|
| 74 |
|
| 75 |
Args:
|
| 76 |
query: The search query."""
|
| 77 |
+
search_docs = WikipediaLoader(
|
| 78 |
+
query=query,
|
| 79 |
+
load_max_docs=2,
|
| 80 |
+
doc_content_chars_max=50_000
|
| 81 |
+
).load()
|
| 82 |
formatted_search_docs = [
|
| 83 |
{
|
| 84 |
"source": doc.metadata["source"],
|
|
|
|
| 85 |
"content": doc.page_content,
|
| 86 |
}
|
| 87 |
for doc in search_docs
|