andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 23:32:39 +00:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

updated browserbase tool

Browse Source
This commit is contained in:
Mish Ushakov
2024-05-20 08:58:25 +00:00
parent 53c7d815ae
commit 5c2d8c4cfa
1 changed files with 22 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -10,20 +10,35 @@ class BrowserbaseLoadTool(BaseTool):
description: str = "Load webpages url 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
project_id: Optional[str] = None
text_content: Optional[bool] = False text_content: Optional[bool] = False
session_id: Optional[str] = None
proxy: Optional[bool] = None
browserbase: Optional[Any] = None browserbase: Optional[Any] = None
def __init__(self, api_key: Optional[str] = None, text_content: Optional[bool] = False, **kwargs): def __init__(
self,
api_key: Optional[str] = None,
project_id: Optional[str] = None,
text_content: Optional[bool] = False,
session_id: Optional[str] = None,
proxy: Optional[bool] = None,
**kwargs,
):
super().__init__(**kwargs) super().__init__(**kwargs)
try: try:
from browserbase import Browserbase # type: ignore 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`"
) )
self.browserbase = Browserbase(api_key=api_key) self.browserbase = Browserbase(api_key, project_id)
self.text_content = text_content self.text_content = text_content
self.session_id = session_id
self.proxy = proxy
def _run(self, url: str): def _run(self, url: str):
return self.browserbase.load_url(url, text_content=self.text_content) return self.browserbase.load_url(
url, self.text_content, self.session_id, self.proxy
)