@dudousxd/nestjs-agent-core 0.31.0 → 0.33.0

package/dist/index.js

@@ -292,6 +292,56 @@ __name(mergeUi, "mergeUi");
 // src/tool-presentation.ts
 var ALL_AGENTS = "*";
 
+// src/spi/chat-queue.ts
+var CHAT_QUEUE_METHODS = [
+  "enqueueMessage",
+  "listQueue",
+  "getQueuedMessage",
+  "updateQueuedMessage",
+  "moveQueuedMessage",
+  "removeQueuedMessage",
+  "clearQueue",
+  "queuePause",
+  "setQueuePause",
+  "activeRunForThread",
+  "claimActiveStream",
+  "releaseActiveStream"
+];
+function isChatQueueStore(store) {
+  const candidate = store;
+  return CHAT_QUEUE_METHODS.every((method) => typeof candidate[method] === "function");
+}
+__name(isChatQueueStore, "isChatQueueStore");
+async function releaseThreadRun(store, threadId, runId) {
+  if (isChatQueueStore(store)) {
+    await store.releaseActiveStream(threadId, runId);
+    return;
+  }
+  await store.setActiveStream(threadId, null);
+}
+__name(releaseThreadRun, "releaseThreadRun");
+function queuedMessageView(message) {
+  return {
+    id: message.id,
+    content: message.content,
+    ...message.attachments !== void 0 && message.attachments.length > 0 ? {
+      attachments: message.attachments
+    } : {},
+    ...message.agentName !== void 0 ? {
+      agentName: message.agentName
+    } : {},
+    ...message.model !== void 0 ? {
+      model: message.model
+    } : {},
+    ...message.interrupt === true ? {
+      interrupt: true
+    } : {},
+    createdAt: message.createdAt,
+    updatedAt: message.updatedAt
+  };
+}
+__name(queuedMessageView, "queuedMessageView");
+
 // src/spi/quota-provider.ts
 function exhaustedWindow(windows) {
   for (const window of windows) {
@@ -4682,6 +4732,9 @@ var InMemoryAgentStore = class {
     __name(this, "InMemoryAgentStore");
   }
   threads = /* @__PURE__ */ new Map();
+  /** Each thread's waiting messages, in run order. */
+  queues = /* @__PURE__ */ new Map();
+  pauses = /* @__PURE__ */ new Map();
   toolCalls = /* @__PURE__ */ new Map();
   usage = [];
   runs = /* @__PURE__ */ new Map();
@@ -4938,6 +4991,149 @@ var InMemoryAgentStore = class {
       }
     }
   }
+  async claimActiveStream(threadId, runId, options = {}) {
+    const row = this.threads.get(threadId);
+    if (row === void 0) {
+      return false;
+    }
+    const holder = row.activeStreamId;
+    if (holder !== void 0 && holder !== runId && holder !== options.replacing) {
+      return false;
+    }
+    row.activeStreamId = runId;
+    return true;
+  }
+  async releaseActiveStream(threadId, runId) {
+    const row = this.threads.get(threadId);
+    if (row?.activeStreamId !== runId) {
+      return false;
+    }
+    delete row.activeStreamId;
+    return true;
+  }
+  async enqueueMessage(input) {
+    const ts = this.now();
+    const message = {
+      id: crypto.randomUUID(),
+      threadId: input.threadId,
+      actor: input.actor,
+      content: input.content,
+      ...input.attachments !== void 0 && input.attachments.length > 0 ? {
+        attachments: input.attachments
+      } : {},
+      ...input.agentName !== void 0 ? {
+        agentName: input.agentName
+      } : {},
+      ...input.model !== void 0 ? {
+        model: input.model
+      } : {},
+      ...input.pageContext !== void 0 ? {
+        pageContext: input.pageContext
+      } : {},
+      ...input.interrupt === true ? {
+        interrupt: true
+      } : {},
+      createdAt: ts,
+      updatedAt: ts
+    };
+    const queue = this.queues.get(input.threadId) ?? [];
+    if (input.at === "head") {
+      queue.unshift(message);
+    } else {
+      queue.push(message);
+    }
+    this.queues.set(input.threadId, queue);
+    return {
+      ...message
+    };
+  }
+  async listQueue(threadId) {
+    return (this.queues.get(threadId) ?? []).map((message) => ({
+      ...message
+    }));
+  }
+  async getQueuedMessage(id) {
+    const found = this.findQueued(id);
+    return found === void 0 ? null : {
+      ...found.message
+    };
+  }
+  async updateQueuedMessage(id, patch) {
+    const found = this.findQueued(id);
+    if (found === void 0) {
+      return null;
+    }
+    const { message } = found;
+    if (patch.content !== void 0) {
+      message.content = patch.content;
+    }
+    if (patch.attachments !== void 0) {
+      if (patch.attachments === null || patch.attachments.length === 0) {
+        delete message.attachments;
+      } else {
+        message.attachments = patch.attachments;
+      }
+    }
+    if (patch.interrupt !== void 0) {
+      if (patch.interrupt) {
+        message.interrupt = true;
+      } else {
+        delete message.interrupt;
+      }
+    }
+    message.updatedAt = this.now();
+    return {
+      ...message
+    };
+  }
+  async moveQueuedMessage(id, index) {
+    const found = this.findQueued(id);
+    if (found === void 0) {
+      return false;
+    }
+    const { queue, position, message } = found;
+    queue.splice(position, 1);
+    const target = Math.max(0, Math.min(queue.length, Math.trunc(index)));
+    queue.splice(target, 0, message);
+    return true;
+  }
+  async removeQueuedMessage(id) {
+    const found = this.findQueued(id);
+    if (found === void 0) {
+      return false;
+    }
+    found.queue.splice(found.position, 1);
+    return true;
+  }
+  async clearQueue(threadId) {
+    const count = this.queues.get(threadId)?.length ?? 0;
+    this.queues.delete(threadId);
+    return count;
+  }
+  async queuePause(threadId) {
+    return this.pauses.get(threadId) ?? null;
+  }
+  async setQueuePause(threadId, pause) {
+    if (pause === null) {
+      this.pauses.delete(threadId);
+    } else {
+      this.pauses.set(threadId, pause);
+    }
+  }
+  findQueued(id) {
+    for (const queue of this.queues.values()) {
+      const position = queue.findIndex((message2) => message2.id === id);
+      const message = queue[position];
+      if (message !== void 0) {
+        return {
+          queue,
+          position,
+          message
+        };
+      }
+    }
+    return void 0;
+  }
   async appendMessage(input) {
     const row = this.threads.get(input.threadId);
     if (row === void 0) {
@@ -5454,6 +5650,7 @@ export {
   gateFollowUps,
   gateTail,
   invokeWithTransientRetry,
+  isChatQueueStore,
   isControlFlowSignal,
   isReplayIntegrityError,
   isToolEnabled,
@@ -5482,12 +5679,14 @@ export {
   publishAgentToolCall,
   publishAgentToolRetry,
   questionOptions,
+  queuedMessageView,
   quotaPeriodRange,
   quotaUsedRatio,
   quotaWarning,
   readElicitationInput,
   readElicitationQuestions,
   releaseGatedFrames,
+  releaseThreadRun,
   rememberInputSchema,
   rememberToolDefinition,
   renderElicitationAnswers,