@axiom-lattice/core 2.1.42 → 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>;
@@ -2290,6 +2313,7 @@ interface IMessageQueueStore {
     markProcessing(messageId: string): Promise<void>;
     markCompleted(messageId: string): Promise<void>;
     clearCompletedMessages(threadId: string): Promise<void>;
+    resetProcessingToPending(threadId: string): Promise<number>;
 }
 /**
  * Agent executor function type
@@ -2847,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>;
@@ -2857,6 +2881,7 @@ declare class InMemoryThreadMessageQueueStore implements IMessageQueueStore {
     markProcessing(messageId: string): Promise<void>;
     markCompleted(messageId: string): Promise<void>;
     clearCompletedMessages(threadId: string): Promise<void>;
+    resetProcessingToPending(threadId: string): Promise<number>;
 }
 
 /**
@@ -4926,6 +4951,7 @@ interface ThreadState {
 interface QueueMessage {
     input: {
         message: string;
+        id?: string;
     };
     command?: CommandParams<any>;
     custom_run_config?: any;
@@ -5007,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
      */
@@ -5048,9 +5075,9 @@ declare class Agent {
     }>;
     /**
      * Start queue processor if not already running
-     * Private method used internally by addMessage
+     * Public method to allow external triggering (e.g., from recovery)
      */
-    private startQueueProcessorIfNeeded;
+    startQueueProcessorIfNeeded(): Promise<void>;
     /**
      * Remove pending message
      */
@@ -5067,6 +5094,12 @@ declare class Agent {
     }[]>;
     get_draw_graph(): Promise<string>;
     getRunStatus(): Promise<ThreadStatus>;
+    /**
+     * Resume task processing after server restart
+     * Resets any stuck processing messages to pending and starts queue processing
+     * Note: Does not rely on LangGraph state as it may be stale after crash/restart
+     */
+    resumeTask(): Promise<void>;
     /**
      * Abort the current agent execution
      * This will cancel any ongoing invoke or stream operations
@@ -5129,6 +5162,20 @@ declare class AgentInstanceManager {
      * Clear all agent instances (use with caution)
      */
     clearAll(): void;
+    /**
+     * Restore agent instances for threads with pending messages after server restart
+     * Queries the message queue store for threads with pending/processing messages
+     * and recreates agent instances to resume processing
+     */
+    restore(): Promise<{
+        restored: number;
+        errors: number;
+    }>;
+    /**
+     * Restore a single thread
+     * Delegates actual recovery logic to Agent.resumeTask()
+     */
+    private restoreThread;
 }
 declare const agentInstanceManager: AgentInstanceManager;
 

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>;
@@ -2290,6 +2313,7 @@ interface IMessageQueueStore {
     markProcessing(messageId: string): Promise<void>;
     markCompleted(messageId: string): Promise<void>;
     clearCompletedMessages(threadId: string): Promise<void>;
+    resetProcessingToPending(threadId: string): Promise<number>;
 }
 /**
  * Agent executor function type
@@ -2847,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>;
@@ -2857,6 +2881,7 @@ declare class InMemoryThreadMessageQueueStore implements IMessageQueueStore {
     markProcessing(messageId: string): Promise<void>;
     markCompleted(messageId: string): Promise<void>;
     clearCompletedMessages(threadId: string): Promise<void>;
+    resetProcessingToPending(threadId: string): Promise<number>;
 }
 
 /**
@@ -4926,6 +4951,7 @@ interface ThreadState {
 interface QueueMessage {
     input: {
         message: string;
+        id?: string;
     };
     command?: CommandParams<any>;
     custom_run_config?: any;
@@ -5007,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
      */
@@ -5048,9 +5075,9 @@ declare class Agent {
     }>;
     /**
      * Start queue processor if not already running
-     * Private method used internally by addMessage
+     * Public method to allow external triggering (e.g., from recovery)
      */
-    private startQueueProcessorIfNeeded;
+    startQueueProcessorIfNeeded(): Promise<void>;
     /**
      * Remove pending message
      */
@@ -5067,6 +5094,12 @@ declare class Agent {
     }[]>;
     get_draw_graph(): Promise<string>;
     getRunStatus(): Promise<ThreadStatus>;
+    /**
+     * Resume task processing after server restart
+     * Resets any stuck processing messages to pending and starts queue processing
+     * Note: Does not rely on LangGraph state as it may be stale after crash/restart
+     */
+    resumeTask(): Promise<void>;
     /**
      * Abort the current agent execution
      * This will cancel any ongoing invoke or stream operations
@@ -5129,6 +5162,20 @@ declare class AgentInstanceManager {
      * Clear all agent instances (use with caution)
      */
     clearAll(): void;
+    /**
+     * Restore agent instances for threads with pending messages after server restart
+     * Queries the message queue store for threads with pending/processing messages
+     * and recreates agent instances to resume processing
+     */
+    restore(): Promise<{
+        restored: number;
+        errors: number;
+    }>;
+    /**
+     * Restore a single thread
+     * Delegates actual recovery logic to Agent.resumeTask()
+     */
+    private restoreThread;
 }
 declare const agentInstanceManager: AgentInstanceManager;