From 5d86a6acf10616f881c814229d930242e81b11af Mon Sep 17 00:00:00 2001 From: Rui Ferreira Date: Thu, 30 Mar 2023 15:34:13 +0100 Subject: [PATCH] Fix wikipedia summaries (#2187) This upsteam wikipedia page loading seems to still have issues. Finding a compromise solution where it does an exact match search and not a search for the completion. See previous PR: https://github.com/hwchase17/langchain/pull/2169 --- langchain/utilities/wikipedia.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/langchain/utilities/wikipedia.py b/langchain/utilities/wikipedia.py index 5256e3d9..6685ead8 100644 --- a/langchain/utilities/wikipedia.py +++ b/langchain/utilities/wikipedia.py @@ -47,7 +47,7 @@ class WikipediaAPIWrapper(BaseModel): def fetch_formatted_page_summary(self, page: str) -> Optional[str]: try: - wiki_page = self.wiki_client.page(pageid=page) + wiki_page = self.wiki_client.page(title=page, auto_suggest=False) return f"Page: {page}\nSummary: {wiki_page.summary}" except ( self.wiki_client.exceptions.PageError,