@giselles-ai/agent 0.1.23 → 0.1.24

package/dist/server/index.d.ts

@@ -3,30 +3,6 @@ import { z } from 'zod';
 import { RelayRequest, RelayResponse } from '@giselles-ai/browser-tool';
 import { RelayRequestSubscription } from '@giselles-ai/browser-tool/relay';
 
-type AgentType$1 = "codex" | "gemini";
-declare class Agent {
-    private _type;
-    private _snapshotId;
-    private _pendingOps;
-    private constructor();
-    static create(type: AgentType$1, options: {
-        snapshotId: string;
-    }): Agent;
-    get type(): AgentType$1;
-    get snapshotId(): string;
-    get dirty(): boolean;
-    addFiles(files: Array<{
-        path: string;
-        content: Buffer;
-    }>): this;
-    setAgentMd(content: string | Buffer): this;
-    runCommands(commands: Array<{
-        cmd: string;
-        args?: string[];
-    }>): this;
-    prepare(): Promise<void>;
-}
-
 type BaseChatRequest = {
     message: string;
     session_id?: string;
@@ -143,6 +119,7 @@ type CloudChatSessionState = {
     chatId: string;
     agentSessionId?: string;
     sandboxId?: string;
+    snapshotId?: string;
     relay?: CloudRelaySession;
     pendingTool?: PendingToolState | null;
     updatedAt: number;
@@ -150,6 +127,7 @@ type CloudChatSessionState = {
 type CloudChatSessionPatch = {
     agentSessionId?: string;
     sandboxId?: string;
+    snapshotId?: string;
     relay?: CloudRelaySession;
     pendingTool?: PendingToolState | null;
 };
@@ -251,4 +229,4 @@ declare function runCloudChat<TRequest extends CloudChatRequest & {
     }) => Promise<void>;
 }): Promise<Response>;
 
-export { Agent, type AgentApiOptions, type AgentApiStoreConfig, type AgentParam, type AgentRequest, type AgentType, type BaseChatRequest, type ChatAgent, type ChatCommand, type CloudChatRequest, type CloudChatRunRequest, type CloudChatSessionPatch, type CloudChatSessionState, type CloudChatStateStore, type CloudRelaySession, type CloudToolName, type CloudToolResult, type CreateAgentOptions, type PendingToolState, type RelaySessionFactoryResult, type RunChatImpl, type RunChatInput, type StdoutMapper, applyCloudChatPatch, cloudChatRunRequestSchema, createAgent, createAgentApi, createCodexAgent, createCodexStdoutMapper, createGeminiAgent, reduceCloudChatEvent, runChat, runCloudChat, toolNameFromRelayRequest };
+export { type AgentApiOptions, type AgentApiStoreConfig, type AgentParam, type AgentRequest, type AgentType, type BaseChatRequest, type ChatAgent, type ChatCommand, type CloudChatRequest, type CloudChatRunRequest, type CloudChatSessionPatch, type CloudChatSessionState, type CloudChatStateStore, type CloudRelaySession, type CloudToolName, type CloudToolResult, type CreateAgentOptions, type PendingToolState, type RelaySessionFactoryResult, type RunChatImpl, type RunChatInput, type StdoutMapper, applyCloudChatPatch, cloudChatRunRequestSchema, createAgent, createAgentApi, createCodexAgent, createCodexStdoutMapper, createGeminiAgent, reduceCloudChatEvent, runChat, runCloudChat, toolNameFromRelayRequest };

package/dist/server/index.js

@@ -1,87 +1,3 @@
-// src/agent.ts
-import { Sandbox } from "@vercel/sandbox";
-var Agent = class _Agent {
-  _type;
-  _snapshotId;
-  _pendingOps = [];
-  constructor(type, snapshotId) {
-    this._type = type;
-    this._snapshotId = snapshotId;
-  }
-  static create(type, options) {
-    const trimmed = options.snapshotId.trim();
-    if (!trimmed) {
-      throw new Error("snapshotId is required.");
-    }
-    return new _Agent(type, trimmed);
-  }
-  get type() {
-    return this._type;
-  }
-  get snapshotId() {
-    return this._snapshotId;
-  }
-  get dirty() {
-    return this._pendingOps.length > 0;
-  }
-  addFiles(files) {
-    if (files.length === 0) {
-      return this;
-    }
-    this._pendingOps.push({
-      kind: "writeFiles",
-      files
-    });
-    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/.codex/AGENTS.md", content: buffer }
-    ]);
-  }
-  runCommands(commands) {
-    for (const command of commands) {
-      this._pendingOps.push({
-        kind: "runCommand",
-        cmd: command.cmd,
-        args: command.args ?? []
-      });
-    }
-    return this;
-  }
-  async prepare() {
-    console.log(
-      `[agent] prepare called, dirty=${this.dirty}, pendingOps=${this._pendingOps.length}`
-    );
-    if (!this.dirty) {
-      return;
-    }
-    const ops = this._pendingOps;
-    const sandbox = await Sandbox.create({
-      source: { type: "snapshot", snapshotId: this._snapshotId }
-    });
-    console.log(
-      `[sandbox] created sandbox=${sandbox.sandboxId} from snapshot=${this._snapshotId}`
-    );
-    for (const op of ops) {
-      switch (op.kind) {
-        case "writeFiles":
-          await sandbox.writeFiles(op.files);
-          break;
-        case "runCommand":
-          await sandbox.runCommand(op.cmd, op.args);
-          break;
-      }
-    }
-    const snapshot = await sandbox.snapshot();
-    console.log(`[agent] prepare done, new snapshotId=${snapshot.snapshotId}`);
-    this._snapshotId = snapshot.snapshotId;
-    this._pendingOps = [];
-  }
-};
-
 // src/agent-api.ts
 import {
   createRelayHandler,
@@ -89,7 +5,7 @@ import {
 } from "@giselles-ai/browser-tool/relay";
 
 // src/build.ts
-import { Sandbox as Sandbox2 } from "@vercel/sandbox";
+import { Sandbox } from "@vercel/sandbox";
 var CACHE_KEY_PREFIX = "agent-build:snapshot:";
 var CACHE_TTL_SEC = 60 * 60 * 24;
 var memoryCache = /* @__PURE__ */ new Map();
@@ -193,7 +109,7 @@ async function buildAgent(input) {
     const response2 = { snapshot_id: cached, cached: true };
     return Response.json(response2);
   }
-  const sandbox = await Sandbox2.create({
+  const sandbox = await Sandbox.create({
     source: { type: "snapshot", snapshotId: baseSnapshotId }
   });
   console.log(
@@ -633,7 +549,7 @@ function createAgent(options) {
 
 // src/chat-run.ts
 import { Writable } from "stream";
-import { Sandbox as Sandbox3 } from "@vercel/sandbox";
+import { Sandbox as Sandbox2 } from "@vercel/sandbox";
 function emitText(controller, text, encoder) {
   if (text.length === 0) {
     return;
@@ -687,23 +603,39 @@ function runChat(input) {
       const mapper = input.agent.createStdoutMapper?.();
       void (async () => {
         try {
-          const sandbox = parsed.sandbox_id ? await Sandbox3.get({ sandboxId: parsed.sandbox_id }) : await (async () => {
-            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."
-              );
-            }
-            const created = await Sandbox3.create({
+          const createFromSnapshot = async (snapshotId2) => {
+            const created = await Sandbox2.create({
               source: {
                 type: "snapshot",
-                snapshotId
+                snapshotId: snapshotId2
               }
             });
             console.log(
-              `[sandbox] created sandbox=${created.sandboxId} from snapshot=${snapshotId}`
+              `[sandbox] created sandbox=${created.sandboxId} from snapshot=${snapshotId2}`
             );
             return created;
+          };
+          const snapshotId = parsed.snapshot_id?.trim() || input.agent.snapshotId?.trim();
+          const sandbox = await (async () => {
+            if (parsed.sandbox_id) {
+              try {
+                return await Sandbox2.get({ sandboxId: parsed.sandbox_id });
+              } catch (error) {
+                if (!snapshotId) {
+                  throw error;
+                }
+                console.log(
+                  `[sandbox] sandbox=${parsed.sandbox_id} expired, recreating from snapshot=${snapshotId}`
+                );
+                return createFromSnapshot(snapshotId);
+              }
+            }
+            if (!snapshotId) {
+              throw new Error(
+                "Agent must provide snapshotId when sandbox_id is not provided."
+              );
+            }
+            return createFromSnapshot(snapshotId);
           })();
           enqueueEvent({ type: "sandbox", sandbox_id: sandbox.sandboxId });
           await input.agent.prepareSandbox({
@@ -740,6 +672,11 @@ function runChat(input) {
             }),
             signal: abortController.signal
           });
+          const snapshot = await sandbox.snapshot();
+          enqueueEvent({
+            type: "snapshot",
+            snapshot_id: snapshot.snapshotId
+          });
         } catch (error) {
           if (abortController.signal.aborted) {
             return;
@@ -855,6 +792,9 @@ function reduceCloudChatEvent(event) {
   if (event.type === "sandbox" && typeof event.sandbox_id === "string") {
     return { sandboxId: event.sandbox_id };
   }
+  if (event.type === "snapshot" && typeof event.snapshot_id === "string") {
+    return { snapshotId: event.snapshot_id };
+  }
   if (event.type === "relay.session" && typeof event.sessionId === "string" && typeof event.token === "string" && typeof event.relayUrl === "string" && typeof event.expiresAt === "number") {
     return {
       relay: {
@@ -895,6 +835,7 @@ function applyCloudChatPatch(input) {
     chatId: input.chatId,
     agentSessionId: input.patch?.agentSessionId ?? input.base?.agentSessionId,
     sandboxId: input.patch?.sandboxId ?? input.base?.sandboxId,
+    snapshotId: input.patch?.snapshotId ?? input.base?.snapshotId,
     relay: input.patch?.relay ?? input.base?.relay,
     pendingTool: input.patch?.pendingTool !== void 0 ? input.patch.pendingTool : input.base?.pendingTool,
     updatedAt: input.now
@@ -1016,6 +957,7 @@ async function runCloudChat(input) {
     ...input.request,
     ...existing?.agentSessionId ? { session_id: existing.agentSessionId } : {},
     ...existing?.sandboxId ? { sandbox_id: existing.sandboxId } : {},
+    ...existing?.snapshotId ? { snapshot_id: existing.snapshotId } : {},
     relay_session_id: relaySession.sessionId,
     relay_token: relaySession.token
   };
@@ -1116,6 +1058,7 @@ async function resumeCloudChat(input) {
     ...input.request,
     ...input.existing.agentSessionId ? { session_id: input.existing.agentSessionId } : {},
     ...input.existing.sandboxId ? { sandbox_id: input.existing.sandboxId } : {},
+    ...input.existing.snapshotId ? { snapshot_id: input.existing.snapshotId } : {},
     relay_session_id: relaySession.sessionId,
     relay_token: relaySession.token
   };
@@ -1584,7 +1527,6 @@ function createAgentApi(options) {
   };
 }
 export {
-  Agent,
   applyCloudChatPatch,
   cloudChatRunRequestSchema,
   createAgent,

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@giselles-ai/agent",
-	"version": "0.1.23",
+	"version": "0.1.24",
 	"type": "module",
 	"sideEffects": false,
 	"license": "Apache-2.0",
@@ -37,7 +37,7 @@
 		"format": "pnpm exec biome check --write ."
 	},
 	"dependencies": {
-		"@giselles-ai/browser-tool": "0.1.23",
+		"@giselles-ai/browser-tool": "0.1.24",
 		"@vercel/sandbox": "1.6.0",
 		"@iarna/toml": "3.0.0",
 		"zod": "4.3.6"