OpenAI Code Interpreter

npm install @declaw/sdk dotenv openai

import { Sandbox } from "@declaw/sdk";
import OpenAI from "openai";

const client = new OpenAI();

const response = await client.chat.completions.create({
  model: "gpt-4o-mini",
  messages: [
    {
      role: "system",
      content:
        "You are a Python code interpreter. When asked a question, respond ONLY with Python code that computes the answer and prints it. No markdown, no explanation, just code.",
    },
    { role: "user", content: "What are the first 20 prime numbers?" },
  ],
  temperature: 0,
});

function stripCodeFences(code: string): string {
  let cleaned = code.trim();
  if (cleaned.startsWith("```")) {
    cleaned = cleaned.split("\n").slice(1).join("\n");
  }
  if (cleaned.endsWith("```")) {
    cleaned = cleaned.split("\n").slice(0, -1).join("\n");
  }
  return cleaned.trim();
}

const code = stripCodeFences(response.choices[0].message.content || "");

const sbx = await Sandbox.create({ template: "python", timeout: 300 });
try {
  await sbx.files.write("/tmp/solution.py", code);
  const result = await sbx.commands.run("python3 /tmp/solution.py", {
    timeout: 30,
  });
  console.log(result.stdout);
  console.log(result.exitCode);
} finally {
  await sbx.kill();
}

const code = `
def primes(n):
    result = []
    candidate = 2
    while len(result) < n:
        if all(candidate % p != 0 for p in result):
            result.append(candidate)
        candidate += 1
    return result

print("First 20 primes:", primes(20))
print("Factorial of 15:", __import__('math').factorial(15))
`.trim();

const sbx = await Sandbox.create({ template: "python", timeout: 300 });
try {
  await sbx.files.write("/tmp/demo.py", code);
  const result = await sbx.commands.run("python3 /tmp/demo.py", { timeout: 30 });
  console.log(result.stdout);
} finally {
  await sbx.kill();
}