andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-02 15:52:34 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Fix pydantic related errors on FirecrawlScrapeWebsiteTool

Browse Source
This commit is contained in:
Carlos Souza
2024-11-28 19:09:44 -05:00
parent 601abb2bc3
commit a94470772f
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/crewai_tools/tools/firecrawl_scrape_website_tool/firecrawl_scrape_website_tool.py
View File

@@ -1,6 +1,6 @@
from typing import TYPE_CHECKING, Any, Dict, Optional, Type 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 from crewai_tools.tools.base_tool import BaseTool
@@ -24,6 +24,7 @@ class FirecrawlScrapeWebsiteToolSchema(BaseModel):
class FirecrawlScrapeWebsiteTool(BaseTool): class FirecrawlScrapeWebsiteTool(BaseTool):
model_config = ConfigDict(arbitrary_types_allowed=True)
name: str = "Firecrawl web scrape tool" name: str = "Firecrawl web scrape tool"
description: str = "Scrape webpages url using Firecrawl and return the contents" description: str = "Scrape webpages url using Firecrawl and return the contents"
args_schema: Type[BaseModel] = FirecrawlScrapeWebsiteToolSchema args_schema: Type[BaseModel] = FirecrawlScrapeWebsiteToolSchema
@@ -61,3 +62,11 @@ class FirecrawlScrapeWebsiteTool(BaseTool):
"timeout": timeout, "timeout": timeout,
} }
return self.firecrawl.scrape_url(url, options) return self.firecrawl.scrape_url(url, options)
try:
from firecrawl import FirecrawlApp
FirecrawlScrapeWebsiteTool.model_rebuild()
except ImportError:
raise ImportError(
"`firecrawl` package not found, please run `pip install firecrawl-py`"
)