andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

More explicit model config

Browse Source
This commit is contained in:
Carlos Souza
2024-11-29 05:57:09 -05:00
parent 49ad43ff08
commit e677a271e5
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -24,7 +24,11 @@ class FirecrawlScrapeWebsiteToolSchema(BaseModel):
class FirecrawlScrapeWebsiteTool(BaseTool): class FirecrawlScrapeWebsiteTool(BaseTool):
model_config = ConfigDict(arbitrary_types_allowed=True) model_config = ConfigDict(
arbitrary_types_allowed=True,
validate_assignment=True,
frozen=False
)
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
@@ -63,6 +67,7 @@ class FirecrawlScrapeWebsiteTool(BaseTool):
} }
return self.firecrawl.scrape_url(url, options) return self.firecrawl.scrape_url(url, options)
try: try:
from firecrawl import FirecrawlApp from firecrawl import FirecrawlApp
# Must rebuild model after class is defined # Must rebuild model after class is defined