updated browserbase integration

This commit is contained in:
Mish Ushakov
2024-04-19 11:40:03 +00:00
parent b80dd1ca8b
commit e0d799c075
2 changed files with 4 additions and 4 deletions

View File

@@ -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`.

View File

@@ -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: