Fix tests

ag2ai · Jan 29, 2025 · b660a43 · b660a43
1 parent ccd1cc5
commit b660a43
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/autogen/tools/experimental/crawl4ai/crawl4ai.py b/autogen/tools/experimental/crawl4ai/crawl4ai.py
@@ -43,8 +43,8 @@ def __init__(
 
         async def crawl4ai_helper(  # type: ignore[no-any-unimported]
             url: str,
-            browser_cfg: Optional[BrowserConfig] = None,
-            crawl_config: Optional[CrawlerRunConfig] = None,
+            browser_cfg: Optional["BrowserConfig"] = None,
+            crawl_config: Optional["CrawlerRunConfig"] = None,
         ) -> Any:
             async with AsyncWebCrawler(config=browser_cfg) as crawler:
                 result = await crawler.arun(
@@ -142,7 +142,7 @@ def _get_crawl_config(  # type: ignore[no-any-unimported]
         instruction: str,
         llm_strategy_kwargs: Optional[dict[str, Any]] = None,
         extraction_model: Optional[Type[BaseModel]] = None,
-    ) -> CrawlerRunConfig:
+    ) -> "CrawlerRunConfig":
         provider, api_key = Crawl4AITool._get_provider_and_api_key(llm_config)
 
         if llm_strategy_kwargs is None: