Scrapegraph-ai/scrapegraphai/docloaders/browser_base.py
2024-08-17 12:21:00 +02:00

51 lines
1.6 KiB
Python

"""
browserbase integration module
"""
from typing import List
from browserbase import Browserbase
def browser_base_fetch(api_key: str, project_id: str, link: List[str]) -> List[str]:
"""
BrowserBase Fetch
This module provides an interface to the BrowserBase API.
The `browser_base_fetch` function takes three arguments:
- `api_key`: The API key provided by BrowserBase.
- `project_id`: The ID of the project on BrowserBase where you want to fetch data from.
- `link`: The URL or link that you want to fetch data from.
It initializes a Browserbase object with the given API key and project ID,
then uses this object to load the specified link.
It returns the result of the loading operation.
Example usage:
```
from browser_base_fetch import browser_base_fetch
result = browser_base_fetch(api_key="your_api_key",
project_id="your_project_id", link="https://example.com")
print(result)
```
Please note that you need to replace "your_api_key" and "your_project_id"
with your actual BrowserBase API key and project ID.
Args:
api_key (str): The API key provided by BrowserBase.
project_id (str): The ID of the project on BrowserBase where you want to fetch data from.
link (str): The URL or link that you want to fetch data from.
Returns:
object: The result of the loading operation.
"""
browserbase = Browserbase(api_key=api_key, project_id=project_id)
result = []
for l in link:
result.append(browserbase.load(l, text_content=True))
return result