@flutchai/flutch-sdk 0.1.5 → 0.1.6

package/dist/index.js

@@ -1,10 +1,12 @@
 import 'reflect-metadata';
-import { Injectable, Post, UseGuards, Req, Controller, Inject, Get, Body, Res, Param, Logger, Module, UnauthorizedException, Headers, NotFoundException, InternalServerErrorException, ForbiddenException, ValidationPipe, HttpException, HttpStatus } from '@nestjs/common';
+import { Injectable, Post, UseGuards, Req, Controller, Inject, Get, Body, Res, Param, Logger, Module, UnauthorizedException, Optional, Headers, NotFoundException, InternalServerErrorException, ForbiddenException, ValidationPipe, HttpException, HttpStatus } from '@nestjs/common';
 import { ApiOperation, ApiResponse, ApiTags, ApiParam, DocumentBuilder, SwaggerModule } from '@nestjs/swagger';
-import * as path from 'path';
+import * as fs from 'fs';
+import * as path2 from 'path';
+import * as os from 'os';
 import { randomBytes, createHash, randomUUID } from 'crypto';
 import { Registry, collectDefaultMetrics, Counter, Histogram, Gauge } from 'prom-client';
-import { NestFactory, ModuleRef, DiscoveryModule } from '@nestjs/core';
+import { NestFactory, MetadataScanner, ModuleRef, DiscoveryModule } from '@nestjs/core';
 import * as net from 'net';
 import { ConfigModule } from '@nestjs/config';
 import * as LangGraph from '@langchain/langgraph';
@@ -432,12 +434,21 @@ var init_ui_endpoints_discovery = __esm({
         this.endpointRegistry = endpointRegistry;
       }
       logger = new Logger(UIEndpointsDiscoveryService.name);
+      async onModuleInit() {
+        await this.discoverUIEndpoints();
+      }
       /**
        * Discover and register all UI endpoint classes
        * Called automatically during module initialization
        */
       async discoverUIEndpoints() {
         this.logger.log("Starting UI endpoints discovery...");
+        if (!this.discoveryService) {
+          this.logger.warn(
+            "DiscoveryService not available, skipping UI endpoints discovery"
+          );
+          return;
+        }
         const providers = this.discoveryService.getProviders();
         let registeredCount = 0;
         let totalEndpoints = 0;
@@ -489,7 +500,9 @@ var init_ui_endpoints_discovery = __esm({
       }
     };
     UIEndpointsDiscoveryService = __decorateClass([
-      Injectable()
+      Injectable(),
+      __decorateParam(0, Optional()),
+      __decorateParam(1, Optional())
     ], UIEndpointsDiscoveryService);
   }
 });
@@ -684,6 +697,523 @@ var init_agent_ui = __esm({
     init_ui_dispatch_controller();
   }
 });
