lavish-axi 0.1.54 → 0.1.55

package/dist/cli.mjs

@@ -7446,13 +7446,19 @@ var SessionStore = class {
     }
     session.layout_warnings = warnings;
     const userMessages = restoring ? [] : acceptedPrompts.filter((prompt) => prompt.tag === "message" && prompt.prompt).map((prompt) => ({ role: "user", text: prompt.prompt, at: (/* @__PURE__ */ new Date()).toISOString() }));
-    session.prompts = [...session.prompts || [], ...acceptedPrompts];
+    const existingPrompts = Array.isArray(session.prompts) ? session.prompts : [];
+    session.prompts = restoring ? [...acceptedPrompts, ...existingPrompts] : [...existingPrompts, ...acceptedPrompts];
     session.chat = [...session.chat || [], ...userMessages];
     if (restoring) {
-      session.artifact_failures = Array.isArray(payload.artifact_failures) ? JSON.parse(JSON.stringify(payload.artifact_failures)) : [];
+      const restoredFailures = Array.isArray(payload.artifact_failures) ? JSON.parse(JSON.stringify(payload.artifact_failures)) : [];
+      const existingFailures = Array.isArray(session.artifact_failures) ? session.artifact_failures : [];
+      session.artifact_failures = mergeArtifactFailures(restoredFailures, existingFailures).failures;
     }
     session.pending_prompts = session.prompts.length;
-    session.dom_snapshot = String(payload.domSnapshot || payload.dom_snapshot || "");
+    const restoredSnapshot = String(payload.domSnapshot || payload.dom_snapshot || "");
+    if (!restoring || existingPrompts.length === 0 && !session.dom_snapshot) {
+      session.dom_snapshot = restoredSnapshot;
+    }
     session.status = shouldEndSession || alreadyEnded ? "ended" : session.prompts.length > 0 || restoring && Array.isArray(session.artifact_failures) && session.artifact_failures.length > 0 ? "feedback" : "open";
     if (shouldEndSession) session.ended_by = "user";
     session.updated_at = (/* @__PURE__ */ new Date()).toISOString();
@@ -7661,17 +7667,11 @@ var SessionStore = class {
       }
       const normalized = normalizeArtifactFailures(payload?.failures);
       const previous = Array.isArray(session.artifact_failures) ? session.artifact_failures : [];
-      const merged = [...previous];
-      let changed = false;
-      for (const failure of normalized) {
-        if (merged.some((item) => item.kind === failure.kind && item.detail === failure.detail)) continue;
-        merged.push(failure);
-        changed = true;
-      }
+      const { failures, changed } = mergeArtifactFailures(previous, normalized);
       if (!changed) {
         return { session, changed: false };
       }
-      session.artifact_failures = merged.slice(-MAX_ARTIFACT_FAILURES);
+      session.artifact_failures = failures;
       if (session.status !== "ended") session.status = "feedback";
       session.updated_at = (/* @__PURE__ */ new Date()).toISOString();
       await this.writeState(state);
@@ -7888,10 +7888,12 @@ function boundAttachmentRefs(normalized, options) {
     }
   }
   if (rejected.length) return rejected;
-  let requestRefs = 0;
-  for (const { prompt } of normalized) requestRefs += prompt.attachments?.length || 0;
-  if (requestRefs > MAX_REQUEST_ATTACHMENT_REFS) {
-    return [{ id: "", name: "", reason: "too-many-in-request" }];
+  if (options.restore !== true) {
+    let requestRefs = 0;
+    for (const { prompt } of normalized) requestRefs += prompt.attachments?.length || 0;
+    if (requestRefs > MAX_REQUEST_ATTACHMENT_REFS) {
+      return [{ id: "", name: "", reason: "too-many-in-request" }];
+    }
   }
   return rejected;
 }
@@ -7966,6 +7968,16 @@ function parsePassSequence(payload) {
   return { present, value: Number.isSafeInteger(value) && value > 0 ? value : null };
 }
 var ARTIFACT_FAILURE_KINDS = /* @__PURE__ */ new Set(["artifact-unavailable", "artifact-asset-unavailable"]);
+function mergeArtifactFailures(earlier, later) {
+  const merged = Array.isArray(earlier) ? [...earlier] : [];
+  let changed = false;
+  for (const failure of Array.isArray(later) ? later : []) {
+    if (merged.some((item) => item.kind === failure.kind && item.detail === failure.detail)) continue;
+    merged.push(failure);
+    changed = true;
+  }
+  return { failures: merged.slice(-MAX_ARTIFACT_FAILURES), changed };
+}
 function normalizeArtifactFailures(failures) {
   if (!Array.isArray(failures)) return [];
   return failures.filter((failure) => failure && typeof failure === "object" && !Array.isArray(failure)).map((failure) => ({
@@ -8637,25 +8649,43 @@ async function serve({
   async function restoreClosedFeedback(key, result) {
     if (result.status !== "feedback") return;
     const prompts = Array.isArray(result.prompts) ? result.prompts : [];
-    const session = await store.queuePrompts(
-      key,
-      {
-        dom_snapshot: result.dom_snapshot || "",
-        prompts,
-        ...Array.isArray(result.artifact_failures) ? { artifact_failures: result.artifact_failures } : {}
-      },
-      {
-        restore: true,
-        resolveAttachment: (sessionKeyValue, id) => resolveAttachment(attachmentStateRoot, sessionKeyValue, id),
-        maxPerPrompt: attachmentConfig.maxPerPrompt,
-        maxPromptBytes: attachmentConfig.maxPromptBytes
-      }
-    );
-    const restoredPrompts = prompts.length === 0 ? [] : session && !session.rejected && !session.conflict && Array.isArray(session.prompts) ? session.prompts.slice(-prompts.length) : null;
+    let session = null;
+    let restoreError = null;
+    try {
+      session = await store.queuePrompts(
+        key,
+        {
+          dom_snapshot: result.dom_snapshot || "",
+          prompts,
+          ...Array.isArray(result.artifact_failures) ? { artifact_failures: result.artifact_failures } : {}
+        },
+        {
+          restore: true,
+          resolveAttachment: (sessionKeyValue, id) => resolveAttachment(attachmentStateRoot, sessionKeyValue, id),
+          maxPerPrompt: attachmentConfig.maxPerPrompt,
+          maxPromptBytes: attachmentConfig.maxPromptBytes
+        }
+      );
+    } catch (error) {
+      restoreError = error;
+    }
+    const restoredPrompts = prompts.length === 0 ? [] : session && !session.rejected && !session.conflict && Array.isArray(session.prompts) ? session.prompts.slice(0, prompts.length) : null;
     const restoredFailures = session && Array.isArray(session.artifact_failures) ? session.artifact_failures : null;
-    if (!restoredPrompts || JSON.stringify(restoredPrompts) !== JSON.stringify(prompts) || Array.isArray(result.artifact_failures) && JSON.stringify(restoredFailures) !== JSON.stringify(result.artifact_failures)) {
+    const failuresRestored = !Array.isArray(result.artifact_failures) || Array.isArray(restoredFailures) && result.artifact_failures.every(
+      (failure) => restoredFailures.some((restoredFailure) => JSON.stringify(restoredFailure) === JSON.stringify(failure))
+    );
+    const persistedNothing = !session || Boolean(session.rejected) || Boolean(session.conflict);
+    if (restoreError) {
+      writeLog(
+        `[lavish] closed poll feedback restore failed; the batch was lost: ${restoreError?.message || restoreError}`
+      );
+    } else if (persistedNothing) {
+      writeLog("[lavish] closed poll feedback restore was refused; nothing was persisted and the batch was lost");
+    } else if (!restoredPrompts || JSON.stringify(restoredPrompts) !== JSON.stringify(prompts) || !failuresRestored) {
       writeLog("[lavish] closed poll feedback restore was incomplete; delivery was not marked");
     }
+    const pendingAfterRestore = Array.isArray(restoredPrompts) && restoredPrompts.length > 0 || Array.isArray(restoredFailures) && restoredFailures.length > 0;
+    if (pendingAfterRestore) events.emit("feedback", key);
   }
   const whiteboardStateRoot = path7.dirname(stateFile2);
   const allowedHostnames = buildAllowedHostnames({ host, linkHost: linkHostName, allowedHosts });
@@ -8800,6 +8830,10 @@ async function serve({
         responding = true;
         try {
           const result = await store.takeFeedback(key);
+          if (requestClosed || req.destroyed || res.writableEnded) {
+            await restoreClosedFeedback(key, result);
+            return;
+          }
           finishFeedbackDelivery(key, result);
           if (streamHeartbeat) {
             res.end(JSON.stringify(result));
@@ -10298,7 +10332,7 @@ var POLL_WAKE_PATH_RULES = Object.freeze([
 ]);
 var POLL_SEND_AND_END_RULE = "`Send & End` ends the session. Its final feedback is still delivered once. After that response, polling stops, and the agent must not reopen the session uninvited.";
 var CODEX_POLL_WAKE_PATH_GUIDANCE = "Codex detected: completed background tasks may not resume Codex automatically, so keep the poll attached to the active turn.";
-var VERSION = "0.1.54";
+var VERSION = "0.1.55";
 function detectInvokingAgent(env = process.env) {
   return ["CODEX_SANDBOX", "CODEX_THREAD_ID"].some((key) => Object.hasOwn(env, key)) ? "codex" : "generic";
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lavish-axi",
-  "version": "0.1.54",
+  "version": "0.1.55",
   "packageManager": "pnpm@11.1.1",
   "description": "HTML is the new markdown. Lavish is the new editor for your HTML artifacts.",
   "type": "module",

package/plugin.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://agent-plugins.org/schemas/1.0.0/plugin.schema.json",
   "name": "lavish-axi",
-  "version": "0.1.54",
+  "version": "0.1.55",
   "description": "HTML is the new markdown. Lavish is the new editor for your HTML artifacts.",
   "author": {
     "name": "Kun Chen",