fix(code-interpreter): preserve Result.data in JSON serialization

.changeset/fix-result-data-serialization.md

@@ -0,0 +1,6 @@
+---
+"@e2b/code-interpreter": patch
+"@e2b/code-interpreter-python": patch
+---
+
+Fix `Result.toJSON()` and `Execution.to_json()` to preserve the `data` field when present, including empty objects. Previously, the JS SDK omitted `data` entirely from serialization, and the Python SDK dropped empty `data` objects due to a truthiness check.

packages/code-interpreter-js/src/messaging.ts

@@ -259,6 +259,7 @@ export class Result {
       latex: this.latex,
       json: this.json,
       javascript: this.javascript,
+      ...(this.data !== undefined ? { data: this.data } : {}),
       ...(Object.keys(this.extra).length > 0 ? { extra: this.extra } : {}),
     }
   }

packages/code-interpreter-js/tests/messaging.test.ts

@@ -2,6 +2,7 @@ import { expect, test } from 'vitest'
 
 import { SandboxError } from '../src'
 import { extractError } from '../src/messaging'
+import { Result } from '../src/messaging'
 
 test('includes the response body in server errors', async () => {
   const error = await extractError(
@@ -25,3 +26,43 @@ test('falls back to the status text for an empty body', async () => {
 
   expect(error).toEqual(new SandboxError('500 Internal Server Error'))
 })
+
+test('Result.toJSON includes data field when present', () => {
+  const result = new Result(
+    {
+      text: 'table',
+      data: { columns: ['answer'], rows: [[42]] },
+    },
+    true
+  )
+
+  const json = result.toJSON()
+  expect(json).toHaveProperty('data')
+  expect(json.data).toEqual({ columns: ['answer'], rows: [[42]] })
+})
+
+test('Result.toJSON includes empty data field', () => {
+  const result = new Result(
+    {
+      text: 'table',
+      data: {},
+    },
+    true
+  )
+
+  const json = result.toJSON()
+  expect(json).toHaveProperty('data')
+  expect(json.data).toEqual({})
+})
+
+test('Result.toJSON omits data field when undefined', () => {
+  const result = new Result(
+    {
+      text: 'table',
+    },
+    true
+  )
+
+  const json = result.toJSON()
+  expect(json).not.toHaveProperty('data')
+})

packages/code-interpreter-python/e2b_code_interpreter/models.py

@@ -196,7 +196,7 @@ def formats(self) -> Iterable[str]:
             formats.append("json")
         if self.javascript:
             formats.append("javascript")
-        if self.data:
+        if self.data is not None:
             formats.append("data")
         if self.chart:
             formats.append("chart")

packages/code-interpreter-python/tests/charts/test_json.py

@@ -1,6 +1,7 @@
 import json
 
 from e2b_code_interpreter.code_interpreter_async import AsyncSandbox
+from e2b_code_interpreter.models import Result, Execution
 
 code = """
 import matplotlib.pyplot as plt
@@ -25,3 +26,32 @@ async def test_scatter_chart(async_sandbox: AsyncSandbox):
     assert isinstance(serialized, str)
 
     assert json.loads(serialized)["results"][0]["chart"]["type"] == "scatter"
+
+
+def test_result_to_json_includes_data_field():
+    """Test that Result.to_json includes data field when present"""
+    result = Result(text="table", data={"columns": ["answer"], "rows": [[42]]})
+    execution = Execution(results=[result])
+    serialized = execution.to_json()
+    data = json.loads(serialized)
+    assert "data" in data["results"][0]
+    assert data["results"][0]["data"] == {"columns": ["answer"], "rows": [[42]]}
+
+
+def test_result_to_json_includes_empty_data_field():
+    """Test that Result.to_json includes empty data field"""
+    result = Result(text="table", data={})
+    execution = Execution(results=[result])
+    serialized = execution.to_json()
+    data = json.loads(serialized)
+    assert "data" in data["results"][0]
+    assert data["results"][0]["data"] == {}
+
+
+def test_result_to_json_omits_data_field_when_none():
+    """Test that Result.to_json omits data field when None"""
+    result = Result(text="table", data=None)
+    execution = Execution(results=[result])
+    serialized = execution.to_json()
+    data = json.loads(serialized)
+    assert "data" not in data["results"][0]