e2b-claude-toolsets 0.0.1__py3-none-any.whl

e2b_claude_toolsets/__init__.py

@@ -0,0 +1,23 @@
+"""Sync and native asyncio E2B drivers for Anthropic browser and computer toolsets."""
+
+from ._async_browser import AsyncBrowserInitializationError, AsyncE2BBrowserToolset
+from ._async_computer import AsyncE2BComputerToolset
+from ._browser import BrowserInitializationError, E2BBrowserToolset
+from ._computer import E2BComputerToolset
+from ._policy import allow_hosts
+from ._uploads import UploadFile
+from ._viewer import LiveView, ViewerInitializationError, live_view
+
+__all__ = [
+    "E2BBrowserToolset",
+    "E2BComputerToolset",
+    "AsyncE2BBrowserToolset",
+    "AsyncE2BComputerToolset",
+    "AsyncBrowserInitializationError",
+    "BrowserInitializationError",
+    "LiveView",
+    "ViewerInitializationError",
+    "allow_hosts",
+    "live_view",
+    "UploadFile",
+]

e2b_claude_toolsets/_async.py

@@ -0,0 +1,26 @@
+"""Finish bounded resource cleanup even if its caller is cancelled again."""
+
+import asyncio
+from collections.abc import Awaitable
+from typing import TypeVar
+
+T = TypeVar("T")
+
+
+async def finish_cleanup(operation: Awaitable[T]) -> T:
+    task = asyncio.ensure_future(operation)
+    cancelled = None
+    while True:
+        try:
+            result = await asyncio.shield(task)
+            break
+        except asyncio.CancelledError as error:
+            if task.cancelled():
+                raise
+            cancelled = error
+            if task.done():
+                result = task.result()
+                break
+    if cancelled is not None:
+        raise cancelled
+    return result