From 39aba4cb48b0fdbfc2dff63ee01ec0d6c6c22078 Mon Sep 17 00:00:00 2001
From: Mish Ushakov <10400064+mishushakov@users.noreply.github.com>
Date: Thu, 25 Apr 2024 12:14:56 +0000
Subject: [PATCH 1/2] updated browserbase load tool

---
 .../tools/browserbase_load_tool/browserbase_load_tool.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py b/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py
index 126219bd3..7bf066287 100644
--- a/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py
+++ b/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py
@@ -1,12 +1,15 @@
-import os
 from crewai_tools import BaseTool
-from typing import Union, Optional
+from typing import Optional, Any
 
 class BrowserbaseLoadTool(BaseTool):
     name: str = "Browserbase web load tool"
     description: str = "Load webpages in a headless browser using Browserbase and return the contents"
+    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:

From f78011e68ca911010b81a4097b153d57eca3162b Mon Sep 17 00:00:00 2001
From: Mish Ushakov <10400064+mishushakov@users.noreply.github.com>
Date: Mon, 29 Apr 2024 11:53:10 +0200
Subject: [PATCH 2/2] added args_schema to browserbase tool

---
 .../tools/browserbase_load_tool/browserbase_load_tool.py     | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py b/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py
index 7bf066287..d29656188 100644
--- a/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py
+++ b/src/crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py
@@ -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