mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 15:48:29 +00:00
* fix: fix Firecrawl Scrape tool * fix: fix Firecrawl Search tool * fix: fix Firecrawl Website tool * tests: adding tests for Firecrawl
19 lines
568 B
Python
19 lines
568 B
Python
import pytest
|
|
|
|
from crewai_tools.tools.firecrawl_crawl_website_tool.firecrawl_crawl_website_tool import (
|
|
FirecrawlCrawlWebsiteTool,
|
|
)
|
|
|
|
@pytest.mark.vcr(filter_headers=["authorization"])
|
|
def test_firecrawl_crawl_tool_integration():
|
|
tool = FirecrawlCrawlWebsiteTool(config={
|
|
"limit": 2,
|
|
"max_discovery_depth": 1,
|
|
"scrape_options": {"formats": ["markdown"]}
|
|
})
|
|
result = tool.run(url="https://firecrawl.dev")
|
|
|
|
assert result is not None
|
|
assert hasattr(result, 'status')
|
|
assert result.status in ["completed", "scraping"]
|