Execution

class Execution {
  public results: Result[] = [];
  public logs = {
    stdout: [] as string[],
    stderr: [] as string[]
  };
  public error?: ExecutionError;
  public executionCount?: number;
  
  constructor(
    public readonly code: string,
    public readonly context: CodeContext
  );
  
  toJSON(): ExecutionResult;
}

interface ExecutionResult {
  code: string;
  logs: {
    stdout: string[];
    stderr: string[];
  };
  error?: ExecutionError;
  executionCount?: number;
  results: Array<{
    text?: string;
    html?: string;
    png?: string;
    jpeg?: string;
    svg?: string;
    latex?: string;
    markdown?: string;
    json?: any;
    chart?: ChartData;
    data?: any;
  }>;
}

import { getSandbox } from '@cloudflare/sandbox';

const sandbox = getSandbox(env.SANDBOX, 'my-sandbox');

const execution = await sandbox.runCode(`
import matplotlib.pyplot as plt
import numpy as np

print("Generating sine wave...")

x = np.linspace(0, 10, 100)
y = np.sin(x)

plt.plot(x, y)
plt.title('Sine Wave')
plt.show()
`);

// Access outputs
console.log('Code:', execution.code);
console.log('Context:', execution.context.id);
console.log('Stdout:', execution.logs.stdout); // ["Generating sine wave..."]
console.log('Results:', execution.results.length); // 1

// Access chart
if (execution.results[0]?.chart) {
  const chart = execution.results[0].chart;
  console.log('Chart type:', chart.type); // "line"
  console.log('Chart library:', chart.library); // "matplotlib"
  
  // Get image
  const image = execution.results[0].png;
  console.log('Has image:', !!image);
}

// Check for errors
if (execution.error) {
  console.error('Error:', execution.error.name);
  console.error('Message:', execution.error.message);
  console.error('Traceback:', execution.error.traceback.join('\n'));
}

const execution = await sandbox.runCode(`
import pandas as pd
import matplotlib.pyplot as plt

# Create and display a dataframe
df = pd.DataFrame({
  'name': ['Alice', 'Bob', 'Charlie'],
  'age': [25, 30, 35]
})
print(df)

# Create a chart
plt.bar(df['name'], df['age'])
plt.show()
`);

// First result is the dataframe (HTML table)
if (execution.results[0]?.html) {
  console.log('Dataframe HTML:', execution.results[0].html);
}

// Second result is the chart
if (execution.results[1]?.chart) {
  console.log('Chart:', execution.results[1].chart.type);
}

const execution = await sandbox.runCode(`
x = 1 / 0  # ZeroDivisionError
`);

if (execution.error) {
  console.error('Execution failed');
  console.error('Error type:', execution.error.name); // "ZeroDivisionError"
  console.error('Message:', execution.error.message);
  
  // Format traceback for display
  const traceback = execution.error.traceback.join('\n');
  console.error('Traceback:\n', traceback);
}

const execution = await sandbox.runCode('print("Hello")');

// Convert to plain object for storage/transmission
const json = execution.toJSON();

// Store in KV or database
await env.KV.put('execution:latest', JSON.stringify(json));

// Or return to client
return Response.json(json);