e2b-code-interpreter 0.0.3__tar.gz → 0.0.5__tar.gz

{e2b_code_interpreter-0.0.3 → e2b_code_interpreter-0.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: e2b-code-interpreter
-Version: 0.0.3
+Version: 0.0.5
 Summary: E2B Code Interpreter - Stateful code execution
 Home-page: https://e2b.dev/
 License: Apache-2.0
@@ -14,7 +14,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: e2b (>=0.14.11)
+Requires-Dist: e2b (>=0.17.0)
 Requires-Dist: pydantic (>1,<3)
 Requires-Dist: websocket-client (>=1.7.0,<2.0.0)
 Project-URL: Bug Tracker, https://github.com/e2b-dev/code-interpreter/issues
@@ -112,45 +112,3 @@ with CodeInterpreter() as sandbox:
     sandbox.notebook.exec_cell(code, on_stdout=print, on_stderr=print, on_result=(lambda result: print(result.text)))
 ```
 
-### Pre-installed Python packages inside the sandbox
-
-The full and always up-to-date list can be found in the [`requirements.txt`](https://github.com/e2b-dev/E2B/blob/stateful-code-interpreter/sandboxes/code-interpreter-stateful/requirements.txt) file.
-
-```text
-# Jupyter server requirements
-jupyter-server==2.13.0
-ipykernel==6.29.3
-ipython==8.22.2
-
-# Other packages
-aiohttp==3.9.3
-beautifulsoup4==4.12.3
-bokeh==3.3.4
-gensim==4.3.2
-imageio==2.34.0
-joblib==1.3.2
-librosa==0.10.1
-matplotlib==3.8.3
-nltk==3.8.1
-numpy==1.26.4
-opencv-python==4.9.0.80
-openpyxl==3.1.2
-pandas==1.5.3
-plotly==5.19.0
-pytest==8.1.0
-python-docx==1.1.0
-pytz==2024.1
-requests==2.26.0
-scikit-image==0.22.0
-scikit-learn==1.4.1.post1
-scipy==1.12.0
-seaborn==0.13.2
-soundfile==0.12.1
-spacy==3.7.4
-textblob==0.18.0
-tornado==6.4
-urllib3==1.26.7
-xarray==2024.2.0
-xlrd==2.0.1
-```
-

{e2b_code_interpreter-0.0.3 → e2b_code_interpreter-0.0.5}/README.md

@@ -88,45 +88,3 @@ print("world")
 with CodeInterpreter() as sandbox:
     sandbox.notebook.exec_cell(code, on_stdout=print, on_stderr=print, on_result=(lambda result: print(result.text)))
 ```
-
-### Pre-installed Python packages inside the sandbox
-
-The full and always up-to-date list can be found in the [`requirements.txt`](https://github.com/e2b-dev/E2B/blob/stateful-code-interpreter/sandboxes/code-interpreter-stateful/requirements.txt) file.
-
-```text
-# Jupyter server requirements
-jupyter-server==2.13.0
-ipykernel==6.29.3
-ipython==8.22.2
-
-# Other packages
-aiohttp==3.9.3
-beautifulsoup4==4.12.3
-bokeh==3.3.4
-gensim==4.3.2
-imageio==2.34.0
-joblib==1.3.2
-librosa==0.10.1
-matplotlib==3.8.3
-nltk==3.8.1
-numpy==1.26.4
-opencv-python==4.9.0.80
-openpyxl==3.1.2
-pandas==1.5.3
-plotly==5.19.0
-pytest==8.1.0
-python-docx==1.1.0
-pytz==2024.1
-requests==2.26.0
-scikit-image==0.22.0
-scikit-learn==1.4.1.post1
-scipy==1.12.0
-seaborn==0.13.2
-soundfile==0.12.1
-spacy==3.7.4
-textblob==0.18.0
-tornado==6.4
-urllib3==1.26.7
-xarray==2024.2.0
-xlrd==2.0.1
-```

e2b_code_interpreter-0.0.5/e2b_code_interpreter/__init__.py

@@ -0,0 +1,4 @@
+from e2b import *
+
+from .main import CodeInterpreter, JupyterExtension
+from .models import Execution, Error, Result, KernelException, MIMEType, Logs

{e2b_code_interpreter-0.0.3 → e2b_code_interpreter-0.0.5}/e2b_code_interpreter/main.py

@@ -2,10 +2,11 @@ from __future__ import annotations
 
 import logging
 import threading
+import requests
+
 from concurrent.futures import Future
 from typing import Any, Callable, List, Optional, Dict
 
-import requests
 from e2b import EnvVars, ProcessMessage, Sandbox
 from e2b.constants import TIMEOUT
 

{e2b_code_interpreter-0.0.3 → e2b_code_interpreter-0.0.5}/e2b_code_interpreter/models.py

@@ -1,6 +1,6 @@
 import copy
 from typing import List, Optional, Iterable, Dict
-from pydantic import BaseModel
+from pydantic import BaseModel, field_serializer
 
 
 class Error(BaseModel):
@@ -79,6 +79,12 @@ class Result:
         self.javascript = data.pop("application/javascript", None)
         self.extra = data
 
+    # Allows to iterate over formats()
+    def __getitem__(self, item):
+        if item in self.raw:
+            return self.raw[item]
+        return getattr(self, item)
+
     def formats(self) -> Iterable[str]:
         """
         Returns all available formats of the result.
@@ -228,6 +234,25 @@ class Execution(BaseModel):
             if d.is_main_result:
                 return d.text
 
+    def to_json(self) -> str:
+        """
+        Returns the JSON representation of the Execution object.
+        """
+        return self.model_dump_json(exclude_none=True)
+
+    @field_serializer("results", when_used="json")
+    def serialize_results(results: List[Result]) -> List[Dict[str, str]]:
+        """
+        Serializes the results to JSON.
+        This method is used by the Pydantic JSON encoder.
+        """
+        serialized = []
+        for result in results:
+            serialized_dict = {key: result[key] for key in result.formats()}
+            serialized_dict['text'] = result.text
+            serialized.append(serialized_dict)
+        return serialized
+
 
 class KernelException(Exception):
     """

{e2b_code_interpreter-0.0.3 → e2b_code_interpreter-0.0.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "e2b-code-interpreter"
-version = "0.0.3"
+version = "0.0.5"
 description = "E2B Code Interpreter - Stateful code execution"
 authors = ["e2b <hello@e2b.dev>"]
 license = "Apache-2.0"
@@ -12,9 +12,9 @@ packages = [{ include = "e2b_code_interpreter" }]
 [tool.poetry.dependencies]
 python = "^3.8"
 
-e2b = ">=0.14.11"
 pydantic = ">1, <3"
 websocket-client = "^1.7.0"
+e2b = ">=0.17.0"
 
 [tool.poetry.group.dev.dependencies]
 black = "^24.3.0"
@@ -22,7 +22,6 @@ pytest = "^7.4.0"
 python-dotenv = "^1.0.0"
 pytest-dotenv = "^0.5.2"
 
-
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

e2b_code_interpreter-0.0.3/e2b_code_interpreter/__init__.py

@@ -1,2 +0,0 @@
-from .main import CodeInterpreter, JupyterExtension
-from .models import Execution, Error, Result, KernelException