@axiom-lattice/gateway 2.1.87 → 2.1.89

package/dist/index.js

@@ -1715,7 +1715,7 @@ async function executeDataQuery(request, reply) {
         message: `Metrics server not registered: ${body.serverKey}. Please register the server first.`
       };
     }
-    const client = import_core12.metricsServerManager.getClient(tenantId, body.serverKey);
+    const client = await import_core12.metricsServerManager.getClient(tenantId, body.serverKey);
     if (isSemanticQuery) {
       return await executeSemanticQuery(client, body, reply);
     } else {
@@ -3915,7 +3915,7 @@ async function testMetricsServerConnection(request, reply) {
     const testKey = `__test_${key}_${Date.now()}`;
     import_core21.metricsServerManager.registerServer(tenantId, testKey, config.config);
     try {
-      const client = import_core21.metricsServerManager.getClient(tenantId, testKey);
+      const client = await import_core21.metricsServerManager.getClient(tenantId, testKey);
       const result = await client.testConnection();
       import_core21.metricsServerManager.removeServer(tenantId, testKey);
       return {
@@ -3966,7 +3966,7 @@ async function listAvailableMetrics(request, reply) {
     if (!import_core21.metricsServerManager.hasServer(tenantId, key)) {
       import_core21.metricsServerManager.registerServer(tenantId, key, config.config);
     }
-    const client = import_core21.metricsServerManager.getClient(tenantId, key);
+    const client = await import_core21.metricsServerManager.getClient(tenantId, key);
     const metrics = await client.listMetrics();
     return {
       success: true,
@@ -4012,7 +4012,7 @@ async function queryMetricsData(request, reply) {
     if (!import_core21.metricsServerManager.hasServer(tenantId, key)) {
       import_core21.metricsServerManager.registerServer(tenantId, key, config.config);
     }
-    const client = import_core21.metricsServerManager.getClient(tenantId, key);
+    const client = await import_core21.metricsServerManager.getClient(tenantId, key);
     const result = await client.queryMetricData(metricName, {
       startTime,
       endTime,
@@ -6354,19 +6354,8 @@ function registerChannelInstallationRoutes(app2) {
   app2.delete("/api/channel-installations/:installationId", deleteChannelInstallation);
 }
 
-// src/bindings/index.ts
-var registryInstance = null;
-function setBindingRegistry(registry) {
-  registryInstance = registry;
-}
-function getBindingRegistry() {
-  if (!registryInstance) {
-    throw new Error("BindingRegistry not initialized. Call setBindingRegistry() first.");
-  }
-  return registryInstance;
-}
-
 // src/controllers/channel-bindings.ts
+var import_core28 = require("@axiom-lattice/core");
 function getTenantId12(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) return userTenantId;
@@ -6376,7 +6365,7 @@ async function getBindingList(request, _reply) {
   const tenantId = getTenantId12(request);
   const { channel, agentId, channelInstallationId, limit, offset } = request.query;
   try {
-    const registry = getBindingRegistry();
+    const registry = (0, import_core28.getBindingRegistry)();
     const bindings = await registry.list({ channel, agentId, tenantId, channelInstallationId, limit, offset });
     return { success: true, message: "Bindings retrieved", data: { records: bindings, total: bindings.length } };
   } catch (error) {
@@ -6387,7 +6376,7 @@ async function getBindingList(request, _reply) {
 async function getBinding(request, reply) {
   const tenantId = getTenantId12(request);
   try {
-    const registry = getBindingRegistry();
+    const registry = (0, import_core28.getBindingRegistry)();
     const bindings = await registry.list({ tenantId });
     const binding = bindings.find((b) => b.id === request.params.id);
     if (!binding || binding.tenantId !== tenantId) {
@@ -6403,7 +6392,7 @@ async function getBinding(request, reply) {
 async function createBinding(request, reply) {
   const tenantId = getTenantId12(request);
   try {
-    const registry = getBindingRegistry();
+    const registry = (0, import_core28.getBindingRegistry)();
     const binding = await registry.create({ ...request.body, tenantId });
     reply.status(201);
     return { success: true, message: "Binding created", data: binding };
@@ -6416,7 +6405,7 @@ async function createBinding(request, reply) {
 async function updateBinding(request, reply) {
   try {
     const tenantId = getTenantId12(request);
-    const registry = getBindingRegistry();
+    const registry = (0, import_core28.getBindingRegistry)();
     const bindings = await registry.list({ tenantId });
     const existing = bindings.find((b) => b.id === request.params.id);
     if (!existing || existing.tenantId !== tenantId) {
@@ -6434,7 +6423,7 @@ async function updateBinding(request, reply) {
 async function deleteBinding(request, reply) {
   try {
     const tenantId = getTenantId12(request);
-    const registry = getBindingRegistry();
+    const registry = (0, import_core28.getBindingRegistry)();
     const bindings = await registry.list({ tenantId });
     const existing = bindings.find((b) => b.id === request.params.id);
     if (!existing || existing.tenantId !== tenantId) {
@@ -6453,7 +6442,7 @@ async function resolveBinding(request, _reply) {
   const tenantId = getTenantId12(request);
   const { channel, senderId, channelInstallationId } = request.query;
   try {
-    const registry = getBindingRegistry();
+    const registry = (0, import_core28.getBindingRegistry)();
     const binding = await registry.resolve({ channel, senderId, channelInstallationId, tenantId });
     if (!binding) {
       return { success: false, message: "No binding found", data: null };
@@ -6690,7 +6679,7 @@ var registerLatticeRoutes = (app2, channelDeps) => {
 };
 
 // src/router/MessageRouter.ts
-var import_core28 = require("@axiom-lattice/core");
+var import_core29 = require("@axiom-lattice/core");
 var import_crypto8 = require("crypto");
 var BindingNotFoundError = class extends Error {
   constructor(message) {
@@ -6700,14 +6689,39 @@ var BindingNotFoundError = class extends Error {
 };
 var MessageRouter = class {
   constructor(config) {
+    /**
+     * Tracks reply subscriptions per thread+channel to avoid duplicate
+     * `subscribeOnce` registrations and ensure proper cleanup.
+     *
+     * Key format: `{threadId}:{adapterChannel}:reply`
+     */
+    this._replySubs = /* @__PURE__ */ new Map();
     this.middlewares = [...config.middlewares];
     this.bindingRegistry = config.bindingRegistry;
     this.adapterRegistry = config.adapterRegistry;
     this.installationStore = config.installationStore;
   }
+  /**
+   * Register an additional middleware at the end of the chain.
+   *
+   * @param middleware - A {@link MessageMiddleware} function
+   */
   use(middleware) {
     this.middlewares.push(middleware);
   }
+  /**
+   * Dispatch an inbound channel message to the bound agent.
+   *
+   * Full pipeline: middleware chain → binding resolution → thread lifecycle
+   * → agent.addMessage() → (async) reply via {@link ChannelAdapter.sendReply}.
+   *
+   * If the message has a {@link InboundMessage.replyTarget}, the router subscribes
+   * to the agent's `reply:ready` event before enqueuing the message, and sends
+   * the AI response back to the channel when it arrives.
+   *
+   * @param message - Normalized inbound message from a channel adapter
+   * @returns {@link DispatchResult} with success status, bindingId, and threadId
+   */
   async dispatch(message) {
     const ctx = {
       inboundMessage: message,
@@ -6806,7 +6820,7 @@ var MessageRouter = class {
         }
         let threadId = ctx.binding.threadMode === "per_conversation" ? void 0 : ctx.binding.threadId;
         if (!threadId) {
-          const threadStore = (0, import_core28.getStoreLattice)("default", "thread").store;
+          const threadStore = (0, import_core29.getStoreLattice)("default", "thread").store;
           const newThreadId = (0, import_crypto8.randomUUID)();
           console.log({
             event: "dispatch:thread:create",
@@ -6847,39 +6861,103 @@ var MessageRouter = class {
           senderId: message.sender.id,
           contentLength: message.content.text.length
         }, "Dispatching to agent");
-        const agent = import_core28.agentInstanceManager.getAgent({
+        const agent = import_core29.agentInstanceManager.getAgent({
           tenant_id: tenantId,
           assistant_id: ctx.binding.agentId,
           thread_id: threadId,
           workspace_id: ctx.binding.workspaceId || "",
           project_id: ctx.binding.projectId || ""
         });
-        const addResult = await agent.addMessage({
-          input: { message: message.content.text },
-          custom_run_config: message.content.metadata || {}
-        });
-        console.log({
-          event: "dispatch:complete",
-          agentId: ctx.binding.agentId,
-          threadId,
-          messageId: addResult?.messageId,
-          result: JSON.stringify(addResult)
-        }, "Agent dispatch complete \u2014 messageId = " + (addResult?.messageId || "N/A"));
         if (message.replyTarget) {
+          const replySubKey = `${threadId}:${message.replyTarget.adapterChannel}:reply`;
           const adapter = this.adapterRegistry.get(message.replyTarget.adapterChannel);
           if (adapter) {
             const installation = await this.installationStore.getInstallationById(
               message.channelInstallationId
             );
             if (installation) {
-              await adapter.sendReply(
-                message.replyTarget,
-                { text: ctx.result || "" },
-                installation
-              );
+              const existing = this._replySubs.get(replySubKey);
+              if (!existing || existing.count === 0) {
+                const timer = setTimeout(() => {
+                  const entry = this._replySubs.get(replySubKey);
+                  if (entry) {
+                    this._replySubs.delete(replySubKey);
+                    console.warn({
+                      event: "dispatch:reply:timeout",
+                      threadId,
+                      channel: message.replyTarget.adapterChannel
+                    }, "Reply subscription timed out \u2014 no reply:ready fired within 1h");
+                  }
+                }, 36e5);
+                this._replySubs.set(replySubKey, { count: 1, timer });
+                console.log({
+                  event: "dispatch:reply:subscribed",
+                  threadId,
+                  channel: message.replyTarget.adapterChannel,
+                  senderId: message.sender.id
+                }, "Subscribed to reply:ready for thread");
+                agent.subscribeOnce("reply:ready", (data) => {
+                  const messages = data.state?.values?.messages;
+                  const lastAI = messages?.filter((m) => m.type === "ai" || m.getType?.() === "ai").pop();
+                  const replyText = lastAI?.content ?? "";
+                  const entry = this._replySubs.get(replySubKey);
+                  if (entry) {
+                    entry.count--;
+                    if (entry.count <= 0) {
+                      clearTimeout(entry.timer);
+                      this._replySubs.delete(replySubKey);
+                    }
+                  }
+                  if (replyText) {
+                    console.log({
+                      event: "dispatch:reply:sending",
+                      threadId,
+                      channel: message.replyTarget.adapterChannel,
+                      replyLength: replyText.length
+                    }, "Sending channel reply");
+                    adapter.sendReply(message.replyTarget, { text: replyText }, installation).then(() => {
+                      console.log({
+                        event: "dispatch:reply:sent",
+                        threadId,
+                        channel: message.replyTarget.adapterChannel
+                      }, "Channel reply sent successfully");
+                    }).catch((err) => console.error({
+                      event: "dispatch:reply:failed",
+                      threadId,
+                      channel: message.replyTarget.adapterChannel,
+                      error: err instanceof Error ? err.message : String(err)
+                    }, "Failed to send channel reply"));
+                  } else {
+                    console.warn({
+                      event: "dispatch:reply:empty",
+                      threadId,
+                      channel: message.replyTarget.adapterChannel
+                    }, "Agent produced no text output \u2014 skipping reply");
+                  }
+                });
+              } else {
+                existing.count++;
+                console.log({
+                  event: "dispatch:reply:incremented",
+                  threadId,
+                  channel: message.replyTarget.adapterChannel,
+                  count: existing.count
+                }, "Incremented reply counter for thread (already subscribed)");
+              }
             }
           }
         }
+        const addResult = await agent.addMessage({
+          input: { message: message.content.text },
+          custom_run_config: message.replyTarget ? { ...message.content.metadata || {}, _replyTarget: message.replyTarget } : message.content.metadata || {}
+        });
+        console.log({
+          event: "dispatch:complete",
+          agentId: ctx.binding.agentId,
+          threadId,
+          messageId: addResult?.messageId,
+          result: JSON.stringify(addResult)
+        }, "Agent dispatch complete \u2014 messageId = " + (addResult?.messageId || "N/A"));
       });
       return {
         success: true,
@@ -7027,7 +7105,7 @@ function createAuditLoggerMiddleware() {
 }
 
 // src/index.ts
-var import_core30 = require("@axiom-lattice/core");
+var import_core31 = require("@axiom-lattice/core");
 
 // src/swagger.ts
 var import_swagger = __toESM(require("@fastify/swagger"));
@@ -7092,7 +7170,7 @@ var configureSwagger = async (app2, customSwaggerConfig, customSwaggerUiConfig)
 };
 
 // src/services/agent_task_consumer.ts
-var import_core29 = require("@axiom-lattice/core");
+var import_core30 = require("@axiom-lattice/core");
 var handleAgentTask = async (taskRequest, retryCount = 0) => {
   const {
     assistant_id,
@@ -7110,18 +7188,18 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
     console.log(
       `\u5F00\u59CB\u5904\u7406\u4EFB\u52A1 [assistant_id: ${assistant_id}, thread_id: ${thread_id}]`
     );
-    const agent = import_core29.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, custom_run_config: runConfig }, import_core29.QueueMode.STEER);
+    const agent = import_core30.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, custom_run_config: runConfig }, import_core30.QueueMode.STEER);
     if (callback_event) {
       agent.subscribeOnce("message:completed", (evt) => {
-        import_core29.eventBus.publish(callback_event, {
+        import_core30.eventBus.publish(callback_event, {
           success: true,
           state: evt.state,
           config: { assistant_id, thread_id, tenant_id }
         });
         if (main_thread_id && main_tenant_id) {
           try {
-            const mainAgent = import_core29.agentInstanceManager.getAgent({
+            const mainAgent = import_core30.agentInstanceManager.getAgent({
               assistant_id: main_assistant_id ?? assistant_id,
               thread_id: main_thread_id,
               tenant_id: main_tenant_id
@@ -7149,7 +7227,7 @@ ${summary}`
         }
       });
       agent.subscribeOnce("message:interrupted", (evt) => {
-        import_core29.eventBus.publish(callback_event, {
+        import_core30.eventBus.publish(callback_event, {
           success: true,
           state: evt.state,
           config: { assistant_id, thread_id, tenant_id }
@@ -7173,7 +7251,7 @@ ${summary}`
       return handleAgentTask(taskRequest, nextRetryCount);
     }
     if (callback_event) {
-      import_core29.eventBus.publish(callback_event, {
+      import_core30.eventBus.publish(callback_event, {
         success: false,
         error: error instanceof Error ? error.message : String(error),
         config: { assistant_id, thread_id, tenant_id }
@@ -7211,7 +7289,7 @@ var _AgentTaskConsumer = class _AgentTaskConsumer {
    * 初始化事件监听和队列轮询
    */
   initialize() {
-    import_core29.eventBus.subscribe(import_core29.AGENT_TASK_EVENT, this.trigger_agent_task.bind(this));
+    import_core30.eventBus.subscribe(import_core30.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");
   }
@@ -7330,7 +7408,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_core29.eventBus.publish(taskRequest.callback_event, {
+        import_core30.eventBus.publish(taskRequest.callback_event, {
           success: false,
           error: error instanceof Error ? error.message : String(error),
           config: {
@@ -7350,7 +7428,7 @@ _AgentTaskConsumer.agent_run_endpoint = "http://localhost:4001/api/runs";
 var AgentTaskConsumer = _AgentTaskConsumer;
 
 // src/index.ts
-var import_core31 = require("@axiom-lattice/core");
+var import_core32 = require("@axiom-lattice/core");
 var import_protocols4 = require("@axiom-lattice/protocols");
 var import_meta = {};
 process.on("unhandledRejection", (reason, promise) => {
@@ -7366,11 +7444,11 @@ var DEFAULT_LOGGER_CONFIG = {
 var loggerLattice = initializeLogger(DEFAULT_LOGGER_CONFIG);
 var logger3 = loggerLattice.client;
 function initializeLogger(config) {
-  if (import_core31.loggerLatticeManager.hasLattice("default")) {
-    import_core31.loggerLatticeManager.removeLattice("default");
+  if (import_core32.loggerLatticeManager.hasLattice("default")) {
+    import_core32.loggerLatticeManager.removeLattice("default");
   }
-  (0, import_core31.registerLoggerLattice)("default", config);
-  return (0, import_core31.getLoggerLattice)("default");
+  (0, import_core32.registerLoggerLattice)("default", config);
+  return (0, import_core32.getLoggerLattice)("default");
 }
 var app = (0, import_fastify.default)({
   logger: false,
@@ -7473,7 +7551,7 @@ app.setErrorHandler((error, request, reply) => {
 });
 function getConfiguredSandboxProvider() {
   const sandboxProviderType = process.env.SANDBOX_PROVIDER_TYPE || "microsandbox-remote";
-  return (0, import_core31.createSandboxProvider)({
+  return (0, import_core32.createSandboxProvider)({
     type: sandboxProviderType,
     remoteBaseURL: process.env.SANDBOX_BASE_URL,
     microsandboxServiceBaseURL: process.env.MICROSANDBOX_SERVICE_BASE_URL,
@@ -7505,8 +7583,7 @@ var start = async (config) => {
       const { getStoreLattice: getStoreLattice16 } = await import("@axiom-lattice/core");
       const bindingStore = getStoreLattice16("default", "channelBinding").store;
       const installationStore = getStoreLattice16("default", "channelInstallation").store;
-      setBindingRegistry(bindingStore);
-      (0, import_core30.setBindingRegistry)(bindingStore);
+      (0, import_core31.setBindingRegistry)(bindingStore);
       const adapterRegistry = new ChannelAdapterRegistry();
       adapterRegistry.register(larkChannelAdapter);
       const router = new MessageRouter({
@@ -7523,16 +7600,8 @@ var start = async (config) => {
     } catch {
     }
     registerLatticeRoutes(app, channelDeps);
-    try {
-      const storeLattice = (0, import_core31.getStoreLattice)("default", "database");
-      const store = storeLattice.store;
-      import_core31.sqlDatabaseManager.setConfigStore(store);
-      logger3.info("Database config store set for SqlDatabaseManager");
-    } catch (error) {
-      logger3.warn("Failed to set database config store: " + (error instanceof Error ? error.message : String(error)));
-    }
-    if (!import_core31.sandboxLatticeManager.hasLattice("default")) {
-      import_core31.sandboxLatticeManager.registerLattice("default", getConfiguredSandboxProvider());
+    if (!import_core32.sandboxLatticeManager.hasLattice("default")) {
+      import_core32.sandboxLatticeManager.registerLattice("default", getConfiguredSandboxProvider());
       logger3.info("Registered sandbox manager from env configuration");
     }
     const target_port = config?.port || Number(process.env.PORT) || 4001;
@@ -7553,7 +7622,7 @@ var start = async (config) => {
     }
     try {
       logger3.info("Starting agent instance recovery...");
-      const restoreStats = await import_core31.agentInstanceManager.restore();
+      const restoreStats = await import_core32.agentInstanceManager.restore();
       logger3.info(`Agent recovery complete: ${restoreStats.restored} threads restored, ${restoreStats.errors} errors`);
     } catch (error) {
       logger3.error("Agent recovery failed", { error });