diff --git a/libs/community/langchain_community/document_loaders/firecrawl.py b/libs/community/langchain_community/document_loaders/firecrawl.py index 3ff3bb3b1e9..2577ce6eda4 100644 --- a/libs/community/langchain_community/document_loaders/firecrawl.py +++ b/libs/community/langchain_community/document_loaders/firecrawl.py @@ -63,7 +63,10 @@ class FireCrawlLoader(BaseLoader): f"Unrecognized mode '{self.mode}'. Expected one of 'crawl', 'scrape'." ) for doc in firecrawl_docs: - yield Document( - page_content=doc.get("markdown", ""), - metadata=doc.get("metadata", {}), - ) + metadata = doc.get("metadata", {}) + if (self.params is not None) and self.params.get( + "extractorOptions", {} + ).get("mode") == "llm-extraction": + metadata["llm_extraction"] = doc.get("llm_extraction") + + yield Document(page_content=doc.get("markdown", ""), metadata=metadata)