@axiom-lattice/core 2.1.43 → 2.1.44

package/dist/index.d.mts

@@ -2253,21 +2253,44 @@ interface ThreadQueueConfig {
 }
 interface PendingMessage {
     id: string;
-    content: any;
+    content: PendingMessageContent;
     type: "human" | "system";
     sequence: number;
     createdAt: Date;
     priority?: number;
-    command?: any;
+    command?: CommandParams<unknown>;
 }
+interface PendingQueueMessageInput {
+    message: string;
+    id?: string;
+    messages?: unknown[];
+    [key: string]: unknown;
+}
+interface PendingQueueMessage {
+    input: PendingQueueMessageInput;
+    command?: CommandParams<unknown>;
+    custom_run_config?: unknown;
+}
+interface HumanPendingMessageContent {
+    id: string;
+    message: string;
+    queueMessage: PendingQueueMessage;
+}
+interface SystemPendingMessageContent {
+    role: "system";
+    content: string;
+    metadata?: Record<string, unknown>;
+}
+type PendingMessageContent = HumanPendingMessageContent | SystemPendingMessageContent | string;
 interface AddMessageParams {
     threadId: string;
     tenantId: string;
     assistantId: string;
-    content: any;
+    content: PendingMessageContent;
     type?: "human" | "system";
     priority?: number;
-    command?: any;
+    command?: CommandParams<unknown>;
+    id?: string;
 }
 interface ThreadInfo {
     tenantId: string;
@@ -2280,7 +2303,7 @@ interface ThreadInfo {
  */
 interface IMessageQueueStore {
     addMessage(params: AddMessageParams): Promise<PendingMessage>;
-    addMessageAtHead(threadId: string, content: any, type?: "human" | "system"): Promise<PendingMessage>;
+    addMessageAtHead(threadId: string, content: PendingMessageContent, type?: "human" | "system", id?: string, command?: CommandParams<unknown>): Promise<PendingMessage>;
     getPendingMessages(threadId: string): Promise<PendingMessage[]>;
     getProcessingMessages(threadId: string): Promise<PendingMessage[]>;
     getQueueSize(threadId: string): Promise<number>;
@@ -2848,7 +2871,7 @@ declare class InMemoryThreadMessageQueueStore implements IMessageQueueStore {
     private generateId;
     private getMessagesForThread;
     addMessage(params: AddMessageParams): Promise<PendingMessage>;
-    addMessageAtHead(threadId: string, content: any, type?: "human" | "system"): Promise<PendingMessage>;
+    addMessageAtHead(threadId: string, content: PendingMessageContent, type?: "human" | "system", id?: string, command?: CommandParams<unknown>): Promise<PendingMessage>;
     getPendingMessages(threadId: string): Promise<PendingMessage[]>;
     getProcessingMessages(threadId: string): Promise<PendingMessage[]>;
     getQueueSize(threadId: string): Promise<number>;
@@ -4928,6 +4951,7 @@ interface ThreadState {
 interface QueueMessage {
     input: {
         message: string;
+        id?: string;
     };
     command?: CommandParams<any>;
     custom_run_config?: any;
@@ -5009,6 +5033,7 @@ declare class Agent {
     queueMode: ThreadQueueConfig;
     private isWaitingForQueueEnd;
     constructor({ assistant_id, thread_id, tenant_id, workspace_id, project_id, custom_run_config, }: AgentThreadInterface);
+    private getHumanPendingContent;
     /**
      * Initialize with message queue store
      */

package/dist/index.d.ts

@@ -2253,21 +2253,44 @@ interface ThreadQueueConfig {
 }
 interface PendingMessage {
     id: string;
-    content: any;
+    content: PendingMessageContent;
     type: "human" | "system";
     sequence: number;
     createdAt: Date;
     priority?: number;
-    command?: any;
+    command?: CommandParams<unknown>;
 }
+interface PendingQueueMessageInput {
+    message: string;
+    id?: string;
+    messages?: unknown[];
+    [key: string]: unknown;
+}
+interface PendingQueueMessage {
+    input: PendingQueueMessageInput;
+    command?: CommandParams<unknown>;
+    custom_run_config?: unknown;
+}
+interface HumanPendingMessageContent {
+    id: string;
+    message: string;
+    queueMessage: PendingQueueMessage;
+}
+interface SystemPendingMessageContent {
+    role: "system";
+    content: string;
+    metadata?: Record<string, unknown>;
+}
+type PendingMessageContent = HumanPendingMessageContent | SystemPendingMessageContent | string;
 interface AddMessageParams {
     threadId: string;
     tenantId: string;
     assistantId: string;
-    content: any;
+    content: PendingMessageContent;
     type?: "human" | "system";
     priority?: number;
-    command?: any;
+    command?: CommandParams<unknown>;
+    id?: string;
 }
 interface ThreadInfo {
     tenantId: string;
@@ -2280,7 +2303,7 @@ interface ThreadInfo {
  */
 interface IMessageQueueStore {
     addMessage(params: AddMessageParams): Promise<PendingMessage>;
-    addMessageAtHead(threadId: string, content: any, type?: "human" | "system"): Promise<PendingMessage>;
+    addMessageAtHead(threadId: string, content: PendingMessageContent, type?: "human" | "system", id?: string, command?: CommandParams<unknown>): Promise<PendingMessage>;
     getPendingMessages(threadId: string): Promise<PendingMessage[]>;
     getProcessingMessages(threadId: string): Promise<PendingMessage[]>;
     getQueueSize(threadId: string): Promise<number>;
@@ -2848,7 +2871,7 @@ declare class InMemoryThreadMessageQueueStore implements IMessageQueueStore {
     private generateId;
     private getMessagesForThread;
     addMessage(params: AddMessageParams): Promise<PendingMessage>;
-    addMessageAtHead(threadId: string, content: any, type?: "human" | "system"): Promise<PendingMessage>;
+    addMessageAtHead(threadId: string, content: PendingMessageContent, type?: "human" | "system", id?: string, command?: CommandParams<unknown>): Promise<PendingMessage>;
     getPendingMessages(threadId: string): Promise<PendingMessage[]>;
     getProcessingMessages(threadId: string): Promise<PendingMessage[]>;
     getQueueSize(threadId: string): Promise<number>;
@@ -4928,6 +4951,7 @@ interface ThreadState {
 interface QueueMessage {
     input: {
         message: string;
+        id?: string;
     };
     command?: CommandParams<any>;
     custom_run_config?: any;
@@ -5009,6 +5033,7 @@ declare class Agent {
     queueMode: ThreadQueueConfig;
     private isWaitingForQueueEnd;
     constructor({ assistant_id, thread_id, tenant_id, workspace_id, project_id, custom_run_config, }: AgentThreadInterface);
+    private getHumanPendingContent;
     /**
      * Initialize with message queue store
      */

package/dist/index.js

@@ -6635,10 +6635,10 @@ var InMemoryThreadMessageQueueStore = class {
     return this.messages.get(threadId);
   }
   async addMessage(params) {
-    const { threadId, tenantId, assistantId, content, type = "human", priority = 0, command } = params;
+    const { threadId, tenantId, assistantId, content, type = "human", priority = 0, command, id } = params;
     const threadMessages = this.getMessagesForThread(threadId);
     const message = {
-      id: this.generateId(),
+      id: id || this.generateId(),
       content,
       type,
       sequence: threadMessages.length,
@@ -6652,7 +6652,7 @@ var InMemoryThreadMessageQueueStore = class {
     threadMessages.push(message);
     return message;
   }
-  async addMessageAtHead(threadId, content, type = "system") {
+  async addMessageAtHead(threadId, content, type = "system", id, command) {
     const threadMessages = this.getMessagesForThread(threadId);
     let tenantId = "default";
     let assistantId = "";
@@ -6662,7 +6662,7 @@ var InMemoryThreadMessageQueueStore = class {
     }
     const nextSequence = threadMessages.length;
     const message = {
-      id: this.generateId(),
+      id: id || this.generateId(),
       content,
       type,
       sequence: nextSequence,
@@ -6670,8 +6670,9 @@ var InMemoryThreadMessageQueueStore = class {
       status: "pending",
       tenantId,
       assistantId,
-      priority: 100
+      priority: 100,
       // High priority for head messages (STEER/Command)
+      command
     };
     threadMessages.push(message);
     return message;
@@ -17098,28 +17099,29 @@ var Agent = class {
           break;
         }
         const firstMessage = pendings[0];
-        const hasCommand = firstMessage.content?.command;
+        const hasCommand = firstMessage.command;
         if (hasCommand) {
           for (const p of pendings) {
             if (signal?.aborted) break;
-            if (!p.content?.command) break;
+            if (!p.command) break;
             await this.queueStore?.markProcessing(p.id);
             const startTime = Date.now();
             this.publish("message:started", {
               type: "message:started",
-              messageId: p.content.id,
-              messageContent: p.content.message,
+              messageId: this.getHumanPendingContent(p).id,
+              messageContent: this.getHumanPendingContent(p).message,
               timestamp: /* @__PURE__ */ new Date(),
               queueMode: this.queueMode.mode
             });
+            const humanContent = p.content;
             const input = {
-              messages: [new import_langchain61.HumanMessage({ id: p.content.id, content: p.content.message })]
+              messages: [new import_langchain61.HumanMessage({ id: humanContent.id, content: humanContent.message })]
             };
-            const queueMessageData = p.content.queueMessage || {};
+            const queueMessageData = humanContent.queueMessage;
             try {
               await this.agentStreamExecutor({
                 input,
-                command: queueMessageData.command || p.content.command,
+                command: p.command,
                 custom_run_config: queueMessageData.custom_run_config
               }, signal);
               await this.queueStore?.markCompleted(p.id);
@@ -17128,7 +17130,7 @@ var Agent = class {
               if (runStatus === "interrupted" /* INTERRUPTED */) {
                 this.publish("message:interrupted", {
                   type: "message:interrupted",
-                  messageId: p.content.id,
+                  messageId: humanContent.id,
                   timestamp: /* @__PURE__ */ new Date(),
                   duration: Date.now() - startTime,
                   state
@@ -17137,13 +17139,13 @@ var Agent = class {
                 this.addChunk({
                   type: "message_completed",
                   data: {
-                    id: p.content.id,
+                    id: humanContent.id,
                     content: ""
                   }
                 });
                 this.publish("message:completed", {
                   type: "message:completed",
-                  messageId: p.content.id,
+                  messageId: humanContent.id,
                   timestamp: /* @__PURE__ */ new Date(),
                   duration: Date.now() - startTime,
                   state
@@ -17154,13 +17156,13 @@ var Agent = class {
               this.addChunk({
                 type: "message_failed",
                 data: {
-                  id: p.content.id,
+                  id: humanContent.id,
                   content: error instanceof Error ? error.message : String(error)
                 }
               });
               this.publish("message:failed", {
                 type: "message:failed",
-                messageId: p.content.id,
+                messageId: humanContent.id,
                 error: error instanceof Error ? error.message : String(error),
                 timestamp: /* @__PURE__ */ new Date()
               });
@@ -17175,20 +17177,21 @@ var Agent = class {
             const startTime = Date.now();
             remainingPendings.forEach((p) => {
               this.queueStore?.markProcessing(p.id);
-              userMessages.push(new import_langchain61.HumanMessage({ id: p.content.id, content: p.content.message }));
+              const humanContent = p.content;
+              userMessages.push(new import_langchain61.HumanMessage({ id: humanContent.id, content: humanContent.message }));
               this.publish("message:started", {
                 type: "message:started",
-                messageId: p.content.id,
-                messageContent: p.content.message,
+                messageId: humanContent.id,
+                messageContent: humanContent.message,
                 timestamp: /* @__PURE__ */ new Date(),
                 queueMode: "collect" /* COLLECT */
               });
             });
-            const firstQueueMessage = remainingPendings[0]?.content?.queueMessage || {};
+            const firstQueueMessage = remainingPendings[0] ? this.getHumanPendingContent(remainingPendings[0]).queueMessage : void 0;
             try {
               await this.agentStreamExecutor({
                 input: { messages: userMessages },
-                custom_run_config: firstQueueMessage.custom_run_config
+                custom_run_config: firstQueueMessage?.custom_run_config
               }, signal);
               const runStatus = await this.getRunStatus();
               const state = await this.getCurrentState();
@@ -17242,16 +17245,17 @@ var Agent = class {
             for (const p of remainingPendings) {
               if (signal?.aborted) break;
               await this.queueStore?.markProcessing(p.id);
-              const message = new import_langchain61.HumanMessage({ id: p.id, content: p.content.message });
+              const humanContent = p.content;
+              const message = new import_langchain61.HumanMessage({ id: humanContent.id, content: humanContent.message });
               const startTime = Date.now();
               this.publish("message:started", {
                 type: "message:started",
-                messageId: p.content.id,
-                messageContent: p.content.message,
+                messageId: humanContent.id,
+                messageContent: humanContent.message,
                 timestamp: /* @__PURE__ */ new Date(),
                 queueMode: "followup" /* FOLLOWUP */
               });
-              const queueMessageData = p.content.queueMessage || {};
+              const queueMessageData = humanContent.queueMessage || {};
               try {
                 await this.agentStreamExecutor({
                   input: { messages: [message] },
@@ -17263,7 +17267,7 @@ var Agent = class {
                 if (runStatus === "interrupted" /* INTERRUPTED */) {
                   this.publish("message:interrupted", {
                     type: "message:interrupted",
-                    messageId: p.content.id,
+                    messageId: humanContent.id,
                     timestamp: /* @__PURE__ */ new Date(),
                     duration: Date.now() - startTime,
                     state
@@ -17272,13 +17276,13 @@ var Agent = class {
                   this.addChunk({
                     type: "message_completed",
                     data: {
-                      id: p.content.id,
+                      id: humanContent.id,
                       content: ""
                     }
                   });
                   this.publish("message:completed", {
                     type: "message:completed",
-                    messageId: p.content.id,
+                    messageId: humanContent.id,
                     timestamp: /* @__PURE__ */ new Date(),
                     duration: Date.now() - startTime,
                     state
@@ -17289,13 +17293,13 @@ var Agent = class {
                 this.addChunk({
                   type: "message_failed",
                   data: {
-                    id: p.content.id,
+                    id: humanContent.id,
                     content: error instanceof Error ? error.message : String(error)
                   }
                 });
                 this.publish("message:failed", {
                   type: "message:failed",
-                  messageId: p.content.id,
+                  messageId: humanContent.id,
                   error: error instanceof Error ? error.message : String(error),
                   timestamp: /* @__PURE__ */ new Date()
                 });
@@ -17313,6 +17317,12 @@ var Agent = class {
     this.project_id = project_id;
     this.custom_run_config = custom_run_config;
   }
+  getHumanPendingContent(message) {
+    if (typeof message.content === "string" || !message.content || !("id" in message.content) || !("message" in message.content)) {
+      throw new Error(`Expected human pending message content for message ${message.id}`);
+    }
+    return message.content;
+  }
   /**
    * Initialize with message queue store
    */
@@ -17427,7 +17437,7 @@ var Agent = class {
    */
   async addMessage(queueMessage, mode) {
     const useMode = mode ?? this.queueMode.mode;
-    const messageId = (0, import_uuid2.v4)();
+    const messageId = queueMessage.input.id || (0, import_uuid2.v4)();
     const isHighPriority = useMode === "steer" /* STEER */ || !!queueMessage.command;
     const store = this.getQueueStore();
     const currentSize = await store.getQueueSize(this.thread_id);
@@ -17446,14 +17456,21 @@ var Agent = class {
       }
     };
     if (isHighPriority) {
-      await store.addMessageAtHead(this.thread_id, content, "human");
+      await store.addMessageAtHead(this.thread_id, content, "human", messageId, queueMessage.command);
+      if (useMode === "steer" /* STEER */) {
+        this.stopQueueProcessor();
+        await store.resetProcessingToPending(this.thread_id);
+        console.log(`[Agent] STEER mode: stopped current processing, reset queue, message ${messageId} will execute next`);
+      }
     } else {
       await store.addMessage({
         threadId: this.thread_id,
         tenantId: this.tenant_id,
         assistantId: this.assistant_id,
         content,
-        type: "human"
+        type: "human",
+        command: queueMessage.command,
+        id: messageId
       });
     }
     this.startQueueProcessorIfNeeded().catch((err) => {
@@ -17486,7 +17503,7 @@ var Agent = class {
     this.publish("thread:busy", {
       type: "thread:busy",
       timestamp: /* @__PURE__ */ new Date(),
-      messageId: firstMessage?.content?.id
+      messageId: firstMessage ? this.getHumanPendingContent(firstMessage).id : void 0
     });
     this.waitingForQueueEnd(this.abortController.signal).catch((error) => {
       console.error(`Queue processing error for thread ${this.thread_id}:`, error);