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 6 Packages Projects Releases Wiki Activity

Merge pull request #167 from Priyanshupareek/patch-1

Browse Source 
Update browserbase_load_tool.py | Fix for Browserbase Tool Initialization Issue
This commit is contained in:
João Moura
2025-01-02 20:45:05 -03:00
committed by GitHub
parent 0d94a8f7d9 954dd43c17
commit fc68b9f6bf
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,8 +1,9 @@
import os
from typing import Any, Optional, Type from typing import Any, Optional, Type
from crewai.tools import BaseTool
from pydantic import BaseModel, Field from pydantic 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")
@@ -14,8 +15,8 @@ class BrowserbaseLoadTool(BaseTool):
"Load webpages url in a headless browser using Browserbase and return the contents" "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] = os.getenv('BROWSERBASE_API_KEY')
project_id: Optional[str] = None project_id: Optional[str] = os.getenv('BROWSERBASE_PROJECT_ID')
text_content: Optional[bool] = False text_content: Optional[bool] = False
session_id: Optional[str] = None session_id: Optional[str] = None
proxy: Optional[bool] = None proxy: Optional[bool] = None
@@ -31,6 +32,8 @@ class BrowserbaseLoadTool(BaseTool):
**kwargs, **kwargs,
): ):
super().__init__(**kwargs) super().__init__(**kwargs)
if not self.api_key:
raise EnvironmentError("BROWSERBASE_API_KEY environment variable is required for initialization")
try: try:
from browserbase import Browserbase # type: ignore from browserbase import Browserbase # type: ignore
except ImportError: except ImportError:
@@ -38,7 +41,7 @@ class BrowserbaseLoadTool(BaseTool):
"`browserbase` package not found, please run `pip install browserbase`" "`browserbase` package not found, please run `pip install browserbase`"
) )
self.browserbase = Browserbase(api_key, project_id) self.browserbase = Browserbase(api_key=self.api_key)
self.text_content = text_content self.text_content = text_content
self.session_id = session_id self.session_id = session_id
self.proxy = proxy self.proxy = proxy