e2b-code-interpreter 0.0.9a0__tar.gz → 0.0.10__tar.gz

{e2b_code_interpreter-0.0.9a0 → e2b_code_interpreter-0.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: e2b-code-interpreter
-Version: 0.0.9a0
+Version: 0.0.10
 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: e2b (>=0.17.1)
-Requires-Dist: pydantic (>1,<3)
+Requires-Dist: pydantic
 Requires-Dist: websocket-client (>=1.7.0,<2.0.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.9a0 → e2b_code_interpreter-0.0.10}/e2b_code_interpreter/main.py

@@ -23,7 +23,7 @@ class CodeInterpreter(Sandbox):
     E2B code interpreter sandbox extension.
     """
 
-    template = "code-interpreter-multikernel"
+    template = "code-interpreter-stateful"
 
     def __init__(
         self,

{e2b_code_interpreter-0.0.9a0 → e2b_code_interpreter-0.0.10}/e2b_code_interpreter/messaging.py

@@ -161,7 +161,11 @@ class JupyterKernelWebSocket:
 
         :param data: The message data
         """
-        parent_msg_ig = data["parent_header"]["msg_id"]
+        parent_msg_ig = data["parent_header"].get("msg_id", None)
+        if parent_msg_ig is None:
+            logger.warning("Parent message ID not found. %s", data)
+            return
+
         logger.debug(f"Received message {data['msg_type']} for {parent_msg_ig}")
 
         cell = self._cells.get(parent_msg_ig)

{e2b_code_interpreter-0.0.9a0 → e2b_code_interpreter-0.0.10}/e2b_code_interpreter/models.py

@@ -1,6 +1,6 @@
 import copy
 from typing import List, Optional, Iterable, Dict
-from pydantic import BaseModel, field_serializer
+from pydantic import BaseModel
 
 
 class Error(BaseModel):
@@ -211,6 +211,19 @@ class Logs(BaseModel):
     "List of strings printed to stderr by prints, subprocesses, etc."
 
 
+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 Execution(BaseModel):
     """
     Represents the result of a cell execution.
@@ -218,6 +231,9 @@ class Execution(BaseModel):
 
     class Config:
         arbitrary_types_allowed = True
+        json_encoders = {
+            List[Result]: serialize_results,
+        }
 
     results: List[Result] = []
     "List of the result of the cell (interactively interpreted last line), display calls (e.g. matplotlib plots)."
@@ -245,19 +261,6 @@ class Execution(BaseModel):
         """
         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.9a0 → e2b_code_interpreter-0.0.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "e2b-code-interpreter"
-version = "0.0.9a0"
+version = "0.0.10"
 description = "E2B Code Interpreter - Stateful code execution"
 authors = ["e2b <hello@e2b.dev>"]
 license = "Apache-2.0"
@@ -12,7 +12,7 @@ packages = [{ include = "e2b_code_interpreter" }]
 [tool.poetry.dependencies]
 python = "^3.8"
 
-pydantic = ">1, <3"
+pydantic = "*"
 websocket-client = "^1.7.0"
 e2b = ">=0.17.1"