Extract RunCodeLanguage type to deduplicate language unions

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: mishushakov

js/src/index.ts

@@ -1,7 +1,7 @@
 export * from 'e2b'
 
 export { Sandbox } from './sandbox'
-export type { Context, RunCodeOpts, CreateCodeContextOpts } from './sandbox'
+export type { Context, RunCodeLanguage, RunCodeOpts, CreateCodeContextOpts } from './sandbox'
 export type {
   Logs,
   ExecutionError,

js/src/sandbox.ts

@@ -33,6 +33,20 @@ export type Context = {
   cwd: string
 }
 
+/* eslint-disable @typescript-eslint/ban-types */
+/**
+ * Supported language for code execution.
+ */
+export type RunCodeLanguage =
+  | 'python'
+  | 'javascript'
+  | 'typescript'
+  | 'r'
+  | 'java'
+  | 'bash'
+  | (string & {})
+/* eslint-enable @typescript-eslint/ban-types */
+
 /**
  * Options for running code.
  */
@@ -88,16 +102,7 @@ export interface CreateCodeContextOpts {
    *
    * @default python
    */
-  /* eslint-disable @typescript-eslint/ban-types */
-  language?:
-    | 'python'
-    | 'javascript'
-    | 'typescript'
-    | 'r'
-    | 'java'
-    | 'bash'
-    | (string & {})
-  /* eslint-enable @typescript-eslint/ban-types */
+  language?: RunCodeLanguage
   /**
    * Timeout for the request in **milliseconds**.
    *
@@ -158,16 +163,7 @@ export class Sandbox extends BaseSandbox {
        *
        * If not defined, the default Python context is used.
        */
-      /* eslint-disable @typescript-eslint/ban-types */
-      language?:
-        | 'python'
-        | 'javascript'
-        | 'typescript'
-        | 'r'
-        | 'java'
-        | 'bash'
-        | (string & {})
-      /* eslint-enable @typescript-eslint/ban-types */
+      language?: RunCodeLanguage
     }
   ): Promise<Execution>
   /**

python/e2b_code_interpreter/__init__.py

@@ -10,4 +10,5 @@
     Logs,
     OutputHandler,
     OutputMessage,
+    RunCodeLanguage,
 )

python/e2b_code_interpreter/code_interpreter_async.py

@@ -1,7 +1,7 @@
 import logging
 import httpx
 
-from typing import Optional, Dict, overload, Union, Literal, List
+from typing import Optional, Dict, overload, Union, List
 from httpx import AsyncClient
 
 from e2b import (
@@ -18,6 +18,7 @@
     Execution,
     ExecutionError,
     Context,
+    RunCodeLanguage,
     Result,
     aextract_exception,
     OutputHandlerWithAsync,
@@ -68,11 +69,7 @@ def _client(self) -> AsyncClient:
     async def run_code(
         self,
         code: str,
-        language: Union[
-            Literal["python", "javascript", "typescript", "r", "java", "bash"],
-            str,
-            None,
-        ] = None,
+        language: RunCodeLanguage = None,
         on_stdout: Optional[OutputHandlerWithAsync[OutputMessage]] = None,
         on_stderr: Optional[OutputHandlerWithAsync[OutputMessage]] = None,
         on_result: Optional[OutputHandlerWithAsync[Result]] = None,
@@ -206,11 +203,7 @@ async def run_code(
     async def create_code_context(
         self,
         cwd: Optional[str] = None,
-        language: Union[
-            Literal["python", "javascript", "typescript", "r", "java", "bash"],
-            str,
-            None,
-        ] = None,
+        language: RunCodeLanguage = None,
         request_timeout: Optional[float] = None,
     ) -> Context:
         """

python/e2b_code_interpreter/code_interpreter_sync.py

@@ -1,7 +1,7 @@
 import logging
 import httpx
 
-from typing import Optional, Dict, overload, Literal, Union, List
+from typing import Optional, Dict, overload, Union, List
 from httpx import Client
 from e2b import Sandbox as BaseSandbox, InvalidArgumentException
 
@@ -13,6 +13,7 @@
 from e2b_code_interpreter.models import (
     ExecutionError,
     Execution,
+    RunCodeLanguage,
     Context,
     Result,
     extract_exception,
@@ -65,11 +66,7 @@ def _client(self) -> Client:
     def run_code(
         self,
         code: str,
-        language: Union[
-            Literal["python", "javascript", "typescript", "r", "java", "bash"],
-            str,
-            None,
-        ] = None,
+        language: RunCodeLanguage = None,
         on_stdout: Optional[OutputHandler[OutputMessage]] = None,
         on_stderr: Optional[OutputHandler[OutputMessage]] = None,
         on_result: Optional[OutputHandler[Result]] = None,
@@ -202,11 +199,7 @@ def run_code(
     def create_code_context(
         self,
         cwd: Optional[str] = None,
-        language: Union[
-            Literal["python", "javascript", "typescript", "r", "java", "bash"],
-            str,
-            None,
-        ] = None,
+        language: RunCodeLanguage = None,
         request_timeout: Optional[float] = None,
     ) -> Context:
         """

python/e2b_code_interpreter/models.py

@@ -6,6 +6,7 @@
 from dataclasses import dataclass, field
 from typing import (
     List,
+    Literal,
     Optional,
     Iterable,
     Dict,
@@ -20,6 +21,12 @@
 
 from .charts import Chart, _deserialize_chart
 
+RunCodeLanguage = Union[
+    Literal["python", "javascript", "typescript", "r", "java", "bash"],
+    str,
+    None,
+]
+
 T = TypeVar("T")
 OutputHandler = Union[Callable[[T], Any],]