+
+// src/service-discovery/file-based.discovery.ts
+var file_based_discovery_exports = {};
+__export(file_based_discovery_exports, {
+  FileBasedDiscovery: () => FileBasedDiscovery
+});
+var FileBasedDiscovery;
+var init_file_based_discovery = __esm({
+  "src/service-discovery/file-based.discovery.ts"() {
+    FileBasedDiscovery = class {
+      logger = new Logger(FileBasedDiscovery.name);
+      servicesDir;
+      constructor() {
+        this.servicesDir = path2.join(os.homedir(), ".flutch", "services");
+        this.ensureServicesDirectory();
+        this.cleanupStaleServices();
+      }
+      /**
+       * Register service
+       */
+      async registerService(name, address, port, metadata, graphTypes = []) {
+        const registration = {
+          name,
+          address,
+          port,
+          metadata,
+          pid: process.pid,
+          timestamp: Date.now(),
+          graphTypes
+        };
+        const serviceFile = path2.join(
+          this.servicesDir,
+          `${name}-${process.pid}.json`
+        );
+        try {
+          await fs.promises.writeFile(
+            serviceFile,
+            JSON.stringify(registration, null, 2)
+          );
+          this.logger.log(`Registered service: ${name} at ${address}:${port}`);
+          process.on("exit", () => this.unregisterService(name));
+          process.on("SIGINT", () => {
+            this.unregisterService(name);
+            process.exit(0);
+          });
+          process.on("SIGTERM", () => {
+            this.unregisterService(name);
+            process.exit(0);
+          });
+        } catch (error) {
+          this.logger.error(`Failed to register service ${name}: ${error.message}`);
+        }
+      }
+      /**
+       * Unregister service
+       */
+      async unregisterService(name) {
+        const serviceFile = path2.join(
+          this.servicesDir,
+          `${name}-${process.pid}.json`
+        );
+        try {
+          if (fs.existsSync(serviceFile)) {
+            await fs.promises.unlink(serviceFile);
+            this.logger.log(`Unregistered service: ${name}`);
+          }
+        } catch (error) {
+          this.logger.error(
+            `Failed to unregister service ${name}: ${error.message}`
+          );
+        }
+      }
+      /**
+       * Get list of services by graph type
+       */
+      async getServices(graphType) {
+        try {
+          const files = await fs.promises.readdir(this.servicesDir);
+          const services = [];
+          for (const file of files) {
+            if (!file.endsWith(".json")) continue;
+            try {
+              const serviceFile = path2.join(this.servicesDir, file);
+              const content = await fs.promises.readFile(serviceFile, "utf-8");
+              const registration = JSON.parse(content);
+              if (!this.isProcessAlive(registration.pid)) {
+                await fs.promises.unlink(serviceFile);
+                continue;
+              }
+              if (registration.graphTypes.includes(graphType) || registration.metadata.graphTypes?.includes(graphType)) {
+                services.push({
+                  name: registration.name,
+                  address: registration.address,
+                  port: registration.port,
+                  metadata: registration.metadata
+                });
+              }
+            } catch (error) {
+              this.logger.warn(
+                `Failed to parse service file ${file}: ${error.message}`
+              );
+            }
+          }
+          return services;
+        } catch (error) {
+          this.logger.error(`Failed to get services: ${error.message}`);
+          return [];
+        }
+      }
+      /**
+       * Find service by name
+       */
+      async findServiceByName(serviceName) {
+        try {
+          const files = await fs.promises.readdir(this.servicesDir);
+          for (const file of files) {
+            if (!file.endsWith(".json")) continue;
+            try {
+              const serviceFile = path2.join(this.servicesDir, file);
+              const content = await fs.promises.readFile(serviceFile, "utf-8");
+              const registration = JSON.parse(content);
+              if (registration.name === serviceName && this.isProcessAlive(registration.pid)) {
+                return {
+                  name: registration.name,
+                  address: registration.address,
+                  port: registration.port,
+                  metadata: registration.metadata
+                };
+              }
+            } catch (error) {
+              this.logger.warn(
+                `Failed to parse service file ${file}: ${error.message}`
+              );
+            }
+          }
+          return null;
+        } catch (error) {
+          this.logger.error(
+            `Failed to find service ${serviceName}: ${error.message}`
+          );
+          return null;
+        }
+      }
+      /**
+       * Ensure services directory exists
+       */
+      ensureServicesDirectory() {
+        try {
+          const amelieDir = path2.join(os.homedir(), ".flutch");
+          if (!fs.existsSync(amelieDir)) {
+            fs.mkdirSync(amelieDir, { recursive: true });
+          }
+          if (!fs.existsSync(this.servicesDir)) {
+            fs.mkdirSync(this.servicesDir, { recursive: true });
+          }
+        } catch (error) {
+          this.logger.error(
+            `Failed to create services directory: ${error.message}`
+          );
+        }
+      }
+      /**
+       * Cleanup stale services
+       */
+      async cleanupStaleServices() {
+        try {
+          const files = await fs.promises.readdir(this.servicesDir);
+          for (const file of files) {
+            if (!file.endsWith(".json")) continue;
+            try {
+              const serviceFile = path2.join(this.servicesDir, file);
+              const content = await fs.promises.readFile(serviceFile, "utf-8");
+              const registration = JSON.parse(content);
+              if (!this.isProcessAlive(registration.pid)) {
+                await fs.promises.unlink(serviceFile);
+                this.logger.debug(`Cleaned up stale service: ${registration.name}`);
+              }
+            } catch (error) {
+              try {
+                await fs.promises.unlink(path2.join(this.servicesDir, file));
+              } catch {
+              }
+            }
+          }
+        } catch (error) {
+          this.logger.error(`Failed to cleanup stale services: ${error.message}`);
+        }
+      }
+      /**
+       * Check if process is still alive
+       */
+      isProcessAlive(pid) {
+        try {
+          process.kill(pid, 0);
+          return true;
+        } catch (error) {
+          return false;
+        }
+      }
+    };
+    FileBasedDiscovery = __decorateClass([
+      Injectable()
+    ], FileBasedDiscovery);
+  }
+});
+
+// src/core/builder-registry.service.ts
+var builder_registry_service_exports = {};
+__export(builder_registry_service_exports, {
+  BuilderRegistryService: () => BuilderRegistryService
+});
+var BuilderRegistryService;
+var init_builder_registry_service = __esm({
+  "src/core/builder-registry.service.ts"() {
+    BuilderRegistryService = class {
+      builders = [];
+      registerBuilder(builder) {
+        const existingBuilder = this.builders.find(
+          (b) => b.graphType === builder.graphType
+        );
+        if (!existingBuilder) {
+          this.builders.push(builder);
+        }
+      }
+      getBuilders() {
+        return this.builders;
+      }
+    };
+    BuilderRegistryService = __decorateClass([
+      Injectable()
+    ], BuilderRegistryService);
+  }
+});
+
+// src/utils/graph-type.utils.ts
+var GraphTypeUtils;
+var init_graph_type_utils = __esm({
+  "src/utils/graph-type.utils.ts"() {
+    GraphTypeUtils = class {
+      /**
+       * Parse full graph type
+       * @param fullType - full graph type
+       * @returns object with type components
+       */
+      static parse(fullType) {
+        if (fullType.includes("::")) {
+          const [baseType, version] = fullType.split("::");
+          const [companyId, name] = baseType.split(".");
+          return { companyId, name, version };
+        }
+        const parts = fullType.split(".");
+        if (parts.length === 1) {
+          return { companyId: "global", name: parts[0] };
+        }
+        return { companyId: parts[0], name: parts[1] };
+      }
+      /**
+       * Build full type from components
+       * @param companyId - company ID
+       * @param name - graph name
+       * @param version - version (optional)
+       * @returns full graph type
+       */
+      static build(companyId, name, version) {
+        const base = `${companyId}.${name}`;
+        return version ? `${base}::${version}` : base;
+      }
+      /**
+       * Normalize graph type for backward compatibility
+       * @param graphType - graph type in any format
+       * @returns normalized type
+       */
+      static normalize(graphType) {
+        const { companyId, name, version } = this.parse(graphType);
+        return this.build(companyId, name, version);
+      }
+      /**
+       * Get base type without version
+       * @param graphType - full graph type
+       * @returns base type
+       */
+      static getBaseType(graphType) {
+        if (graphType.includes("::")) {
+          return graphType.split("::")[0];
+        }
+        const { companyId, name } = this.parse(graphType);
+        return `${companyId}.${name}`;
+      }
+      /**
+       * Extract version from graph type
+       * @param graphType - full graph type
+       * @returns version or undefined
+       */
+      static getVersion(graphType) {
+        return graphType.includes("::") ? graphType.split("::")[1] : void 0;
+      }
+      /**
+       * Validate version (basic semver check)
+       * @param version - version to check
+       * @returns true if version is valid
+       */
+      static isValidVersion(version) {
+        const semverRegex = /^\d+\.\d+\.\d+(-[a-zA-Z0-9.-]+)?$/;
+        return semverRegex.test(version);
+      }
+      /**
+       * Check if type is system (global)
+       * @param graphType - graph type
+       * @returns true if graph is system
+       */
+      static isSystemGraph(graphType) {
+        const { companyId } = this.parse(graphType);
+        return companyId === "global";
+      }
+      /**
+       * Compare versions (simple comparison for sorting)
+       * @param a - first version
+       * @param b - second version
+       * @returns -1, 0, 1 for sorting
+       */
+      static compareVersions(a, b) {
+        const parseVersion = (v) => {
+          const [main, prerelease] = v.split("-");
+          const [major, minor, patch] = main.split(".").map(Number);
+          return { major, minor, patch, prerelease };
+        };
+        const versionA = parseVersion(a);
+        const versionB = parseVersion(b);
+        if (versionA.major !== versionB.major) {
+          return versionA.major - versionB.major;
+        }
+        if (versionA.minor !== versionB.minor) {
+          return versionA.minor - versionB.minor;
+        }
+        if (versionA.patch !== versionB.patch) {
+          return versionA.patch - versionB.patch;
+        }
+        if (versionA.prerelease && !versionB.prerelease) return -1;
+        if (!versionA.prerelease && versionB.prerelease) return 1;
+        if (versionA.prerelease && versionB.prerelease) {
+          return versionA.prerelease.localeCompare(versionB.prerelease);
+        }
+        return 0;
+      }
+    };
+  }
+});
+
+// src/graph/versioning/versioned-graph.service.ts
+var versioned_graph_service_exports = {};
+__export(versioned_graph_service_exports, {
+  VersionedGraphService: () => VersionedGraphService
+});
+var VersionedGraphService;
+var init_versioned_graph_service = __esm({
+  "src/graph/versioning/versioned-graph.service.ts"() {
+    init_graph_type_utils();
+    VersionedGraphService = class {
+      logger = new Logger(VersionedGraphService.name);
+      versionConfigs = /* @__PURE__ */ new Map();
+      constructor() {
+      }
+      /**
+       * Register versioning configuration
+       */
+      registerVersioning(config) {
+        this.versionConfigs.set(config.baseGraphType, config);
+        this.logger.log(
+          `Registered versioning for ${config.baseGraphType} with ${config.versions.length} versions`
+        );
+      }
+      /**
+       * Resolve graph version
+       */
+      async resolveVersion(graphType, options = {}) {
+        const parsed = GraphTypeUtils.parse(graphType);
+        const baseType = GraphTypeUtils.getBaseType(graphType);
+        const requestedVersion = GraphTypeUtils.getVersion(graphType) || options.requestedVersion;
+        const config = this.versionConfigs.get(baseType);
+        if (!config) {
+          throw new Error(`No versioning configuration found for ${baseType}`);
+        }
+        const route = this.findBestVersionRoute(config, requestedVersion, options);
+        if (!route) {
+          throw new Error(
+            `No compatible version found for ${graphType} with options: ${JSON.stringify(options)}`
+          );
+        }
+        const fullGraphType = GraphTypeUtils.build(
+          parsed.companyId,
+          parsed.name,
+          route.version
+        );
+        return {
+          version: route.version,
+          builderClass: route.builderClass,
+          fullGraphType
+        };
+      }
+      /**
+       * Create builder for specified version
+       */
+      async createVersionedBuilder(graphType, moduleRef, options = {}) {
+        const resolution = await this.resolveVersion(graphType, options);
+        try {
+          if (!moduleRef) {
+            throw new Error(
+              "ModuleRef is not available - falling back to direct instantiation"
+            );
+          }
+          const builder = await moduleRef.create(resolution.builderClass);
+          this.logger.debug(
+            `Created versioned builder for ${resolution.fullGraphType}`
+          );
+          return builder;
+        } catch (error) {
+          this.logger.error(
+            `Failed to create builder for ${resolution.fullGraphType}: ${error.message}`
+          );
+          throw new Error(`Failed to create versioned builder: ${error.message}`);
+        }
+      }
+      /**
+       * Get all available versions for a graph
+       */
+      getAvailableVersions(baseGraphType) {
+        const config = this.versionConfigs.get(baseGraphType);
+        if (!config) {
+          return [];
+        }
+        return config.versions.map((route) => route.version).sort((a, b) => GraphTypeUtils.compareVersions(b, a));
+      }
+      /**
+       * Check version support
+       */
+      isVersionSupported(graphType, options = {}) {
+        try {
+          const baseType = GraphTypeUtils.getBaseType(graphType);
+          const requestedVersion = GraphTypeUtils.getVersion(graphType);
+          const config = this.versionConfigs.get(baseType);
+          if (!config) {
+            return false;
+          }
+          const route = this.findBestVersionRoute(
+            config,
+            requestedVersion,
+            options
+          );
+          return route !== null;
+        } catch {
+          return false;
+        }
+      }
+      /**
+       * Find best version
+       */
+      findBestVersionRoute(config, requestedVersion, options = {}) {
+        const { strict = false } = options;
+        let candidates = [...config.versions];
+        if (requestedVersion) {
+          const exactMatch = candidates.find(
+            (route) => route.version === requestedVersion
+          );
+          if (exactMatch) {
+            return exactMatch;
+          }
+          if (strict) {
+            return null;
+          }
+          const compatibleVersions = candidates.filter(
+            (route) => this.isVersionCompatible(requestedVersion, route.version)
+          );
+          if (compatibleVersions.length > 0) {
+            return compatibleVersions.sort(
+              (a, b) => GraphTypeUtils.compareVersions(b.version, a.version)
+            )[0];
+          }
+        }
+        const defaultRoute = candidates.find((route) => route.isDefault);
+        if (defaultRoute) {
+          return defaultRoute;
+        }
+        if (config.defaultVersionStrategy === "latest" || !config.defaultVersionStrategy) {
+          const sortedVersions = candidates.sort(
+            (a, b) => GraphTypeUtils.compareVersions(b.version, a.version)
+          );
+          return sortedVersions[0] || null;
+        }
+        return null;
+      }
+      /**
+       * Check version compatibility (simplified semantics)
+       */
+      isVersionCompatible(requested, available) {
+        try {
+          const requestedParts = requested.split(".").map(Number);
+          const availableParts = available.split(".").map(Number);
+          if (requestedParts[0] !== availableParts[0]) {
+            return false;
+          }
+          return GraphTypeUtils.compareVersions(available, requested) >= 0;
+        } catch {
+          return false;
+        }
+      }
+      /**
+       * Get versioning configuration information
+       */
+      getVersioningInfo(baseGraphType) {
+        return this.versionConfigs.get(baseGraphType) || null;
+      }
+    };
+    VersionedGraphService = __decorateClass([
+      Injectable()
+    ], VersionedGraphService);
+  }
+});
 var CallbackStore = class {
   constructor(redis) {
     this.redis = redis;
@@ -2525,7 +3055,7 @@ var _AbstractGraphBuilder = class _AbstractGraphBuilder {
   /**
    * Path to root graph manifest (defaults to graph.manifest.json in root)
    */
-  manifestPath = path.join(
+  manifestPath = path2.join(
     process.cwd(),
     "graph.manifest.json"
   );
@@ -2605,10 +3135,10 @@ var _AbstractGraphBuilder = class _AbstractGraphBuilder {
       return null;
     }
     try {
-      const fs = await import('fs/promises');
-      const path2 = await import('path');
-      const manifestFullPath = path2.resolve(this.manifestPath);
-      const manifestContent = await fs.readFile(manifestFullPath, "utf-8");
+      const fs2 = await import('fs/promises');
+      const path3 = await import('path');
+      const manifestFullPath = path3.resolve(this.manifestPath);
+      const manifestContent = await fs2.readFile(manifestFullPath, "utf-8");
       const manifest = JSON.parse(manifestContent);
       this.manifest = manifest;
       return manifest;
@@ -2627,10 +3157,10 @@ var _AbstractGraphBuilder = class _AbstractGraphBuilder {
       return null;
     }
     try {
-      const fs = __require("fs");
-      const path2 = __require("path");
-      const manifestFullPath = path2.resolve(this.manifestPath);
-      const manifestContent = fs.readFileSync(manifestFullPath, "utf-8");
+      const fs2 = __require("fs");
+      const path3 = __require("path");
+      const manifestFullPath = path3.resolve(this.manifestPath);
+      const manifestContent = fs2.readFileSync(manifestFullPath, "utf-8");
       const manifest = JSON.parse(manifestContent);
       this.manifest = manifest;
       return manifest;
@@ -2670,12 +3200,12 @@ var _AbstractGraphBuilder = class _AbstractGraphBuilder {
     let configSchema = null;
     if (versionConfig.configSchemaPath) {
       try {
-        const fs = await import('fs/promises');
-        const schemaPath = path.resolve(
+        const fs2 = await import('fs/promises');
+        const schemaPath = path2.resolve(
           process.cwd(),
           versionConfig.configSchemaPath
         );
-        const schemaContent = await fs.readFile(schemaPath, "utf-8");
+        const schemaContent = await fs2.readFile(schemaPath, "utf-8");
         const schemaData = JSON.parse(schemaContent);
         configSchema = schemaData.schema;
       } catch (error) {
@@ -3140,6 +3670,20 @@ UniversalGraphService = __decorateClass([
   __decorateParam(2, Inject("GRAPH_ENGINE")),
   __decorateParam(3, Inject(EndpointRegistry))
 ], UniversalGraphService);
+function setupRedisMock() {
+  if (process.env.NODE_ENV === "development" && !process.env.KUBERNETES_SERVICE_HOST) {
+    console.log("[REDIS_MOCK] Intercepting ioredis requires for development");
+    const Module2 = __require("module");
+    const originalRequire = Module2.prototype.require;
+    Module2.prototype.require = function(...args) {
+      if (args[0] === "ioredis") {
+        console.log("[REDIS_MOCK] Redirecting ioredis to ioredis-mock");
+        return originalRequire.apply(this, ["ioredis-mock"]);
+      }
+      return originalRequire.apply(this, args);
+    };
+  }
+}
 async function isPortAvailable(port) {
   return new Promise((resolve2) => {
     const server = net.createServer();
@@ -3161,10 +3705,110 @@ async function findAvailablePort(startPort) {
     `No available ports found in range ${startPort}-${startPort + 99}`
   );
 }
+async function registerWithServiceDiscovery(AppModule, port, logger2, app) {
+  try {
+    logger2.debug(
+      "[SERVICE_DISCOVERY] Starting service discovery registration..."
+    );
+    const { FileBasedDiscovery: FileBasedDiscovery2 } = await Promise.resolve().then(() => (init_file_based_discovery(), file_based_discovery_exports));
+    logger2.debug(
+      "[SERVICE_DISCOVERY] FileBasedDiscovery imported successfully"
+    );
+    const discovery = new FileBasedDiscovery2();
+    logger2.debug("[SERVICE_DISCOVERY] FileBasedDiscovery instance created");
+    const serviceName = AppModule.name.replace("Module", "").toLowerCase();
+    let graphTypes = [];
+    try {
+      if (app) {
+        const { BuilderRegistryService: BuilderRegistryService2 } = await Promise.resolve().then(() => (init_builder_registry_service(), builder_registry_service_exports));
+        const builderRegistry = app.get(BuilderRegistryService2, {
+          strict: false
+        });
+        logger2.debug(
+          `BuilderRegistryService found in DI: ${!!builderRegistry}`
+        );
+        if (builderRegistry) {
+          const builders = builderRegistry.getBuilders();
+          logger2.debug(`Found ${builders.length} builders in registry`);
+          const baseGraphTypes = builders.map((builder) => builder.graphType);
+          logger2.debug(`Base graph types: [${baseGraphTypes.join(", ")}]`);
+          try {
+            logger2.debug("Attempting to import VersionedGraphService...");
+            const { VersionedGraphService: VersionedGraphService2 } = await Promise.resolve().then(() => (init_versioned_graph_service(), versioned_graph_service_exports));
+            logger2.debug("VersionedGraphService imported successfully");
+            const versionedService = app.get(VersionedGraphService2, {
+              strict: false
+            });
+            logger2.debug(
+              `VersionedGraphService found in DI: ${!!versionedService}`
+            );
+            if (versionedService) {
+              const allVersionedTypes = [];
+              for (const baseType of baseGraphTypes) {
+                logger2.debug(`Getting versions for base type: ${baseType}`);
+                const versions = versionedService.getAvailableVersions(baseType);
+                logger2.debug(
+                  `Found ${versions.length} versions for ${baseType}: [${versions.join(", ")}]`
+                );
+                if (versions.length > 0) {
+                  allVersionedTypes.push(baseType);
+                  versions.forEach((version) => {
+                    allVersionedTypes.push(`${baseType}::${version}`);
+                  });
+                } else {
+                  allVersionedTypes.push(baseType);
+                }
+              }
+              graphTypes = allVersionedTypes;
+              logger2.debug(
+                `Found ${baseGraphTypes.length} base types with ${graphTypes.length} total versioned types: ${graphTypes.join(", ")}`
+              );
+            } else {
+              graphTypes = baseGraphTypes;
+              logger2.debug(
+                `VersionedGraphService not found in DI container, using base types: ${graphTypes.join(", ")}`
+              );
+            }
+          } catch (error) {
+            graphTypes = baseGraphTypes;
+            logger2.debug(
+              `Failed to get versioned types, using base types: ${error.message}`
+            );
+            logger2.debug(`Stack trace: ${error.stack}`);
+          }
+        }
+      }
+    } catch (error) {
+      logger2.debug(`Failed to get graph types from builders: ${error.message}`);
+    }
+    if (graphTypes.length === 0) {
+      logger2.debug("No builders found, using service name as graph type");
+      graphTypes = [serviceName];
+    }
+    await discovery.registerService(
+      serviceName,
+      "localhost",
+      port,
+      {
+        graphTypes,
+        environment: "development",
+        startTime: (/* @__PURE__ */ new Date()).toISOString(),
+        version: process.env.npm_package_version || "1.0.0"
+      },
+      graphTypes
+    );
+    logger2.log(
+      `\u{1F4E1} Service registered with discovery: ${serviceName} (types: ${graphTypes.join(", ")})`
+    );
+  } catch (error) {
+    logger2.warn(`Failed to register with service discovery: ${error.message}`);
+  }
+}
 async function bootstrap(AppModule, options = {}) {
+  setupRedisMock();
   const app = await NestFactory.create(AppModule);
   const logger2 = new Logger("Bootstrap");
-  const requestedPort = options.port || parseInt(process.env.PORT || "3000", 10);
+  const requestedPort = options.port || parseInt(process.env.PORT || "3100", 10);
   const port = await findAvailablePort(requestedPort);
   const globalPrefix = options.globalPrefix;
   if (port !== requestedPort) {
@@ -3191,57 +3835,21 @@ async function bootstrap(AppModule, options = {}) {
   const baseUrl = globalPrefix ? `http://localhost:${port}/${globalPrefix}` : `http://localhost:${port}`;
   logger2.log(`\u{1F680} Graph service is running on: ${baseUrl}`);
   logger2.log(`\u{1F4DA} API Documentation: http://localhost:${port}/api/docs`);
-  if (process.env.NODE_ENV !== "production") ;
-  return app;
-}
-var BuilderRegistryService = class {
-  builders = [];
-  registerBuilder(builder) {
-    const existingBuilder = this.builders.find(
-      (b) => b.graphType === builder.graphType
+  if (process.env.NODE_ENV !== "production") {
+    logger2.debug(
+      `Attempting service discovery registration (NODE_ENV: ${process.env.NODE_ENV || "undefined"})`
+    );
+    await registerWithServiceDiscovery(AppModule, port, logger2, app);
+  } else {
+    logger2.debug(
+      `Skipping service discovery registration (NODE_ENV: ${process.env.NODE_ENV})`
     );
-    if (!existingBuilder) {
-      this.builders.push(builder);
-    }
-  }
-  getBuilders() {
-    return this.builders;
-  }
-};
-BuilderRegistryService = __decorateClass([
-  Injectable()
-], BuilderRegistryService);
-var GraphEngineType = /* @__PURE__ */ ((GraphEngineType2) => {
-  GraphEngineType2["LANGGRAPH"] = "langgraph";
-  GraphEngineType2["LANGFLOW"] = "langflow";
-  GraphEngineType2["FLOWISE"] = "flowise";
-  return GraphEngineType2;
-})(GraphEngineType || {});
-var GraphEngineFactory = class {
-  constructor(langgraph) {
-    this.langgraph = langgraph;
-  }
-  /**
-   * Get engine for the specified type
-   */
-  getEngine(engineType) {
-    switch (engineType) {
-      case "langgraph" /* LANGGRAPH */:
-        return this.langgraph;
-      // Will add other types in the future
-      // case GraphEngineType.LANGFLOW:
-      //   return new LangFlowEngine();
-      // case GraphEngineType.FLOWISE:
-      //   return new FlowiseEngine();
-      default:
-        throw new Error(`Unsupported graph engine type: ${engineType}`);
-    }
   }
-};
-GraphEngineFactory = __decorateClass([
-  Injectable(),
-  __decorateParam(0, Inject())
-], GraphEngineFactory);
+  return app;
+}
+
+// src/core/index.ts
+init_builder_registry_service();
 
 // src/messages/attachments.ts
 var AttachmentType = /* @__PURE__ */ ((AttachmentType2) => {
@@ -3623,274 +4231,8 @@ ${result.errors.join("\n")}`
   }
 };
 
-// src/utils/graph-type.utils.ts
-var GraphTypeUtils = class {
-  /**
-   * Parse full graph type
-   * @param fullType - full graph type
-   * @returns object with type components
-   */
-  static parse(fullType) {
-    if (fullType.includes("::")) {
-      const [baseType, version] = fullType.split("::");
-      const [companyId, name] = baseType.split(".");
-      return { companyId, name, version };
-    }
-    const parts = fullType.split(".");
-    if (parts.length === 1) {
-      return { companyId: "global", name: parts[0] };
-    }
-    return { companyId: parts[0], name: parts[1] };
-  }
-  /**
-   * Build full type from components
-   * @param companyId - company ID
-   * @param name - graph name
-   * @param version - version (optional)
-   * @returns full graph type
-   */
-  static build(companyId, name, version) {
-    const base = `${companyId}.${name}`;
-    return version ? `${base}::${version}` : base;
-  }
-  /**
-   * Normalize graph type for backward compatibility
-   * @param graphType - graph type in any format
-   * @returns normalized type
-   */
-  static normalize(graphType) {
-    const { companyId, name, version } = this.parse(graphType);
-    return this.build(companyId, name, version);
-  }
-  /**
-   * Get base type without version
-   * @param graphType - full graph type
-   * @returns base type
-   */
-  static getBaseType(graphType) {
-    if (graphType.includes("::")) {
-      return graphType.split("::")[0];
-    }
-    const { companyId, name } = this.parse(graphType);
-    return `${companyId}.${name}`;
-  }
-  /**
-   * Extract version from graph type
-   * @param graphType - full graph type
-   * @returns version or undefined
-   */
-  static getVersion(graphType) {
-    return graphType.includes("::") ? graphType.split("::")[1] : void 0;
-  }
-  /**
-   * Validate version (basic semver check)
-   * @param version - version to check
-   * @returns true if version is valid
-   */
-  static isValidVersion(version) {
-    const semverRegex = /^\d+\.\d+\.\d+(-[a-zA-Z0-9.-]+)?$/;
-    return semverRegex.test(version);
-  }
-  /**
-   * Check if type is system (global)
-   * @param graphType - graph type
-   * @returns true if graph is system
-   */
-  static isSystemGraph(graphType) {
-    const { companyId } = this.parse(graphType);
-    return companyId === "global";
-  }
-  /**
-   * Compare versions (simple comparison for sorting)
-   * @param a - first version
-   * @param b - second version
-   * @returns -1, 0, 1 for sorting
-   */
-  static compareVersions(a, b) {
-    const parseVersion = (v) => {
-      const [main, prerelease] = v.split("-");
-      const [major, minor, patch] = main.split(".").map(Number);
-      return { major, minor, patch, prerelease };
-    };
-    const versionA = parseVersion(a);
-    const versionB = parseVersion(b);
-    if (versionA.major !== versionB.major) {
-      return versionA.major - versionB.major;
-    }
-    if (versionA.minor !== versionB.minor) {
-      return versionA.minor - versionB.minor;
-    }
-    if (versionA.patch !== versionB.patch) {
-      return versionA.patch - versionB.patch;
-    }
-    if (versionA.prerelease && !versionB.prerelease) return -1;
-    if (!versionA.prerelease && versionB.prerelease) return 1;
-    if (versionA.prerelease && versionB.prerelease) {
-      return versionA.prerelease.localeCompare(versionB.prerelease);
-    }
-    return 0;
-  }
-};
-
-// src/graph/versioning/versioned-graph.service.ts
-var VersionedGraphService = class {
-  logger = new Logger(VersionedGraphService.name);
-  versionConfigs = /* @__PURE__ */ new Map();
-  constructor() {
-  }
-  /**
-   * Register versioning configuration
-   */
-  registerVersioning(config) {
-    this.versionConfigs.set(config.baseGraphType, config);
-    this.logger.log(
-      `Registered versioning for ${config.baseGraphType} with ${config.versions.length} versions`
-    );
-  }
-  /**
-   * Resolve graph version
-   */
-  async resolveVersion(graphType, options = {}) {
-    const parsed = GraphTypeUtils.parse(graphType);
-    const baseType = GraphTypeUtils.getBaseType(graphType);
-    const requestedVersion = GraphTypeUtils.getVersion(graphType) || options.requestedVersion;
-    const config = this.versionConfigs.get(baseType);
-    if (!config) {
-      throw new Error(`No versioning configuration found for ${baseType}`);
-    }
-    const route = this.findBestVersionRoute(config, requestedVersion, options);
-    if (!route) {
-      throw new Error(
-        `No compatible version found for ${graphType} with options: ${JSON.stringify(options)}`
-      );
-    }
-    const fullGraphType = GraphTypeUtils.build(
-      parsed.companyId,
-      parsed.name,
-      route.version
-    );
-    return {
-      version: route.version,
-      builderClass: route.builderClass,
-      fullGraphType
-    };
-  }
-  /**
-   * Create builder for specified version
-   */
-  async createVersionedBuilder(graphType, moduleRef, options = {}) {
-    const resolution = await this.resolveVersion(graphType, options);
-    try {
-      if (!moduleRef) {
-        throw new Error(
-          "ModuleRef is not available - falling back to direct instantiation"
-        );
-      }
-      const builder = await moduleRef.create(resolution.builderClass);
-      this.logger.debug(
-        `Created versioned builder for ${resolution.fullGraphType}`
-      );
-      return builder;
-    } catch (error) {
-      this.logger.error(
-        `Failed to create builder for ${resolution.fullGraphType}: ${error.message}`
-      );
-      throw new Error(`Failed to create versioned builder: ${error.message}`);
-    }
-  }
-  /**
-   * Get all available versions for a graph
-   */
-  getAvailableVersions(baseGraphType) {
-    const config = this.versionConfigs.get(baseGraphType);
-    if (!config) {
-      return [];
-    }
-    return config.versions.map((route) => route.version).sort((a, b) => GraphTypeUtils.compareVersions(b, a));
-  }
-  /**
-   * Check version support
-   */
-  isVersionSupported(graphType, options = {}) {
-    try {
-      const baseType = GraphTypeUtils.getBaseType(graphType);
-      const requestedVersion = GraphTypeUtils.getVersion(graphType);
-      const config = this.versionConfigs.get(baseType);
-      if (!config) {
-        return false;
-      }
-      const route = this.findBestVersionRoute(
-        config,
-        requestedVersion,
-        options
-      );
-      return route !== null;
-    } catch {
-      return false;
-    }
-  }
-  /**
-   * Find best version
-   */
-  findBestVersionRoute(config, requestedVersion, options = {}) {
-    const { strict = false } = options;
-    let candidates = [...config.versions];
-    if (requestedVersion) {
-      const exactMatch = candidates.find(
-        (route) => route.version === requestedVersion
-      );
-      if (exactMatch) {
-        return exactMatch;
-      }
-      if (strict) {
-        return null;
-      }
-      const compatibleVersions = candidates.filter(
-        (route) => this.isVersionCompatible(requestedVersion, route.version)
-      );
-      if (compatibleVersions.length > 0) {
-        return compatibleVersions.sort(
-          (a, b) => GraphTypeUtils.compareVersions(b.version, a.version)
-        )[0];
-      }
-    }
-    const defaultRoute = candidates.find((route) => route.isDefault);
-    if (defaultRoute) {
-      return defaultRoute;
-    }
-    if (config.defaultVersionStrategy === "latest" || !config.defaultVersionStrategy) {
-      const sortedVersions = candidates.sort(
-        (a, b) => GraphTypeUtils.compareVersions(b.version, a.version)
-      );
-      return sortedVersions[0] || null;
-    }
-    return null;
-  }
-  /**
-   * Check version compatibility (simplified semantics)
-   */
-  isVersionCompatible(requested, available) {
-    try {
-      const requestedParts = requested.split(".").map(Number);
-      const availableParts = available.split(".").map(Number);
-      if (requestedParts[0] !== availableParts[0]) {
-        return false;
-      }
-      return GraphTypeUtils.compareVersions(available, requested) >= 0;
-    } catch {
-      return false;
-    }
-  }
-  /**
-   * Get versioning configuration information
-   */
-  getVersioningInfo(baseGraphType) {
-    return this.versionConfigs.get(baseGraphType) || null;
-  }
-};
-VersionedGraphService = __decorateClass([
-  Injectable()
-], VersionedGraphService);
+// src/graph/versioning/index.ts
+init_versioned_graph_service();
 var logger = new Logger("ApiCallTracer");
 var DEFAULT_TRACER_OPTIONS = {
   maxStringLength: 5e3,
@@ -4034,8 +4376,36 @@ function sanitizeTraceError(error, options) {
     raw: sanitizeTraceData(error, 0, /* @__PURE__ */ new WeakSet(), options)
   };
 }
-
-// src/engines/langgraph/event-processor.utils.ts
+var GraphEngineType = /* @__PURE__ */ ((GraphEngineType2) => {
+  GraphEngineType2["LANGGRAPH"] = "langgraph";
+  GraphEngineType2["LANGFLOW"] = "langflow";
+  GraphEngineType2["FLOWISE"] = "flowise";
+  return GraphEngineType2;
+})(GraphEngineType || {});
+var GraphEngineFactory = class {
+  constructor(langgraph) {
+    this.langgraph = langgraph;
+  }
+  /**
+   * Get engine for the specified type
+   */
+  getEngine(engineType) {
+    switch (engineType) {
+      case "langgraph" /* LANGGRAPH */:
+        return this.langgraph;
+      // Will add other types in the future
+      // case GraphEngineType.LANGFLOW:
+      //   return new LangFlowEngine();
+      // case GraphEngineType.FLOWISE:
+      //   return new FlowiseEngine();
+      default:
+        throw new Error(`Unsupported graph engine type: ${engineType}`);
+    }
+  }
+};
+GraphEngineFactory = __decorateClass([
+  Injectable()
+], GraphEngineFactory);
 var EventProcessor = class {
   logger = new Logger(EventProcessor.name);
   /**
@@ -4568,12 +4938,11 @@ var LangGraphEngine = class {
   }
 };
 LangGraphEngine = __decorateClass([
-  Injectable(),
-  __decorateParam(0, Inject())
+  Injectable()
 ], LangGraphEngine);
 
 // src/core/universal-graph.module.ts
-init_ui_dispatch_controller();
+init_builder_registry_service();
 init_agent_ui();
 function createMetaBuilder(config, versionedGraphService, moduleRef, callbackRegistry, endpointRegistry) {
   const className = `${config.baseGraphType.replace(/\./g, "")}VersionRouter`;
@@ -4642,6 +5011,8 @@ function createMetaBuilder(config, versionedGraphService, moduleRef, callbackReg
 var UniversalGraphModule = class {
   static forRoot(options) {
     const providers = [
+      // Discovery services from @nestjs/core
+      MetadataScanner,
       // Event processor for stream handling
       EventProcessor,
       // Graph engines
@@ -4671,14 +5042,8 @@ var UniversalGraphModule = class {
         provide: CallbackRegistry,
         useClass: CallbackRegistry
       },
-      {
-        provide: EndpointRegistry,
-        useClass: EndpointRegistry
-      },
-      {
-        provide: UIEndpointsDiscoveryService,
-        useClass: UIEndpointsDiscoveryService
-      },
+      EndpointRegistry,
+      UIEndpointsDiscoveryService,
       {
         provide: CallbackACL,
         useClass: CallbackACL
@@ -4755,14 +5120,6 @@ var UniversalGraphModule = class {
         },
         inject: [CallbackRegistry]
       },
-      {
-        provide: "UI_ENDPOINTS_DISCOVERY",
-        useFactory: async (discoveryService) => {
-          await discoveryService.discoverUIEndpoints();
-          return true;
-        },
-        inject: [UIEndpointsDiscoveryService]
-      },
       {
         provide: "GRAPH_ENGINE",
         useFactory: (factory) => {
@@ -5182,6 +5539,94 @@ var McpToolFilter = class _McpToolFilter {
     }
   }
 };
+var McpRuntimeHttpClient = class {
+  logger = new Logger(McpRuntimeHttpClient.name);
+  httpClient;
+  baseUrl;
+  constructor(mcpRuntimeUrl) {
+    this.baseUrl = mcpRuntimeUrl || process.env.MCP_RUNTIME_URL || "http://localhost:3004";
+    this.httpClient = axios2.create({
+      baseURL: this.baseUrl,
+      timeout: 3e4
+      // 30 seconds
+    });
+    this.logger.log(
+      `MCP Runtime HTTP Client initialized with URL: ${this.baseUrl}`
+    );
+  }
+  /**
+   * Get all available tools from MCP Runtime
+   */
+  async getTools() {
+    try {
+      this.logger.debug("Fetching available tools from MCP runtime");
+      const response = await this.httpClient.get("/tools/list");
+      const tools = Array.isArray(response.data) ? response.data : [];
+      this.logger.log(`Retrieved ${tools.length} tools from MCP runtime`);
+      return tools;
+    } catch (error) {
+      this.logger.error("Failed to fetch tools from MCP runtime:", error);
+      throw new Error(`Failed to fetch tools: ${error.message}`);
+    }
+  }
+  /**
+   * Execute a tool by name with given arguments
+   */
+  async executeTool(name, args, context) {
+    try {
+      this.logger.debug(`Executing tool: ${name} with args:`, args);
+      const payload = {
+        name,
+        arguments: args || {}
+      };
+      if (context) {
+        payload.context = context;
+      }
+      const response = await this.httpClient.post("/tools/execute", payload);
+      this.logger.log(`Tool ${name} executed successfully`);
+      return response.data;
+    } catch (error) {
+      this.logger.error(`Failed to execute tool ${name}:`, error);
+      if (error.response) {
+        return {
+          success: false,
+          error: error.response.data.message || error.response.data.error || "Tool execution failed"
+        };
+      }
+      return {
+        success: false,
+        error: error.message || "Unknown error occurred"
+      };
+    }
+  }
+  /**
+   * Get tool execution statistics from MCP Runtime
+   */
+  async getToolStats() {
+    try {
+      const response = await this.httpClient.get("/tools/stats");
+      return response.data;
+    } catch (error) {
+      this.logger.error("Failed to fetch tool stats:", error);
+      return null;
+    }
+  }
+  /**
+   * Health check for MCP Runtime service
+   */
+  async isHealthy() {
+    try {
+      const response = await this.httpClient.get("/", { timeout: 5e3 });
+      return response.status === 200;
+    } catch (error) {
+      this.logger.warn("MCP Runtime health check failed:", error.message);
+      return false;
+    }
+  }
+};
+McpRuntimeHttpClient = __decorateClass([
+  Injectable()
+], McpRuntimeHttpClient);
 
 // src/models/enums.ts
 var ModelProvider = /* @__PURE__ */ ((ModelProvider2) => {
@@ -6343,6 +6788,28 @@ RetrieverService = __decorateClass([
   Injectable()
 ], RetrieverService);
 
-export { AbstractGraphBuilder, AttachmentType, GraphController as BaseGraphServiceController, UniversalGraphModule as BaseGraphServiceModule, BuilderRegistryService, Callback, CallbackACL, CallbackAuditAction, CallbackAuditor, CallbackController, CallbackMetrics, CallbackPatchService, CallbackRateLimiter, CallbackRegistry, CallbackStore, CallbackTokenGuard, ChatFeature, DEFAULT_TRACER_OPTIONS, ENDPOINT_METADATA_KEY, Endpoint, EndpointRegistry, EventProcessor, GraphController, GraphEngineFactory, GraphEngineType, GraphManifestSchema, GraphManifestValidator, GraphServiceTokens, GraphTypeUtils, IdempotencyManager, IdempotencyStatus, LangGraphEngine, McpConverter, McpToolFilter, ModelInitializer, ModelProvider, ModelType, RetrieverSearchType, RetrieverService, SmartCallbackRouter, StreamChannel, TelegramPatchHandler, UIDispatchController, UIEndpoint, UIEndpointsDiscoveryService, UniversalCallbackService, UniversalGraphModule, UniversalGraphService, VersionedGraphService, VoyageAIRerank, WebPatchHandler, WithCallbacks, WithEndpoints, WithUIEndpoints, bootstrap, createEndpointDescriptors, findCallbackMethod, findEndpointMethod, getCallbackMetadata, getEndpointMetadata, getUIEndpointClassMetadata, getUIEndpointMethodsMetadata, hasCallbacks, hasUIEndpoints, prepareModelWithTools, registerFinanceExampleCallback, registerUIEndpointsFromClass, sanitizeTraceData, traceApiCall };
+// src/utils/index.ts
+init_graph_type_utils();
+
+// src/service-discovery/index.ts
+init_file_based_discovery();
+var StaticDiscovery = class {
+  constructor(services) {
+    this.services = services;
+  }
+  /**
+   * Get list of services by category
+   */
+  async getServices(category) {
+    return this.services.filter(
+      (service) => service.category === category || service.metadata?.category === category
+    );
+  }
+};
+StaticDiscovery = __decorateClass([
+  Injectable()
+], StaticDiscovery);
+
+export { AbstractGraphBuilder, AttachmentType, GraphController as BaseGraphServiceController, UniversalGraphModule as BaseGraphServiceModule, BuilderRegistryService, Callback, CallbackACL, CallbackAuditAction, CallbackAuditor, CallbackController, CallbackMetrics, CallbackPatchService, CallbackRateLimiter, CallbackRegistry, CallbackStore, CallbackTokenGuard, ChatFeature, DEFAULT_TRACER_OPTIONS, ENDPOINT_METADATA_KEY, Endpoint, EndpointRegistry, EventProcessor, FileBasedDiscovery, GraphController, GraphEngineFactory, GraphEngineType, GraphManifestSchema, GraphManifestValidator, GraphServiceTokens, GraphTypeUtils, IdempotencyManager, IdempotencyStatus, LangGraphEngine, McpConverter, McpRuntimeHttpClient, McpToolFilter, ModelInitializer, ModelProvider, ModelType, RetrieverSearchType, RetrieverService, SmartCallbackRouter, StaticDiscovery, StreamChannel, TelegramPatchHandler, UIDispatchController, UIEndpoint, UIEndpointsDiscoveryService, UniversalCallbackService, UniversalGraphModule, UniversalGraphService, VersionedGraphService, VoyageAIRerank, WebPatchHandler, WithCallbacks, WithEndpoints, WithUIEndpoints, bootstrap, createEndpointDescriptors, findCallbackMethod, findEndpointMethod, getCallbackMetadata, getEndpointMetadata, getUIEndpointClassMetadata, getUIEndpointMethodsMetadata, hasCallbacks, hasUIEndpoints, prepareModelWithTools, registerFinanceExampleCallback, registerUIEndpointsFromClass, sanitizeTraceData, traceApiCall };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map