andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 16:18:30 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Update browserbase_load_tool.py

Browse Source
This commit is contained in:
Priyanshupareek
2025-01-02 00:54:48 +05:30
committed by GitHub
parent 0d94a8f7d9
commit 4c7ce3a945
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,8 +1,9 @@
from typing import Any, Optional, Type from typing import Any, Optional, Type
import os
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
@@ -38,7 +39,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