@axiom-lattice/core 2.1.44 → 2.1.45

package/dist/index.d.mts

@@ -2259,6 +2259,7 @@ interface PendingMessage {
     createdAt: Date;
     priority?: number;
     command?: CommandParams<unknown>;
+    custom_run_config?: unknown;
 }
 interface PendingQueueMessageInput {
     message: string;
@@ -2290,6 +2291,7 @@ interface AddMessageParams {
     type?: "human" | "system";
     priority?: number;
     command?: CommandParams<unknown>;
+    custom_run_config?: unknown;
     id?: string;
 }
 interface ThreadInfo {
@@ -2303,7 +2305,7 @@ interface ThreadInfo {
  */
 interface IMessageQueueStore {
     addMessage(params: AddMessageParams): Promise<PendingMessage>;
-    addMessageAtHead(threadId: string, content: PendingMessageContent, type?: "human" | "system", id?: string, command?: CommandParams<unknown>): Promise<PendingMessage>;
+    addMessageAtHead(params: AddMessageParams): Promise<PendingMessage>;
     getPendingMessages(threadId: string): Promise<PendingMessage[]>;
     getProcessingMessages(threadId: string): Promise<PendingMessage[]>;
     getQueueSize(threadId: string): Promise<number>;
@@ -2464,7 +2466,7 @@ declare class InMemoryThreadStore implements ThreadStore {
     /**
      * Get all threads for a specific tenant and assistant
      */
-    getThreadsByAssistantId(tenantId: string, assistantId: string): Promise<Thread[]>;
+    getThreadsByAssistantId(tenantId: string, assistantId: string, metadataFilter?: Record<string, string>): Promise<Thread[]>;
     /**
      * Get a thread by ID for a specific tenant
      */
@@ -2871,7 +2873,7 @@ declare class InMemoryThreadMessageQueueStore implements IMessageQueueStore {
     private generateId;
     private getMessagesForThread;
     addMessage(params: AddMessageParams): Promise<PendingMessage>;
-    addMessageAtHead(threadId: string, content: PendingMessageContent, type?: "human" | "system", id?: string, command?: CommandParams<unknown>): Promise<PendingMessage>;
+    addMessageAtHead(params: AddMessageParams): Promise<PendingMessage>;
     getPendingMessages(threadId: string): Promise<PendingMessage[]>;
     getProcessingMessages(threadId: string): Promise<PendingMessage[]>;
     getQueueSize(threadId: string): Promise<number>;

package/dist/index.d.ts

@@ -2259,6 +2259,7 @@ interface PendingMessage {
     createdAt: Date;
     priority?: number;
     command?: CommandParams<unknown>;
+    custom_run_config?: unknown;
 }
 interface PendingQueueMessageInput {
     message: string;
@@ -2290,6 +2291,7 @@ interface AddMessageParams {
     type?: "human" | "system";
     priority?: number;
     command?: CommandParams<unknown>;
+    custom_run_config?: unknown;
     id?: string;
 }
 interface ThreadInfo {
@@ -2303,7 +2305,7 @@ interface ThreadInfo {
  */
 interface IMessageQueueStore {
     addMessage(params: AddMessageParams): Promise<PendingMessage>;
-    addMessageAtHead(threadId: string, content: PendingMessageContent, type?: "human" | "system", id?: string, command?: CommandParams<unknown>): Promise<PendingMessage>;
+    addMessageAtHead(params: AddMessageParams): Promise<PendingMessage>;
     getPendingMessages(threadId: string): Promise<PendingMessage[]>;
     getProcessingMessages(threadId: string): Promise<PendingMessage[]>;
     getQueueSize(threadId: string): Promise<number>;
@@ -2464,7 +2466,7 @@ declare class InMemoryThreadStore implements ThreadStore {
     /**
      * Get all threads for a specific tenant and assistant
      */
-    getThreadsByAssistantId(tenantId: string, assistantId: string): Promise<Thread[]>;
+    getThreadsByAssistantId(tenantId: string, assistantId: string, metadataFilter?: Record<string, string>): Promise<Thread[]>;
     /**
      * Get a thread by ID for a specific tenant
      */
@@ -2871,7 +2873,7 @@ declare class InMemoryThreadMessageQueueStore implements IMessageQueueStore {
     private generateId;
     private getMessagesForThread;
     addMessage(params: AddMessageParams): Promise<PendingMessage>;
-    addMessageAtHead(threadId: string, content: PendingMessageContent, type?: "human" | "system", id?: string, command?: CommandParams<unknown>): Promise<PendingMessage>;
+    addMessageAtHead(params: AddMessageParams): Promise<PendingMessage>;
     getPendingMessages(threadId: string): Promise<PendingMessage[]>;
     getProcessingMessages(threadId: string): Promise<PendingMessage[]>;
     getQueueSize(threadId: string): Promise<number>;

package/dist/index.js

@@ -5200,7 +5200,7 @@ var InMemoryThreadStore = class {
   /**
    * Get all threads for a specific tenant and assistant
    */
-  async getThreadsByAssistantId(tenantId, assistantId) {
+  async getThreadsByAssistantId(tenantId, assistantId, metadataFilter) {
     const tenantThreads = this.threads.get(tenantId);
     if (!tenantThreads) {
       return [];
@@ -5209,7 +5209,15 @@ var InMemoryThreadStore = class {
     if (!assistantThreads) {
       return [];
     }
-    return Array.from(assistantThreads.values());
+    let threads = Array.from(assistantThreads.values());
+    if (metadataFilter && Object.keys(metadataFilter).length > 0) {
+      threads = threads.filter(
+        (thread) => Object.entries(metadataFilter).every(
+          ([key, value]) => thread.metadata?.[key] === value
+        )
+      );
+    }
+    return threads;
   }
   /**
    * Get a thread by ID for a specific tenant
@@ -6635,7 +6643,7 @@ var InMemoryThreadMessageQueueStore = class {
     return this.messages.get(threadId);
   }
   async addMessage(params) {
-    const { threadId, tenantId, assistantId, content, type = "human", priority = 0, command, id } = params;
+    const { threadId, tenantId, assistantId, content, type = "human", priority = 0, command, custom_run_config, id } = params;
     const threadMessages = this.getMessagesForThread(threadId);
     const message = {
       id: id || this.generateId(),
@@ -6647,18 +6655,20 @@ var InMemoryThreadMessageQueueStore = class {
       tenantId,
       assistantId,
       priority,
-      command
+      command,
+      custom_run_config
     };
     threadMessages.push(message);
     return message;
   }
-  async addMessageAtHead(threadId, content, type = "system", id, command) {
+  async addMessageAtHead(params) {
+    const { threadId, tenantId, assistantId, content, type = "system", id, command, custom_run_config } = params;
     const threadMessages = this.getMessagesForThread(threadId);
-    let tenantId = "default";
-    let assistantId = "";
-    if (threadMessages.length > 0) {
-      tenantId = threadMessages[0].tenantId || "default";
-      assistantId = threadMessages[0].assistantId || "";
+    let resolvedTenantId = tenantId || "default";
+    let resolvedAssistantId = assistantId || "";
+    if (!tenantId && !assistantId && threadMessages.length > 0) {
+      resolvedTenantId = threadMessages[0].tenantId || "default";
+      resolvedAssistantId = threadMessages[0].assistantId || "";
     }
     const nextSequence = threadMessages.length;
     const message = {
@@ -6668,11 +6678,12 @@ var InMemoryThreadMessageQueueStore = class {
       sequence: nextSequence,
       createdAt: /* @__PURE__ */ new Date(),
       status: "pending",
-      tenantId,
-      assistantId,
+      tenantId: resolvedTenantId,
+      assistantId: resolvedAssistantId,
       priority: 100,
       // High priority for head messages (STEER/Command)
-      command
+      command,
+      custom_run_config
     };
     threadMessages.push(message);
     return message;
@@ -17099,6 +17110,7 @@ var Agent = class {
           break;
         }
         const firstMessage = pendings[0];
+        const files = firstMessage.content?.queueMessage?.input?.files;
         const hasCommand = firstMessage.command;
         if (hasCommand) {
           for (const p of pendings) {
@@ -17117,12 +17129,15 @@ var Agent = class {
             const input = {
               messages: [new import_langchain61.HumanMessage({ id: humanContent.id, content: humanContent.message })]
             };
+            if (files) {
+              input.files = files;
+            }
             const queueMessageData = humanContent.queueMessage;
             try {
               await this.agentStreamExecutor({
                 input,
                 command: p.command,
-                custom_run_config: queueMessageData.custom_run_config
+                custom_run_config: p.custom_run_config ?? queueMessageData.custom_run_config
               }, signal);
               await this.queueStore?.markCompleted(p.id);
               const runStatus = await this.getRunStatus();
@@ -17188,10 +17203,16 @@ var Agent = class {
               });
             });
             const firstQueueMessage = remainingPendings[0] ? this.getHumanPendingContent(remainingPendings[0]).queueMessage : void 0;
+            const input = {
+              messages: userMessages
+            };
+            if (files) {
+              input.files = files;
+            }
             try {
               await this.agentStreamExecutor({
-                input: { messages: userMessages },
-                custom_run_config: firstQueueMessage?.custom_run_config
+                input,
+                custom_run_config: remainingPendings[0]?.custom_run_config ?? firstQueueMessage?.custom_run_config
               }, signal);
               const runStatus = await this.getRunStatus();
               const state = await this.getCurrentState();
@@ -17256,10 +17277,16 @@ var Agent = class {
                 queueMode: "followup" /* FOLLOWUP */
               });
               const queueMessageData = humanContent.queueMessage || {};
+              const input = {
+                messages: [message]
+              };
+              if (files) {
+                input.files = files;
+              }
               try {
                 await this.agentStreamExecutor({
-                  input: { messages: [message] },
-                  custom_run_config: queueMessageData.custom_run_config
+                  input,
+                  custom_run_config: p.custom_run_config ?? queueMessageData.custom_run_config
                 }, signal);
                 await this.queueStore?.markCompleted(p.id);
                 const runStatus = await this.getRunStatus();
@@ -17456,7 +17483,16 @@ var Agent = class {
       }
     };
     if (isHighPriority) {
-      await store.addMessageAtHead(this.thread_id, content, "human", messageId, queueMessage.command);
+      await store.addMessageAtHead({
+        threadId: this.thread_id,
+        tenantId: this.tenant_id,
+        assistantId: this.assistant_id,
+        content,
+        type: "human",
+        command: queueMessage.command,
+        custom_run_config: queueMessage.custom_run_config,
+        id: messageId
+      });
       if (useMode === "steer" /* STEER */) {
         this.stopQueueProcessor();
         await store.resetProcessingToPending(this.thread_id);
@@ -17470,6 +17506,7 @@ var Agent = class {
         content,
         type: "human",
         command: queueMessage.command,
+        custom_run_config: queueMessage.custom_run_config,
         id: messageId
       });
     }
@@ -17529,11 +17566,13 @@ var Agent = class {
         timestamp: (/* @__PURE__ */ new Date()).toISOString()
       }
     };
-    await this.queueStore.addMessageAtHead(
-      thread.threadId,
-      reminderContent,
-      "system"
-    );
+    await this.queueStore.addMessageAtHead({
+      threadId: thread.threadId,
+      tenantId: thread.tenantId,
+      assistantId: thread.assistantId,
+      content: reminderContent,
+      type: "system"
+    });
   }
   async getCurrentState() {
     const { runnable_agent } = await this.getLatticeClientAndRuntimeConfig();