diff --git a/backend/open_webui/retrieval/web/firecrawl.py b/backend/open_webui/retrieval/web/firecrawl.py index 00de5de9cb..2d9b104bca 100644 --- a/backend/open_webui/retrieval/web/firecrawl.py +++ b/backend/open_webui/retrieval/web/firecrawl.py @@ -17,9 +17,9 @@ def search_firecrawl( filter_list: Optional[List[str]] = None, ) -> List[SearchResult]: try: - from firecrawl import Firecrawl + from firecrawl import FirecrawlApp - firecrawl = Firecrawl(api_key=firecrawl_api_key, api_url=firecrawl_url) + firecrawl = FirecrawlApp(api_key=firecrawl_api_key, api_url=firecrawl_url) response = firecrawl.search( query=query, limit=count, ignore_invalid_urls=True, timeout=count * 3 ) diff --git a/backend/open_webui/retrieval/web/utils.py b/backend/open_webui/retrieval/web/utils.py index 74c4a734da..91699a157b 100644 --- a/backend/open_webui/retrieval/web/utils.py +++ b/backend/open_webui/retrieval/web/utils.py @@ -41,7 +41,6 @@ from open_webui.config import ( ) from open_webui.env import SRC_LOG_LEVELS -from firecrawl import Firecrawl log = logging.getLogger(__name__) log.setLevel(SRC_LOG_LEVELS["RAG"]) @@ -227,7 +226,9 @@ class SafeFireCrawlLoader(BaseLoader, RateLimitMixin, URLProcessingMixin): self.params, ) try: - firecrawl = Firecrawl(api_key=self.api_key, api_url=self.api_url) + from firecrawl import FirecrawlApp + + firecrawl = FirecrawlApp(api_key=self.api_key, api_url=self.api_url) result = firecrawl.batch_scrape( self.web_paths, formats=["markdown"], @@ -266,7 +267,9 @@ class SafeFireCrawlLoader(BaseLoader, RateLimitMixin, URLProcessingMixin): self.params, ) try: - firecrawl = Firecrawl(api_key=self.api_key, api_url=self.api_url) + from firecrawl import FirecrawlApp + + firecrawl = FirecrawlApp(api_key=self.api_key, api_url=self.api_url) result = firecrawl.batch_scrape( self.web_paths, formats=["markdown"],