npm - @axiom-lattice/gateway - Versions diffs - 2.1.43 → 2.1.45 - Mend

@axiom-lattice/gateway 2.1.43 → 2.1.45

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (18) hide show

package/.turbo/turbo-build.log +8 -8
package/CHANGELOG.md +18 -0
package/dist/index.d.mts +1 -1
package/dist/index.d.ts +1 -1
package/dist/index.js +234 -561
package/dist/index.js.map +1 -1
package/dist/index.mjs +65 -399
package/dist/index.mjs.map +1 -1
package/package.json +7 -4
package/src/controllers/assistant.ts +5 -4
package/src/controllers/memory.ts +17 -13
package/src/controllers/run.ts +62 -52
package/src/controllers/thread_status.ts +228 -0
package/src/index.ts +8 -1
package/src/routes/index.ts +18 -0
package/src/services/agent_task_consumer.ts +12 -107
package/src/__tests__/agent_service.test.ts +0 -238
package/src/services/agent_service.ts +0 -375

package/dist/index.js CHANGED Viewed

@@ -41,269 +41,10 @@ var import_websocket = __toESM(require("@fastify/websocket"));
 var import_static = __toESM(require("@fastify/static"));
 var import_path = __toESM(require("path"));
-// src/services/agent_service.ts
-var import_messages = require("@langchain/core/messages");
-var import_langgraph = require("@langchain/langgraph");
-var import_uuid = require("uuid");
-var import_core = require("@axiom-lattice/core");
-async function checkAgentExists(tenant_id, assistant_id) {
-  try {
-    const agentLattice = import_core.agentLatticeManager.getAgentLatticeWithTenant(tenant_id, assistant_id);
-    return agentLattice !== void 0;
-  } catch {
-    return false;
-  }
-}
-function getOrCreateChunkBuffer() {
-  if (!(0, import_core.hasChunkBuffer)("default")) {
-    const buffer = new import_core.InMemoryChunkBuffer({
-      ttl: 60 * 60 * 1e3,
-      // 1 hour TTL
-      cleanupInterval: 5 * 60 * 1e3
-      // Clean every 5 minutes
-    });
-    (0, import_core.registerChunkBuffer)("default", buffer);
-  }
-  return (0, import_core.getChunkBuffer)("default");
-}
-async function agent_invoke({
-  input,
-  thread_id,
-  assistant_id,
-  tenant_id,
-  workspace_id,
-  project_id,
-  command,
-  run_id,
-  custom_run_config
-}) {
-  const agentLattice = import_core.agentLatticeManager.getAgentLatticeWithTenant(tenant_id, assistant_id);
-  const runnable_agent = agentLattice?.client;
-  const { message, ...rest } = input;
-  const humanMessage = new import_messages.HumanMessage(message || "");
-  const messages = [humanMessage];
-  if (!runnable_agent) {
-    throw new Error(`Agent ${assistant_id} not found`);
-  }
-  const runConfig = {
-    ...agentLattice?.config?.runConfig || {},
-    tenantId: tenant_id,
-    workspaceId: workspace_id,
-    projectId: project_id,
-    ...custom_run_config || {},
-    assistant_id
-  };
-  const result = await runnable_agent.invoke(
-    command ? new import_langgraph.Command(command) : { ...rest, messages, "x-tenant-id": tenant_id },
-    {
-      context: {
-        runConfig
-      },
-      configurable: {
-        thread_id,
-        run_id: run_id || (0, import_uuid.v4)(),
-        "x-tenant-id": tenant_id,
-        "x-workspace-id": workspace_id,
-        "x-project-id": project_id,
-        "x-request-id": run_id,
-        "x-thread-id": thread_id,
-        "x-assistant-id": assistant_id,
-        runConfig
-      },
-      recursionLimit: 200
-    }
-  );
-  const data = result.messages.map((message2) => {
-    const { type, data: data2 } = message2.toDict();
-    return {
-      ...data2,
-      role: type
-    };
-  });
-  return { messages: data };
-}
-async function agent_stream({
-  input,
-  thread_id,
-  command,
-  tenant_id,
-  workspace_id,
-  project_id,
-  assistant_id,
-  run_id,
-  custom_run_config
-}) {
-  const runnable_agent = await (0, import_core.getAgentClient)(tenant_id, assistant_id);
-  const agentLattice = import_core.agentLatticeManager.getAgentLatticeWithTenant(tenant_id, assistant_id);
-  const { message, ...rest } = input;
-  let messages = [];
-  if (!command) {
-    const humanMessage = new import_messages.HumanMessage(message);
-    messages = [humanMessage];
-  }
-  const chunkBuffer = getOrCreateChunkBuffer();
-  const runConfig = {
-    ...agentLattice?.config?.runConfig || {},
-    tenantId: tenant_id,
-    workspaceId: workspace_id,
-    projectId: project_id,
-    ...custom_run_config || {},
-    assistant_id
-  };
-  try {
-    if (!runnable_agent) {
-      throw new Error(`Agent ${assistant_id} not found`);
-    }
-    const agentStream = await runnable_agent.stream(
-      command ? new import_langgraph.Command(command) : {
-        ...rest,
-        messages,
-        "x-tenant-id": tenant_id
-      },
-      {
-        context: {
-          runConfig
-        },
-        configurable: {
-          thread_id,
-          run_id: run_id || (0, import_uuid.v4)(),
-          "x-tenant-id": tenant_id,
-          "x-workspace-id": workspace_id,
-          "x-project-id": project_id,
-          "x-request-id": run_id,
-          "x-thread-id": thread_id,
-          "x-assistant-id": assistant_id,
-          runConfig
-          // Inject runConfig for tools to access
-        },
-        streamMode: ["updates", "messages"],
-        subgraphs: false,
-        recursionLimit: 200
-      }
-    );
-    return {
-      [Symbol.asyncIterator]: async function* () {
-        try {
-          for await (const chunk of agentStream) {
-            let data;
-            let chunkContent = "";
-            if (chunk[0] === "updates") {
-              const update = chunk[1];
-              const values = Object.values(update);
-              const messages2 = values[0]?.messages;
-              if (messages2?.[0]?.tool_call_id) {
-                data = messages2[0].toDict();
-              }
-            } else if (chunk[0] === "messages") {
-              const messages2 = chunk[1];
-              data = messages2?.[0]?.toDict();
-            }
-            if (chunk?.[1]?.__interrupt__) {
-              data = {
-                type: "interrupt",
-                id: chunk?.[1]?.__interrupt__[0].id,
-                data: { content: chunk?.[1]?.__interrupt__[0].value }
-              };
-            }
-            if (data) {
-              if (data.type !== "interrupt") {
-                await chunkBuffer.addChunk(thread_id, data);
-              }
-              yield data;
-            }
-          }
-          await chunkBuffer.completeThread(thread_id);
-        } catch (error) {
-          console.error("Stream error:", error);
-          await chunkBuffer.abortThread(thread_id);
-          throw error;
-        }
-      }
-    };
-  } catch (error) {
-    await chunkBuffer.abortThread(thread_id);
-    throw error;
-  }
-}
-async function agent_state({
-  assistant_id,
-  thread_id,
-  tenant_id
-}) {
-  const runnable_agent = await (0, import_core.getAgentClient)(tenant_id, assistant_id);
-  if (!runnable_agent) {
-    throw new Error(`Agent ${assistant_id} not found`);
-  }
-  const state = await runnable_agent.getState({
-    configurable: { thread_id, subgraphs: false }
-  });
-  return state;
-}
-async function agent_messages({
-  thread_id,
-  tenant_id,
-  assistant_id
-}) {
-  const runnable_agent = await (0, import_core.getAgentClient)(tenant_id, assistant_id);
-  if (!runnable_agent) {
-    throw new Error(`Agent ${assistant_id} not found`);
-  }
-  const state = await runnable_agent.getState({
-    configurable: { thread_id, subgraphs: false }
-  });
-  const messages = state.values.messages || [];
-  const filteredMessages = (0, import_messages.filterMessages)(messages, {
-    includeTypes: ["ai", "human", "tool"]
-    //["human", "ai", "tool"],
-  });
-  let messagesArray = filteredMessages.map((message) => ({
-    id: message.id,
-    role: message.getType(),
-    content: message.content,
-    ...message.lc_kwargs
-  }));
-  const new_messages = messagesArray;
-  return new_messages;
-}
-async function draw_graph(assistant_id, tenant_id) {
-  const runnable_agent = await (0, import_core.getAgentClient)(tenant_id, assistant_id);
-  if (!runnable_agent) {
-    throw new Error(`Agent ${assistant_id} not found`);
-  }
-  const drawableGraph = await runnable_agent.getGraphAsync();
-  const image = await drawableGraph.drawMermaid();
-  return image;
-}
-async function resume_stream({
-  thread_id,
-  message_id,
-  known_content,
-  poll_interval = 100
-}) {
-  const chunkBuffer = getOrCreateChunkBuffer();
-  const stream = await chunkBuffer.getNewChunksSinceContentIterator(
-    thread_id,
-    message_id,
-    known_content
-  );
-  return {
-    [Symbol.asyncIterator]: async function* () {
-      try {
-        for await (const chunk of stream) {
-          yield chunk;
-        }
-      } catch (error) {
-        console.error("Resume stream error:", error);
-        throw error;
-      }
-    }
-  };
-}
 // src/controllers/assistant.ts
-var import_core2 = require("@axiom-lattice/core");
+var import_core = require("@axiom-lattice/core");
 var import_crypto = require("crypto");
-var import_core3 = require("@axiom-lattice/core");
+var import_core2 = require("@axiom-lattice/core");
 function getTenantId(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
@@ -328,11 +69,11 @@ function convertAgentConfigToAssistant(config) {
 }
 async function getAssistantList(request, reply) {
   const tenantId = getTenantId(request);
-  const agentConfigs = await import_core3.agentLatticeManager.getAllAgentConfigsByTenant(tenantId);
+  const agentConfigs = await import_core2.agentLatticeManager.getAllAgentConfigsByTenant(tenantId);
   const codeConfiguredAssistants = agentConfigs.map(
     convertAgentConfigToAssistant
   );
-  const storeLattice = (0, import_core2.getStoreLattice)("default", "assistant");
+  const storeLattice = (0, import_core.getStoreLattice)("default", "assistant");
   const assistantStore = storeLattice.store;
   const storedAssistants = await assistantStore.getAllAssistants(tenantId);
   const assistantMap = /* @__PURE__ */ new Map();
@@ -355,11 +96,11 @@ async function getAssistantList(request, reply) {
 async function getAssistant(request, reply) {
   const { id } = request.params;
   const tenantId = getTenantId(request);
-  const storeLattice = (0, import_core2.getStoreLattice)("default", "assistant");
+  const storeLattice = (0, import_core.getStoreLattice)("default", "assistant");
   const assistantStore = storeLattice.store;
   let assistant = await assistantStore.getAssistantById(tenantId, id);
   if (!assistant) {
-    const agentConfig = await import_core3.agentLatticeManager.getAgentConfigWithTenant(tenantId, id);
+    const agentConfig = await import_core2.agentLatticeManager.getAgentConfigWithTenant(tenantId, id);
     if (agentConfig) {
       assistant = convertAgentConfigToAssistant(agentConfig);
     }
@@ -377,7 +118,7 @@ async function getAssistant(request, reply) {
   };
 }
 async function upsertAssistant(tenantId, id, data, reply, requireFields = false) {
-  const storeLattice = (0, import_core2.getStoreLattice)("default", "assistant");
+  const storeLattice = (0, import_core.getStoreLattice)("default", "assistant");
   const assistantStore = storeLattice.store;
   const exists = await assistantStore.hasAssistant(tenantId, id);
   let assistant;
@@ -389,7 +130,7 @@ async function upsertAssistant(tenantId, id, data, reply, requireFields = false)
         message: "Failed to update assistant"
       });
     }
-    import_core3.eventBus.publish("assistant:updated", { id: assistant.id, name: assistant.name, tenantId });
+    import_core2.eventBus.publish("assistant:updated", { id: assistant.id, name: assistant.name, tenantId });
     return {
       success: true,
       message: "Updated assistant",
@@ -406,7 +147,7 @@ async function upsertAssistant(tenantId, id, data, reply, requireFields = false)
     }
   }
   assistant = await assistantStore.createAssistant(tenantId, id, data);
-  import_core3.eventBus.publish("assistant:created", { id: assistant.id, name: assistant.name, tenantId });
+  import_core2.eventBus.publish("assistant:created", { id: assistant.id, name: assistant.name, tenantId });
   return reply.status(201).send({
     success: true,
     message: "Created assistant",
@@ -434,9 +175,9 @@ async function updateAssistant(request, reply) {
 async function deleteAssistant(request, reply) {
   const tenantId = getTenantId(request);
   const { id } = request.params;
-  const storeLattice = (0, import_core2.getStoreLattice)("default", "assistant");
+  const storeLattice = (0, import_core.getStoreLattice)("default", "assistant");
   const assistantStore = storeLattice.store;
-  const agentConfig = await import_core3.agentLatticeManager.getAgentConfigWithTenant(tenantId, id);
+  const agentConfig = await import_core2.agentLatticeManager.getAgentConfigWithTenant(tenantId, id);
   const isCodeConfigured = !!agentConfig;
   if (isCodeConfigured) {
     const exists2 = await assistantStore.hasAssistant(tenantId, id);
@@ -447,7 +188,7 @@ async function deleteAssistant(request, reply) {
       });
     }
     await assistantStore.deleteAssistant(tenantId, id);
-    import_core3.eventBus.publish("assistant:deleted", { id, tenantId });
+    import_core2.eventBus.publish("assistant:deleted", { id, tenantId });
     return {
       success: true,
       message: "Deleted assistant from store (code-configured registration remains)"
@@ -467,7 +208,7 @@ async function deleteAssistant(request, reply) {
       message: "Failed to delete assistant"
     });
   }
-  import_core3.eventBus.publish("assistant:deleted", { id, tenantId });
+  import_core2.eventBus.publish("assistant:deleted", { id, tenantId });
   return {
     success: true,
     message: "Successfully deleted assistant"
@@ -477,7 +218,8 @@ var getAgentGraph = async (request, reply) => {
   try {
     const { assistantId } = request.params;
     const tenant_id = getTenantId(request);
-    const imageData = await draw_graph(assistantId, tenant_id);
+    const agent = import_core.agentInstanceManager.getAgent({ assistant_id: assistantId, tenant_id, thread_id: "" });
+    const imageData = await agent.get_draw_graph();
     reply.header("Content-Type", "application/json").send({
       image: imageData
     });
@@ -490,7 +232,8 @@ var getAgentGraph = async (request, reply) => {
 };
 // src/controllers/run.ts
-var import_uuid2 = require("uuid");
+var import_uuid = require("uuid");
+var import_core3 = require("@axiom-lattice/core");
 var createRun = async (request, reply) => {
   try {
     const {
@@ -505,7 +248,7 @@ var createRun = async (request, reply) => {
     const tenant_id = request.headers["x-tenant-id"];
     const workspace_id = request.headers["x-workspace-id"];
     const project_id = request.headers["x-project-id"];
-    const x_request_id = request.headers["x-request-id"] || (0, import_uuid2.v4)();
+    const x_request_id = request.headers["x-request-id"] || (0, import_uuid.v4)();
     if (!assistant_id) {
       reply.status(400).send({
         success: false,
@@ -513,26 +256,15 @@ var createRun = async (request, reply) => {
       });
       return;
     }
+    const agent = import_core3.agentInstanceManager.getAgent({
+      assistant_id,
+      thread_id,
+      tenant_id,
+      workspace_id,
+      project_id,
+      custom_run_config
+    });
     if (streaming) {
-      const agentExists = await checkAgentExists(tenant_id, assistant_id);
-      if (!agentExists) {
-        reply.status(404).send({
-          success: false,
-          error: `Agent ${assistant_id} not found for tenant ${tenant_id}`
-        });
-        return;
-      }
-      const stream = await agent_stream({
-        assistant_id,
-        input,
-        thread_id,
-        command,
-        tenant_id,
-        workspace_id,
-        project_id,
-        run_id: x_request_id,
-        custom_run_config
-      });
       reply.hijack();
       reply.raw.writeHead(200, {
         "Content-Type": "text/event-stream",
@@ -541,9 +273,13 @@ var createRun = async (request, reply) => {
         "Access-Control-Allow-Origin": "*"
       });
       try {
-        let chunkCount = 0;
+        const result = await agent.addMessage({
+          input,
+          command
+        });
+        const stream = agent.chunkStream(result.messageId);
         for await (const chunk of stream) {
-          chunkCount++;
+          console.log(input.message, chunk.data.content);
           const success = reply.raw.write(`data: ${JSON.stringify(chunk)}
 `);
@@ -555,7 +291,7 @@ var createRun = async (request, reply) => {
         const errorEvent = {
           type: "error",
           data: {
-            id: (0, import_uuid2.v4)(),
+            id: (0, import_uuid.v4)(),
             content: error.message || "Stream processing error"
           }
         };
@@ -566,18 +302,15 @@ var createRun = async (request, reply) => {
         reply.raw.end();
       }
     } else {
-      const result = await agent_invoke({
-        assistant_id,
-        input,
-        command,
-        thread_id,
-        tenant_id,
-        workspace_id,
-        project_id,
-        run_id: x_request_id,
-        custom_run_config
+      const { message: msg, ...restInputNonStream } = input;
+      const result = await agent.invoke({
+        input: { message: msg, ...restInputNonStream },
+        command
+      });
+      reply.status(200).send({
+        success: true,
+        ...result
       });
-      reply.status(200).send(result);
     }
   } catch (error) {
     reply.status(500).send({
@@ -604,12 +337,12 @@ var resumeStream = async (request, reply) => {
       "Access-Control-Allow-Origin": "*"
     });
     try {
-      const stream = await resume_stream({
+      const agent = new import_core3.Agent({
+        assistant_id: "",
         thread_id,
-        message_id,
-        known_content,
-        poll_interval: poll_interval || 100
+        tenant_id: ""
       });
+      const stream = agent.chunkStream(message_id, known_content);
       for await (const chunk of stream) {
         reply.raw.write(`data: ${JSON.stringify(chunk)}
@@ -619,7 +352,7 @@ var resumeStream = async (request, reply) => {
       const errorEvent = {
         type: "error",
         data: {
-          id: (0, import_uuid2.v4)(),
+          id: (0, import_uuid.v4)(),
           content: error.message || "Resume stream processing error"
         }
       };
@@ -638,6 +371,7 @@ var resumeStream = async (request, reply) => {
 };
 // src/controllers/memory.ts
+var import_core4 = require("@axiom-lattice/core");
 var setMemoryItem = async (request, reply) => {
   try {
     const { assistantId, key } = request.params;
@@ -702,11 +436,8 @@ var getAllMemoryItems = async (request, reply) => {
       });
       return;
     }
-    const result = await agent_messages({
-      assistant_id: assistantId,
-      thread_id,
-      tenant_id
-    });
+    const agent = import_core4.agentInstanceManager.getAgent({ assistant_id: assistantId, tenant_id, thread_id });
+    const result = await agent.getCurrentMessages();
     if (!result) {
       reply.status(500).send(result);
       return;
@@ -737,16 +468,18 @@ var getAgentState = async (request, reply) => {
       return;
     }
     const tenant_id = request.headers["x-tenant-id"];
-    const result = await agent_state({
-      assistant_id: assistantId,
-      thread_id,
-      tenant_id
-    });
+    const agent = import_core4.agentInstanceManager.getAgent({ assistant_id: assistantId, tenant_id, thread_id });
+    const result = await agent.getCurrentState();
+    const pendingMessages = await agent.getPendingMessages();
     if (!result) {
       reply.status(500).send(result);
       return;
     }
-    reply.send(result);
+    const mergedResult = {
+      ...result,
+      pendingMessages
+    };
+    reply.send(mergedResult);
   } catch (error) {
     reply.status(500).send({
       success: false,
@@ -799,7 +532,7 @@ var clearMemory = async (request, reply) => {
 };
 // src/controllers/agent_task.ts
-var import_core4 = require("@axiom-lattice/core");
+var import_core5 = require("@axiom-lattice/core");
 var triggerAgentTask = async (request, reply) => {
   try {
     const { assistant_id, thread_id, input, command } = request.body;
@@ -818,7 +551,7 @@ var triggerAgentTask = async (request, reply) => {
       });
       return;
     }
-    const agentManager = import_core4.AgentManager.getInstance();
+    const agentManager = import_core5.AgentManager.getInstance();
     const result = await agentManager.callAgentInQueue({
       assistant_id,
       thread_id,
@@ -838,7 +571,7 @@ var triggerAgentTask = async (request, reply) => {
 };
 // src/controllers/threads.ts
-var import_core5 = require("@axiom-lattice/core");
+var import_core6 = require("@axiom-lattice/core");
 var import_crypto2 = require("crypto");
 function getTenantId2(request) {
   const userTenantId = request.user?.tenantId;
@@ -850,7 +583,7 @@ function getTenantId2(request) {
 async function getThreadList(request, reply) {
   const tenantId = getTenantId2(request);
   const { assistantId } = request.params;
-  const storeLattice = (0, import_core5.getStoreLattice)("default", "thread");
+  const storeLattice = (0, import_core6.getStoreLattice)("default", "thread");
   const threadStore = storeLattice.store;
   const threads = await threadStore.getThreadsByAssistantId(tenantId, assistantId);
   return {
@@ -865,7 +598,7 @@ async function getThreadList(request, reply) {
 async function getThread(request, reply) {
   const tenantId = getTenantId2(request);
   const { threadId } = request.params;
-  const storeLattice = (0, import_core5.getStoreLattice)("default", "thread");
+  const storeLattice = (0, import_core6.getStoreLattice)("default", "thread");
   const threadStore = storeLattice.store;
   const thread = await threadStore.getThreadById(tenantId, threadId);
   if (!thread) {
@@ -885,7 +618,7 @@ async function createThread(request, reply) {
   const { assistantId } = request.params;
   const data = request.body;
   const threadId = (0, import_crypto2.randomUUID)();
-  const storeLattice = (0, import_core5.getStoreLattice)("default", "thread");
+  const storeLattice = (0, import_core6.getStoreLattice)("default", "thread");
   const threadStore = storeLattice.store;
   const newThread = await threadStore.createThread(tenantId, assistantId, threadId, data);
   return reply.status(201).send({
@@ -898,7 +631,7 @@ async function updateThread(request, reply) {
   const tenantId = getTenantId2(request);
   const { threadId } = request.params;
   const updates = request.body;
-  const storeLattice = (0, import_core5.getStoreLattice)("default", "thread");
+  const storeLattice = (0, import_core6.getStoreLattice)("default", "thread");
   const threadStore = storeLattice.store;
   const exists = await threadStore.hasThread(tenantId, threadId);
   if (!exists) {
@@ -927,7 +660,7 @@ async function updateThread(request, reply) {
 async function deleteThread(request, reply) {
   const tenantId = getTenantId2(request);
   const { threadId } = request.params;
-  const storeLattice = (0, import_core5.getStoreLattice)("default", "thread");
+  const storeLattice = (0, import_core6.getStoreLattice)("default", "thread");
   const threadStore = storeLattice.store;
   const exists = await threadStore.hasThread(tenantId, threadId);
   if (!exists) {
@@ -950,7 +683,7 @@ async function deleteThread(request, reply) {
 }
 // src/controllers/schedules.ts
-var import_core6 = require("@axiom-lattice/core");
+var import_core7 = require("@axiom-lattice/core");
 function getTenantId3(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
@@ -959,14 +692,14 @@ function getTenantId3(request) {
   return request.headers["x-tenant-id"] || "default";
 }
 function getScheduleLattice() {
-  const keys = import_core6.scheduleLatticeManager.getLatticeKeys();
+  const keys = import_core7.scheduleLatticeManager.getLatticeKeys();
   if (keys.length === 0) {
     return null;
   }
-  if (import_core6.scheduleLatticeManager.hasLattice("default")) {
-    return import_core6.scheduleLatticeManager.getScheduleLattice("default");
+  if (import_core7.scheduleLatticeManager.hasLattice("default")) {
+    return import_core7.scheduleLatticeManager.getScheduleLattice("default");
   }
-  return import_core6.scheduleLatticeManager.getScheduleLattice(keys[0]);
+  return import_core7.scheduleLatticeManager.getScheduleLattice(keys[0]);
 }
 async function getThreadSchedules(request, reply) {
   const { assistantId, threadId } = request.params;
@@ -1227,7 +960,7 @@ var ConfigService = class {
 var configService = new ConfigService();
 // src/services/queue_service.ts
-var import_core7 = require("@axiom-lattice/core");
+var import_core8 = require("@axiom-lattice/core");
 var import_protocols = require("@axiom-lattice/protocols");
 var import_queue_redis = require("@axiom-lattice/queue-redis");
 var DEFAULT_QUEUE_KEY = "default";
@@ -1246,8 +979,8 @@ var setQueueServiceType = (type) => {
       redisPassword: process.env.REDIS_PASSWORD
     } : void 0
   };
-  if (import_core7.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
-    import_core7.queueLatticeManager.removeLattice(DEFAULT_QUEUE_KEY);
+  if (import_core8.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
+    import_core8.queueLatticeManager.removeLattice(DEFAULT_QUEUE_KEY);
   }
   let client;
   if (type === "redis") {
@@ -1256,13 +989,13 @@ var setQueueServiceType = (type) => {
       redisPassword: process.env.REDIS_PASSWORD
     });
   }
-  (0, import_core7.registerQueueLattice)(DEFAULT_QUEUE_KEY, config, client);
+  (0, import_core8.registerQueueLattice)(DEFAULT_QUEUE_KEY, config, client);
 };
 var getQueueService = () => {
-  if (!import_core7.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
+  if (!import_core8.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
     setQueueServiceType(queueServiceType);
   }
-  return (0, import_core7.getQueueLattice)(DEFAULT_QUEUE_KEY);
+  return (0, import_core8.getQueueLattice)(DEFAULT_QUEUE_KEY);
 };
 var popAgentTaskFromQueue = async () => {
   const queue = getQueueService();
@@ -1343,10 +1076,10 @@ async function getConfig(request, reply) {
 }
 // src/controllers/models.ts
-var import_core8 = require("@axiom-lattice/core");
+var import_core9 = require("@axiom-lattice/core");
 async function getModels(request, reply) {
   try {
-    const allLattices = import_core8.modelLatticeManager.getAllLattices();
+    const allLattices = import_core9.modelLatticeManager.getAllLattices();
     const models = allLattices.map((lattice) => {
       const config = lattice.client.config || {};
       return {
@@ -1396,8 +1129,8 @@ async function updateModels(request, reply) {
         continue;
       }
       try {
-        if (import_core8.modelLatticeManager.hasLattice(modelConfig.key)) {
-          import_core8.modelLatticeManager.removeLattice(modelConfig.key);
+        if (import_core9.modelLatticeManager.hasLattice(modelConfig.key)) {
+          import_core9.modelLatticeManager.removeLattice(modelConfig.key);
         }
         const llmConfig = {
           provider: modelConfig.provider,
@@ -1410,7 +1143,7 @@ async function updateModels(request, reply) {
           timeout: modelConfig.timeout,
           maxRetries: modelConfig.maxRetries
         };
-        (0, import_core8.registerModelLattice)(modelConfig.key, llmConfig);
+        (0, import_core9.registerModelLattice)(modelConfig.key, llmConfig);
         registeredModels.push(modelConfig.key);
       } catch (error) {
         errors.push(
@@ -1468,8 +1201,8 @@ async function getHealth(request, reply) {
 }
 // src/controllers/skills.ts
-var import_core9 = require("@axiom-lattice/core");
 var import_core10 = require("@axiom-lattice/core");
+var import_core11 = require("@axiom-lattice/core");
 function getTenantId4(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
@@ -1500,7 +1233,7 @@ function serializeSkill(skill) {
 async function getSkillList(request, reply) {
   try {
     const tenantId = getTenantId4(request);
-    const storeLattice = (0, import_core9.getStoreLattice)("default", "skill");
+    const storeLattice = (0, import_core10.getStoreLattice)("default", "skill");
     const skillStore = storeLattice.store;
     const skills = await skillStore.getAllSkills(tenantId);
     const serializedSkills = skills.map(serializeSkill);
@@ -1527,7 +1260,7 @@ async function getSkill(request, reply) {
   try {
     const tenantId = getTenantId4(request);
     const { id } = request.params;
-    const storeLattice = (0, import_core9.getStoreLattice)("default", "skill");
+    const storeLattice = (0, import_core10.getStoreLattice)("default", "skill");
     const skillStore = storeLattice.store;
     const skill = await skillStore.getSkillById(tenantId, id);
     if (!skill) {
@@ -1564,7 +1297,7 @@ async function createSkill(request, reply) {
       });
     }
     try {
-      (0, import_core10.validateSkillName)(data.name);
+      (0, import_core11.validateSkillName)(data.name);
     } catch (error) {
       return reply.status(400).send({
         success: false,
@@ -1579,7 +1312,7 @@ async function createSkill(request, reply) {
       });
     }
     const tenantId = getTenantId4(request);
-    const storeLattice = (0, import_core9.getStoreLattice)("default", "skill");
+    const storeLattice = (0, import_core10.getStoreLattice)("default", "skill");
     const skillStore = storeLattice.store;
     const exists = await skillStore.hasSkill(tenantId, id);
     if (exists) {
@@ -1607,7 +1340,7 @@ async function updateSkill(request, reply) {
     const updates = request.body;
     if (updates.name !== void 0) {
       try {
-        (0, import_core10.validateSkillName)(updates.name);
+        (0, import_core11.validateSkillName)(updates.name);
       } catch (error) {
         return reply.status(400).send({
           success: false,
@@ -1616,7 +1349,7 @@ async function updateSkill(request, reply) {
       }
     }
     const tenantId = getTenantId4(request);
-    const storeLattice = (0, import_core9.getStoreLattice)("default", "skill");
+    const storeLattice = (0, import_core10.getStoreLattice)("default", "skill");
     const skillStore = storeLattice.store;
     const exists = await skillStore.hasSkill(tenantId, id);
     if (!exists) {
@@ -1648,7 +1381,7 @@ async function deleteSkill(request, reply) {
   try {
     const { id } = request.params;
     const tenantId = getTenantId4(request);
-    const storeLattice = (0, import_core9.getStoreLattice)("default", "skill");
+    const storeLattice = (0, import_core10.getStoreLattice)("default", "skill");
     const skillStore = storeLattice.store;
     const exists = await skillStore.hasSkill(tenantId, id);
     if (!exists) {
@@ -1689,7 +1422,7 @@ async function searchSkillsByMetadata(request, reply) {
       });
     }
     const tenantId = getTenantId4(request);
-    const storeLattice = (0, import_core9.getStoreLattice)("default", "skill");
+    const storeLattice = (0, import_core10.getStoreLattice)("default", "skill");
     const skillStore = storeLattice.store;
     const skills = await skillStore.searchByMetadata(tenantId, key, value);
     const serializedSkills = skills.map(serializeSkill);
@@ -1726,7 +1459,7 @@ async function filterSkillsByCompatibility(request, reply) {
       });
     }
     const tenantId = getTenantId4(request);
-    const storeLattice = (0, import_core9.getStoreLattice)("default", "skill");
+    const storeLattice = (0, import_core10.getStoreLattice)("default", "skill");
     const skillStore = storeLattice.store;
     const skills = await skillStore.filterByCompatibility(tenantId, compatibility);
     const serializedSkills = skills.map(serializeSkill);
@@ -1763,7 +1496,7 @@ async function filterSkillsByLicense(request, reply) {
       });
     }
     const tenantId = getTenantId4(request);
-    const storeLattice = (0, import_core9.getStoreLattice)("default", "skill");
+    const storeLattice = (0, import_core10.getStoreLattice)("default", "skill");
     const skillStore = storeLattice.store;
     const skills = await skillStore.filterByLicense(tenantId, license);
     const serializedSkills = skills.map(serializeSkill);
@@ -1788,7 +1521,7 @@ async function filterSkillsByLicense(request, reply) {
 }
 // src/controllers/tools.ts
-var import_core11 = require("@axiom-lattice/core");
+var import_core12 = require("@axiom-lattice/core");
 function serializeSchema(schema) {
   if (!schema) {
     return void 0;
@@ -1832,7 +1565,7 @@ function serializeSchema(schema) {
 }
 async function getToolConfigs(request, reply) {
   try {
-    const allLattices = import_core11.toolLatticeManager.getAllLattices();
+    const allLattices = import_core12.toolLatticeManager.getAllLattices();
     const toolConfigs = allLattices.map((lattice) => {
       const config = { ...lattice.config };
       const serializedSchema = config.schema ? serializeSchema(config.schema) : void 0;
@@ -1870,7 +1603,7 @@ async function getToolConfigs(request, reply) {
 }
 // src/controllers/data-query.ts
-var import_core12 = require("@axiom-lattice/core");
+var import_core13 = require("@axiom-lattice/core");
 function getTenantId5(request) {
   return request.headers["x-tenant-id"] || "default";
 }
@@ -1901,7 +1634,7 @@ async function executeDataQuery(request, reply) {
         message: "Cannot provide both metrics and customSql. Use one query type only."
       };
     }
-    const storeLattice = (0, import_core12.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core13.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     if (!body.serverKey) {
       reply.code(400);
@@ -1932,14 +1665,14 @@ async function executeDataQuery(request, reply) {
         message: "datasourceId is required"
       };
     }
-    if (!import_core12.metricsServerManager.hasServer(tenantId, body.serverKey)) {
+    if (!import_core13.metricsServerManager.hasServer(tenantId, body.serverKey)) {
       reply.code(400);
       return {
         success: false,
         message: `Metrics server not registered: ${body.serverKey}. Please register the server first.`
       };
     }
-    const client = import_core12.metricsServerManager.getClient(tenantId, body.serverKey);
+    const client = import_core13.metricsServerManager.getClient(tenantId, body.serverKey);
     if (isSemanticQuery) {
       return await executeSemanticQuery(client, body, reply);
     } else {
@@ -2350,7 +2083,7 @@ var getHealthSchema = {
 var import_stream = require("stream");
 // src/services/sandbox_service.ts
-var import_core13 = require("@axiom-lattice/core");
+var import_core14 = require("@axiom-lattice/core");
 var ERROR_HTML = `<!DOCTYPE html>
 <html lang="zh-CN">
 <head>
@@ -2462,7 +2195,7 @@ var ERROR_HTML = `<!DOCTYPE html>
 </html>`;
 var SandboxService = class {
   getFilesystemIsolatedLevel(tenantId, assistantId) {
-    const agentLattice = import_core13.agentLatticeManager.getAgentLatticeWithTenant(tenantId, assistantId);
+    const agentLattice = import_core14.agentLatticeManager.getAgentLatticeWithTenant(tenantId, assistantId);
     if (!agentLattice) {
       return null;
     }
@@ -2487,10 +2220,10 @@ var SandboxService = class {
         sandboxName = "global";
         break;
     }
-    return (0, import_core13.normalizeSandboxName)(sandboxName);
+    return (0, import_core14.normalizeSandboxName)(sandboxName);
   }
   getTargetUrl(sandboxName) {
-    const sandboxManager = (0, import_core13.getSandBoxManager)();
+    const sandboxManager = (0, import_core14.getSandBoxManager)();
     return `${sandboxManager.getBaseURL()}/sandbox/${sandboxName}`;
   }
   async getVncHtml(sandboxName) {
@@ -2535,7 +2268,7 @@ var SandboxService = class {
 var sandboxService = new SandboxService();
 // src/controllers/sandbox.ts
-var import_core14 = require("@axiom-lattice/core");
+var import_core15 = require("@axiom-lattice/core");
 function getFilenameFromPath(path3) {
   const segments = path3.replace(/\/+$/, "").split("/");
   return segments[segments.length - 1] || "download";
@@ -2577,7 +2310,7 @@ function registerSandboxProxyRoutes(app2) {
         threadId,
         isolatedLevel
       );
-      const sandboxManager = (0, import_core14.getSandBoxManager)();
+      const sandboxManager = (0, import_core15.getSandBoxManager)();
       const sandbox = await sandboxManager.createSandbox(sandboxName);
       try {
         const data = await request.file();
@@ -2624,7 +2357,7 @@ function registerSandboxProxyRoutes(app2) {
         threadId,
         isolatedLevel
       );
-      const sandboxManager = (0, import_core14.getSandBoxManager)();
+      const sandboxManager = (0, import_core15.getSandBoxManager)();
       const sandbox = await sandboxManager.createSandbox(sandboxName);
       try {
         const resolvedPath = filePath.startsWith("/home/gem") ? filePath : `/home/gem/${filePath.replace(/^\//, "")}`;
@@ -2680,14 +2413,14 @@ function registerSandboxProxyRoutes(app2) {
 var fs = __toESM(require("fs/promises"));
 var path = __toESM(require("path"));
 var import_stream2 = require("stream");
-var import_core15 = require("@axiom-lattice/core");
 var import_core16 = require("@axiom-lattice/core");
 var import_core17 = require("@axiom-lattice/core");
-var import_uuid3 = require("uuid");
+var import_core18 = require("@axiom-lattice/core");
+var import_uuid2 = require("uuid");
 var WorkspaceController = class {
   constructor() {
-    this.workspaceStore = (0, import_core15.getStoreLattice)("default", "workspace").store;
-    this.projectStore = (0, import_core15.getStoreLattice)("default", "project").store;
+    this.workspaceStore = (0, import_core16.getStoreLattice)("default", "workspace").store;
+    this.projectStore = (0, import_core16.getStoreLattice)("default", "project").store;
   }
   getTenantId(request) {
     const userTenantId = request.user?.tenantId;
@@ -2709,7 +2442,7 @@ var WorkspaceController = class {
   async createWorkspace(request, reply) {
     const tenantId = this.getTenantId(request);
     const data = request.body;
-    const id = (0, import_uuid3.v4)();
+    const id = (0, import_uuid2.v4)();
     const workspace = await this.workspaceStore.createWorkspace(
       tenantId,
       id,
@@ -2769,7 +2502,7 @@ var WorkspaceController = class {
     const tenantId = this.getTenantId(request);
     const { workspaceId } = request.params;
     const data = request.body;
-    const id = (0, import_uuid3.v4)();
+    const id = (0, import_uuid2.v4)();
     const project = await this.projectStore.createProject(
       tenantId,
       workspaceId,
@@ -2820,11 +2553,11 @@ var WorkspaceController = class {
       throw new Error("Workspace not found");
     }
     if (workspace.storageType === "sandbox") {
-      const sandboxManager = (0, import_core17.getSandBoxManager)();
+      const sandboxManager = (0, import_core18.getSandBoxManager)();
       const sandboxName = "global";
       const sandbox = await sandboxManager.createSandbox(sandboxName);
       return {
-        backend: new import_core16.SandboxFilesystem({
+        backend: new import_core17.SandboxFilesystem({
           sandboxInstance: sandbox,
           workingDirectory: `/tenants/${tenantId}/workspaces/${workspaceId}/${projectId}`
         }),
@@ -2832,7 +2565,7 @@ var WorkspaceController = class {
       };
     } else {
       return {
-        backend: new import_core16.FilesystemBackend({
+        backend: new import_core17.FilesystemBackend({
           rootDir: `/lattice_store/tenants/${tenantId}/workspaces/${workspaceId}/${projectId}`,
           virtualMode: true
         }),
@@ -2877,7 +2610,7 @@ var WorkspaceController = class {
       const { workspace } = await this.getBackend(tenantId, workspaceId, projectId);
       const resolvedPath = filePath.startsWith("/") ? filePath : `/${filePath}`;
       if (workspace.storageType === "sandbox") {
-        const sandboxManager = (0, import_core17.getSandBoxManager)();
+        const sandboxManager = (0, import_core18.getSandBoxManager)();
         const sandbox = await sandboxManager.createSandbox("global");
         const realPath = path.join("/home/gem/tenants", tenantId, "workspaces", workspaceId, projectId, resolvedPath);
         const filename2 = this.getFilenameFromPath(resolvedPath);
@@ -2946,7 +2679,7 @@ var WorkspaceController = class {
       const { workspace } = await this.getBackend(tenantId, workspaceId, projectId);
       const resolvedPath = filePath.startsWith("/") ? filePath : `/${filePath}`;
       if (workspace.storageType === "sandbox") {
-        const sandboxManager = (0, import_core17.getSandBoxManager)();
+        const sandboxManager = (0, import_core18.getSandBoxManager)();
         const sandbox = await sandboxManager.createSandbox("global");
         const realPath = path.join("/home/gem/tenants", tenantId, "workspaces", workspaceId, projectId, resolvedPath);
         const filename2 = this.getFilenameFromPath(resolvedPath);
@@ -3116,7 +2849,7 @@ var WorkspaceController = class {
         return reply.status(400).send({ success: false, error: "Invalid path parameter" });
       }
       if (workspace.storageType === "sandbox") {
-        const sandboxManager = (0, import_core17.getSandBoxManager)();
+        const sandboxManager = (0, import_core18.getSandBoxManager)();
         const sandboxName = "global";
         const sandbox = await sandboxManager.createSandbox(sandboxName);
         const baseDir = path.join("/home/gem/tenants", tenantId, "workspaces", workspaceId, projectId);
@@ -3222,7 +2955,7 @@ function registerWorkspaceRoutes(app2) {
 }
 // src/controllers/database-configs.ts
-var import_core18 = require("@axiom-lattice/core");
+var import_core19 = require("@axiom-lattice/core");
 var import_crypto3 = require("crypto");
 function getTenantId6(request) {
   const userTenantId = request.user?.tenantId;
@@ -3234,7 +2967,7 @@ function getTenantId6(request) {
 async function getDatabaseConfigList(request, reply) {
   const tenantId = getTenantId6(request);
   try {
-    const storeLattice = (0, import_core18.getStoreLattice)("default", "database");
+    const storeLattice = (0, import_core19.getStoreLattice)("default", "database");
     const store = storeLattice.store;
     const configs = await store.getAllConfigs(tenantId);
     console.log("Backend: getAllConfigs returned:", configs);
@@ -3265,7 +2998,7 @@ async function getDatabaseConfig(request, reply) {
   const tenantId = getTenantId6(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core18.getStoreLattice)("default", "database");
+    const storeLattice = (0, import_core19.getStoreLattice)("default", "database");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3291,7 +3024,7 @@ async function createDatabaseConfig(request, reply) {
   const tenantId = getTenantId6(request);
   const body = request.body;
   try {
-    const storeLattice = (0, import_core18.getStoreLattice)("default", "database");
+    const storeLattice = (0, import_core19.getStoreLattice)("default", "database");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, body.key);
     if (existing) {
@@ -3304,7 +3037,7 @@ async function createDatabaseConfig(request, reply) {
     const id = body.id || (0, import_crypto3.randomUUID)();
     const config = await store.createConfig(tenantId, id, body);
     try {
-      import_core18.sqlDatabaseManager.registerDatabase(tenantId, config.key, config.config);
+      import_core19.sqlDatabaseManager.registerDatabase(tenantId, config.key, config.config);
     } catch (error) {
       console.warn("Failed to auto-register database:", error);
     }
@@ -3327,7 +3060,7 @@ async function updateDatabaseConfig(request, reply) {
   const { key } = request.params;
   const updates = request.body;
   try {
-    const storeLattice = (0, import_core18.getStoreLattice)("default", "database");
+    const storeLattice = (0, import_core19.getStoreLattice)("default", "database");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, key);
     if (!existing) {
@@ -3346,7 +3079,7 @@ async function updateDatabaseConfig(request, reply) {
     }
     if (updates.config) {
       try {
-        import_core18.sqlDatabaseManager.registerDatabase(tenantId, updated.key, updated.config);
+        import_core19.sqlDatabaseManager.registerDatabase(tenantId, updated.key, updated.config);
       } catch (error) {
         console.warn("Failed to re-register database:", error);
       }
@@ -3368,7 +3101,7 @@ async function deleteDatabaseConfig(request, reply) {
   const tenantId = getTenantId6(request);
   const { keyOrId } = request.params;
   try {
-    const storeLattice = (0, import_core18.getStoreLattice)("default", "database");
+    const storeLattice = (0, import_core19.getStoreLattice)("default", "database");
     const store = storeLattice.store;
     console.log("Delete request - keyOrId:", keyOrId);
     let config = await store.getConfigByKey(tenantId, keyOrId);
@@ -3395,8 +3128,8 @@ async function deleteDatabaseConfig(request, reply) {
       };
     }
     try {
-      if (import_core18.sqlDatabaseManager.hasDatabase(tenantId, configKey)) {
-        await import_core18.sqlDatabaseManager.removeDatabase(tenantId, configKey);
+      if (import_core19.sqlDatabaseManager.hasDatabase(tenantId, configKey)) {
+        await import_core19.sqlDatabaseManager.removeDatabase(tenantId, configKey);
       }
     } catch (error) {
       console.warn("Failed to remove from SqlDatabaseManager:", error);
@@ -3417,7 +3150,7 @@ async function testDatabaseConnection(request, reply) {
   const tenantId = getTenantId6(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core18.getStoreLattice)("default", "database");
+    const storeLattice = (0, import_core19.getStoreLattice)("default", "database");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3428,16 +3161,16 @@ async function testDatabaseConnection(request, reply) {
       };
     }
     const testKey = `__test_${key}_${Date.now()}`;
-    import_core18.sqlDatabaseManager.registerDatabase(tenantId, testKey, config.config);
+    import_core19.sqlDatabaseManager.registerDatabase(tenantId, testKey, config.config);
     const startTime = Date.now();
-    const db = await import_core18.sqlDatabaseManager.getDatabase(tenantId, testKey);
+    const db = await import_core19.sqlDatabaseManager.getDatabase(tenantId, testKey);
     try {
       await db.connect();
       await db.listTables();
       const latency = Date.now() - startTime;
       await new Promise((resolve) => setTimeout(resolve, 100));
       await db.disconnect();
-      await import_core18.sqlDatabaseManager.removeDatabase(tenantId, testKey);
+      await import_core19.sqlDatabaseManager.removeDatabase(tenantId, testKey);
       return {
         success: true,
         message: "Connection test successful",
@@ -3449,7 +3182,7 @@ async function testDatabaseConnection(request, reply) {
     } catch (error) {
       try {
         await db.disconnect();
-        await import_core18.sqlDatabaseManager.removeDatabase(tenantId, testKey);
+        await import_core19.sqlDatabaseManager.removeDatabase(tenantId, testKey);
       } catch {
       }
       return {
@@ -3501,7 +3234,7 @@ function registerDatabaseConfigRoutes(app2) {
 }
 // src/controllers/metrics-configs.ts
-var import_core19 = require("@axiom-lattice/core");
+var import_core20 = require("@axiom-lattice/core");
 var import_crypto4 = require("crypto");
 function getTenantId7(request) {
   const userTenantId = request.user?.tenantId;
@@ -3513,7 +3246,7 @@ function getTenantId7(request) {
 async function getMetricsServerConfigList(request, reply) {
   const tenantId = getTenantId7(request);
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     const configs = await store.getAllConfigs(tenantId);
     return {
@@ -3540,7 +3273,7 @@ async function getMetricsServerConfig(request, reply) {
   const tenantId = getTenantId7(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3566,7 +3299,7 @@ async function createMetricsServerConfig(request, reply) {
   const tenantId = getTenantId7(request);
   const body = request.body;
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, body.key);
     if (existing) {
@@ -3595,7 +3328,7 @@ async function createMetricsServerConfig(request, reply) {
     };
     const config = await store.createConfig(tenantId, id, configData);
     try {
-      import_core19.metricsServerManager.registerServer(tenantId, config.key, config.config);
+      import_core20.metricsServerManager.registerServer(tenantId, config.key, config.config);
     } catch (error) {
       console.warn("Failed to auto-register metrics server:", error);
     }
@@ -3618,7 +3351,7 @@ async function updateMetricsServerConfig(request, reply) {
   const { key } = request.params;
   const updates = request.body;
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, key);
     if (!existing) {
@@ -3646,7 +3379,7 @@ async function updateMetricsServerConfig(request, reply) {
     }
     if (updates.config) {
       try {
-        import_core19.metricsServerManager.registerServer(tenantId, updated.key, updated.config);
+        import_core20.metricsServerManager.registerServer(tenantId, updated.key, updated.config);
       } catch (error) {
         console.warn("Failed to re-register metrics server:", error);
       }
@@ -3668,7 +3401,7 @@ async function deleteMetricsServerConfig(request, reply) {
   const tenantId = getTenantId7(request);
   const { keyOrId } = request.params;
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     let config = await store.getConfigByKey(tenantId, keyOrId);
     let configKey = keyOrId;
@@ -3693,8 +3426,8 @@ async function deleteMetricsServerConfig(request, reply) {
       };
     }
     try {
-      if (import_core19.metricsServerManager.hasServer(tenantId, configKey)) {
-        import_core19.metricsServerManager.removeServer(tenantId, configKey);
+      if (import_core20.metricsServerManager.hasServer(tenantId, configKey)) {
+        import_core20.metricsServerManager.removeServer(tenantId, configKey);
       }
     } catch (error) {
       console.warn("Failed to remove from MetricsServerManager:", error);
@@ -3715,7 +3448,7 @@ async function testMetricsServerConnection(request, reply) {
   const tenantId = getTenantId7(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3726,11 +3459,11 @@ async function testMetricsServerConnection(request, reply) {
       };
     }
     const testKey = `__test_${key}_${Date.now()}`;
-    import_core19.metricsServerManager.registerServer(tenantId, testKey, config.config);
+    import_core20.metricsServerManager.registerServer(tenantId, testKey, config.config);
     try {
-      const client = import_core19.metricsServerManager.getClient(tenantId, testKey);
+      const client = import_core20.metricsServerManager.getClient(tenantId, testKey);
       const result = await client.testConnection();
-      import_core19.metricsServerManager.removeServer(tenantId, testKey);
+      import_core20.metricsServerManager.removeServer(tenantId, testKey);
       return {
         success: true,
         message: result.connected ? "Connection test successful" : "Connection test failed",
@@ -3738,7 +3471,7 @@ async function testMetricsServerConnection(request, reply) {
       };
     } catch (error) {
       try {
-        import_core19.metricsServerManager.removeServer(tenantId, testKey);
+        import_core20.metricsServerManager.removeServer(tenantId, testKey);
       } catch {
       }
       return {
@@ -3766,7 +3499,7 @@ async function listAvailableMetrics(request, reply) {
   const tenantId = getTenantId7(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3776,10 +3509,10 @@ async function listAvailableMetrics(request, reply) {
         message: "Metrics server configuration not found"
       };
     }
-    if (!import_core19.metricsServerManager.hasServer(tenantId, key)) {
-      import_core19.metricsServerManager.registerServer(tenantId, key, config.config);
+    if (!import_core20.metricsServerManager.hasServer(tenantId, key)) {
+      import_core20.metricsServerManager.registerServer(tenantId, key, config.config);
     }
-    const client = import_core19.metricsServerManager.getClient(tenantId, key);
+    const client = import_core20.metricsServerManager.getClient(tenantId, key);
     const metrics = await client.listMetrics();
     return {
       success: true,
@@ -3805,7 +3538,7 @@ async function queryMetricsData(request, reply) {
   const { key } = request.params;
   const { metricName, startTime, endTime, step, labels } = request.body;
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3822,10 +3555,10 @@ async function queryMetricsData(request, reply) {
         message: "metricName is required"
       };
     }
-    if (!import_core19.metricsServerManager.hasServer(tenantId, key)) {
-      import_core19.metricsServerManager.registerServer(tenantId, key, config.config);
+    if (!import_core20.metricsServerManager.hasServer(tenantId, key)) {
+      import_core20.metricsServerManager.registerServer(tenantId, key, config.config);
     }
-    const client = import_core19.metricsServerManager.getClient(tenantId, key);
+    const client = import_core20.metricsServerManager.getClient(tenantId, key);
     const result = await client.queryMetricData(metricName, {
       startTime,
       endTime,
@@ -3852,7 +3585,7 @@ async function getDataSources(request, reply) {
   const tenantId = getTenantId7(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3870,7 +3603,7 @@ async function getDataSources(request, reply) {
       };
     }
     const semanticConfig = config.config;
-    const client = new import_core19.SemanticMetricsClient(semanticConfig);
+    const client = new import_core20.SemanticMetricsClient(semanticConfig);
     const allDatasources = await client.getDataSources();
     const selectedIds = semanticConfig.selectedDataSources || [];
     const filteredDatasources = selectedIds.length > 0 ? allDatasources.filter((ds) => selectedIds.includes(String(ds.id))) : allDatasources;
@@ -3893,7 +3626,7 @@ async function getDatasourceMetrics(request, reply) {
   const tenantId = getTenantId7(request);
   const { key, datasourceId } = request.params;
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3911,7 +3644,7 @@ async function getDatasourceMetrics(request, reply) {
       };
     }
     const semanticConfig = config.config;
-    const client = new import_core19.SemanticMetricsClient(semanticConfig);
+    const client = new import_core20.SemanticMetricsClient(semanticConfig);
     const metrics = await client.getDatasourceMetrics(datasourceId);
     return {
       success: true,
@@ -3931,7 +3664,7 @@ async function querySemanticMetrics(request, reply) {
   const { key } = request.params;
   const body = request.body;
   try {
-    const storeLattice = (0, import_core19.getStoreLattice)("default", "metrics");
+    const storeLattice = (0, import_core20.getStoreLattice)("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3956,7 +3689,7 @@ async function querySemanticMetrics(request, reply) {
       };
     }
     const semanticConfig = config.config;
-    const client = new import_core19.SemanticMetricsClient(semanticConfig);
+    const client = new import_core20.SemanticMetricsClient(semanticConfig);
     const result = await client.semanticQuery(body);
     const columnNames = result.columns.map((col) => col.name);
     const allDataPoints = [];
@@ -4016,7 +3749,7 @@ async function testSemanticDataSources(request, reply) {
       password: body.password,
       headers: body.headers
     };
-    const client = new import_core19.SemanticMetricsClient(testConfig);
+    const client = new import_core20.SemanticMetricsClient(testConfig);
     const datasources = await client.getDataSources();
     return {
       success: true,
@@ -4052,7 +3785,7 @@ async function testDatasourceMetrics(request, reply) {
       password: body.password,
       headers: body.headers
     };
-    const client = new import_core19.SemanticMetricsClient(testConfig);
+    const client = new import_core20.SemanticMetricsClient(testConfig);
     const metrics = await client.getDatasourceMetrics(datasourceId);
     return {
       success: true,
@@ -4084,7 +3817,7 @@ function registerMetricsServerConfigRoutes(app2) {
 }
 // src/controllers/mcp-configs.ts
-var import_core20 = require("@axiom-lattice/core");
+var import_core21 = require("@axiom-lattice/core");
 var import_crypto5 = require("crypto");
 function getTenantId8(request) {
   const userTenantId = request.user?.tenantId;
@@ -4096,7 +3829,7 @@ function getTenantId8(request) {
 async function getMcpServerConfigList(request, reply) {
   const tenantId = getTenantId8(request);
   try {
-    const storeLattice = (0, import_core20.getStoreLattice)("default", "mcp");
+    const storeLattice = (0, import_core21.getStoreLattice)("default", "mcp");
     const store = storeLattice.store;
     const configs = await store.getAllConfigs(tenantId);
     return {
@@ -4123,7 +3856,7 @@ async function getMcpServerConfig(request, reply) {
   const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core20.getStoreLattice)("default", "mcp");
+    const storeLattice = (0, import_core21.getStoreLattice)("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4149,7 +3882,7 @@ async function createMcpServerConfig(request, reply) {
   const tenantId = getTenantId8(request);
   const body = request.body;
   try {
-    const storeLattice = (0, import_core20.getStoreLattice)("default", "mcp");
+    const storeLattice = (0, import_core21.getStoreLattice)("default", "mcp");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, body.key);
     if (existing) {
@@ -4189,7 +3922,7 @@ async function updateMcpServerConfig(request, reply) {
   const { key } = request.params;
   const updates = request.body;
   try {
-    const storeLattice = (0, import_core20.getStoreLattice)("default", "mcp");
+    const storeLattice = (0, import_core21.getStoreLattice)("default", "mcp");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, key);
     if (!existing) {
@@ -4209,8 +3942,8 @@ async function updateMcpServerConfig(request, reply) {
     }
     if (shouldReconnect) {
       try {
-        if (import_core20.mcpManager.hasServer(key)) {
-          await import_core20.mcpManager.removeServer(key);
+        if (import_core21.mcpManager.hasServer(key)) {
+          await import_core21.mcpManager.removeServer(key);
         }
         await connectAndRegisterTools(updated);
         await store.updateConfig(tenantId, existing.id, { status: "connected" });
@@ -4238,7 +3971,7 @@ async function deleteMcpServerConfig(request, reply) {
   const tenantId = getTenantId8(request);
   const { keyOrId } = request.params;
   try {
-    const storeLattice = (0, import_core20.getStoreLattice)("default", "mcp");
+    const storeLattice = (0, import_core21.getStoreLattice)("default", "mcp");
     const store = storeLattice.store;
     let config = await store.getConfigByKey(tenantId, keyOrId);
     let configKey = keyOrId;
@@ -4256,8 +3989,8 @@ async function deleteMcpServerConfig(request, reply) {
       };
     }
     try {
-      if (import_core20.mcpManager.hasServer(configKey)) {
-        await import_core20.mcpManager.removeServer(configKey);
+      if (import_core21.mcpManager.hasServer(configKey)) {
+        await import_core21.mcpManager.removeServer(configKey);
       }
     } catch (error) {
       console.warn("Failed to remove from MCP manager:", error);
@@ -4285,7 +4018,7 @@ async function testMcpServerConnection(request, reply) {
   const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core20.getStoreLattice)("default", "mcp");
+    const storeLattice = (0, import_core21.getStoreLattice)("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4299,11 +4032,11 @@ async function testMcpServerConnection(request, reply) {
     try {
       const testKey = `__test_${key}_${Date.now()}`;
       const connection = convertToConnection(config.config);
-      import_core20.mcpManager.addServer(testKey, connection);
-      await import_core20.mcpManager.connect();
-      const tools = await import_core20.mcpManager.getAllTools();
+      import_core21.mcpManager.addServer(testKey, connection);
+      await import_core21.mcpManager.connect();
+      const tools = await import_core21.mcpManager.getAllTools();
       const latency = Date.now() - startTime;
-      await import_core20.mcpManager.removeServer(testKey);
+      await import_core21.mcpManager.removeServer(testKey);
       return {
         success: true,
         message: "Connection test successful",
@@ -4338,7 +4071,7 @@ async function listMcpServerTools(request, reply) {
   const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core20.getStoreLattice)("default", "mcp");
+    const storeLattice = (0, import_core21.getStoreLattice)("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4348,10 +4081,10 @@ async function listMcpServerTools(request, reply) {
         message: "MCP server configuration not found"
       };
     }
-    if (!import_core20.mcpManager.hasServer(key)) {
+    if (!import_core21.mcpManager.hasServer(key)) {
       await connectAndRegisterTools(config);
     }
-    const tools = await import_core20.mcpManager.getAllTools();
+    const tools = await import_core21.mcpManager.getAllTools();
     return {
       success: true,
       message: "Tools retrieved successfully",
@@ -4371,7 +4104,7 @@ async function connectMcpServer(request, reply) {
   const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core20.getStoreLattice)("default", "mcp");
+    const storeLattice = (0, import_core21.getStoreLattice)("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4392,7 +4125,7 @@ async function connectMcpServer(request, reply) {
     };
   } catch (error) {
     console.error("Failed to connect MCP server:", error);
-    const storeLattice = (0, import_core20.getStoreLattice)("default", "mcp");
+    const storeLattice = (0, import_core21.getStoreLattice)("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (config) {
@@ -4408,7 +4141,7 @@ async function disconnectMcpServer(request, reply) {
   const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = (0, import_core20.getStoreLattice)("default", "mcp");
+    const storeLattice = (0, import_core21.getStoreLattice)("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4418,8 +4151,8 @@ async function disconnectMcpServer(request, reply) {
         message: "MCP server configuration not found"
       };
     }
-    if (import_core20.mcpManager.hasServer(key)) {
-      await import_core20.mcpManager.removeServer(key);
+    if (import_core21.mcpManager.hasServer(key)) {
+      await import_core21.mcpManager.removeServer(key);
     }
     const updated = await store.updateConfig(tenantId, config.id, {
       status: "disconnected"
@@ -4449,10 +4182,10 @@ async function testMcpServerTools(request, reply) {
     }
     const testKey = `__test_${Date.now()}`;
     const connection = convertToConnection(body.config);
-    import_core20.mcpManager.addServer(testKey, connection);
-    await import_core20.mcpManager.connect();
-    const tools = await import_core20.mcpManager.getAllTools();
-    await import_core20.mcpManager.removeServer(testKey);
+    import_core21.mcpManager.addServer(testKey, connection);
+    await import_core21.mcpManager.connect();
+    const tools = await import_core21.mcpManager.getAllTools();
+    await import_core21.mcpManager.removeServer(testKey);
     return {
       success: true,
       message: "Tools retrieved successfully",
@@ -4489,14 +4222,14 @@ function convertToConnection(config) {
 }
 async function connectAndRegisterTools(config) {
   const connection = convertToConnection(config.config);
-  import_core20.mcpManager.addServer(config.key, connection);
-  await import_core20.mcpManager.connect();
-  const allTools = await import_core20.mcpManager.getAllTools();
+  import_core21.mcpManager.addServer(config.key, connection);
+  await import_core21.mcpManager.connect();
+  const allTools = await import_core21.mcpManager.getAllTools();
   const selectedTools = allTools.filter(
     (tool) => config.selectedTools.includes(tool.name)
   );
   for (const tool of selectedTools) {
-    import_core20.toolLatticeManager.registerExistingTool(tool.name, tool);
+    import_core21.toolLatticeManager.registerExistingTool(tool.name, tool);
   }
 }
 function registerMcpServerConfigRoutes(app2) {
@@ -4513,11 +4246,11 @@ function registerMcpServerConfigRoutes(app2) {
 }
 // src/controllers/users.ts
-var import_core21 = require("@axiom-lattice/core");
-var import_uuid4 = require("uuid");
+var import_core22 = require("@axiom-lattice/core");
+var import_uuid3 = require("uuid");
 var UsersController = class {
   constructor() {
-    this.userStore = (0, import_core21.getStoreLattice)("default", "user").store;
+    this.userStore = (0, import_core22.getStoreLattice)("default", "user").store;
   }
   async listUsers(request, reply) {
     const { email } = request.query;
@@ -4530,7 +4263,7 @@ var UsersController = class {
   }
   async createUser(request, reply) {
     const data = request.body;
-    const id = (0, import_uuid4.v4)();
+    const id = (0, import_uuid3.v4)();
     const existingUser = await this.userStore.getUserByEmail(data.email);
     if (existingUser) {
       return reply.status(409).send({
@@ -4595,11 +4328,11 @@ function registerUserRoutes(app2) {
 }
 // src/controllers/tenants.ts
-var import_core22 = require("@axiom-lattice/core");
-var import_uuid5 = require("uuid");
+var import_core23 = require("@axiom-lattice/core");
+var import_uuid4 = require("uuid");
 var TenantsController = class {
   constructor() {
-    this.tenantStore = (0, import_core22.getStoreLattice)("default", "tenant").store;
+    this.tenantStore = (0, import_core23.getStoreLattice)("default", "tenant").store;
   }
   // ==================== Tenant CRUD ====================
   async listTenants(request, reply) {
@@ -4608,7 +4341,7 @@ var TenantsController = class {
   }
   async createTenant(request, reply) {
     const data = request.body;
-    const id = (0, import_uuid5.v4)();
+    const id = (0, import_uuid4.v4)();
     const tenant = await this.tenantStore.createTenant(id, data);
     return reply.status(201).send({ success: true, data: tenant });
   }
@@ -4663,8 +4396,8 @@ function registerTenantRoutes(app2) {
 }
 // src/controllers/auth.ts
-var import_core23 = require("@axiom-lattice/core");
-var import_uuid6 = require("uuid");
+var import_core24 = require("@axiom-lattice/core");
+var import_uuid5 = require("uuid");
 var defaultAuthConfig = {
   autoApproveUsers: true,
   allowTenantRegistration: true,
@@ -4672,9 +4405,9 @@ var defaultAuthConfig = {
 };
 var AuthController = class {
   constructor(config = {}) {
-    this.userStore = (0, import_core23.getStoreLattice)("default", "user").store;
-    this.tenantStore = (0, import_core23.getStoreLattice)("default", "tenant").store;
-    this.userTenantLinkStore = (0, import_core23.getStoreLattice)("default", "userTenantLink").store;
+    this.userStore = (0, import_core24.getStoreLattice)("default", "user").store;
+    this.tenantStore = (0, import_core24.getStoreLattice)("default", "tenant").store;
+    this.userTenantLinkStore = (0, import_core24.getStoreLattice)("default", "userTenantLink").store;
     this.config = { ...defaultAuthConfig, ...config };
   }
   async register(request, reply) {
@@ -4687,7 +4420,7 @@ var AuthController = class {
           error: "User with this email already exists"
         });
       }
-      const userId = (0, import_uuid6.v4)();
+      const userId = (0, import_uuid5.v4)();
       const userStatus = this.config.autoApproveUsers ? "active" : "pending";
       const userData = {
         email,
@@ -5188,7 +4921,7 @@ var configureSwagger = async (app2, customSwaggerConfig, customSwaggerUiConfig)
 };
 // src/services/agent_task_consumer.ts
-var import_core24 = require("@axiom-lattice/core");
+var import_core25 = require("@axiom-lattice/core");
 var handleAgentTask = async (taskRequest, retryCount = 0) => {
   const {
     assistant_id,
@@ -5205,82 +4938,18 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
     );
     const apiUrl = AgentTaskConsumer.agent_run_endpoint;
     console.log(`apiUrl: ${apiUrl}`);
-    const response = await fetch(apiUrl, {
-      method: "POST",
-      body: JSON.stringify({
-        assistant_id,
-        streaming: true,
-        ...input,
-        thread_id,
-        command,
-        custom_run_config: runConfig
-      }),
-      headers: {
-        "Content-Type": "application/json",
-        "x-tenant-id": tenant_id,
-        "x-workspace-id": runConfig?.workspaceId,
-        "x-project-id": runConfig?.projectId
-      }
-    }).catch((err) => {
-      console.error(`fetch\u8BF7\u6C42\u5931\u8D25: ${err.message || String(err)}`);
-      throw new Error(`fetch\u5931\u8D25: ${err.message || String(err)}`);
-    });
-    if (!response.ok) {
-      throw new Error(`API\u8BF7\u6C42\u5931\u8D25: ${response.status} ${response.statusText}`);
-    }
-    const contentType = response.headers.get("content-type");
-    if (contentType?.includes("text/event-stream")) {
-      const reader = response.body?.getReader();
-      const decoder = new TextDecoder();
-      if (!reader) {
-        throw new Error("Response body is not readable");
-      }
-      let buffer = "";
-      let streamEnded = false;
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) {
-            streamEnded = true;
-            console.log(
-              `SSE\u6D41\u5DF2\u7ED3\u675F [assistant_id: ${assistant_id}, thread_id: ${thread_id}]`
-            );
-            break;
-          }
-        }
-      } catch (streamError) {
-        console.error("Error reading SSE stream:", streamError);
-        throw streamError;
-      } finally {
-        reader.releaseLock();
-      }
-      if (callback_event) {
-        const state = await agent_state({ assistant_id, thread_id, tenant_id });
-        import_core24.eventBus.publish(callback_event, {
-          success: true,
-          state,
-          config: { assistant_id, thread_id, tenant_id }
-        });
-      }
-      console.log(
-        `\u4EFB\u52A1\u5904\u7406\u6210\u529F [assistant_id: ${assistant_id}, thread_id: ${thread_id}]`
-      );
-      return true;
-    } else {
-      await response.text();
-      if (callback_event) {
-        const state = await agent_state({ assistant_id, thread_id, tenant_id });
-        import_core24.eventBus.publish(callback_event, {
+    const agent = import_core25.agentInstanceManager.getAgent({ assistant_id, thread_id, tenant_id, workspace_id: runConfig?.workspaceId, project_id: runConfig?.projectId, custom_run_config: runConfig });
+    await agent.addMessage({ input, command }, import_core25.QueueMode.STEER);
+    if (callback_event) {
+      agent.subscribeOnce("message:completed", (evt) => {
+        import_core25.eventBus.publish(callback_event, {
           success: true,
-          state,
+          state: evt.state,
           config: { assistant_id, thread_id, tenant_id }
         });
-      }
-      console.log(
-        `\u4EFB\u52A1\u5904\u7406\u6210\u529F [assistant_id: ${assistant_id}, thread_id: ${thread_id}]`
-      );
-      return true;
+      });
     }
+    return true;
   } catch (error) {
     console.error(
       `Agent\u4EFB\u52A1\u6267\u884C\u5931\u8D25: ${assistant_id}, \u7EBF\u7A0B: ${thread_id}`,
@@ -5297,7 +4966,7 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
       return handleAgentTask(taskRequest, nextRetryCount);
     }
     if (callback_event) {
-      import_core24.eventBus.publish(callback_event, {
+      import_core25.eventBus.publish(callback_event, {
         success: false,
         error: error instanceof Error ? error.message : String(error),
         config: { assistant_id, thread_id, tenant_id }
@@ -5335,7 +5004,7 @@ var _AgentTaskConsumer = class _AgentTaskConsumer {
    * 初始化事件监听和队列轮询
    */
   initialize() {
-    import_core24.eventBus.subscribe(import_core24.AGENT_TASK_EVENT, this.trigger_agent_task.bind(this));
+    import_core25.eventBus.subscribe(import_core25.AGENT_TASK_EVENT, this.trigger_agent_task.bind(this));
     this.startPollingQueue();
     console.log("Agent\u4EFB\u52A1\u6D88\u8D39\u8005\u5DF2\u542F\u52A8\u5E76\u76D1\u542C\u4EFB\u52A1\u4E8B\u4EF6\u548C\u961F\u5217");
   }
@@ -5454,7 +5123,7 @@ var _AgentTaskConsumer = class _AgentTaskConsumer {
     handleAgentTask(taskRequest).catch((error) => {
       console.error("\u5904\u7406Agent\u4EFB\u52A1\u65F6\u53D1\u751F\u672A\u6355\u83B7\u7684\u9519\u8BEF:", error);
       if (taskRequest.callback_event) {
-        import_core24.eventBus.publish(taskRequest.callback_event, {
+        import_core25.eventBus.publish(taskRequest.callback_event, {
           success: false,
           error: error instanceof Error ? error.message : String(error),
           config: {
@@ -5474,7 +5143,7 @@ _AgentTaskConsumer.agent_run_endpoint = "http://localhost:4001/api/runs";
 var AgentTaskConsumer = _AgentTaskConsumer;
 // src/index.ts
-var import_core25 = require("@axiom-lattice/core");
+var import_core26 = require("@axiom-lattice/core");
 var import_protocols2 = require("@axiom-lattice/protocols");
 process.on("unhandledRejection", (reason, promise) => {
   console.error("\u672A\u5904\u7406\u7684Promise\u62D2\u7EDD:", reason);
@@ -5489,11 +5158,11 @@ var DEFAULT_LOGGER_CONFIG = {
 var loggerLattice = initializeLogger(DEFAULT_LOGGER_CONFIG);
 var logger = loggerLattice.client;
 function initializeLogger(config) {
-  if (import_core25.loggerLatticeManager.hasLattice("default")) {
-    import_core25.loggerLatticeManager.removeLattice("default");
+  if (import_core26.loggerLatticeManager.hasLattice("default")) {
+    import_core26.loggerLatticeManager.removeLattice("default");
   }
-  (0, import_core25.registerLoggerLattice)("default", config);
-  return (0, import_core25.getLoggerLattice)("default");
+  (0, import_core26.registerLoggerLattice)("default", config);
+  return (0, import_core26.getLoggerLattice)("default");
 }
 var app = (0, import_fastify.default)({
   logger: false,
@@ -5533,7 +5202,6 @@ app.addHook("onResponse", (request, reply, done) => {
     "x-tenant-id": getHeaderValue(request.headers["x-tenant-id"]),
     "x-request-id": getHeaderValue(request.headers["x-request-id"])
   };
-  loggerLattice.info(`${request.method} ${request.url} - ${reply.statusCode}`);
   done();
 });
 app.register(import_cors.default, {
@@ -5594,16 +5262,16 @@ var start = async (config) => {
     app.decorate("loggerLattice", loggerLattice);
     registerLatticeRoutes(app);
     try {
-      const storeLattice = (0, import_core25.getStoreLattice)("default", "database");
+      const storeLattice = (0, import_core26.getStoreLattice)("default", "database");
       const store = storeLattice.store;
-      import_core25.sqlDatabaseManager.setConfigStore(store);
+      import_core26.sqlDatabaseManager.setConfigStore(store);
       logger.info("Database config store set for SqlDatabaseManager");
     } catch (error) {
       logger.warn("Failed to set database config store: " + (error instanceof Error ? error.message : String(error)));
     }
-    if (!import_core25.sandboxLatticeManager.hasLattice("default")) {
+    if (!import_core26.sandboxLatticeManager.hasLattice("default")) {
       const sandboxBaseURL = process.env.SANDBOX_BASE_URL || "http://localhost:8080";
-      import_core25.sandboxLatticeManager.registerLattice("default", {
+      import_core26.sandboxLatticeManager.registerLattice("default", {
         baseURL: sandboxBaseURL
       });
       logger.info(`Registered sandbox manager with baseURL: ${sandboxBaseURL}`);
@@ -5611,6 +5279,11 @@ var start = async (config) => {
     const target_port = config?.port || Number(process.env.PORT) || 4001;
     await app.listen({ port: target_port, host: "0.0.0.0" });
     logger.info(`Lattice Gateway is running on port: ${target_port}`);
+    try {
+      logger.info("AgentLifecycleManager initialized");
+    } catch (error) {
+      logger.warn("Failed to initialize AgentLifecycleManager", { error });
+    }
     const queueServiceConfig = config?.queueServiceConfig;
     if (queueServiceConfig) {
       setQueueServiceType(queueServiceConfig.type);