andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-08 15:48:29 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Merge pull request #28 from mishushakov/main

Browse Source 
Updated Browserbase load tool
This commit is contained in:
João Moura
2024-05-02 02:23:41 -03:00
committed by GitHub
parent ed58694675 f78011e68c
commit e6da49754d
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,12 +1,20 @@
import os
from crewai_tools import BaseTool
from typing import Union, Optional
from typing import Optional, Any
from pydantic.v1 import BaseModel, Field
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"
args_schema: Type[BaseModel] = BrowserbaseLoadToolSchema
api_key: Optional[str] = None
text_content: Optional[bool] = False
browserbase: Optional[Any] = None
def __init__(self, api_key: Optional[str] = None, text_content: bool = False):
def __init__(self, api_key: Optional[str] = None, text_content: Optional[bool] = False, **kwargs):
super().__init__(**kwargs)
try:
from browserbase import Browserbase
except ImportError: