added args_schema to browserbase tool

This commit is contained in:
Mish Ushakov
2024-04-29 11:53:10 +02:00
committed by GitHub
parent 39aba4cb48
commit f78011e68c

View File

@@ -1,9 +1,14 @@
from crewai_tools import BaseTool
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