@axiom-lattice/gateway 2.1.87 → 2.1.89

package/dist/index.mjs

@@ -1649,7 +1649,7 @@ async function executeDataQuery(request, reply) {
         message: `Metrics server not registered: ${body.serverKey}. Please register the server first.`
       };
     }
-    const client = metricsServerManager.getClient(tenantId, body.serverKey);
+    const client = await metricsServerManager.getClient(tenantId, body.serverKey);
     if (isSemanticQuery) {
       return await executeSemanticQuery(client, body, reply);
     } else {
@@ -3856,7 +3856,7 @@ async function testMetricsServerConnection(request, reply) {
     const testKey = `__test_${key}_${Date.now()}`;
     metricsServerManager2.registerServer(tenantId, testKey, config.config);
     try {
-      const client = metricsServerManager2.getClient(tenantId, testKey);
+      const client = await metricsServerManager2.getClient(tenantId, testKey);
       const result = await client.testConnection();
       metricsServerManager2.removeServer(tenantId, testKey);
       return {
@@ -3907,7 +3907,7 @@ async function listAvailableMetrics(request, reply) {
     if (!metricsServerManager2.hasServer(tenantId, key)) {
       metricsServerManager2.registerServer(tenantId, key, config.config);
     }
-    const client = metricsServerManager2.getClient(tenantId, key);
+    const client = await metricsServerManager2.getClient(tenantId, key);
     const metrics = await client.listMetrics();
     return {
       success: true,
@@ -3953,7 +3953,7 @@ async function queryMetricsData(request, reply) {
     if (!metricsServerManager2.hasServer(tenantId, key)) {
       metricsServerManager2.registerServer(tenantId, key, config.config);
     }
-    const client = metricsServerManager2.getClient(tenantId, key);
+    const client = await metricsServerManager2.getClient(tenantId, key);
     const result = await client.queryMetricData(metricName, {
       startTime,
       endTime,
@@ -6299,19 +6299,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
+import { getBindingRegistry } from "@axiom-lattice/core";
 function getTenantId12(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) return userTenantId;
@@ -6648,14 +6637,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,
@@ -6802,32 +6816,96 @@ var MessageRouter = class {
           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,
@@ -6975,7 +7053,7 @@ function createAuditLoggerMiddleware() {
 }
 
 // src/index.ts
-import { setBindingRegistry as setCoreBindingRegistry } from "@axiom-lattice/core";
+import { setBindingRegistry } from "@axiom-lattice/core";
 
 // src/swagger.ts
 import swagger from "@fastify/swagger";
@@ -7303,8 +7381,6 @@ import {
   getLoggerLattice,
   loggerLatticeManager,
   sandboxLatticeManager as sandboxLatticeManager2,
-  sqlDatabaseManager as sqlDatabaseManager2,
-  getStoreLattice as getStoreLattice15,
   agentInstanceManager as agentInstanceManager8,
   createSandboxProvider
 } from "@axiom-lattice/core";
@@ -7464,7 +7540,6 @@ var start = async (config) => {
       const bindingStore = getStoreLattice16("default", "channelBinding").store;
       const installationStore = getStoreLattice16("default", "channelInstallation").store;
       setBindingRegistry(bindingStore);
-      setCoreBindingRegistry(bindingStore);
       const adapterRegistry = new ChannelAdapterRegistry();
       adapterRegistry.register(larkChannelAdapter);
       const router = new MessageRouter({
@@ -7481,14 +7556,6 @@ var start = async (config) => {
     } catch {
     }
     registerLatticeRoutes(app, channelDeps);
-    try {
-      const storeLattice = getStoreLattice15("default", "database");
-      const store = storeLattice.store;
-      sqlDatabaseManager2.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 (!sandboxLatticeManager2.hasLattice("default")) {
       sandboxLatticeManager2.registerLattice("default", getConfiguredSandboxProvider());
       logger3.info("Registered sandbox manager from env configuration");