Merge pull request #132 from caike/fix-firecrawl-scrape-pydantic-errors

Fix pydantic related errors on FirecrawlScrapeWebsiteTool

src/crewai_tools/tools/firecrawl_scrape_website_tool/firecrawl_scrape_website_tool.py

@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING, Any, Dict, Optional, Type
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, ConfigDict
 
 from crewai_tools.tools.base_tool import BaseTool
 
@@ -24,6 +24,11 @@ class FirecrawlScrapeWebsiteToolSchema(BaseModel):
 
 
 class FirecrawlScrapeWebsiteTool(BaseTool):
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        validate_assignment=True,
+        frozen=False
+    )
     name: str = "Firecrawl web scrape tool"
     description: str = "Scrape webpages url using Firecrawl and return the contents"
     args_schema: Type[BaseModel] = FirecrawlScrapeWebsiteToolSchema
@@ -61,3 +66,14 @@ class FirecrawlScrapeWebsiteTool(BaseTool):
             "timeout": timeout,
         }
         return self.firecrawl.scrape_url(url, options)
+
+
+try:
+    from firecrawl import FirecrawlApp
+    # Must rebuild model after class is defined
+    FirecrawlScrapeWebsiteTool.model_rebuild()
+except ImportError:
+    """
+    When this tool is not used, then exception can be ignored.
+    """
+    pass