@axiom-lattice/gateway 2.1.35 → 2.1.37

package/dist/index.mjs

@@ -1,7 +1,3 @@
-import {
-  configService
-} from "./chunk-FSASG3SB.mjs";
-
 // src/index.ts
 import fastify from "fastify";
 import cors from "@fastify/cors";
@@ -24,31 +20,6 @@ import {
   getChunkBuffer,
   hasChunkBuffer
 } from "@axiom-lattice/core";
-async function fetchDatabaseConfigs(baseURL, apiKey, tenantId) {
-  try {
-    const headers = {};
-    if (apiKey) {
-      headers["Authorization"] = `Bearer ${apiKey}`;
-    }
-    if (tenantId) {
-      headers["x-tenant-id"] = tenantId;
-    }
-    const response = await fetch(`${baseURL}/api/database-configs`, { headers });
-    if (response.ok) {
-      const data = await response.json();
-      if (data.success && data.data && Array.isArray(data.data.records)) {
-        return data.data.records.map((record) => ({
-          key: record.key,
-          name: record.name,
-          description: record.description
-        }));
-      }
-    }
-  } catch (error) {
-    console.error("Failed to fetch database configs:", error);
-  }
-  return [];
-}
 function getOrCreateChunkBuffer() {
   if (!hasChunkBuffer("default")) {
     const buffer = new InMemoryChunkBuffer({
@@ -80,14 +51,6 @@ async function agent_invoke({
   if (!runnable_agent) {
     throw new Error(`Agent ${assistant_id} not found`);
   }
-  const { configService: configService2 } = await import("./config-F3FCBSPH.mjs");
-  const gatewayConfig = configService2.getConfig();
-  const databaseConfigs = await fetchDatabaseConfigs(
-    gatewayConfig.baseURL || "http://localhost:4001",
-    void 0,
-    tenant_id
-  );
-  global.__DATABASE_CONFIGS__ = databaseConfigs;
   const runConfig = {
     ...agentLattice?.config?.runConfig || {},
     workspaceId: workspace_id,
@@ -141,14 +104,6 @@ async function agent_stream({
     messages = [humanMessage];
   }
   const chunkBuffer = getOrCreateChunkBuffer();
-  const { configService: configService2 } = await import("./config-F3FCBSPH.mjs");
-  const gatewayConfig = configService2.getConfig();
-  const databaseConfigs = await fetchDatabaseConfigs(
-    gatewayConfig.baseURL || "http://localhost:4001",
-    void 0,
-    tenant_id
-  );
-  global.__DATABASE_CONFIGS__ = databaseConfigs;
   const runConfig = {
     ...agentLattice?.config?.runConfig || {},
     workspaceId: workspace_id,
@@ -1087,6 +1042,78 @@ async function resumeScheduledTask(request, reply) {
   }
 }
 
+// src/config.ts
+var ConfigService = class {
+  constructor() {
+    this.config = this.loadFromEnv();
+  }
+  /**
+   * Load configuration from environment variables
+   */
+  loadFromEnv() {
+    return {
+      port: process.env.PORT ? Number(process.env.PORT) : void 0,
+      queueServiceType: process.env.QUEUE_SERVICE_TYPE,
+      redisUrl: process.env.REDIS_URL,
+      redisPassword: process.env.REDIS_PASSWORD,
+      queueName: process.env.QUEUE_NAME
+    };
+  }
+  /**
+   * Update configuration from JSON object
+   * This will update both the internal config and process.env
+   */
+  updateConfig(jsonConfig) {
+    for (const [key, value] of Object.entries(jsonConfig)) {
+      if (value !== null && value !== void 0) {
+        if (typeof value === "object" && !Array.isArray(value)) {
+          for (const [nestedKey, nestedValue] of Object.entries(value)) {
+            const envKey = `${key.toUpperCase()}_${nestedKey.toUpperCase()}`;
+            process.env[envKey] = String(nestedValue);
+          }
+        } else {
+          process.env[key.toUpperCase()] = String(value);
+        }
+      }
+    }
+    this.config = this.loadFromEnv();
+    this.config = this.deepMerge(this.config, jsonConfig);
+  }
+  /**
+   * Deep merge two objects
+   */
+  deepMerge(target, source) {
+    const output = { ...target };
+    if (this.isObject(target) && this.isObject(source)) {
+      Object.keys(source).forEach((key) => {
+        if (this.isObject(source[key])) {
+          if (!(key in target)) {
+            Object.assign(output, { [key]: source[key] });
+          } else {
+            output[key] = this.deepMerge(target[key], source[key]);
+          }
+        } else {
+          Object.assign(output, { [key]: source[key] });
+        }
+      });
+    }
+    return output;
+  }
+  /**
+   * Check if value is a plain object
+   */
+  isObject(item) {
+    return item && typeof item === "object" && !Array.isArray(item);
+  }
+  /**
+   * Get current configuration
+   */
+  getConfig() {
+    return { ...this.config };
+  }
+};
+var configService = new ConfigService();
+
 // src/services/queue_service.ts
 import {
   queueLatticeManager,
@@ -3356,12 +3383,14 @@ async function getDataSources(request, reply) {
     }
     const semanticConfig = config.config;
     const client = new SemanticMetricsClient(semanticConfig);
-    const datasources = await client.getDataSources();
+    const allDatasources = await client.getDataSources();
+    const selectedIds = semanticConfig.selectedDataSources || [];
+    const filteredDatasources = selectedIds.length > 0 ? allDatasources.filter((ds) => selectedIds.includes(String(ds.id))) : allDatasources;
     return {
       success: true,
       message: "Data sources retrieved successfully",
       data: {
-        datasources
+        datasources: filteredDatasources
       }
     };
   } catch (error) {