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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: e2b-code-interpreter
-Version: 0.0.4
+Version: 0.0.5
 Summary: E2B Code Interpreter - Stateful code execution
 Home-page: https://e2b.dev/
 License: Apache-2.0

{e2b_code_interpreter-0.0.4 → 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):
@@ -234,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.4 → e2b_code_interpreter-0.0.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "e2b-code-interpreter"
-version = "0.0.4"
+version = "0.0.5"
 description = "E2B Code Interpreter - Stateful code execution"
 authors = ["e2b <hello@e2b.dev>"]
 license = "Apache-2.0"