@theaiinc/yggdrasil 0.1.0 → 0.2.1

package/dist/src/services/agent-manager.js

@@ -1,227 +0,0 @@
-import axios from 'axios';
-import { getLogger } from './logger';
-import { nanoid } from 'nanoid';
-/**
- * Manages agent registration, health checks, and lifecycle
- * Uses stable, well-documented libraries for reliability
- */
-export class AgentManager {
-    agents = new Map();
-    logger = getLogger();
-    healthCheckInterval = null;
-    healthCheckPath;
-    constructor(healthCheckPath = '/health') {
-        this.healthCheckPath = healthCheckPath;
-    }
-    /**
-     * Register a new agent
-     */
-    registerAgent(url, sessionId) {
-        const agentId = nanoid();
-        const agentInfo = {
-            id: agentId,
-            url,
-            health: {
-                status: 'unknown',
-                lastCheck: new Date(),
-                responseTime: 0,
-                errorCount: 0,
-                consecutiveFailures: 0,
-            },
-            metrics: {
-                cpuUsage: 0,
-                memoryUsage: 0,
-                requestCount: 0,
-                errorRate: 0,
-                averageResponseTime: 0,
-                activeConnections: 0,
-            },
-            lastSeen: new Date(),
-            ...(sessionId && { sessionId }),
-        };
-        this.agents.set(agentId, agentInfo);
-        this.logger.info('Agent registered', { agentId, url, sessionId });
-        return agentId;
-    }
-    /**
-     * Unregister an agent
-     */
-    unregisterAgent(agentId) {
-        const removed = this.agents.delete(agentId);
-        if (removed) {
-            this.logger.info('Agent unregistered', { agentId });
-        }
-        return removed;
-    }
-    /**
-     * Get all healthy agents
-     */
-    getHealthyAgents() {
-        return Array.from(this.agents.values()).filter(agent => agent.health.status === 'healthy');
-    }
-    /**
-     * Get agent by ID
-     */
-    getAgent(agentId) {
-        return this.agents.get(agentId);
-    }
-    /**
-     * Get agent by session ID
-     */
-    getAgentBySession(sessionId) {
-        return Array.from(this.agents.values()).find(agent => agent.sessionId === sessionId);
-    }
-    /**
-     * Start health check monitoring
-     */
-    startHealthChecks(intervalMs = 30000) {
-        if (this.healthCheckInterval) {
-            clearInterval(this.healthCheckInterval);
-        }
-        this.healthCheckInterval = setInterval(() => {
-            this.performHealthChecks();
-        }, intervalMs);
-        this.logger.info('Health checks started', { intervalMs });
-    }
-    /**
-     * Stop health check monitoring
-     */
-    stopHealthChecks() {
-        if (this.healthCheckInterval) {
-            clearInterval(this.healthCheckInterval);
-            this.healthCheckInterval = null;
-            this.logger.info('Health checks stopped');
-        }
-    }
-    /**
-     * Perform health check on a specific agent
-     */
-    async checkAgentHealth(agentId) {
-        const agent = this.agents.get(agentId);
-        if (!agent) {
-            return {
-                agentId,
-                healthy: false,
-                responseTime: 0,
-                error: 'Agent not found',
-                timestamp: new Date(),
-            };
-        }
-        const startTime = Date.now();
-        try {
-            const response = await axios.get(`${agent.url}${this.healthCheckPath}`, { timeout: 5000 });
-            const responseTime = Date.now() - startTime;
-            const healthy = response.status === 200;
-            // Update agent health
-            agent.health = {
-                status: healthy ? 'healthy' : 'unhealthy',
-                lastCheck: new Date(),
-                responseTime,
-                errorCount: healthy
-                    ? agent.health.errorCount
-                    : agent.health.errorCount + 1,
-                consecutiveFailures: healthy ? 0 : agent.health.consecutiveFailures + 1,
-            };
-            agent.lastSeen = new Date();
-            this.logger.info('Agent health check completed', {
-                agentId,
-                healthy,
-                responseTime,
-                status: agent.health.status,
-            });
-            return {
-                agentId,
-                healthy,
-                responseTime,
-                timestamp: new Date(),
-            };
-        }
-        catch (error) {
-            const responseTime = Date.now() - startTime;
-            const errorMessage = error instanceof Error ? error.message : 'Unknown error';
-            // Update agent health
-            agent.health = {
-                status: 'unhealthy',
-                lastCheck: new Date(),
-                responseTime,
-                errorCount: agent.health.errorCount + 1,
-                consecutiveFailures: agent.health.consecutiveFailures + 1,
-            };
-            this.logger.warn('Agent health check failed', {
-                agentId,
-                error: errorMessage,
-                responseTime,
-            });
-            return {
-                agentId,
-                healthy: false,
-                responseTime,
-                error: errorMessage,
-                timestamp: new Date(),
-            };
-        }
-    }
-    /**
-     * Perform health checks on all agents
-     */
-    async performHealthChecks() {
-        const promises = Array.from(this.agents.keys()).map(agentId => this.checkAgentHealth(agentId));
-        try {
-            await Promise.allSettled(promises);
-        }
-        catch (error) {
-            this.logger.error('Health check batch failed', { error });
-        }
-    }
-    /**
-     * Update agent metrics
-     */
-    updateAgentMetrics(agentId, metrics) {
-        const agent = this.agents.get(agentId);
-        if (agent) {
-            agent.metrics = { ...agent.metrics, ...metrics };
-            agent.lastSeen = new Date();
-        }
-    }
-    /**
-     * Get all agents
-     */
-    getAllAgents() {
-        return Array.from(this.agents.values());
-    }
-    /**
-     * Get agent count by health status
-     */
-    getAgentCounts() {
-        const agents = this.getAllAgents();
-        return {
-            total: agents.length,
-            healthy: agents.filter(a => a.health.status === 'healthy').length,
-            unhealthy: agents.filter(a => a.health.status === 'unhealthy').length,
-        };
-    }
-    /**
-     * Clean up stale agents (not seen for more than threshold)
-     */
-    cleanupStaleAgents(thresholdMs = 300000) {
-        const now = new Date();
-        const staleAgents = [];
-        for (const [agentId, agent] of this.agents.entries()) {
-            const timeSinceLastSeen = now.getTime() - agent.lastSeen.getTime();
-            if (timeSinceLastSeen > thresholdMs) {
-                staleAgents.push(agentId);
-            }
-        }
-        staleAgents.forEach(agentId => {
-            this.unregisterAgent(agentId);
-        });
-        if (staleAgents.length > 0) {
-            this.logger.info('Cleaned up stale agents', {
-                count: staleAgents.length,
-                agentIds: staleAgents,
-            });
-        }
-        return staleAgents.length;
-    }
-}
-//# sourceMappingURL=agent-manager.js.map

