Spaces:
Sleeping
Sleeping
Update tools.py
Browse files
tools.py
CHANGED
|
@@ -23,8 +23,12 @@ def wiki_loader(query: str, lang: str='en', load_max_docs: int=2):
|
|
| 23 |
loader = WikipediaLoader(query=query, lang=lang, load_max_docs=load_max_docs)
|
| 24 |
# Load the documents
|
| 25 |
documents = loader.load()
|
| 26 |
-
|
| 27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
except Exception as e:
|
| 29 |
print(f"An error occurred: {e}")
|
| 30 |
return []
|
|
|
|
| 23 |
loader = WikipediaLoader(query=query, lang=lang, load_max_docs=load_max_docs)
|
| 24 |
# Load the documents
|
| 25 |
documents = loader.load()
|
| 26 |
+
result = []
|
| 27 |
+
for doc in documents:
|
| 28 |
+
content = doc.page_content
|
| 29 |
+
result.append(content)
|
| 30 |
+
print(result)
|
| 31 |
+
return result
|
| 32 |
except Exception as e:
|
| 33 |
print(f"An error occurred: {e}")
|
| 34 |
return []
|