mirror of
https://github.com/harivansh-afk/sandbox-agent.git
synced 2026-04-18 04:04:24 +00:00
63 lines
2.2 KiB
TypeScript
63 lines
2.2 KiB
TypeScript
import Docker from "dockerode";
|
|
import { SandboxAgent } from "sandbox-agent";
|
|
import { detectAgent, buildInspectorUrl, waitForHealth } from "@sandbox-agent/example-shared";
|
|
|
|
const IMAGE = "alpine:latest";
|
|
const PORT = 3000;
|
|
|
|
const docker = new Docker({ socketPath: "/var/run/docker.sock" });
|
|
|
|
// Pull image if needed
|
|
try {
|
|
await docker.getImage(IMAGE).inspect();
|
|
} catch {
|
|
console.log(`Pulling ${IMAGE}...`);
|
|
await new Promise<void>((resolve, reject) => {
|
|
docker.pull(IMAGE, (err: Error | null, stream: NodeJS.ReadableStream) => {
|
|
if (err) return reject(err);
|
|
docker.modem.followProgress(stream, (err: Error | null) => err ? reject(err) : resolve());
|
|
});
|
|
});
|
|
}
|
|
|
|
console.log("Starting container...");
|
|
const container = await docker.createContainer({
|
|
Image: IMAGE,
|
|
Cmd: ["sh", "-c", [
|
|
"apk add --no-cache curl ca-certificates libstdc++ libgcc bash",
|
|
"curl -fsSL https://releases.rivet.dev/sandbox-agent/0.2.x/install.sh | sh",
|
|
"sandbox-agent install-agent claude",
|
|
"sandbox-agent install-agent codex",
|
|
`sandbox-agent server --no-token --host 0.0.0.0 --port ${PORT}`,
|
|
].join(" && ")],
|
|
Env: [
|
|
process.env.ANTHROPIC_API_KEY ? `ANTHROPIC_API_KEY=${process.env.ANTHROPIC_API_KEY}` : "",
|
|
process.env.OPENAI_API_KEY ? `OPENAI_API_KEY=${process.env.OPENAI_API_KEY}` : "",
|
|
].filter(Boolean),
|
|
ExposedPorts: { [`${PORT}/tcp`]: {} },
|
|
HostConfig: {
|
|
AutoRemove: true,
|
|
PortBindings: { [`${PORT}/tcp`]: [{ HostPort: `${PORT}` }] },
|
|
},
|
|
});
|
|
await container.start();
|
|
|
|
const baseUrl = `http://127.0.0.1:${PORT}`;
|
|
await waitForHealth({ baseUrl });
|
|
|
|
const client = await SandboxAgent.connect({ baseUrl });
|
|
const session = await client.createSession({ agent: detectAgent(), sessionInit: { cwd: "/root", mcpServers: [] } });
|
|
const sessionId = session.id;
|
|
|
|
console.log(` UI: ${buildInspectorUrl({ baseUrl, sessionId })}`);
|
|
console.log(" Press Ctrl+C to stop.");
|
|
|
|
const keepAlive = setInterval(() => {}, 60_000);
|
|
const cleanup = async () => {
|
|
clearInterval(keepAlive);
|
|
try { await container.stop({ t: 5 }); } catch {}
|
|
try { await container.remove({ force: true }); } catch {}
|
|
process.exit(0);
|
|
};
|
|
process.once("SIGINT", cleanup);
|
|
process.once("SIGTERM", cleanup);
|