package/dist/src/services/agent-manager.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"agent-manager.js","sourceRoot":"","sources":["../../../src/services/agent-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAwB,MAAM,OAAO,CAAC;AAE7C,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC;;;GAGG;AACH,MAAM,OAAO,YAAY;IACf,MAAM,GAA2B,IAAI,GAAG,EAAE,CAAC;IAC3C,MAAM,GAAG,SAAS,EAAE,CAAC;IACrB,mBAAmB,GAA0C,IAAI,CAAC;IAClE,eAAe,CAAS;IAEhC,YAAY,kBAA0B,SAAS;QAC7C,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IACzC,CAAC;IAED;;OAEG;IACI,aAAa,CAAC,GAAW,EAAE,SAAkB;QAClD,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC;QACzB,MAAM,SAAS,GAAc;YAC3B,EAAE,EAAE,OAAO;YACX,GAAG;YACH,MAAM,EAAE;gBACN,MAAM,EAAE,SAAS;gBACjB,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,YAAY,EAAE,CAAC;gBACf,UAAU,EAAE,CAAC;gBACb,mBAAmB,EAAE,CAAC;aACvB;YACD,OAAO,EAAE;gBACP,QAAQ,EAAE,CAAC;gBACX,WAAW,EAAE,CAAC;gBACd,YAAY,EAAE,CAAC;gBACf,SAAS,EAAE,CAAC;gBACZ,mBAAmB,EAAE,CAAC;gBACtB,iBAAiB,EAAE,CAAC;aACrB;YACD,QAAQ,EAAE,IAAI,IAAI,EAAE;YACpB,GAAG,CAAC,SAAS,IAAI,EAAE,SAAS,EAAE,CAAC;SAChC,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QACpC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,CAAC;QAClE,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACI,eAAe,CAAC,OAAe;QACpC,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACI,gBAAgB;QACrB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAC5C,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS,CAC3C,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,QAAQ,CAAC,OAAe;QAC7B,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACI,iBAAiB,CAAC,SAAiB;QACxC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAC1C,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,CACvC,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,iBAAiB,CAAC,aAAqB,KAAK;QACjD,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,aAAa,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC1C,CAAC;QAED,IAAI,CAAC,mBAAmB,GAAG,WAAW,CAAC,GAAG,EAAE;YAC1C,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC7B,CAAC,EAAE,UAAU,CAAC,CAAC;QAEf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED;;OAEG;IACI,gBAAgB;QACrB,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,aAAa,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YACxC,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC;YAChC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,gBAAgB,CAAC,OAAe;QAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO;gBACL,OAAO;gBACP,OAAO,EAAE,KAAK;gBACd,YAAY,EAAE,CAAC;gBACf,KAAK,EAAE,iBAAiB;gBACxB,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAkB,MAAM,KAAK,CAAC,GAAG,CAC7C,GAAG,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC,eAAe,EAAE,EACrC,EAAE,OAAO,EAAE,IAAI,EAAE,CAClB,CAAC;YAEF,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC5C,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,KAAK,GAAG,CAAC;YAExC,sBAAsB;YACtB,KAAK,CAAC,MAAM,GAAG;gBACb,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW;gBACzC,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,YAAY;gBACZ,UAAU,EAAE,OAAO;oBACjB,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU;oBACzB,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC;gBAC/B,mBAAmB,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,mBAAmB,GAAG,CAAC;aACxE,CAAC;YAEF,KAAK,CAAC,QAAQ,GAAG,IAAI,IAAI,EAAE,CAAC;YAE5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,8BAA8B,EAAE;gBAC/C,OAAO;gBACP,OAAO;gBACP,YAAY;gBACZ,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM;aAC5B,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO;gBACP,OAAO;gBACP,YAAY;gBACZ,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC5C,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YAE3D,sBAAsB;YACtB,KAAK,CAAC,MAAM,GAAG;gBACb,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,YAAY;gBACZ,UAAU,EAAE,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC;gBACvC,mBAAmB,EAAE,KAAK,CAAC,MAAM,CAAC,mBAAmB,GAAG,CAAC;aAC1D,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE;gBAC5C,OAAO;gBACP,KAAK,EAAE,YAAY;gBACnB,YAAY;aACb,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO;gBACP,OAAO,EAAE,KAAK;gBACd,YAAY;gBACZ,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,mBAAmB;QAC/B,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAC5D,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAC/B,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAED;;OAEG;IACI,kBAAkB,CACvB,OAAe,EACf,OAA8B;QAE9B,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvC,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,CAAC,OAAO,GAAG,EAAE,GAAG,KAAK,CAAC,OAAO,EAAE,GAAG,OAAO,EAAE,CAAC;YACjD,KAAK,CAAC,QAAQ,GAAG,IAAI,IAAI,EAAE,CAAC;QAC9B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,YAAY;QACjB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACI,cAAc;QAKnB,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACnC,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,MAAM;YACpB,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,MAAM;YACjE,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,MAAM;SACtE,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,kBAAkB,CAAC,cAAsB,MAAM;QACpD,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,WAAW,GAAa,EAAE,CAAC;QAEjC,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;YACrD,MAAM,iBAAiB,GAAG,GAAG,CAAC,OAAO,EAAE,GAAG,KAAK,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACnE,IAAI,iBAAiB,GAAG,WAAW,EAAE,CAAC;gBACpC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAC5B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE;gBAC1C,KAAK,EAAE,WAAW,CAAC,MAAM;gBACzB,QAAQ,EAAE,WAAW;aACtB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,WAAW,CAAC,MAAM,CAAC;IAC5B,CAAC;CACF"}

package/dist/src/services/load-balancer.d.ts

@@ -1,62 +0,0 @@
-import { AgentInfo, RequestContext } from '@/types';
-/**
- * Load balancer for distributing requests across healthy agents
- * Implements multiple algorithms and session affinity
- */
-export declare class LoadBalancer {
-    private logger;
-    private currentIndex;
-    private sessionMap;
-    private algorithm;
-    private sessionAffinity;
-    private stickySessionTimeout;
-    constructor(algorithm?: 'round-robin' | 'least-connections' | 'ip-hash' | 'weighted', sessionAffinity?: boolean, stickySessionTimeout?: number);
-    /**
-     * Select an agent for the given request
-     */
-    selectAgent(agents: AgentInfo[], context: RequestContext): AgentInfo | null;
-    /**
-     * Round-robin algorithm
-     */
-    private roundRobin;
-    /**
-     * Least connections algorithm
-     */
-    private leastConnections;
-    /**
-     * IP hash algorithm (uses request ID as hash input)
-     */
-    private ipHash;
-    /**
-     * Weighted algorithm (based on agent health and performance)
-     */
-    private weighted;
-    /**
-     * Simple hash function for IP hash algorithm
-     */
-    private hashCode;
-    /**
-     * Clean up stale session mappings
-     */
-    private cleanupStaleSessions;
-    /**
-     * Remove session mapping
-     */
-    removeSession(sessionId: string): void;
-    /**
-     * Get session statistics
-     */
-    getSessionStats(): {
-        totalSessions: number;
-        sessionMapSize: number;
-    };
-    /**
-     * Update algorithm
-     */
-    setAlgorithm(algorithm: 'round-robin' | 'least-connections' | 'ip-hash' | 'weighted'): void;
-    /**
-     * Update session affinity settings
-     */
-    setSessionAffinity(enabled: boolean, timeout?: number): void;
-}
-//# sourceMappingURL=load-balancer.d.ts.map

