From cf6231384c447780d07446fff6f1c83d7338c91c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Thu, 2 May 2024 03:56:14 -0300 Subject: [PATCH] quick fix --- .../tools/browserbase_load_tool/browserbase_load_tool.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py b/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py index d29656188..48f3dacef 100644 --- a/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py +++ b/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py @@ -1,13 +1,13 @@ -from crewai_tools import BaseTool -from typing import Optional, Any +from typing import Optional, Any, Type from pydantic.v1 import BaseModel, Field +from crewai_tools.tools.base_tool import BaseTool class BrowserbaseLoadToolSchema(BaseModel): url: str = Field(description="Website URL") class BrowserbaseLoadTool(BaseTool): name: str = "Browserbase web load tool" - description: str = "Load webpages in a headless browser using Browserbase and return the contents" + description: str = "Load webpages url in a headless browser using Browserbase and return the contents" args_schema: Type[BaseModel] = BrowserbaseLoadToolSchema api_key: Optional[str] = None text_content: Optional[bool] = False @@ -16,7 +16,7 @@ class BrowserbaseLoadTool(BaseTool): def __init__(self, api_key: Optional[str] = None, text_content: Optional[bool] = False, **kwargs): super().__init__(**kwargs) try: - from browserbase import Browserbase + from browserbase import Browserbase # type: ignore except ImportError: raise ImportError( "`browserbase` package not found, please run `pip install browserbase`"