@axiom-lattice/gateway 2.1.44 → 2.1.46

package/dist/index.js

@@ -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,14 @@ var createRun = async (request, reply) => {
         "Access-Control-Allow-Origin": "*"
       });
       try {
-        let chunkCount = 0;
+        const result = await agent.addMessage({
+          input,
+          command,
+          custom_run_config
+        });
+        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 +292,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 +303,16 @@ var createRun = async (request, reply) => {
         reply.raw.end();
       }
     } else {
-      const result = await agent_invoke({
-        assistant_id,
-        input,
+      const { message: msg, ...restInputNonStream } = input;
+      const result = await agent.invoke({
+        input: { message: msg, ...restInputNonStream },
         command,
-        thread_id,
-        tenant_id,
-        workspace_id,
-        project_id,
-        run_id: x_request_id,
         custom_run_config
       });
-      reply.status(200).send(result);
+      reply.status(200).send({
+        success: true,
+        ...result
+      });
     }
   } catch (error) {
     reply.status(500).send({
@@ -604,12 +339,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 +354,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 +373,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 +438,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 +470,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 +534,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 +553,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 +573,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 +585,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 +600,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 +620,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 +633,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 +662,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 +685,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 +694,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 +962,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 +981,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 +991,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,16 +1078,18 @@ 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 || {};
+      const displayName = config.displayName || lattice.key.split("-").map((word) => word.charAt(0).toUpperCase() + word.slice(1)).join(" ");
       return {
         key: lattice.key,
         model: config.model || "",
         provider: config.provider || "openai",
+        displayName,
         streaming: config.streaming || false,
         apiKey: config.apiKey || "",
         baseURL: config.baseURL || "",
@@ -1396,12 +1133,13 @@ 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,
           model: modelConfig.model,
+          displayName: modelConfig.displayName,
           streaming: modelConfig.streaming ?? false,
           apiKey: modelConfig.apiKey,
           baseURL: modelConfig.baseURL,
@@ -1410,7 +1148,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 +1206,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 +1238,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 +1265,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 +1302,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 +1317,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 +1345,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 +1354,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 +1386,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 +1427,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 +1464,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 +1501,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 +1526,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 +1570,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 +1608,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 +1639,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 +1670,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 +2088,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 +2200,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 +2225,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 +2273,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 +2315,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 +2362,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 +2418,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 +2447,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 +2507,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 +2558,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 +2570,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 +2615,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 +2684,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 +2854,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 +2960,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 +2972,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 +3003,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 +3029,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 +3042,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 +3065,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 +3084,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 +3106,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 +3133,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 +3155,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 +3166,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 +3187,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 +3239,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 +3251,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 +3278,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 +3304,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 +3333,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 +3356,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 +3384,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 +3406,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 +3431,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 +3453,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 +3464,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 +3476,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 +3504,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 +3514,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 +3543,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 +3560,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 +3590,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 +3608,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 +3631,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 +3649,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 +3669,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 +3694,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 +3754,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 +3790,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 +3822,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 +3834,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 +3861,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 +3887,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 +3927,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 +3947,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 +3976,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 +3994,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 +4023,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 +4037,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 +4076,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 +4086,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 +4109,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 +4130,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 +4146,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 +4156,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 +4187,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 +4227,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 +4251,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 +4268,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 +4333,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 +4346,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 +4401,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 +4410,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 +4425,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 +4926,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 +4943,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 +4971,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 +5009,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 +5128,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 +5148,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 +5163,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 +5207,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 +5267,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 +5284,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);