package/dist/src/services/load-balancer.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"load-balancer.d.ts","sourceRoot":"","sources":["../../../src/services/load-balancer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAGpD;;;GAGG;AACH,qBAAa,YAAY;IACvB,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,YAAY,CAAK;IACzB,OAAO,CAAC,UAAU,CAA6B;IAC/C,OAAO,CAAC,SAAS,CAIF;IACf,OAAO,CAAC,eAAe,CAAU;IACjC,OAAO,CAAC,oBAAoB,CAAS;gBAGnC,SAAS,GACL,aAAa,GACb,mBAAmB,GACnB,SAAS,GACT,UAA0B,EAC9B,eAAe,GAAE,OAAc,EAC/B,oBAAoB,GAAE,MAAgB;IAOxC;;OAEG;IACI,WAAW,CAChB,MAAM,EAAE,SAAS,EAAE,EACnB,OAAO,EAAE,cAAc,GACtB,SAAS,GAAG,IAAI;IA2EnB;;OAEG;IACH,OAAO,CAAC,UAAU;IASlB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAQxB;;OAEG;IACH,OAAO,CAAC,MAAM;IASd;;OAEG;IACH,OAAO,CAAC,QAAQ;IA+BhB;;OAEG;IACH,OAAO,CAAC,QAAQ;IAUhB;;OAEG;IACH,OAAO,CAAC,oBAAoB;IA+B5B;;OAEG;IACI,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAI7C;;OAEG;IACI,eAAe,IAAI;QAAE,aAAa,EAAE,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,CAAA;KAAE;IAO3E;;OAEG;IACI,YAAY,CACjB,SAAS,EAAE,aAAa,GAAG,mBAAmB,GAAG,SAAS,GAAG,UAAU,GACtE,IAAI;IAKP;;OAEG;IACI,kBAAkB,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI;CAUpE"}

package/dist/src/services/load-balancer.js

@@ -1,221 +0,0 @@
-import { getLogger } from './logger';
-/**
- * Load balancer for distributing requests across healthy agents
- * Implements multiple algorithms and session affinity
- */
-export class LoadBalancer {
-    logger = getLogger();
-    currentIndex = 0;
-    sessionMap = new Map(); // sessionId -> agentId
-    algorithm;
-    sessionAffinity;
-    stickySessionTimeout;
-    constructor(algorithm = 'round-robin', sessionAffinity = true, stickySessionTimeout = 3600000 // 1 hour
-    ) {
-        this.algorithm = algorithm;
-        this.sessionAffinity = sessionAffinity;
-        this.stickySessionTimeout = stickySessionTimeout;
-    }
-    /**
-     * Select an agent for the given request
-     */
-    selectAgent(agents, context) {
-        if (agents.length === 0) {
-            this.logger.warn('No healthy agents available');
-            return null;
-        }
-        // Filter to only healthy agents
-        const healthyAgents = agents.filter(agent => agent.health.status === 'healthy');
-        if (healthyAgents.length === 0) {
-            this.logger.warn('No healthy agents available');
-            return null;
-        }
-        // Check session affinity first
-        if (this.sessionAffinity && context.sessionId) {
-            const sessionAgentId = this.sessionMap.get(context.sessionId);
-            if (sessionAgentId) {
-                const sessionAgent = healthyAgents.find(agent => agent.id === sessionAgentId);
-                if (sessionAgent) {
-                    this.logger.debug('Using session affinity', {
-                        sessionId: context.sessionId,
-                        agentId: sessionAgent.id,
-                    });
-                    return sessionAgent;
-                }
-                else {
-                    // Remove stale session mapping
-                    this.sessionMap.delete(context.sessionId);
-                }
-            }
-        }
-        // Select agent based on algorithm
-        let selectedAgent = null;
-        try {
-            switch (this.algorithm) {
-                case 'round-robin':
-                    selectedAgent = this.roundRobin(healthyAgents);
-                    break;
-                case 'least-connections':
-                    selectedAgent = this.leastConnections(healthyAgents);
-                    break;
-                case 'ip-hash':
-                    selectedAgent = this.ipHash(healthyAgents, context);
-                    break;
-                case 'weighted':
-                    selectedAgent = this.weighted(healthyAgents);
-                    break;
-                default:
-                    selectedAgent = this.roundRobin(healthyAgents);
-            }
-        }
-        catch (error) {
-            this.logger.error('Agent selection failed', {
-                error,
-                algorithm: this.algorithm,
-            });
-            return null;
-        }
-        // Update session mapping if session affinity is enabled
-        if (this.sessionAffinity && context.sessionId && selectedAgent) {
-            this.sessionMap.set(context.sessionId, selectedAgent.id);
-            // Clean up old session mappings periodically
-            this.cleanupStaleSessions();
-        }
-        return selectedAgent;
-    }
-    /**
-     * Round-robin algorithm
-     */
-    roundRobin(agents) {
-        const agent = agents[this.currentIndex % agents.length];
-        if (!agent) {
-            throw new Error('No agents available for round-robin selection');
-        }
-        this.currentIndex = (this.currentIndex + 1) % agents.length;
-        return agent;
-    }
-    /**
-     * Least connections algorithm
-     */
-    leastConnections(agents) {
-        return agents.reduce((min, current) => current.metrics.activeConnections < min.metrics.activeConnections
-            ? current
-            : min);
-    }
-    /**
-     * IP hash algorithm (uses request ID as hash input)
-     */
-    ipHash(agents, context) {
-        const hash = this.hashCode(context.id);
-        const agent = agents[Math.abs(hash) % agents.length];
-        if (!agent) {
-            throw new Error('No agents available for IP hash selection');
-        }
-        return agent;
-    }
-    /**
-     * Weighted algorithm (based on agent health and performance)
-     */
-    weighted(agents) {
-        const weights = agents.map(agent => {
-            let weight = 1;
-            // Reduce weight for agents with high error rates
-            if (agent.metrics.errorRate > 0.1) {
-                weight *= 0.5;
-            }
-            // Reduce weight for agents with high response times
-            if (agent.metrics.averageResponseTime > 1000) {
-                weight *= 0.7;
-            }
-            // Increase weight for agents with low CPU usage
-            if (agent.metrics.cpuUsage < 0.5) {
-                weight *= 1.2;
-            }
-            return { agent, weight };
-        });
-        // Sort by weight and return the highest weighted agent
-        weights.sort((a, b) => b.weight - a.weight);
-        const selectedAgent = weights[0]?.agent;
-        if (!selectedAgent) {
-            throw new Error('No agents available for weighted selection');
-        }
-        return selectedAgent;
-    }
-    /**
-     * Simple hash function for IP hash algorithm
-     */
-    hashCode(str) {
-        let hash = 0;
-        for (let i = 0; i < str.length; i++) {
-            const char = str.charCodeAt(i);
-            hash = (hash << 5) - hash + char;
-            hash = hash & hash; // Convert to 32-bit integer
-        }
-        return hash;
-    }
-    /**
-     * Clean up stale session mappings
-     */
-    cleanupStaleSessions() {
-        const staleSessions = [];
-        // This is a simplified cleanup - in a real implementation,
-        // you'd want to track session creation times
-        if (this.sessionMap.size > 1000) {
-            // If we have too many sessions, clean up some randomly
-            const sessions = Array.from(this.sessionMap.keys());
-            const toRemove = Math.floor(sessions.length * 0.1); // Remove 10%
-            for (let i = 0; i < toRemove; i++) {
-                const randomIndex = Math.floor(Math.random() * sessions.length);
-                const sessionId = sessions[randomIndex];
-                if (sessionId) {
-                    staleSessions.push(sessionId);
-                    sessions.splice(randomIndex, 1);
-                }
-            }
-        }
-        staleSessions.forEach(sessionId => {
-            this.sessionMap.delete(sessionId);
-        });
-        if (staleSessions.length > 0) {
-            this.logger.debug('Cleaned up stale sessions', {
-                count: staleSessions.length,
-            });
-        }
-    }
-    /**
-     * Remove session mapping
-     */
-    removeSession(sessionId) {
-        this.sessionMap.delete(sessionId);
-    }
-    /**
-     * Get session statistics
-     */
-    getSessionStats() {
-        return {
-            totalSessions: this.sessionMap.size,
-            sessionMapSize: this.sessionMap.size,
-        };
-    }
-    /**
-     * Update algorithm
-     */
-    setAlgorithm(algorithm) {
-        this.algorithm = algorithm;
-        this.logger.info('Load balancer algorithm updated', { algorithm });
-    }
-    /**
-     * Update session affinity settings
-     */
-    setSessionAffinity(enabled, timeout) {
-        this.sessionAffinity = enabled;
-        if (timeout) {
-            this.stickySessionTimeout = timeout;
-        }
-        this.logger.info('Session affinity settings updated', {
-            enabled: this.sessionAffinity,
-            timeout: this.stickySessionTimeout,
-        });
-    }
-}
-//# sourceMappingURL=load-balancer.js.map

