@auvira.ai/sdk 0.3.1 → 0.5.0

package/dist/runner/mapStreamEvents.js

@@ -1,5 +1,6 @@
 import { getAssistantEventText } from "../agent/assistantEvent.js";
 import { updateJobResultContext } from "./buildJobResult.js";
+import { getSandboxNonStepToolTrace, isSandboxStepTool, } from "./sandboxStepTools.js";
 export function createMapStreamContext(sessionResumed) {
     return {
         toolSeq: 0,
@@ -50,6 +51,18 @@ export function mapAuviraStreamEventToProgress(event, ctx) {
             ];
         }
         case "tool.start": {
+            if (!isSandboxStepTool(event.name)) {
+                const trace = getSandboxNonStepToolTrace(event.name, "start");
+                return [
+                    {
+                        type: "activity_trace",
+                        entry: {
+                            ...trace,
+                            detail: `${trace.detail} ${JSON.stringify(event.input).slice(0, 200)}`.trim(),
+                        },
+                    },
+                ];
+            }
             ctx.toolSeq += 1;
             const stepId = `tool_${ctx.toolSeq}`;
             const queue = ctx.openToolsByName.get(event.name) ?? [];
@@ -76,6 +89,18 @@ export function mapAuviraStreamEventToProgress(event, ctx) {
             ];
         }
         case "tool.end": {
+            if (!isSandboxStepTool(event.name)) {
+                const trace = getSandboxNonStepToolTrace(event.name, "end");
+                return [
+                    {
+                        type: "activity_trace",
+                        entry: {
+                            ...trace,
+                            detail: event.ok ? trace.detail : `${trace.detail} (failed)`,
+                        },
+                    },
+                ];
+            }
             const queue = ctx.openToolsByName.get(event.name) ?? [];
             const stepId = queue.shift();
             if (!stepId) {
@@ -198,6 +223,30 @@ export function mapAuviraStreamEventToProgress(event, ctx) {
                 },
             ];
         }
+        case "agent.thinking":
+            return [
+                {
+                    type: "thinking",
+                    turn: event.turn,
+                    phase: event.phase,
+                    summary: event.summary,
+                    nextTools: event.nextTools,
+                    reason: event.reason,
+                    confidence: event.confidence,
+                },
+            ];
+        case "agent.raw_model_trace":
+            return [
+                {
+                    type: "activity_trace",
+                    entry: {
+                        kind: "raw_trace",
+                        label: `Raw model trace (${event.provider})`,
+                        detail: event.reasoningDetails,
+                        truncated: event.capped,
+                    },
+                },
+            ];
         default:
             return [];
     }

package/dist/runner/mapStreamEvents.js.map

@@ -1 +1 @@
-{"version":3,"file":"mapStreamEvents.js","sourceRoot":"","sources":["../../src/runner/mapStreamEvents.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAe7D,MAAM,UAAU,sBAAsB,CACpC,cAAuB;IAEvB,OAAO;QACL,OAAO,EAAE,CAAC;QACV,eAAe,EAAE,IAAI,GAAG,EAAE;QAC1B,mBAAmB,EAAE,KAAK;QAC1B,SAAS,EAAE,EAAE,cAAc,EAAE;KAC9B,CAAC;AACJ,CAAC;AAED,+DAA+D;AAC/D,MAAM,UAAU,mBAAmB,CACjC,GAAqB,EACrB,MAAmD;IAEnD,MAAM,MAAM,GAA2B,EAAE,CAAC;IAE1C,KAAK,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,CAAC;QAC9D,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,EAAG,CAAC;YAC9B,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,MAAM;gBACZ,EAAE,EAAE,MAAM;gBACV,KAAK,EAAE,QAAQ;gBACf,MAAM;gBACN,QAAQ;aACT,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,GAAG,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IAC5B,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,sEAAsE;AACtE,MAAM,UAAU,8BAA8B,CAC5C,KAAoB,EACpB,GAAqB;IAErB,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QACrE,sBAAsB,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC/C,CAAC;IACD,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,IAAI,KAAK,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;QACvE,sBAAsB,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,oBAAoB,CAAC,CAAC,CAAC;YAC1B,IAAI,GAAG,CAAC,mBAAmB,EAAE,CAAC;gBAC5B,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,GAAG,CAAC,mBAAmB,GAAG,IAAI,CAAC;YAC/B,OAAO;gBACL;oBACE,IAAI,EAAE,MAAM;oBACZ,EAAE,EAAE,gBAAgB;oBACpB,KAAK,EAAE,oBAAoB;oBAC3B,MAAM,EAAE,WAAW;iBACpB;aACF,CAAC;QACJ,CAAC;QACD,KAAK,YAAY,CAAC,CAAC,CAAC;YAClB,GAAG,CAAC,OAAO,IAAI,CAAC,CAAC;YACjB,MAAM,MAAM,GAAG,QAAQ,GAAG,CAAC,OAAO,EAAE,CAAC;YAErC,MAAM,KAAK,GAAG,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACxD,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnB,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAE3C,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,YAAY;wBAClB,QAAQ,EAAE,KAAK,CAAC,IAAI;wBACpB,KAAK,EAAE,KAAK,CAAC,IAAI;wBACjB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;wBACjD,MAAM;qBACP;iBACF;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,EAAE,EAAE,MAAM;oBACV,KAAK,EAAE,KAAK,CAAC,IAAI;oBACjB,MAAM,EAAE,QAAQ;oBAChB,QAAQ,EAAE,KAAK,CAAC,IAAI;iBACrB;aACF,CAAC;QACJ,CAAC;QACD,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,KAAK,GAAG,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACxD,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;YAE7B,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC,yCAAyC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBACpE,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,GAAG,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzC,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAC7C,CAAC;YAED,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,UAAU;wBAChB,QAAQ,EAAE,KAAK,CAAC,IAAI;wBACpB,KAAK,EAAE,KAAK,CAAC,IAAI;wBACjB,MAAM,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;wBACzC,MAAM;qBACP;iBACF;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,EAAE,EAAE,MAAM;oBACV,KAAK,EAAE,KAAK,CAAC,IAAI;oBACjB,MAAM,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;oBACzC,QAAQ,EAAE,KAAK,CAAC,IAAI;iBACrB;aACF,CAAC;QACJ,CAAC;QACD,KAAK,WAAW;YACd,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,KAAK,EAAE,KAAK,CAAC,KAAK;wBAClB,SAAS,EAAE,KAAK,CAAC,SAAS;wBAC1B,KAAK,EAAE,KAAK,CAAC,IAAI;qBAClB;iBACF;aACF,CAAC;QACJ,KAAK,UAAU;YACb,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,KAAK,CAAC,KAAK;wBAClB,SAAS,EAAE,KAAK,CAAC,SAAS;wBAC1B,KAAK,EAAE,aAAa;wBACpB,MAAM,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;qBACtC;iBACF;aACF,CAAC;QACJ,KAAK,cAAc;YACjB,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,iBAAiB;wBACxB,MAAM,EACJ,KAAK,CAAC,MAAM,KAAK,uBAAuB;4BACtC,CAAC,CAAC,uDAAuD;4BACzD,CAAC,CAAC,qBAAqB,KAAK,CAAC,MAAM,EAAE;qBAC1C;iBACF;aACF,CAAC;QACJ,KAAK,kBAAkB;YACrB,OAAO;gBACL,GAAG,mBAAmB,CAAC,GAAG,EAAE,WAAW,CAAC;gBACxC;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,cAAc;wBACrB,MAAM,EAAE,GAAG,KAAK,CAAC,MAAM,KAAK,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;qBAC5D;iBACF;aACF,CAAC;QACJ,KAAK,YAAY;YACf,OAAO;gBACL,GAAG,mBAAmB,CAAC,GAAG,EAAE,QAAQ,CAAC;gBACrC;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,YAAY;wBACnB,MAAM,EAAE,KAAK,CAAC,OAAO;qBACtB;iBACF;aACF,CAAC;QACJ,KAAK,eAAe;YAClB,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,YAAY;wBACnB,MAAM,EAAE,cAAc,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,WAAW,EAAE;qBAC3D;iBACF;aACF,CAAC;QACJ,KAAK,WAAW,CAAC,CAAC,CAAC;YACjB,MAAM,IAAI,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC1C,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;gBACjB,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,sBAAsB;wBAC7B,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;qBAC3B;iBACF;aACF,CAAC;QACJ,CAAC;QACD;YACE,OAAO,EAAE,CAAC;IACd,CAAC;AACH,CAAC;AAED,iDAAiD;AACjD,MAAM,UAAU,4BAA4B,CAC1C,IAAY;IAEZ,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAY,CAAC;QAC9C,IACE,OAAO,MAAM,KAAK,QAAQ;YAC1B,MAAM,KAAK,IAAI;YACf,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,EACnB,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,IAAI,GAAI,MAA2B,CAAC,IAAI,CAAC;QAC/C,IAAI,IAAI,KAAK,UAAU,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;YAC7C,OAAO,MAAkC,CAAC;QAC5C,CAAC;QACD,IAAI,IAAI,KAAK,QAAQ,IAAI,QAAQ,IAAI,MAAM,EAAE,CAAC;YAC5C,OAAO,MAAkC,CAAC;QAC5C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC"}
+{"version":3,"file":"mapStreamEvents.js","sourceRoot":"","sources":["../../src/runner/mapStreamEvents.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAG7D,OAAO,EACL,0BAA0B,EAC1B,iBAAiB,GAClB,MAAM,uBAAuB,CAAC;AAa/B,MAAM,UAAU,sBAAsB,CACpC,cAAuB;IAEvB,OAAO;QACL,OAAO,EAAE,CAAC;QACV,eAAe,EAAE,IAAI,GAAG,EAAE;QAC1B,mBAAmB,EAAE,KAAK;QAC1B,SAAS,EAAE,EAAE,cAAc,EAAE;KAC9B,CAAC;AACJ,CAAC;AAED,+DAA+D;AAC/D,MAAM,UAAU,mBAAmB,CACjC,GAAqB,EACrB,MAAmD;IAEnD,MAAM,MAAM,GAA2B,EAAE,CAAC;IAE1C,KAAK,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,CAAC;QAC9D,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,EAAG,CAAC;YAC9B,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,MAAM;gBACZ,EAAE,EAAE,MAAM;gBACV,KAAK,EAAE,QAAQ;gBACf,MAAM;gBACN,QAAQ;aACT,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,GAAG,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IAC5B,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,sEAAsE;AACtE,MAAM,UAAU,8BAA8B,CAC5C,KAAoB,EACpB,GAAqB;IAErB,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QACrE,sBAAsB,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC/C,CAAC;IACD,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,IAAI,KAAK,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;QACvE,sBAAsB,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,oBAAoB,CAAC,CAAC,CAAC;YAC1B,IAAI,GAAG,CAAC,mBAAmB,EAAE,CAAC;gBAC5B,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,GAAG,CAAC,mBAAmB,GAAG,IAAI,CAAC;YAC/B,OAAO;gBACL;oBACE,IAAI,EAAE,MAAM;oBACZ,EAAE,EAAE,gBAAgB;oBACpB,KAAK,EAAE,oBAAoB;oBAC3B,MAAM,EAAE,WAAW;iBACpB;aACF,CAAC;QACJ,CAAC;QACD,KAAK,YAAY,CAAC,CAAC,CAAC;YAClB,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnC,MAAM,KAAK,GAAG,0BAA0B,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAC9D,OAAO;oBACL;wBACE,IAAI,EAAE,gBAAgB;wBACtB,KAAK,EAAE;4BACL,GAAG,KAAK;4BACR,MAAM,EAAE,GAAG,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE;yBAC9E;qBACF;iBACF,CAAC;YACJ,CAAC;YAED,GAAG,CAAC,OAAO,IAAI,CAAC,CAAC;YACjB,MAAM,MAAM,GAAG,QAAQ,GAAG,CAAC,OAAO,EAAE,CAAC;YAErC,MAAM,KAAK,GAAG,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACxD,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnB,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAE3C,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,YAAY;wBAClB,QAAQ,EAAE,KAAK,CAAC,IAAI;wBACpB,KAAK,EAAE,KAAK,CAAC,IAAI;wBACjB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;wBACjD,MAAM;qBACP;iBACF;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,EAAE,EAAE,MAAM;oBACV,KAAK,EAAE,KAAK,CAAC,IAAI;oBACjB,MAAM,EAAE,QAAQ;oBAChB,QAAQ,EAAE,KAAK,CAAC,IAAI;iBACrB;aACF,CAAC;QACJ,CAAC;QACD,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnC,MAAM,KAAK,GAAG,0BAA0B,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBAC5D,OAAO;oBACL;wBACE,IAAI,EAAE,gBAAgB;wBACtB,KAAK,EAAE;4BACL,GAAG,KAAK;4BACR,MAAM,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,WAAW;yBAC7D;qBACF;iBACF,CAAC;YACJ,CAAC;YAED,MAAM,KAAK,GAAG,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACxD,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;YAE7B,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC,yCAAyC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBACpE,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,GAAG,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzC,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAC7C,CAAC;YAED,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,UAAU;wBAChB,QAAQ,EAAE,KAAK,CAAC,IAAI;wBACpB,KAAK,EAAE,KAAK,CAAC,IAAI;wBACjB,MAAM,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;wBACzC,MAAM;qBACP;iBACF;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,EAAE,EAAE,MAAM;oBACV,KAAK,EAAE,KAAK,CAAC,IAAI;oBACjB,MAAM,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;oBACzC,QAAQ,EAAE,KAAK,CAAC,IAAI;iBACrB;aACF,CAAC;QACJ,CAAC;QACD,KAAK,WAAW;YACd,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,KAAK,EAAE,KAAK,CAAC,KAAK;wBAClB,SAAS,EAAE,KAAK,CAAC,SAAS;wBAC1B,KAAK,EAAE,KAAK,CAAC,IAAI;qBAClB;iBACF;aACF,CAAC;QACJ,KAAK,UAAU;YACb,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,KAAK,CAAC,KAAK;wBAClB,SAAS,EAAE,KAAK,CAAC,SAAS;wBAC1B,KAAK,EAAE,aAAa;wBACpB,MAAM,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;qBACtC;iBACF;aACF,CAAC;QACJ,KAAK,cAAc;YACjB,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,iBAAiB;wBACxB,MAAM,EACJ,KAAK,CAAC,MAAM,KAAK,uBAAuB;4BACtC,CAAC,CAAC,uDAAuD;4BACzD,CAAC,CAAC,qBAAqB,KAAK,CAAC,MAAM,EAAE;qBAC1C;iBACF;aACF,CAAC;QACJ,KAAK,kBAAkB;YACrB,OAAO;gBACL,GAAG,mBAAmB,CAAC,GAAG,EAAE,WAAW,CAAC;gBACxC;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,cAAc;wBACrB,MAAM,EAAE,GAAG,KAAK,CAAC,MAAM,KAAK,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;qBAC5D;iBACF;aACF,CAAC;QACJ,KAAK,YAAY;YACf,OAAO;gBACL,GAAG,mBAAmB,CAAC,GAAG,EAAE,QAAQ,CAAC;gBACrC;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,YAAY;wBACnB,MAAM,EAAE,KAAK,CAAC,OAAO;qBACtB;iBACF;aACF,CAAC;QACJ,KAAK,eAAe;YAClB,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,YAAY;wBACnB,MAAM,EAAE,cAAc,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,WAAW,EAAE;qBAC3D;iBACF;aACF,CAAC;QACJ,KAAK,WAAW,CAAC,CAAC,CAAC;YACjB,MAAM,IAAI,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC1C,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;gBACjB,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,sBAAsB;wBAC7B,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;qBAC3B;iBACF;aACF,CAAC;QACJ,CAAC;QACD,KAAK,gBAAgB;YACnB,OAAO;gBACL;oBACE,IAAI,EAAE,UAAU;oBAChB,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,OAAO,EAAE,KAAK,CAAC,OAAO;oBACtB,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,UAAU,EAAE,KAAK,CAAC,UAAU;iBAC7B;aACF,CAAC;QACJ,KAAK,uBAAuB;YAC1B,OAAO;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE;wBACL,IAAI,EAAE,WAAW;wBACjB,KAAK,EAAE,oBAAoB,KAAK,CAAC,QAAQ,GAAG;wBAC5C,MAAM,EAAE,KAAK,CAAC,gBAAgB;wBAC9B,SAAS,EAAE,KAAK,CAAC,MAAM;qBACxB;iBACF;aACF,CAAC;QACJ;YACE,OAAO,EAAE,CAAC;IACd,CAAC;AACH,CAAC;AAED,iDAAiD;AACjD,MAAM,UAAU,4BAA4B,CAC1C,IAAY;IAEZ,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAY,CAAC;QAC9C,IACE,OAAO,MAAM,KAAK,QAAQ;YAC1B,MAAM,KAAK,IAAI;YACf,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,EACnB,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,IAAI,GAAI,MAA2B,CAAC,IAAI,CAAC;QAC/C,IAAI,IAAI,KAAK,UAAU,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;YAC7C,OAAO,MAAkC,CAAC;QAC5C,CAAC;QACD,IAAI,IAAI,KAAK,QAAQ,IAAI,QAAQ,IAAI,MAAM,EAAE,CAAC;YAC5C,OAAO,MAAkC,CAAC;QAC5C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/runner/sandboxStepTools.d.ts

@@ -0,0 +1,12 @@
+/** Tools that should not create tool_N step rows in sandbox NDJSON. */
+export declare const SANDBOX_NON_STEP_TOOLS: Set<string>;
+/** Returns true when a tool should appear in the host step list (tool_N rows). */
+export declare function isSandboxStepTool(toolName: string): boolean;
+/** Human-friendly activity trace for infrastructure tools hidden from the step list. */
+export declare function getSandboxNonStepToolTrace(toolName: string, phase: "start" | "end"): {
+    kind: "info";
+    label: string;
+    detail: string;
+    toolName: string;
+};
+//# sourceMappingURL=sandboxStepTools.d.ts.map

package/dist/runner/sandboxStepTools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sandboxStepTools.d.ts","sourceRoot":"","sources":["../../src/runner/sandboxStepTools.ts"],"names":[],"mappings":"AAAA,uEAAuE;AACvE,eAAO,MAAM,sBAAsB,aAEjC,CAAC;AAEH,kFAAkF;AAClF,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAE3D;AAED,wFAAwF;AACxF,wBAAgB,0BAA0B,CACxC,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,OAAO,GAAG,KAAK,GACrB;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,CAmBnE"}

package/dist/runner/sandboxStepTools.js

@@ -0,0 +1,28 @@
+/** Tools that should not create tool_N step rows in sandbox NDJSON. */
+export const SANDBOX_NON_STEP_TOOLS = new Set([
+    "model_completion",
+]);
+/** Returns true when a tool should appear in the host step list (tool_N rows). */
+export function isSandboxStepTool(toolName) {
+    return !SANDBOX_NON_STEP_TOOLS.has(toolName);
+}
+/** Human-friendly activity trace for infrastructure tools hidden from the step list. */
+export function getSandboxNonStepToolTrace(toolName, phase) {
+    if (toolName === "model_completion") {
+        return {
+            kind: "info",
+            label: "Planning",
+            detail: phase === "start"
+                ? "Agent is deciding what to do next..."
+                : "Planning step finished",
+            toolName,
+        };
+    }
+    return {
+        kind: "info",
+        label: toolName,
+        detail: phase === "start" ? "Started" : "Finished",
+        toolName,
+    };
+}
+//# sourceMappingURL=sandboxStepTools.js.map

package/dist/runner/sandboxStepTools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sandboxStepTools.js","sourceRoot":"","sources":["../../src/runner/sandboxStepTools.ts"],"names":[],"mappings":"AAAA,uEAAuE;AACvE,MAAM,CAAC,MAAM,sBAAsB,GAAG,IAAI,GAAG,CAAC;IAC5C,kBAAkB;CACnB,CAAC,CAAC;AAEH,kFAAkF;AAClF,MAAM,UAAU,iBAAiB,CAAC,QAAgB;IAChD,OAAO,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AAC/C,CAAC;AAED,wFAAwF;AACxF,MAAM,UAAU,0BAA0B,CACxC,QAAgB,EAChB,KAAsB;IAEtB,IAAI,QAAQ,KAAK,kBAAkB,EAAE,CAAC;QACpC,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,UAAU;YACjB,MAAM,EACJ,KAAK,KAAK,OAAO;gBACf,CAAC,CAAC,sCAAsC;gBACxC,CAAC,CAAC,wBAAwB;YAC9B,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,QAAQ;QACf,MAAM,EAAE,KAAK,KAAK,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU;QAClD,QAAQ;KACT,CAAC;AACJ,CAAC"}

package/dist/util/minimaxImageClient.d.ts

@@ -0,0 +1,25 @@
+export interface MiniMaxImageResult {
+    buffer: Buffer;
+    mimeType: string;
+}
+export interface GenerateImageApiInput {
+    apiKey: string;
+    prompt: string;
+    aspectRatio?: string;
+    referencePath?: string;
+    signal?: AbortSignal;
+    timeoutMs?: number;
+}
+/** Returns true when image generation is enabled (`AUVIRA_IMAGE_GEN_ENABLED=true`). */
+export declare function isImageGenEnabled(): boolean;
+/** Max image generations per agent run. */
+export declare function getImageGenMaxPerRun(): number;
+/** Resolves MiniMax image API base URL (never chat/OpenRouter URL). */
+export declare function resolveImageApiBaseUrl(): string;
+/** Calls MiniMax image-01 text-to-image API. */
+export declare function generateTextToImage(input: GenerateImageApiInput): Promise<MiniMaxImageResult>;
+/** Calls MiniMax image-01 image-to-image API with subject reference. */
+export declare function generateImageToImage(input: GenerateImageApiInput & {
+    referencePath: string;
+}): Promise<MiniMaxImageResult>;
+//# sourceMappingURL=minimaxImageClient.d.ts.map

package/dist/util/minimaxImageClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"minimaxImageClient.d.ts","sourceRoot":"","sources":["../../src/util/minimaxImageClient.ts"],"names":[],"mappings":"AAkBA,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,uFAAuF;AACvF,wBAAgB,iBAAiB,IAAI,OAAO,CAE3C;AAED,2CAA2C;AAC3C,wBAAgB,oBAAoB,IAAI,MAAM,CAI7C;AAMD,uEAAuE;AACvE,wBAAgB,sBAAsB,IAAI,MAAM,CAY/C;AAqBD,gDAAgD;AAChD,wBAAsB,mBAAmB,CACvC,KAAK,EAAE,qBAAqB,GAC3B,OAAO,CAAC,kBAAkB,CAAC,CAK7B;AAED,wEAAwE;AACxE,wBAAsB,oBAAoB,CACxC,KAAK,EAAE,qBAAqB,GAAG;IAAE,aAAa,EAAE,MAAM,CAAA;CAAE,GACvD,OAAO,CAAC,kBAAkB,CAAC,CAM7B"}

package/dist/util/minimaxImageClient.js

@@ -0,0 +1,114 @@
+import { readFile } from "node:fs/promises";
+import { sniffImageMime } from "../agent/tools/assetPathValidation.js";
+import { postJsonWithRetry } from "./modelApiClient.js";
+const GLOBAL_IMAGE_BASE = "https://api.minimax.io/v1";
+const CN_IMAGE_BASE = "https://api.minimaxi.com/v1";
+/** Returns true when image generation is enabled (`AUVIRA_IMAGE_GEN_ENABLED=true`). */
+export function isImageGenEnabled() {
+    return process.env.AUVIRA_IMAGE_GEN_ENABLED?.trim() === "true";
+}
+/** Max image generations per agent run. */
+export function getImageGenMaxPerRun() {
+    const raw = process.env.AUVIRA_IMAGE_GEN_MAX_PER_RUN?.trim();
+    const parsed = raw ? Number.parseInt(raw, 10) : 2;
+    return Number.isFinite(parsed) && parsed > 0 ? parsed : 2;
+}
+function getImageModelId() {
+    return process.env.AUVIRA_IMAGE_MODEL?.trim() || "image-01";
+}
+/** Resolves MiniMax image API base URL (never chat/OpenRouter URL). */
+export function resolveImageApiBaseUrl() {
+    const explicit = process.env.AUVIRA_IMAGE_BASE_URL?.trim();
+    if (explicit) {
+        return explicit.replace(/\/+$/, "");
+    }
+    const chatBase = process.env.CUSTOM_MODEL_BASE_URL?.trim() ?? "";
+    if (chatBase.includes("minimaxi.com")) {
+        return CN_IMAGE_BASE;
+    }
+    return GLOBAL_IMAGE_BASE;
+}
+function mapProviderError(statusCode, message) {
+    if (statusCode === 1002) {
+        return `MiniMax image rate limit: ${message}`;
+    }
+    if (statusCode === 1008) {
+        return `MiniMax image insufficient balance: ${message}`;
+    }
+    if (statusCode === 1026) {
+        return `MiniMax image sensitive content: ${message}`;
+    }
+    return `MiniMax image error (${statusCode}): ${message}`;
+}
+async function referenceToDataUrl(referencePath) {
+    const buffer = await readFile(referencePath);
+    const mime = sniffImageMime(buffer) ?? "image/png";
+    return `data:${mime};base64,${buffer.toString("base64")}`;
+}
+/** Calls MiniMax image-01 text-to-image API. */
+export async function generateTextToImage(input) {
+    return callImageGeneration({
+        ...input,
+        subjectReference: undefined,
+    });
+}
+/** Calls MiniMax image-01 image-to-image API with subject reference. */
+export async function generateImageToImage(input) {
+    const imageFile = await referenceToDataUrl(input.referencePath);
+    return callImageGeneration({
+        ...input,
+        subjectReference: [{ type: "character", image_file: imageFile }],
+    });
+}
+async function callImageGeneration(input) {
+    const baseUrl = resolveImageApiBaseUrl();
+    const body = {
+        model: getImageModelId(),
+        prompt: input.prompt.slice(0, 1500),
+        aspect_ratio: input.aspectRatio ?? "16:9",
+        response_format: "base64",
+        n: 1,
+        prompt_optimizer: false,
+    };
+    if (input.subjectReference?.length) {
+        body.subject_reference = input.subjectReference;
+    }
+    const result = await postJsonWithRetry({
+        url: `${baseUrl}/image_generation`,
+        headers: {
+            Authorization: `Bearer ${input.apiKey}`,
+            "Content-Type": "application/json",
+        },
+        body,
+        timeoutMs: input.timeoutMs ?? 120_000,
+        signal: input.signal,
+    }, 0);
+    if (!result.ok || !result.data) {
+        const rateLimit = result.rateLimit;
+        throw new Error(rateLimit?.message ?? result.errorMessage ?? "Image generation request failed");
+    }
+    const statusCode = result.data.base_resp?.status_code ?? 0;
+    const statusMsg = result.data.base_resp?.status_msg ?? "unknown";
+    if (statusCode !== 0) {
+        if (statusCode === 1002) {
+            const info = {
+                isRateLimited: true,
+                providerStatusCode: statusCode,
+                message: statusMsg,
+                headers: {},
+            };
+            throw Object.assign(new Error(mapProviderError(statusCode, statusMsg)), {
+                rateLimit: info,
+            });
+        }
+        throw new Error(mapProviderError(statusCode, statusMsg));
+    }
+    const b64 = result.data.data?.image_base64?.[0];
+    if (!b64) {
+        throw new Error("Image generation returned no image_base64 data");
+    }
+    const buffer = Buffer.from(b64, "base64");
+    const mimeType = sniffImageMime(buffer) ?? "image/png";
+    return { buffer, mimeType };
+}
+//# sourceMappingURL=minimaxImageClient.js.map

package/dist/util/minimaxImageClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"minimaxImageClient.js","sourceRoot":"","sources":["../../src/util/minimaxImageClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,uCAAuC,CAAC;AACvE,OAAO,EAAE,iBAAiB,EAAsB,MAAM,qBAAqB,CAAC;AAE5E,MAAM,iBAAiB,GAAG,2BAA2B,CAAC;AACtD,MAAM,aAAa,GAAG,6BAA6B,CAAC;AA2BpD,uFAAuF;AACvF,MAAM,UAAU,iBAAiB;IAC/B,OAAO,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,IAAI,EAAE,KAAK,MAAM,CAAC;AACjE,CAAC;AAED,2CAA2C;AAC3C,MAAM,UAAU,oBAAoB;IAClC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,4BAA4B,EAAE,IAAI,EAAE,CAAC;IAC7D,MAAM,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAClD,OAAO,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,SAAS,eAAe;IACtB,OAAO,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,IAAI,EAAE,IAAI,UAAU,CAAC;AAC9D,CAAC;AAED,uEAAuE;AACvE,MAAM,UAAU,sBAAsB;IACpC,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,IAAI,EAAE,CAAC;IAC3D,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACtC,CAAC;IAED,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IACjE,IAAI,QAAQ,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;QACtC,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED,SAAS,gBAAgB,CAAC,UAAkB,EAAE,OAAe;IAC3D,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;QACxB,OAAO,6BAA6B,OAAO,EAAE,CAAC;IAChD,CAAC;IACD,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;QACxB,OAAO,uCAAuC,OAAO,EAAE,CAAC;IAC1D,CAAC;IACD,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;QACxB,OAAO,oCAAoC,OAAO,EAAE,CAAC;IACvD,CAAC;IACD,OAAO,wBAAwB,UAAU,MAAM,OAAO,EAAE,CAAC;AAC3D,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,aAAqB;IACrD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,aAAa,CAAC,CAAC;IAC7C,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC;IACnD,OAAO,QAAQ,IAAI,WAAW,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;AAC5D,CAAC;AAED,gDAAgD;AAChD,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,KAA4B;IAE5B,OAAO,mBAAmB,CAAC;QACzB,GAAG,KAAK;QACR,gBAAgB,EAAE,SAAS;KAC5B,CAAC,CAAC;AACL,CAAC;AAED,wEAAwE;AACxE,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,KAAwD;IAExD,MAAM,SAAS,GAAG,MAAM,kBAAkB,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IAChE,OAAO,mBAAmB,CAAC;QACzB,GAAG,KAAK;QACR,gBAAgB,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;KACjE,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,mBAAmB,CAChC,KAEC;IAED,MAAM,OAAO,GAAG,sBAAsB,EAAE,CAAC;IACzC,MAAM,IAAI,GAA4B;QACpC,KAAK,EAAE,eAAe,EAAE;QACxB,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;QACnC,YAAY,EAAE,KAAK,CAAC,WAAW,IAAI,MAAM;QACzC,eAAe,EAAE,QAAQ;QACzB,CAAC,EAAE,CAAC;QACJ,gBAAgB,EAAE,KAAK;KACxB,CAAC;IAEF,IAAI,KAAK,CAAC,gBAAgB,EAAE,MAAM,EAAE,CAAC;QACnC,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC;IAClD,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,iBAAiB,CACpC;QACE,GAAG,EAAE,GAAG,OAAO,mBAAmB;QAClC,OAAO,EAAE;YACP,aAAa,EAAE,UAAU,KAAK,CAAC,MAAM,EAAE;YACvC,cAAc,EAAE,kBAAkB;SACnC;QACD,IAAI;QACJ,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,OAAO;QACrC,MAAM,EAAE,KAAK,CAAC,MAAM;KACrB,EACD,CAAC,CACF,CAAC;IAEF,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACnC,MAAM,IAAI,KAAK,CACb,SAAS,EAAE,OAAO,IAAI,MAAM,CAAC,YAAY,IAAI,iCAAiC,CAC/E,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,IAAI,CAAC,CAAC;IAC3D,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,IAAI,SAAS,CAAC;IACjE,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;QACrB,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;YACxB,MAAM,IAAI,GAAkB;gBAC1B,aAAa,EAAE,IAAI;gBACnB,kBAAkB,EAAE,UAAU;gBAC9B,OAAO,EAAE,SAAS;gBAClB,OAAO,EAAE,EAAE;aACZ,CAAC;YACF,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,gBAAgB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,EAAE;gBACtE,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC;IAChD,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACpE,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;IAC1C,MAAM,QAAQ,GAAG,cAAc,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC;IACvD,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;AAC9B,CAAC"}

package/dist/workspace/referenceImages.d.ts

@@ -0,0 +1,14 @@
+export declare const AUVIRA_DIR = ".auvira";
+export declare const LEGACY_SCREENSHOT_RELATIVE = ".auvira/input-screenshot.png";
+export type ReferenceImageSource = "primary" | `index:${number}`;
+/** Parses tool source param into a reference index (primary = 0). */
+export declare function parseReferenceImageSource(source: string): number;
+/** Returns expected relative path for a reference image index (without resolving ext). */
+export declare function referenceImageBaseName(index: number): string;
+/** Resolves a reference source to an existing file under .auvira in the workspace. */
+export declare function resolveReferenceImagePath(workspacePath: string, source: string): Promise<{
+    absolutePath: string;
+    relativePath: string;
+    mimeType: string;
+}>;
+//# sourceMappingURL=referenceImages.d.ts.map

package/dist/workspace/referenceImages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"referenceImages.d.ts","sourceRoot":"","sources":["../../src/workspace/referenceImages.ts"],"names":[],"mappings":"AAOA,eAAO,MAAM,UAAU,YAAY,CAAC;AACpC,eAAO,MAAM,0BAA0B,iCAAuC,CAAC;AAE/E,MAAM,MAAM,oBAAoB,GAAG,SAAS,GAAG,SAAS,MAAM,EAAE,CAAC;AAEjE,qEAAqE;AACrE,wBAAgB,yBAAyB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAYhE;AAMD,0FAA0F;AAC1F,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAE5D;AAED,sFAAsF;AACtF,wBAAsB,yBAAyB,CAC7C,aAAa,EAAE,MAAM,EACrB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC;IAAE,YAAY,EAAE,MAAM,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC,CAuC3E"}

package/dist/workspace/referenceImages.js

@@ -0,0 +1,60 @@
+import { promises as fs } from "node:fs";
+import { join } from "node:path";
+import { mimeFromPath, sniffImageMime, } from "../agent/tools/assetPathValidation.js";
+export const AUVIRA_DIR = ".auvira";
+export const LEGACY_SCREENSHOT_RELATIVE = `${AUVIRA_DIR}/input-screenshot.png`;
+/** Parses tool source param into a reference index (primary = 0). */
+export function parseReferenceImageSource(source) {
+    const trimmed = source.trim();
+    if (trimmed === "primary") {
+        return 0;
+    }
+    const match = /^index:(\d+)$/.exec(trimmed);
+    if (!match) {
+        throw new Error(`Invalid reference source "${source}"; use "primary" or "index:N"`);
+    }
+    return Number.parseInt(match[1], 10);
+}
+function referenceFilePrefix(index) {
+    return `input-image-${index}`;
+}
+/** Returns expected relative path for a reference image index (without resolving ext). */
+export function referenceImageBaseName(index) {
+    return `${AUVIRA_DIR}/${referenceFilePrefix(index)}`;
+}
+/** Resolves a reference source to an existing file under .auvira in the workspace. */
+export async function resolveReferenceImagePath(workspacePath, source) {
+    const index = parseReferenceImageSource(source);
+    const auviraDir = join(workspacePath, AUVIRA_DIR);
+    const prefix = referenceFilePrefix(index);
+    let entries;
+    try {
+        entries = await fs.readdir(auviraDir);
+    }
+    catch {
+        throw new Error(`No reference images found in ${AUVIRA_DIR} for ${source}`);
+    }
+    const matches = entries
+        .filter((name) => name === `${prefix}.png` || name.startsWith(`${prefix}.`))
+        .sort();
+    if (matches.length === 0 && index === 0) {
+        const legacy = entries.find((name) => name === "input-screenshot.png");
+        if (legacy) {
+            const relativePath = `${AUVIRA_DIR}/${legacy}`;
+            const absolutePath = join(auviraDir, legacy);
+            const buffer = await fs.readFile(absolutePath);
+            const mimeType = sniffImageMime(buffer) ?? mimeFromPath(relativePath) ?? "image/png";
+            return { absolutePath, relativePath, mimeType };
+        }
+    }
+    if (matches.length === 0) {
+        throw new Error(`Reference image not found for ${source}`);
+    }
+    const fileName = matches[0];
+    const relativePath = `${AUVIRA_DIR}/${fileName}`;
+    const absolutePath = join(auviraDir, fileName);
+    const buffer = await fs.readFile(absolutePath);
+    const mimeType = sniffImageMime(buffer) ?? mimeFromPath(relativePath) ?? "image/png";
+    return { absolutePath, relativePath, mimeType };
+}
+//# sourceMappingURL=referenceImages.js.map

package/dist/workspace/referenceImages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"referenceImages.js","sourceRoot":"","sources":["../../src/workspace/referenceImages.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EACL,YAAY,EACZ,cAAc,GACf,MAAM,uCAAuC,CAAC;AAE/C,MAAM,CAAC,MAAM,UAAU,GAAG,SAAS,CAAC;AACpC,MAAM,CAAC,MAAM,0BAA0B,GAAG,GAAG,UAAU,uBAAuB,CAAC;AAI/E,qEAAqE;AACrE,MAAM,UAAU,yBAAyB,CAAC,MAAc;IACtD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;IAC9B,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,CAAC,CAAC;IACX,CAAC;IACD,MAAM,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC5C,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CACb,6BAA6B,MAAM,+BAA+B,CACnE,CAAC;IACJ,CAAC;IACD,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,EAAE,CAAC,CAAC;AACxC,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAa;IACxC,OAAO,eAAe,KAAK,EAAE,CAAC;AAChC,CAAC;AAED,0FAA0F;AAC1F,MAAM,UAAU,sBAAsB,CAAC,KAAa;IAClD,OAAO,GAAG,UAAU,IAAI,mBAAmB,CAAC,KAAK,CAAC,EAAE,CAAC;AACvD,CAAC;AAED,sFAAsF;AACtF,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,aAAqB,EACrB,MAAc;IAEd,MAAM,KAAK,GAAG,yBAAyB,CAAC,MAAM,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;IAClD,MAAM,MAAM,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;IAE1C,IAAI,OAAiB,CAAC;IACtB,IAAI,CAAC;QACH,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,KAAK,CAAC,gCAAgC,UAAU,QAAQ,MAAM,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,MAAM,OAAO,GAAG,OAAO;SACpB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,GAAG,MAAM,MAAM,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,MAAM,GAAG,CAAC,CAAC;SAC3E,IAAI,EAAE,CAAC;IAEV,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,sBAAsB,CAAC,CAAC;QACvE,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,YAAY,GAAG,GAAG,UAAU,IAAI,MAAM,EAAE,CAAC;YAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YAC7C,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YAC/C,MAAM,QAAQ,GACZ,cAAc,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,IAAI,WAAW,CAAC;YACtE,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC;QAClD,CAAC;IACH,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,iCAAiC,MAAM,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,MAAM,QAAQ,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;IAC7B,MAAM,YAAY,GAAG,GAAG,UAAU,IAAI,QAAQ,EAAE,CAAC;IACjD,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/C,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAC/C,MAAM,QAAQ,GACZ,cAAc,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,IAAI,WAAW,CAAC;IACtE,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC;AAClD,CAAC"}

package/dist/workspace/workspaceCheckpoint.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"workspaceCheckpoint.d.ts","sourceRoot":"","sources":["../../src/workspace/workspaceCheckpoint.ts"],"names":[],"mappings":"AAYA,MAAM,WAAW,mBAAmB;IAClC,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,IAAI,EAAE,MAAM,CAAC;CACd;AAMD,MAAM,WAAW,mBAAmB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,IAAI,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACrC,OAAO,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1B,eAAe,CAAC,IAAI,IAAI,CAAC;CAC1B;AA8RD,uEAAuE;AACvE,wBAAgB,kCAAkC,CAChD,UAAU,CAAC,EAAE,mBAAmB,GAC/B,IAAI,CAEN;AAED,oEAAoE;AACpE,wBAAsB,yBAAyB,CAAC,KAAK,EAAE;IACrD,QAAQ,EAAE,MAAM,CAAC;IACjB,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;CACzB,GAAG,OAAO,CAAC,mBAAmB,CAAC,CA+C/B"}
+{"version":3,"file":"workspaceCheckpoint.d.ts","sourceRoot":"","sources":["../../src/workspace/workspaceCheckpoint.ts"],"names":[],"mappings":"AAiBA,MAAM,WAAW,mBAAmB;IAClC,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,IAAI,EAAE,MAAM,CAAC;CACd;AAiBD,MAAM,WAAW,mBAAmB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,IAAI,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACrC,OAAO,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1B,eAAe,CAAC,IAAI,IAAI,CAAC;CAC1B;AAicD,uEAAuE;AACvE,wBAAgB,kCAAkC,CAChD,UAAU,CAAC,EAAE,mBAAmB,GAC/B,IAAI,CAEN;AAED,oEAAoE;AACpE,wBAAsB,yBAAyB,CAAC,KAAK,EAAE;IACrD,QAAQ,EAAE,MAAM,CAAC;IACjB,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;CACzB,GAAG,OAAO,CAAC,mBAAmB,CAAC,CA+C/B"}

