github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-08 06:00:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Harrison/wiki update (#622)

Browse Source 
Co-authored-by: Rubens Mau <rubensmau@gmail.com>
This commit is contained in:
Harrison Chase 2023-01-15 16:45:16 -08:00 committed by GitHub
parent 1ac3319e45
commit 49b3d6c78c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/docstore/wikipedia.py
View File

@ -30,7 +30,10 @@ class Wikipedia(Docstore):
try: try:
page_content = wikipedia.page(search).content page_content = wikipedia.page(search).content
result: Union[str, Document] = Document(page_content=page_content) url = wikipedia.page(search).url
result: Union[str, Document] = Document(
page_content=page_content, metadata={"page": url}
)
except wikipedia.PageError: except wikipedia.PageError:
result = f"Could not find [{search}]. Similar: {wikipedia.search(search)}" result = f"Could not find [{search}]. Similar: {wikipedia.search(search)}"
except wikipedia.DisambiguationError: except wikipedia.DisambiguationError: