@giselles-ai/sandbox-agent 0.1.12 → 0.1.14

package/dist/index.d.ts

@@ -17,6 +17,7 @@ declare class Agent {
         path: string;
         content: Buffer;
     }>): this;
+    setAgentMd(content: string | Buffer): this;
     runCommands(commands: Array<{
         cmd: string;
         args?: string[];
@@ -28,6 +29,7 @@ type BaseChatRequest = {
     message: string;
     session_id?: string;
     sandbox_id?: string;
+    snapshot_id?: string;
 };
 type ChatCommand = {
     cmd: string;
@@ -61,6 +63,7 @@ declare const codexRequestSchema: z.ZodObject<{
     message: z.ZodString;
     session_id: z.ZodOptional<z.ZodString>;
     sandbox_id: z.ZodOptional<z.ZodString>;
+    snapshot_id: z.ZodOptional<z.ZodString>;
     relay_session_id: z.ZodOptional<z.ZodString>;
     relay_token: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
@@ -86,6 +89,7 @@ declare const geminiRequestSchema: z.ZodObject<{
     message: z.ZodString;
     session_id: z.ZodOptional<z.ZodString>;
     sandbox_id: z.ZodOptional<z.ZodString>;
+    snapshot_id: z.ZodOptional<z.ZodString>;
     relay_session_id: z.ZodOptional<z.ZodString>;
     relay_token: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;

package/dist/index.js

@@ -34,6 +34,13 @@ var Agent = class _Agent {
     });
     return this;
   }
+  setAgentMd(content) {
+    const buffer = typeof content === "string" ? Buffer.from(content) : content;
+    console.log(`[agent] setAgentMd called, content length=${buffer.length}`);
+    return this.addFiles([
+      { path: "/home/vercel-sandbox/AGENTS.md", content: buffer }
+    ]);
+  }
   runCommands(commands) {
     for (const command of commands) {
       this._pendingOps.push({
@@ -45,6 +52,9 @@ var Agent = class _Agent {
     return this;
   }
   async prepare() {
+    console.log(
+      `[agent] prepare called, dirty=${this.dirty}, pendingOps=${this._pendingOps.length}`
+    );
     if (!this.dirty) {
       return;
     }
@@ -66,6 +76,7 @@ var Agent = class _Agent {
       }
     }
     const snapshot = await sandbox.snapshot();
+    console.log(`[agent] prepare done, new snapshotId=${snapshot.snapshotId}`);
     this._snapshotId = snapshot.snapshotId;
     this._pendingOps = [];
   }
@@ -180,6 +191,7 @@ var codexRequestSchema = z.object({
   message: z.string().min(1),
   session_id: z.string().min(1).optional(),
   sandbox_id: z.string().min(1).optional(),
+  snapshot_id: z.string().min(1).optional(),
   relay_session_id: z.string().min(1).optional(),
   relay_token: z.string().min(1).optional()
 });
@@ -248,8 +260,6 @@ function createCodexAgent(options = {}) {
   const browserToolRelayUrl = options.tools?.browser?.relayUrl?.trim();
   if (browserToolEnabled) {
     requiredEnv(env, "BROWSER_TOOL_RELAY_URL");
-    requiredEnv(env, "BROWSER_TOOL_RELAY_SESSION_ID");
-    requiredEnv(env, "BROWSER_TOOL_RELAY_TOKEN");
   }
   if (browserToolEnabled && !browserToolRelayUrl) {
     throw new Error("tools.browser.relayUrl is empty.");
@@ -261,9 +271,13 @@ function createCodexAgent(options = {}) {
       if (!browserToolEnabled) {
         return;
       }
-      requiredEnv(env, "VERCEL_OIDC_TOKEN");
       assertBrowserToolRelayCredentials(_input.input);
-      await patchCodexConfigTransportEnv(_input.sandbox, env);
+      const patchEnv = {
+        ...env,
+        BROWSER_TOOL_RELAY_SESSION_ID: _input.input.relay_session_id,
+        BROWSER_TOOL_RELAY_TOKEN: _input.input.relay_token
+      };
+      await patchCodexConfigTransportEnv(_input.sandbox, patchEnv);
     },
     createCommand({ input }) {
       const args = ["exec"];
@@ -290,6 +304,7 @@ var geminiRequestSchema = z2.object({
   message: z2.string().min(1),
   session_id: z2.string().min(1).optional(),
   sandbox_id: z2.string().min(1).optional(),
+  snapshot_id: z2.string().min(1).optional(),
   relay_session_id: z2.string().min(1).optional(),
   relay_token: z2.string().min(1).optional()
 });
@@ -361,8 +376,6 @@ function createGeminiAgent(options = {}) {
   const browserToolRelayUrl = options.tools?.browser?.relayUrl?.trim();
   if (browserToolEnabled) {
     requiredEnv2(env, "BROWSER_TOOL_RELAY_URL");
-    requiredEnv2(env, "BROWSER_TOOL_RELAY_SESSION_ID");
-    requiredEnv2(env, "BROWSER_TOOL_RELAY_TOKEN");
   }
   if (browserToolEnabled && !browserToolRelayUrl) {
     throw new Error("tools.browser.relayUrl is empty.");
@@ -374,9 +387,13 @@ function createGeminiAgent(options = {}) {
       if (!browserToolEnabled) {
         return;
       }
-      requiredEnv2(env, "VERCEL_OIDC_TOKEN");
       assertBrowserToolRelayCredentials2(input);
-      await patchGeminiSettingsTransportEnv(sandbox, env);
+      const patchEnv = {
+        ...env,
+        BROWSER_TOOL_RELAY_SESSION_ID: input.relay_session_id,
+        BROWSER_TOOL_RELAY_TOKEN: input.relay_token
+      };
+      await patchGeminiSettingsTransportEnv(sandbox, patchEnv);
     },
     createCommand({ input }) {
       const args = [
@@ -458,7 +475,7 @@ function runChat(input) {
       void (async () => {
         try {
           const sandbox = parsed.sandbox_id ? await Sandbox2.get({ sandboxId: parsed.sandbox_id }) : await (async () => {
-            const snapshotId = input.agent.snapshotId?.trim();
+            const snapshotId = parsed.snapshot_id?.trim() || input.agent.snapshotId?.trim();
             if (!snapshotId) {
               throw new Error(
                 "Agent must provide snapshotId when sandbox_id is not provided."

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@giselles-ai/sandbox-agent",
-	"version": "0.1.12",
+	"version": "0.1.14",
 	"type": "module",
 	"sideEffects": false,
 	"license": "Apache-2.0",