@axiom-lattice/core 2.1.40 → 2.1.41

package/dist/index.d.mts

@@ -5005,6 +5005,7 @@ declare class Agent {
     project_id: string | undefined;
     custom_run_config: any;
     queueMode: ThreadQueueConfig;
+    private isWaitingForQueueEnd;
     constructor({ assistant_id, thread_id, tenant_id, workspace_id, project_id, custom_run_config, }: AgentThreadInterface);
     /**
      * Initialize with message queue store
@@ -5024,6 +5025,7 @@ declare class Agent {
     private agentExecutor;
     getPendingMessages(): Promise<PendingMessage[]>;
     private agentStreamExecutor;
+    private tryToCompleteThread;
     private waitingForQueueEnd;
     private getQueueStore;
     private getDefaultQueueConfig;

package/dist/index.d.ts

@@ -5005,6 +5005,7 @@ declare class Agent {
     project_id: string | undefined;
     custom_run_config: any;
     queueMode: ThreadQueueConfig;
+    private isWaitingForQueueEnd;
     constructor({ assistant_id, thread_id, tenant_id, workspace_id, project_id, custom_run_config, }: AgentThreadInterface);
     /**
      * Initialize with message queue store
@@ -5024,6 +5025,7 @@ declare class Agent {
     private agentExecutor;
     getPendingMessages(): Promise<PendingMessage[]>;
     private agentStreamExecutor;
+    private tryToCompleteThread;
     private waitingForQueueEnd;
     private getQueueStore;
     private getDefaultQueueConfig;

package/dist/index.js

@@ -12264,17 +12264,11 @@ ${BASE_PROMPT}` : BASE_PROMPT;
     todoListMiddleware(),
     // Enables filesystem operations and optional long-term memory storage
     createFilesystemMiddleware({ backend: filesystemBackend }),
-    createClawMiddleware({
-      backend: filesystemBackend
-    }),
     // Enables delegation to specialized subagents for complex tasks
     createSubAgentMiddleware({
       defaultModel: model,
       defaultTools: tools,
       defaultMiddleware: [
-        createClawMiddleware({
-          backend: filesystemBackend
-        }),
         // Subagent middleware: Todo list management
         todoListMiddleware(),
         // Subagent middleware: Filesystem operations
@@ -14732,6 +14726,7 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
     buffer2.status = "active" /* ACTIVE */;
   }
   async completeThread(threadId) {
+    console.log("completeThread");
     const buffer2 = this.getBufferIfValid(threadId);
     if (buffer2) {
       buffer2.status = "completed" /* COMPLETED */;
@@ -14890,7 +14885,6 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
           if (!chunk) continue;
           if (chunk.data?.id === messageId) {
             startYieldChunk = true;
-            accumulatedContent += chunk.data?.content || "";
           }
           if (startYieldChunk) {
             yield chunk;
@@ -16968,6 +16962,7 @@ var Agent = class {
     this.chunkBuffer = getChunkBuffer("default");
     this.abortController = null;
     this.queueMode = this.getDefaultQueueConfig();
+    this.isWaitingForQueueEnd = false;
     this.agentExecutor = async ({ input, command, custom_run_config }, signal) => {
       const { runnable_agent, runConfig } = await this.getLatticeClientAndRuntimeConfig(custom_run_config);
       const { messages, ...rest } = input;
@@ -17114,7 +17109,7 @@ var Agent = class {
       while (!signal?.aborted) {
         const pendings = await this.getPendingMessages();
         if (pendings.length === 0) {
-          await this.chunkBuffer.completeThread(this.thread_id);
+          await this.tryToCompleteThread();
           const state = await this.getCurrentState();
           const pendingCount = await this.getQueueStore().getQueueSize(this.thread_id);
           this.publish("thread:idle", {
@@ -17141,11 +17136,13 @@ var Agent = class {
               queueMode: this.queueMode.mode
             });
             const input = {
-              messages: [new import_langchain61.HumanMessage({ id: p.content.id, content: p.content.message })],
-              command: p.content.command
+              messages: [new import_langchain61.HumanMessage({ id: p.content.id, content: p.content.message })]
             };
             try {
-              await this.agentStreamExecutor({ input }, signal);
+              await this.agentStreamExecutor({
+                input,
+                command: p.content.command
+              }, signal);
               await this.queueStore?.markCompleted(p.id);
               const state = await this.getCurrentState();
               this.publish("message:completed", {
@@ -17322,6 +17319,11 @@ var Agent = class {
     const store = this.getQueueStore();
     return await store.getPendingMessages(this.thread_id);
   }
+  async tryToCompleteThread() {
+    setTimeout(async () => {
+      await this.chunkBuffer.completeThread(this.thread_id);
+    }, 100);
+  }
   getQueueStore() {
     if (!this.queueStore) {
       try {