watasu-code-interpreter 0.1.51__tar.gz → 0.1.52__tar.gz

{watasu_code_interpreter-0.1.51 → watasu_code_interpreter-0.1.52}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: watasu-code-interpreter
-Version: 0.1.51
+Version: 0.1.52
 Summary: Code Interpreter SDK for Watasu
 Project-URL: Repository, https://github.com/watasuio/sdk
 License-Expression: MIT OR Apache-2.0

{watasu_code_interpreter-0.1.51 → watasu_code_interpreter-0.1.52}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "watasu-code-interpreter"
-version = "0.1.51"
+version = "0.1.52"
 description = "Code Interpreter SDK for Watasu"
 readme = "README.md"
 requires-python = ">=3.9"

{watasu_code_interpreter-0.1.51 → watasu_code_interpreter-0.1.52}/src/watasu_code_interpreter/__init__.py

@@ -14,6 +14,7 @@ from .models import (
     OutputHandler,
     OutputMessage,
     Result,
+    RunCodeLanguage,
 )
 
 _code_interpreter_all = [
@@ -32,6 +33,7 @@ _code_interpreter_all = [
     "OutputHandler",
     "OutputMessage",
     "Result",
+    "RunCodeLanguage",
     "Sandbox",
 ]
 

{watasu_code_interpreter-0.1.51 → watasu_code_interpreter-0.1.52}/src/watasu_code_interpreter/main.py

@@ -46,7 +46,7 @@ class Sandbox(BaseSandbox):
                 "code": code,
                 "language": language,
                 "context_id": _context_id(context),
-                "env_vars": envs,
+                "envs": envs,
                 "timeout_seconds": timeout,
             }
         )
@@ -120,12 +120,74 @@ class AsyncSandbox(BaseAsyncSandbox):
     default_template = Sandbox.default_template
 
     @classmethod
-    async def create(cls, *args: Any, **kwargs: Any) -> "AsyncSandbox":
+    async def create(
+        cls,
+        template: Optional[str] = None,
+        timeout: Optional[int] = None,
+        metadata: Optional[Dict[str, str]] = None,
+        envs: Optional[Dict[str, str]] = None,
+        secure: bool = True,
+        allow_internet_access: bool = True,
+        mcp: Optional[Dict[str, Any]] = None,
+        network=None,
+        volume_mounts: Optional[Dict[str, Any]] = None,
+        lifecycle=None,
+        **opts: ApiParams,
+    ) -> "AsyncSandbox":
         """Create a code-interpreter sandbox and return async helpers."""
 
-        return cls(sync_sandbox=await asyncio.to_thread(Sandbox.create, *args, **kwargs))
+        return cls(
+            sync_sandbox=await asyncio.to_thread(
+                Sandbox.create,
+                template=template,
+                timeout=timeout,
+                metadata=metadata,
+                envs=envs,
+                secure=secure,
+                allow_internet_access=allow_internet_access,
+                mcp=mcp,
+                network=network,
+                volume_mounts=volume_mounts,
+                lifecycle=lifecycle,
+                **opts,
+            )
+        )
+
+    @classmethod
+    async def beta_create(
+        cls,
+        template: Optional[str] = None,
+        timeout: Optional[int] = None,
+        auto_pause: bool = False,
+        metadata: Optional[Dict[str, str]] = None,
+        envs: Optional[Dict[str, str]] = None,
+        secure: bool = True,
+        allow_internet_access: bool = True,
+        mcp: Optional[Dict[str, Any]] = None,
+        network=None,
+        volume_mounts: Optional[Dict[str, Any]] = None,
+        lifecycle=None,
+        **opts: ApiParams,
+    ) -> "AsyncSandbox":
+        """Create a code-interpreter sandbox with beta lifecycle options."""
 
-    beta_create = create
+        return cls(
+            sync_sandbox=await asyncio.to_thread(
+                Sandbox.beta_create,
+                template=template,
+                timeout=timeout,
+                auto_pause=auto_pause,
+                metadata=metadata,
+                envs=envs,
+                secure=secure,
+                allow_internet_access=allow_internet_access,
+                mcp=mcp,
+                network=network,
+                volume_mounts=volume_mounts,
+                lifecycle=lifecycle,
+                **opts,
+            )
+        )
 
     async def _connect_instance(
         self, timeout: Optional[int] = None, **opts: ApiParams

{watasu_code_interpreter-0.1.51 → watasu_code_interpreter-0.1.52}/src/watasu_code_interpreter/models.py

@@ -3,12 +3,16 @@ from __future__ import annotations
 import json as jsonlib
 import time
 from dataclasses import dataclass, field
-from typing import Any, Callable, Dict, List, Optional, TypeVar
+from typing import Any, Callable, Dict, List, Literal, Optional, TypeVar, Union
 
 from .charts import ChartTypes, _deserialize_chart
 
 T = TypeVar("T")
 OutputHandler = Callable[[T], Any]
+RunCodeLanguage = Union[
+    Literal["python", "javascript", "typescript", "r", "java", "bash"],
+    str,
+]
 
 
 class MIMEType(str):