Squashed 'packages/tools/' content from commit 78317b9c

git-subtree-dir: packages/tools
git-subtree-split: 78317b9c127f18bd040c1d77e3c0840cdc9a5b38

crewai_tools/tools/browserbase_load_tool/browserbase_load_tool.py

@@ -0,0 +1,67 @@
+import os
+from typing import Any, Optional, Type, List
+
+from crewai.tools import BaseTool, EnvVar
+from pydantic 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 url in a headless browser using Browserbase and return the contents"
+    args_schema: Type[BaseModel] = BrowserbaseLoadToolSchema
+    api_key: Optional[str] = os.getenv("BROWSERBASE_API_KEY")
+    project_id: Optional[str] = os.getenv("BROWSERBASE_PROJECT_ID")
+    text_content: Optional[bool] = False
+    session_id: Optional[str] = None
+    proxy: Optional[bool] = None
+    browserbase: Optional[Any] = None
+    package_dependencies: List[str] = ["browserbase"]
+    env_vars: List[EnvVar] = [
+        EnvVar(name="BROWSERBASE_API_KEY", description="API key for Browserbase services", required=False),
+        EnvVar(name="BROWSERBASE_PROJECT_ID", description="Project ID for Browserbase services", required=False),
+    ]
+
+    def __init__(
+        self,
+        api_key: Optional[str] = None,
+        project_id: Optional[str] = None,
+        text_content: Optional[bool] = False,
+        session_id: Optional[str] = None,
+        proxy: Optional[bool] = None,
+        **kwargs,
+    ):
+        super().__init__(**kwargs)
+        if not self.api_key:
+            raise EnvironmentError(
+                "BROWSERBASE_API_KEY environment variable is required for initialization"
+            )
+        try:
+            from browserbase import Browserbase  # type: ignore
+        except ImportError:
+            import click
+
+            if click.confirm(
+                "`browserbase` package not found, would you like to install it?"
+            ):
+                import subprocess
+
+                subprocess.run(["uv", "add", "browserbase"], check=True)
+                from browserbase import Browserbase  # type: ignore
+            else:
+                raise ImportError(
+                    "`browserbase` package not found, please run `uv add browserbase`"
+                )
+
+        self.browserbase = Browserbase(api_key=self.api_key)
+        self.text_content = text_content
+        self.session_id = session_id
+        self.proxy = proxy
+
+    def _run(self, url: str):
+        return self.browserbase.load_url(
+            url, self.text_content, self.session_id, self.proxy
+        )