fixes

Author: maticzav

examples/async_retrieve.py

@@ -1,14 +1,14 @@
-#!/usr/bin/env -S rye run python
+#!/usr/bin/env -S poetry run python
 
 import asyncio
-import os
 from typing import List
 
 from pydantic import BaseModel
+from utils import API_KEY
 
 from browser_use import AsyncBrowserUse
 
-client = AsyncBrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
+client = AsyncBrowserUse(api_key=API_KEY)
 
 
 # Regular Task

examples/async_run.py

@@ -1,15 +1,14 @@
-#!/usr/bin/env -S rye run python
+#!/usr/bin/env -S poetry run python
 
 import asyncio
-import os
 from typing import List
 
 from pydantic import BaseModel
+from utils import API_KEY
 
 from browser_use import AsyncBrowserUse
 
-# gets API Key from environment variable BROWSER_USE_API_KEY
-client = AsyncBrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
+client = AsyncBrowserUse(api_key=API_KEY)
 
 
 # Regular Task

examples/async_stream.py

@@ -1,14 +1,14 @@
-#!/usr/bin/env -S rye run python
+#!/usr/bin/env -S poetry run python
 
 import asyncio
-import os
 from typing import List
 
 from pydantic import BaseModel
+from utils import API_KEY
 
 from browser_use import AsyncBrowserUse
 
-client = AsyncBrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
+client = AsyncBrowserUse(api_key=API_KEY)
 
 
 # Regular Task

examples/retrieve.py

@@ -1,14 +1,14 @@
-#!/usr/bin/env -S rye run python
+#!/usr/bin/env -S poetry run python
 
-import os
 import time
 from typing import List
 
 from pydantic import BaseModel
+from utils import API_KEY
 
 from browser_use import BrowserUse
 
-client = BrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
+client = BrowserUse(api_key=API_KEY)
 
 
 # Regular Task

examples/run.py

@@ -1,13 +1,13 @@
-#!/usr/bin/env -S rye run python
+#!/usr/bin/env -S poetry run python
 
-import os
 from typing import List
 
 from pydantic import BaseModel
+from utils import API_KEY
 
 from browser_use import BrowserUse
 
-client = BrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
+client = BrowserUse(api_key=API_KEY)
 
 
 # Regular Task

examples/stream.py

@@ -1,13 +1,13 @@
-#!/usr/bin/env -S rye run python
+#!/usr/bin/env -S poetry run python
 
-import os
 from typing import List
 
 from pydantic import BaseModel
+from utils import API_KEY
 
 from browser_use import BrowserUse
 
-client = BrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
+client = BrowserUse(api_key=API_KEY)
 
 
 # Regular Task

examples/utils.py

@@ -0,0 +1,7 @@
+import os
+
+_API_KEY = os.getenv("BROWSER_USE_API_KEY")
+if not _API_KEY:
+    raise RuntimeError("BROWSER_USE_API_KEY environment variable is not set")
+
+API_KEY = _API_KEY

examples/webhooks.py

@@ -1,4 +1,4 @@
-#!/usr/bin/env -S rye run python
+#!/usr/bin/env -S poetry run python
 
 from datetime import datetime
 from typing import Any, Dict, Tuple