package/dist/src/services/load-balancer.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"load-balancer.js","sourceRoot":"","sources":["../../../src/services/load-balancer.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAErC;;;GAGG;AACH,MAAM,OAAO,YAAY;IACf,MAAM,GAAG,SAAS,EAAE,CAAC;IACrB,YAAY,GAAG,CAAC,CAAC;IACjB,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC,CAAC,uBAAuB;IAC/D,SAAS,CAIF;IACP,eAAe,CAAU;IACzB,oBAAoB,CAAS;IAErC,YACE,YAIiB,aAAa,EAC9B,kBAA2B,IAAI,EAC/B,uBAA+B,OAAO,CAAC,SAAS;;QAEhD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;IACnD,CAAC;IAED;;OAEG;IACI,WAAW,CAChB,MAAmB,EACnB,OAAuB;QAEvB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAChD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,gCAAgC;QAChC,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CACjC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS,CAC3C,CAAC;QAEF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAChD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,+BAA+B;QAC/B,IAAI,IAAI,CAAC,eAAe,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YAC9C,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC9D,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,YAAY,GAAG,aAAa,CAAC,IAAI,CACrC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,cAAc,CACrC,CAAC;gBACF,IAAI,YAAY,EAAE,CAAC;oBACjB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE;wBAC1C,SAAS,EAAE,OAAO,CAAC,SAAS;wBAC5B,OAAO,EAAE,YAAY,CAAC,EAAE;qBACzB,CAAC,CAAC;oBACH,OAAO,YAAY,CAAC;gBACtB,CAAC;qBAAM,CAAC;oBACN,+BAA+B;oBAC/B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;QACH,CAAC;QAED,kCAAkC;QAClC,IAAI,aAAa,GAAqB,IAAI,CAAC;QAE3C,IAAI,CAAC;YACH,QAAQ,IAAI,CAAC,SAAS,EAAE,CAAC;gBACvB,KAAK,aAAa;oBAChB,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;oBAC/C,MAAM;gBACR,KAAK,mBAAmB;oBACtB,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;oBACrD,MAAM;gBACR,KAAK,SAAS;oBACZ,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;oBACpD,MAAM;gBACR,KAAK,UAAU;oBACb,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;oBAC7C,MAAM;gBACR;oBACE,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;YACnD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE;gBAC1C,KAAK;gBACL,SAAS,EAAE,IAAI,CAAC,SAAS;aAC1B,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QACd,CAAC;QAED,wDAAwD;QACxD,IAAI,IAAI,CAAC,eAAe,IAAI,OAAO,CAAC,SAAS,IAAI,aAAa,EAAE,CAAC;YAC/D,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC;YAEzD,6CAA6C;YAC7C,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC9B,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,MAAmB;QACpC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QACxD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QACD,IAAI,CAAC,YAAY,GAAG,CAAC,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5D,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,MAAmB;QAC1C,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CACpC,OAAO,CAAC,OAAO,CAAC,iBAAiB,GAAG,GAAG,CAAC,OAAO,CAAC,iBAAiB;YAC/D,CAAC,CAAC,OAAO;YACT,CAAC,CAAC,GAAG,CACR,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,MAAmB,EAAE,OAAuB;QACzD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACvC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QACrD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC/D,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,QAAQ,CAAC,MAAmB;QAClC,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;YACjC,IAAI,MAAM,GAAG,CAAC,CAAC;YAEf,iDAAiD;YACjD,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;gBAClC,MAAM,IAAI,GAAG,CAAC;YAChB,CAAC;YAED,oDAAoD;YACpD,IAAI,KAAK,CAAC,OAAO,CAAC,mBAAmB,GAAG,IAAI,EAAE,CAAC;gBAC7C,MAAM,IAAI,GAAG,CAAC;YAChB,CAAC;YAED,gDAAgD;YAChD,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,GAAG,GAAG,EAAE,CAAC;gBACjC,MAAM,IAAI,GAAG,CAAC;YAChB,CAAC;YAED,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,uDAAuD;QACvD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,aAAa,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;QACxC,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QACD,OAAO,aAAa,CAAC;IACvB,CAAC;IAED;;OAEG;IACK,QAAQ,CAAC,GAAW;QAC1B,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC/B,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;YACjC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,4BAA4B;QAClD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,oBAAoB;QAC1B,MAAM,aAAa,GAAa,EAAE,CAAC;QAEnC,2DAA2D;QAC3D,6CAA6C;QAC7C,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,GAAG,IAAI,EAAE,CAAC;YAChC,uDAAuD;YACvD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;YACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,aAAa;YAEjE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;gBAClC,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;gBAChE,MAAM,SAAS,GAAG,QAAQ,CAAC,WAAW,CAAC,CAAC;gBACxC,IAAI,SAAS,EAAE,CAAC;oBACd,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBAC9B,QAAQ,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC;QACH,CAAC;QAED,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;YAChC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE;gBAC7C,KAAK,EAAE,aAAa,CAAC,MAAM;aAC5B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACI,aAAa,CAAC,SAAiB;QACpC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACI,eAAe;QACpB,OAAO;YACL,aAAa,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;YACnC,cAAc,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;SACrC,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,YAAY,CACjB,SAAuE;QAEvE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACI,kBAAkB,CAAC,OAAgB,EAAE,OAAgB;QAC1D,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC;QAC/B,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,oBAAoB,GAAG,OAAO,CAAC;QACtC,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mCAAmC,EAAE;YACpD,OAAO,EAAE,IAAI,CAAC,eAAe;YAC7B,OAAO,EAAE,IAAI,CAAC,oBAAoB;SACnC,CAAC,CAAC;IACL,CAAC;CACF"}