Support recursive sitemaps in SitemapLoader (#3146)

A (very) simple addition to support multiple sitemap urls.

---------

Co-authored-by: Johann-Peter Hartmann <johann-peter.hartmann@mayflower.de>
This commit is contained in:
Johann-Peter Hartmann 2023-04-22 17:48:04 +02:00 committed by GitHub
parent 215dcc2d26
commit 7e79f8c136
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -61,6 +61,13 @@ class SitemapLoader(WebBaseLoader):
}
)
for sitemap in soup.find_all("sitemap"):
loc = sitemap.find("loc")
if not loc:
continue
soup_child = self.scrape_all([loc.text], "xml")[0]
els.extend(self.parse_sitemap(soup_child))
return els
def load(self) -> List[Document]: