openmates 0.14.8-alpha.7 → 0.14.8-alpha.9

package/dist/{chunk-S2QEBRF3.js → chunk-G7DDNKJL.js}

@@ -1214,6 +1214,31 @@ function websocketProtocolError(envelope) {
   }
   return null;
 }
+function errorFrameBelongsToAiResponse(envelope, userMessageId, chatId, recoveryTurnId) {
+  if (envelope.type === "client_update_required") return true;
+  if (envelope.type !== "error") return true;
+  const payload = envelope.payload && typeof envelope.payload === "object" ? envelope.payload : {};
+  let scoped = false;
+  const errorChatId = typeof payload.chat_id === "string" ? payload.chat_id : null;
+  if (errorChatId) {
+    scoped = true;
+    if (errorChatId !== chatId) return false;
+  }
+  const errorMessageId = typeof payload.user_message_id === "string" ? payload.user_message_id : typeof payload.userMessageId === "string" ? payload.userMessageId : typeof payload.message_id === "string" ? payload.message_id : null;
+  if (errorMessageId) {
+    scoped = true;
+    if (errorMessageId !== userMessageId) return false;
+  }
+  const errorTurnId = typeof payload.turn_id === "string" ? payload.turn_id : null;
+  if (errorTurnId) {
+    scoped = true;
+    if (recoveryTurnId !== errorTurnId) return false;
+  }
+  if (typeof payload.job_id === "string") {
+    scoped = true;
+  }
+  return !scoped || Boolean(errorChatId || errorMessageId || errorTurnId);
+}
 function parseTaskProposals(value) {
   if (!Array.isArray(value)) return [];
   return value.flatMap((item) => {
@@ -1278,6 +1303,21 @@ function parsePendingTaskUpdateJobs(value) {
     return [job];
   });
 }
+function parseAvailableRecoveryJobs(value) {
+  if (!Array.isArray(value)) return [];
+  return value.flatMap((item) => {
+    if (!item || typeof item !== "object" || Array.isArray(item)) return [];
+    const raw = item;
+    if (typeof raw.job_id !== "string" || typeof raw.chat_id !== "string" || typeof raw.turn_id !== "string" || typeof raw.assistant_message_id !== "string" || typeof raw.chat_key_version !== "number") return [];
+    return [{
+      job_id: raw.job_id,
+      chat_id: raw.chat_id,
+      turn_id: raw.turn_id,
+      assistant_message_id: raw.assistant_message_id,
+      chat_key_version: raw.chat_key_version
+    }];
+  });
+}
 var OpenMatesWsClient = class {
   socket;
   passiveTaskUpdateJobs = /* @__PURE__ */ new Map();
@@ -1698,6 +1738,9 @@ var OpenMatesWsClient = class {
           const type = parsed.type;
           const protocolError = websocketProtocolError(parsed);
           if (protocolError) {
+            if (!errorFrameBelongsToAiResponse(parsed, userMessageId, chatId, options?.recoveryTurnId)) {
+              return;
+            }
             cleanup();
             reject(protocolError);
             return;
@@ -1740,6 +1783,16 @@ var OpenMatesWsClient = class {
             maybeResolve();
             return;
           }
+          if (type === "recovery_jobs_available") {
+            const job = parseAvailableRecoveryJobs(p.jobs).find(
+              (candidate) => candidate.chat_id === chatId && (!options?.recoveryTurnId || candidate.turn_id === options.recoveryTurnId)
+            );
+            if (!job) return;
+            recoveryJobId = job.job_id;
+            messageId = job.assistant_message_id;
+            scheduleResolve(latestContent);
+            return;
+          }
           if (type === "ai_message_update") {
             const msgId = p.user_message_id ?? p.userMessageId;
             if (msgId !== userMessageId && p.chat_id !== chatId) return;
@@ -5078,7 +5131,8 @@ var OpenMatesClient = class _OpenMatesClient {
     if (params.precollectResponse && !params.incognito) {
       precollectedResponse = ws.collectAiResponse(messageId, chatId, {
         onStream: params.onStream,
-        timeoutMs: params.responseTimeoutMs
+        timeoutMs: params.responseTimeoutMs,
+        recoveryTurnId: savedTurnId
       });
     }
     const confirmed = ws.waitForMessage(
@@ -5493,7 +5547,8 @@ var OpenMatesClient = class _OpenMatesClient {
       try {
         const resp = await (precollectedResponse ?? ws.collectAiResponse(messageId, chatId, {
           ...streamOpts,
-          timeoutMs: params.responseTimeoutMs
+          timeoutMs: params.responseTimeoutMs,
+          recoveryTurnId: savedTurnId
         }));
         assistantMessageId = resp.messageId;
         assistant = resp.content;

package/dist/cli.js

@@ -2,7 +2,7 @@
 import {
   getExtForLang,
   serializeToYaml
-} from "./chunk-S2QEBRF3.js";
+} from "./chunk-G7DDNKJL.js";
 import "./chunk-AXNRPVLE.js";
 export {
   getExtForLang,

package/dist/index.js

@@ -29,7 +29,7 @@ import {
   renderSupportInfo,
   serializeToYaml,
   unwrapApiKeyMasterKey
-} from "./chunk-S2QEBRF3.js";
+} from "./chunk-G7DDNKJL.js";
 import "./chunk-AXNRPVLE.js";
 
 // src/generated/appSkills.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "openmates",
-  "version": "0.14.8-alpha.7",
+  "version": "0.14.8-alpha.9",
   "description": "OpenMates CLI and SDK",
   "type": "module",
   "main": "dist/index.js",