e2b-code-interpreter 0.0.11b2__tar.gz → 0.0.11b4__tar.gz

{e2b_code_interpreter-0.0.11b2 → e2b_code_interpreter-0.0.11b4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: e2b-code-interpreter
-Version: 0.0.11b2
+Version: 0.0.11b4
 Summary: E2B Code Interpreter - Stateful code execution
 Home-page: https://e2b.dev/
 License: Apache-2.0
@@ -15,7 +15,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: attrs (>=21.3.0)
-Requires-Dist: e2b (==0.17.2a37)
+Requires-Dist: e2b (>=0.17.2a50,<0.18.0)
 Requires-Dist: httpx (>=0.20.0,<0.28.0)
 Project-URL: Bug Tracker, https://github.com/e2b-dev/code-interpreter/issues
 Project-URL: Repository, https://github.com/e2b-dev/e2b-code-interpreter/tree/python

{e2b_code_interpreter-0.0.11b2 → e2b_code_interpreter-0.0.11b4}/e2b_code_interpreter/code_interpreter_async.py

@@ -1,7 +1,7 @@
 import logging
 import httpx
 
-from typing import Optional, List
+from typing import Optional, List, Dict
 from httpx import AsyncHTTPTransport, AsyncClient
 
 from e2b import AsyncSandbox, ConnectionConfig
@@ -52,6 +52,7 @@ class JupyterExtension:
         on_stdout: Optional[OutputHandler[OutputMessage]] = None,
         on_stderr: Optional[OutputHandler[OutputMessage]] = None,
         on_result: Optional[OutputHandler[Result]] = None,
+        envs: Optional[Dict[str, str]] = None,
         timeout: Optional[float] = None,
         request_timeout: Optional[float] = None,
     ) -> Execution:
@@ -67,6 +68,7 @@ class JupyterExtension:
                 json={
                     "code": code,
                     "context_id": kernel_id,
+                    "env_vars": envs,
                 },
                 timeout=(request_timeout, timeout, request_timeout, request_timeout),
             ) as response:
@@ -97,14 +99,17 @@ class JupyterExtension:
         cwd: Optional[str] = None,
         kernel_name: Optional[str] = None,
         request_timeout: Optional[float] = None,
+        envs: Optional[Dict[str, str]] = None,
     ) -> str:
-        logger.debug(f"Creating new kernel: {kernel_name}")
+        logger.debug(f"Creating new kernel {kernel_name}")
 
         data = {}
         if kernel_name:
             data["name"] = kernel_name
         if cwd:
             data["cwd"] = cwd
+        if envs:
+            data["env_vars"] = envs
 
         try:
             response = await self._client.post(
@@ -127,9 +132,10 @@ class JupyterExtension:
         kernel_id: Optional[str] = None,
         request_timeout: Optional[float] = None,
     ) -> None:
+        kernel_id = kernel_id or DEFAULT_KERNEL_ID
+
         logger.debug(f"Shutting down a kernel with id {kernel_id}")
 
-        kernel_id = kernel_id or DEFAULT_KERNEL_ID
         try:
             response = await self._client.delete(
                 url=f"{self._url}/contexts/{kernel_id}",
@@ -147,10 +153,10 @@ class JupyterExtension:
         kernel_id: Optional[str] = None,
         request_timeout: Optional[float] = None,
     ) -> None:
-        logger.debug(f"Restarting kernel: {kernel_id}")
-
         kernel_id = kernel_id or DEFAULT_KERNEL_ID
 
+        logger.debug(f"Restarting kernel {kernel_id}")
+
         try:
             response = await self._client.post(
                 f"{self._url}/contexts/{kernel_id}/restart",

{e2b_code_interpreter-0.0.11b2 → e2b_code_interpreter-0.0.11b4}/e2b_code_interpreter/code_interpreter_sync.py

@@ -51,6 +51,7 @@ class JupyterExtension:
         on_stdout: Optional[OutputHandler[OutputMessage]] = None,
         on_stderr: Optional[OutputHandler[OutputMessage]] = None,
         on_result: Optional[OutputHandler[Result]] = None,
+        envs: Optional[Dict[str, str]] = None,
         timeout: Optional[float] = None,
         request_timeout: Optional[float] = None,
     ) -> Execution:
@@ -66,6 +67,7 @@ class JupyterExtension:
                 json={
                     "code": code,
                     "context_id": kernel_id,
+                    "env_vars": envs,
                 },
                 timeout=(request_timeout, timeout, request_timeout, request_timeout),
             ) as response:
@@ -96,7 +98,7 @@ class JupyterExtension:
         kernel_name: Optional[str] = None,
         request_timeout: Optional[float] = None,
     ) -> str:
-        logger.debug(f"Creating new kernel: {kernel_name}")
+        logger.debug(f"Creating new kernel {kernel_name}")
 
         data = {}
         if kernel_name:
@@ -125,10 +127,10 @@ class JupyterExtension:
         kernel_id: Optional[str] = None,
         request_timeout: Optional[float] = None,
     ) -> None:
-        logger.debug(f"Shutting down a kernel with id {kernel_id}")
-
         kernel_id = kernel_id or DEFAULT_KERNEL_ID
 
+        logger.debug(f"Shutting down a kernel with id {kernel_id}")
+
         try:
             response = self._client.delete(
                 url=f"{self._url}/contexts/{kernel_id}",
@@ -145,9 +147,10 @@ class JupyterExtension:
         kernel_id: Optional[str] = None,
         request_timeout: Optional[float] = None,
     ) -> None:
-        logger.debug(f"Creating new kernel for language: {kernel_id}")
-
         kernel_id = kernel_id or DEFAULT_KERNEL_ID
+
+        logger.debug(f"Restarting kernel {kernel_id}")
+
         try:
             response = self._client.post(
                 f"{self._url}/contexts/{kernel_id}/restart",
@@ -164,6 +167,8 @@ class JupyterExtension:
         self,
         request_timeout: Optional[float] = None,
     ) -> List[Kernel]:
+        logger.debug("Listing kernels")
+
         try:
             response = self._client.get(
                 f"{self._url}/contexts",
@@ -192,6 +197,7 @@ class CodeInterpreter(Sandbox):
         template: Optional[str] = None,
         timeout: Optional[int] = None,
         metadata: Optional[Dict[str, str]] = None,
+        envs: Optional[Dict[str, str]] = None,
         api_key: Optional[str] = None,
         domain: Optional[str] = None,
         debug: Optional[bool] = None,
@@ -202,6 +208,7 @@ class CodeInterpreter(Sandbox):
             template=template,
             timeout=timeout,
             metadata=metadata,
+            envs=envs,
             api_key=api_key,
             domain=domain,
             debug=debug,

{e2b_code_interpreter-0.0.11b2 → e2b_code_interpreter-0.0.11b4}/e2b_code_interpreter/models.py

@@ -136,10 +136,13 @@ class Result:
 
         :return: The text representation of the data.
         """
-        return self.text
+        return self.__repr__()
 
     def __repr__(self) -> str:
-        return f"Result({self.text})"
+        if self.text:
+            return f"Result({self.text})"
+        else:
+            return "Result(Formats: " + ", ".join(self.formats()) + ")"
 
     def _repr_html_(self) -> Optional[str]:
         """

{e2b_code_interpreter-0.0.11b2 → e2b_code_interpreter-0.0.11b4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "e2b-code-interpreter"
-version = "0.0.11b2"
+version = "0.0.11b4"
 description = "E2B Code Interpreter - Stateful code execution"
 authors = ["e2b <hello@e2b.dev>"]
 license = "Apache-2.0"
@@ -14,7 +14,7 @@ python = "^3.8"
 
 httpx = ">=0.20.0,<0.28.0"
 attrs = ">=21.3.0"
-e2b = "0.17.2a37"
+e2b = "^0.17.2a50"
 
 [tool.poetry.group.dev.dependencies]
 black = "^24.3.0"