refac

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
         )

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"],