package/dist/workspace/workspaceCheckpoint.js

@@ -1,7 +1,8 @@
-import { randomUUID } from "node:crypto";
+import { createHash, randomUUID } from "node:crypto";
 import { promises as fs } from "node:fs";
 import { join, resolve } from "node:path";
 import { buildUnifiedPatch, capPatch, getDiffMaxChars, } from "../agent/patchUtils.js";
+import { isBinaryAssetPath, mimeFromPath, sniffImageMime, } from "../agent/tools/assetPathValidation.js";
 import { toRepoRelativePath } from "../agent/tools/pathGuard.js";
 import { fileExists } from "../tools/fileExists.js";
 import { matchesPathGlobFilters } from "../util/globMatch.js";
@@ -45,6 +46,83 @@ function isEditableFile(relPath) {
     }
     return EDITABLE_EXTENSIONS.has(lower.slice(dot));
 }
+function formatBytes(size) {
+    if (size < 1024) {
+        return `${size} B`;
+    }
+    const kb = size / 1024;
+    if (kb < 1024) {
+        return `${kb.toFixed(1)} KB`;
+    }
+    return `${(kb / 1024).toFixed(1)} MB`;
+}
+async function readBinarySnapshotEntry(repoPath, relPath) {
+    const absolute = join(repoPath, relPath);
+    if (!(await fileExists(absolute))) {
+        return {
+            kind: "binary",
+            sha256: null,
+            byteSize: 0,
+            mimeType: "application/octet-stream",
+            buffer: null,
+        };
+    }
+    const stat = await fs.stat(absolute);
+    if (!stat.isFile()) {
+        return {
+            kind: "binary",
+            sha256: null,
+            byteSize: 0,
+            mimeType: "application/octet-stream",
+            buffer: null,
+        };
+    }
+    const buffer = await fs.readFile(absolute);
+    const mimeType = sniffImageMime(buffer) ?? mimeFromPath(relPath) ?? "application/octet-stream";
+    return {
+        kind: "binary",
+        sha256: createHash("sha256").update(buffer).digest("hex"),
+        byteSize: buffer.byteLength,
+        mimeType,
+        buffer,
+    };
+}
+function buildBinaryDiffLine(relPath, before, after) {
+    if (before.sha256 === null && after.sha256 !== null) {
+        return `Binary asset added: ${relPath} (${after.mimeType}, ${formatBytes(after.byteSize)})`;
+    }
+    if (before.sha256 !== null && after.sha256 === null) {
+        return `Binary asset deleted: ${relPath}`;
+    }
+    if (before.sha256 !== after.sha256) {
+        return `Binary asset changed: ${relPath} (sha256 ${before.sha256?.slice(0, 12)}… → ${after.sha256?.slice(0, 12)}…)`;
+    }
+    return "";
+}
+function isBinaryAssetFile(relPath) {
+    return isBinaryAssetPath(relPath);
+}
+function entriesEqual(before, after) {
+    if (before.kind !== after.kind) {
+        return false;
+    }
+    if (before.kind === "text" && after.kind === "text") {
+        return before.content === after.content;
+    }
+    if (before.kind === "binary" && after.kind === "binary") {
+        return before.sha256 === after.sha256;
+    }
+    return false;
+}
+function buildEntryDiffLine(relPath, before, after) {
+    if (before.kind === "binary" && after.kind === "binary") {
+        return buildBinaryDiffLine(relPath, before, after);
+    }
+    if (before.kind === "text" && after.kind === "text") {
+        return buildCheckpointHunk(relPath, before.content, after.content);
+    }
+    return `Asset changed: ${relPath}`;
+}
 function buildCheckpointHunk(relPath, oldContent, newContent) {
     if (oldContent === null && newContent === null) {
         return "";
@@ -66,21 +144,32 @@ function buildCheckpointHunk(relPath, oldContent, newContent) {
     return buildUnifiedPatch(relPath, oldContent ?? "", newContent ?? "");
 }
 async function readSnapshotEntry(repoPath, relPath) {
+    if (isBinaryAssetFile(relPath)) {
+        return readBinarySnapshotEntry(repoPath, relPath);
+    }
     const absolute = join(repoPath, relPath);
     if (!(await fileExists(absolute))) {
-        return { content: null };
+        return { kind: "text", content: null };
     }
     const stat = await fs.stat(absolute);
     if (!stat.isFile()) {
-        return { content: null };
+        return { kind: "text", content: null };
     }
-    return { content: await fs.readFile(absolute, "utf8") };
+    return { kind: "text", content: await fs.readFile(absolute, "utf8") };
 }
 async function snapshotAllowedPaths(repoPath, allowedWritePaths) {
     const snapshot = new Map();
     for (const rawPath of allowedWritePaths) {
         const relPath = normalizeRelPath(rawPath);
-        if (!relPath || relPath.endsWith("/")) {
+        if (!relPath) {
+            continue;
+        }
+        if (relPath.endsWith("/")) {
+            const absoluteDir = join(repoPath, relPath);
+            if (!(await fileExists(absoluteDir))) {
+                continue;
+            }
+            await walkSnapshot(absoluteDir, repoPath, snapshot);
             continue;
         }
         snapshot.set(relPath, await readSnapshotEntry(repoPath, relPath));
@@ -99,14 +188,19 @@ async function walkSnapshot(dir, repoPath, snapshot, globFilters) {
             await walkSnapshot(fullPath, repoPath, snapshot, globFilters);
             continue;
         }
-        if (!entry.isFile() || !isEditableFile(relPath)) {
+        if (!entry.isFile() || (!isEditableFile(relPath) && !isBinaryAssetFile(relPath))) {
             continue;
         }
         if (globFilters &&
             !matchesPathGlobFilters(relPath, globFilters.includeGlobs, globFilters.excludeGlobs)) {
             continue;
         }
+        if (isBinaryAssetFile(relPath)) {
+            snapshot.set(relPath, await readBinarySnapshotEntry(repoPath, relPath));
+            continue;
+        }
         snapshot.set(relPath, {
+            kind: "text",
             content: await fs.readFile(fullPath, "utf8"),
         });
     }
@@ -127,25 +221,35 @@ async function diffSnapshotAsync(repoPath, snapshot, allowedWritePaths, globFilt
     const hunks = [];
     const seen = new Set();
     for (const relPath of snapshot.keys()) {
-        const before = snapshot.get(relPath) ?? { content: null };
+        const before = snapshot.get(relPath) ?? { kind: "text", content: null };
         const after = await readCurrentEntry(repoPath, relPath);
-        const beforeContent = before.content;
-        const afterContent = after.content;
-        if (beforeContent === afterContent) {
+        if (entriesEqual(before, after)) {
             continue;
         }
         seen.add(relPath);
         changedFiles.push(relPath);
-        hunks.push(buildCheckpointHunk(relPath, beforeContent, afterContent));
+        hunks.push(buildEntryDiffLine(relPath, before, after));
     }
     for (const relPath of await discoverNewFiles(repoPath, snapshot, allowedWritePaths, globFilters)) {
         if (seen.has(relPath)) {
             continue;
         }
         const after = await readCurrentEntry(repoPath, relPath);
-        if (after.content !== null) {
+        const emptyBefore = after.kind === "binary"
+            ? {
+                kind: "binary",
+                sha256: null,
+                byteSize: 0,
+                mimeType: "application/octet-stream",
+                buffer: null,
+            }
+            : { kind: "text", content: null };
+        const hasContent = after.kind === "text"
+            ? after.content !== null
+            : after.sha256 !== null;
+        if (hasContent) {
             changedFiles.push(relPath);
-            hunks.push(buildCheckpointHunk(relPath, null, after.content));
+            hunks.push(buildEntryDiffLine(relPath, emptyBefore, after));
         }
     }
     changedFiles.sort();
@@ -161,7 +265,24 @@ async function discoverNewFiles(repoPath, snapshot, allowedWritePaths, globFilte
         const added = [];
         for (const rawPath of allowedWritePaths) {
             const relPath = normalizeRelPath(rawPath);
-            if (!relPath || relPath.endsWith("/") || snapshot.has(relPath)) {
+            if (!relPath) {
+                continue;
+            }
+            if (relPath.endsWith("/")) {
+                const absoluteDir = join(repoPath, relPath);
+                if (!(await fileExists(absoluteDir))) {
+                    continue;
+                }
+                const discovered = new Map();
+                await walkSnapshot(absoluteDir, repoPath, discovered, globFilters);
+                for (const discoveredPath of discovered.keys()) {
+                    if (!snapshot.has(discoveredPath)) {
+                        added.push(discoveredPath);
+                    }
+                }
+                continue;
+            }
+            if (snapshot.has(relPath)) {
                 continue;
             }
             const absolute = join(repoPath, relPath);
@@ -186,14 +307,25 @@ async function restoreSnapshot(repoPath, snapshot, globFilters) {
     await walkSnapshot(resolve(repoPath), repoPath, current, globFilters);
     for (const [relPath, entry] of snapshot.entries()) {
         const absolute = join(repoPath, relPath);
-        if (entry.content === null) {
+        if (entry.kind === "text") {
+            if (entry.content === null) {
+                if (await fileExists(absolute)) {
+                    await fs.unlink(absolute);
+                }
+                continue;
+            }
+            await fs.mkdir(join(absolute, ".."), { recursive: true });
+            await fs.writeFile(absolute, entry.content, "utf8");
+            continue;
+        }
+        if (entry.buffer === null) {
             if (await fileExists(absolute)) {
                 await fs.unlink(absolute);
             }
             continue;
         }
         await fs.mkdir(join(absolute, ".."), { recursive: true });
-        await fs.writeFile(absolute, entry.content, "utf8");
+        await fs.writeFile(absolute, entry.buffer);
     }
     for (const relPath of current.keys()) {
         if (!snapshot.has(relPath)) {