From e0d799c075972414d328ee4d9e362e286c90f0ef Mon Sep 17 00:00:00 2001
From: Mish Ushakov <10400064+mishushakov@users.noreply.github.com>
Date: Fri, 19 Apr 2024 11:40:03 +0000
Subject: [PATCH] updated browserbase integration

---
 src/crewai_tools/tools/browserbase_load_tool/README.md        | 4 ++--
 .../tools/browserbase_load_tool/browserbase_load_tool.py      | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/crewai_tools/tools/browserbase_load_tool/README.md b/src/crewai_tools/tools/browserbase_load_tool/README.md
index 0007feb91..a2866f9a8 100644
--- a/src/crewai_tools/tools/browserbase_load_tool/README.md
+++ b/src/crewai_tools/tools/browserbase_load_tool/README.md
@@ -6,7 +6,7 @@
 
 ## Installation
 
-- Get an API key from [browserbase.com](https://browserbase.com) and set it in environment variables (`BROWSERBASE_KEY`).
+- Get an API key from [browserbase.com](https://browserbase.com) and set it in environment variables (`BROWSERBASE_API_KEY`).
 - Install the [Browserbase SDK](http://github.com/browserbase/python-sdk) along with `crewai[tools]` package:
 
 ```
@@ -25,5 +25,5 @@ tool = BrowserbaseLoadTool()
 
 ## Arguments
 
-- `api_key`: Optional. Specifies Browserbase API key. Defaults is the `BROWSERBASE_KEY` environment variable.
+- `api_key`: Optional. Specifies Browserbase API key. Defaults is the `BROWSERBASE_API_KEY` environment variable.
 - `text_content`: Optional. Load pages as readable text. Default is `False`.
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 03ee53f99..126219bd3 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,12 @@
 import os
 from crewai_tools import BaseTool
-from typing import Union
+from typing import Union, Optional
 
 class BrowserbaseLoadTool(BaseTool):
     name: str = "Browserbase web load tool"
     description: str = "Load webpages in a headless browser using Browserbase and return the contents"
 
-    def __init__(self, api_key: str = os.environ["BROWSERBASE_KEY"], text_content: bool = False):
+    def __init__(self, api_key: Optional[str] = None, text_content: bool = False):
         try:
             from browserbase import Browserbase
         except ImportError: