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

quick fix

Browse Source
This commit is contained in:
João Moura
2024-05-02 03:56:14 -03:00
parent a95cbfdc6a
commit cf6231384c
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py
View File

@@ -1,13 +1,13 @@
from crewai_tools import BaseTool from typing import Optional, Any, Type
from typing import Optional, Any
from pydantic.v1 import BaseModel, Field from pydantic.v1 import BaseModel, Field
from crewai_tools.tools.base_tool import BaseTool
class BrowserbaseLoadToolSchema(BaseModel): class BrowserbaseLoadToolSchema(BaseModel):
url: str = Field(description="Website URL") url: str = Field(description="Website URL")
class BrowserbaseLoadTool(BaseTool): class BrowserbaseLoadTool(BaseTool):
name: str = "Browserbase web load tool" 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 args_schema: Type[BaseModel] = BrowserbaseLoadToolSchema
api_key: Optional[str] = None api_key: Optional[str] = None
text_content: Optional[bool] = False 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): def __init__(self, api_key: Optional[str] = None, text_content: Optional[bool] = False, **kwargs):
super().__init__(**kwargs) super().__init__(**kwargs)
try: try:
from browserbase import Browserbase from browserbase import Browserbase # type: ignore
except ImportError: except ImportError:
raise ImportError( raise ImportError(
"`browserbase` package not found, please run `pip install browserbase`" "`browserbase` package not found, please run `pip install browserbase`"