diff --git a/langchain/docstore/wikipedia.py b/langchain/docstore/wikipedia.py index 2325ffd9..5575b8c7 100644 --- a/langchain/docstore/wikipedia.py +++ b/langchain/docstore/wikipedia.py @@ -30,7 +30,10 @@ class Wikipedia(Docstore): try: 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: result = f"Could not find [{search}]. Similar: {wikipedia.search(search)}" except wikipedia.DisambiguationError: