@giselles-ai/agent 0.1.23 → 0.1.25

package/dist/{chunk-TAQV6ZOB.js → chunk-JGH5DCA7.js}

@@ -7,7 +7,13 @@ function computeConfigHash(config) {
     files: (config.files ?? []).map((f) => ({
       path: f.path,
       content: f.content
-    }))
+    })),
+    setup: config.setup?.script ?? null,
+    env: config.env ? Object.fromEntries(
+      Object.entries(config.env).sort(
+        ([a], [b]) => a < b ? -1 : a > b ? 1 : 0
+      )
+    ) : null
   });
   return createHash("sha256").update(payload).digest("hex").slice(0, 16);
 }
@@ -50,7 +56,9 @@ async function requestBuild(agent, options) {
   const requestBody = {
     config_hash: configHash,
     agent_type: agent.agentType ?? "gemini",
-    files
+    files,
+    setup_script: agent.setup?.script ?? null,
+    env: agent.env ?? null
   };
   const requestHeaders = {
     "content-type": "application/json",

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { A as AgentConfig, D as DefinedAgent } from './types-Cto-uejG.js';
-export { a as AgentFile } from './types-Cto-uejG.js';
+import { A as AgentConfig, D as DefinedAgent } from './types-cRejduhB.js';
+export { a as AgentFile, b as AgentSetup } from './types-cRejduhB.js';
 
 declare function defineAgent(config: AgentConfig): DefinedAgent;
 

package/dist/index.js

@@ -1,7 +1,7 @@
 import {
   computeConfigHash,
   requestBuild
-} from "./chunk-TAQV6ZOB.js";
+} from "./chunk-JGH5DCA7.js";
 
 // src/define-agent.ts
 function defineAgent(config) {
@@ -9,6 +9,8 @@ function defineAgent(config) {
     agentType: config.agentType ?? "gemini",
     agentMd: config.agentMd,
     files: config.files ?? [],
+    env: config.env ?? {},
+    setup: config.setup,
     get snapshotId() {
       const id = process.env?.GISELLE_AGENT_SNAPSHOT_ID;
       if (!id) {

package/dist/next/index.d.ts

@@ -1,5 +1,5 @@
 import { NextConfig } from 'next';
-import { A as AgentConfig } from '../types-Cto-uejG.js';
+import { A as AgentConfig } from '../types-cRejduhB.js';
 
 type GiselleAgentPluginOptions = {
     /** Base URL for the agent API. Default: process.env.GISELLE_AGENT_BASE_URL ?? "https://studio.giselles.ai/agent-api" */

package/dist/next/index.js

@@ -1,9 +1,9 @@
 import {
+  computeConfigHash,
   requestBuild
-} from "../chunk-TAQV6ZOB.js";
+} from "../chunk-JGH5DCA7.js";
 
 // src/next/with-giselle-agent.ts
-import crypto from "crypto";
 import fs from "fs";
 import { createRequire } from "module";
 import path from "path";
@@ -18,12 +18,7 @@ function resolveBaseUrl(options) {
   return (options?.baseUrl ?? process.env.GISELLE_AGENT_BASE_URL ?? "https://studio.giselles.ai/agent-api").replace(/\/+$/, "");
 }
 function getSnapshotFile(agent) {
-  const key = {
-    agentType: agent.agentType,
-    agentMd: agent.agentMd,
-    files: agent.files
-  };
-  const hash = crypto.createHash("sha256").update(JSON.stringify(key)).digest("hex").slice(0, 16);
+  const hash = computeConfigHash(agent);
   return path.join(process.cwd(), ".next", "giselle", hash);
 }
 function hasRunRecently() {

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;
 };
@@ -169,6 +147,7 @@ declare const cloudChatRunRequestSchema: z.ZodObject<{
         gemini: "gemini";
     }>;
     snapshot_id: z.ZodString;
+    env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
     tool_results: z.ZodOptional<z.ZodArray<z.ZodObject<{
         toolCallId: z.ZodString;
         toolName: z.ZodEnum<{
@@ -251,4 +230,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();
@@ -135,6 +51,8 @@ function parseBuildRequest(body) {
   const configHash = record.config_hash;
   const agentType = record.agent_type;
   const files = record.files;
+  const setupScript = record.setup_script;
+  const rawEnv = record.env;
   if (typeof configHash !== "string" || !configHash.trim()) {
     return null;
   }
@@ -145,6 +63,8 @@ function parseBuildRequest(body) {
     return null;
   }
   const parsedFiles = [];
+  let parsedSetupScript = null;
+  let parsedEnv = null;
   for (const file of files) {
     if (!file || typeof file !== "object" || Array.isArray(file)) {
       return null;
@@ -158,10 +78,30 @@ function parseBuildRequest(body) {
       content: recordFile.content
     });
   }
+  if (setupScript !== void 0 && setupScript !== null) {
+    if (typeof setupScript !== "string") {
+      return null;
+    }
+    parsedSetupScript = setupScript;
+  }
+  if (rawEnv !== void 0 && rawEnv !== null) {
+    if (typeof rawEnv !== "object" || Array.isArray(rawEnv)) {
+      return null;
+    }
+    const envRecord = rawEnv;
+    for (const value of Object.values(envRecord)) {
+      if (typeof value !== "string") {
+        return null;
+      }
+    }
+    parsedEnv = envRecord;
+  }
   return {
     config_hash: configHash.trim(),
     agent_type: agentType,
-    files: parsedFiles
+    files: parsedFiles,
+    setup_script: parsedSetupScript,
+    env: parsedEnv
   };
 }
 async function buildAgent(input) {
@@ -193,7 +133,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(
@@ -207,6 +147,21 @@ async function buildAgent(input) {
       }))
     );
   }
+  if (parsed.setup_script) {
+    console.log("[agent-build] running setup script...");
+    const result = await sandbox.runCommand({
+      cmd: "bash",
+      args: ["-lc", parsed.setup_script],
+      ...parsed.env ? { env: parsed.env } : {}
+    });
+    if (result.exitCode !== 0) {
+      const stderr = typeof result.stderr === "string" ? result.stderr : "";
+      throw new Error(
+        `Setup script failed (exit ${result.exitCode}): ${stderr}`
+      );
+    }
+    console.log("[agent-build] setup script completed");
+  }
   const snapshot = await sandbox.snapshot();
   console.log(`[agent-build] snapshot created: ${snapshot.snapshotId}`);
   await cache.set(cacheKey, snapshot.snapshotId);
@@ -443,7 +398,12 @@ function createCodexAgent(options = {}) {
       return {
         cmd: "codex",
         args,
-        env: { CODEX_API_KEY: apiKey }
+        env: Object.fromEntries(
+          Object.entries({
+            ...env,
+            CODEX_API_KEY: apiKey
+          }).filter((e) => e[1] != null)
+        )
       };
     },
     createStdoutMapper() {
@@ -585,9 +545,12 @@ function createGeminiAgent(options = {}) {
       return {
         cmd: "gemini",
         args,
-        env: {
-          GEMINI_API_KEY: requiredEnv2(env, "GEMINI_API_KEY")
-        }
+        env: Object.fromEntries(
+          Object.entries({
+            ...env,
+            GEMINI_API_KEY: requiredEnv2(env, "GEMINI_API_KEY")
+          }).filter((e) => e[1] != null)
+        )
       };
     }
   };
@@ -633,7 +596,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 +650,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 +719,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 +839,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: {
@@ -882,6 +869,7 @@ var cloudChatRunRequestSchema = z3.object({
   message: z3.string().min(1),
   agent_type: z3.enum(["gemini", "codex"]),
   snapshot_id: z3.string().min(1),
+  env: z3.record(z3.string(), z3.string()).optional(),
   tool_results: z3.array(
     z3.object({
       toolCallId: z3.string().min(1),
@@ -895,6 +883,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 +1005,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 +1106,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
   };
@@ -1503,12 +1494,17 @@ function createAgentApi(options) {
           chat_id: parsed.data.chat_id,
           agent_type: parsed.data.agent_type,
           snapshot_id: parsed.data.snapshot_id,
+          env: parsed.data.env,
           tool_results: parsed.data.tool_results
         },
         agent: {
           ...agentOptions,
           type: parsed.data.agent_type,
-          snapshotId: parsed.data.snapshot_id
+          snapshotId: parsed.data.snapshot_id,
+          env: {
+            ...parsed.data.env,
+            ...agentOptions.env
+          }
         },
         signal: request.signal,
         store: await getStore(),
@@ -1584,7 +1580,6 @@ function createAgentApi(options) {
   };
 }
 export {
-  Agent,
   applyCloudChatPatch,
   cloudChatRunRequestSchema,
   createAgent,

package/dist/{types-Cto-uejG.d.ts → types-cRejduhB.d.ts}

@@ -2,20 +2,32 @@ type AgentFile = {
     path: string;
     content: string;
 };
+type AgentSetup = {
+    /** Shell script to run inside the sandbox during build. Executed as `bash -lc`. */
+    script: string;
+};
 type AgentConfig = {
     /** Agent type. Defaults to "gemini". */
     agentType?: "gemini" | "codex";
     /** Content for AGENTS.md in the sandbox. */
     agentMd?: string;
+    /** Environment variables passed to the sandbox at build and run time. */
+    env?: Record<string, string>;
     /** Additional files to write into the sandbox. */
     files?: AgentFile[];
+    /** Setup configuration for the sandbox build phase. */
+    setup?: AgentSetup;
 };
 type DefinedAgent = {
     readonly agentType: "gemini" | "codex";
     readonly agentMd?: string;
     readonly files: AgentFile[];
+    /** Setup configuration. Undefined when no setup is configured. */
+    readonly setup?: AgentSetup;
+    /** Environment variables. */
+    readonly env: Record<string, string>;
     /** Snapshot ID resolved from env at runtime. Throws if not set. */
     readonly snapshotId: string;
 };
 
-export type { AgentConfig as A, DefinedAgent as D, AgentFile as a };
+export type { AgentConfig as A, DefinedAgent as D, AgentFile as a, AgentSetup as b };

package/package.json

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