@juspay/neurolink 9.32.0 → 9.33.0

package/dist/client/observability/exporterRegistry.js

@@ -1,413 +0,0 @@
-/**
- * Exporter Registry
- * Manages multiple observability exporters with circuit breaker protection
- */
-import { logger } from "../utils/logger.js";
-import { AlwaysSampler } from "./sampling/samplers.js";
-/** Default timeout for exporter API calls (30 seconds) */
-const DEFAULT_EXPORT_TIMEOUT_MS = 30_000;
-/**
- * Wrap a promise with a timeout. Rejects with a descriptive error if the
- * promise does not settle within `timeoutMs` milliseconds.
- */
-function withExportTimeout(promise, timeoutMs, label) {
-    return new Promise((resolve, reject) => {
-        const timer = setTimeout(() => {
-            reject(new Error(`Export to '${label}' timed out after ${timeoutMs}ms`));
-        }, timeoutMs);
-        promise.then((v) => {
-            clearTimeout(timer);
-            resolve(v);
-        }, (e) => {
-            clearTimeout(timer);
-            reject(e);
-        });
-    });
-}
-/**
- * Registry for managing multiple observability exporters
- * Includes circuit breaker protection to prevent cascading failures
- */
-export class ExporterRegistry {
-    exporters = new Map();
-    defaultExporter = null;
-    sampler = new AlwaysSampler();
-    circuitBreakers = new Map();
-    circuitBreakerConfig = {
-        failureThreshold: 5,
-        resetTimeout: 30000, // 30 seconds
-    };
-    /**
-     * Register an exporter
-     */
-    register(exporter) {
-        this.exporters.set(exporter.getName(), exporter);
-    }
-    /**
-     * Unregister an exporter
-     */
-    unregister(name) {
-        return this.exporters.delete(name);
-    }
-    /**
-     * Get an exporter by name
-     */
-    get(name) {
-        return this.exporters.get(name);
-    }
-    /**
-     * Get all registered exporter names
-     */
-    getNames() {
-        return Array.from(this.exporters.keys());
-    }
-    /**
-     * Get total exporter count
-     */
-    getCount() {
-        return this.exporters.size;
-    }
-    /**
-     * Set the default exporter
-     */
-    setDefault(name) {
-        if (!this.exporters.has(name)) {
-            throw new Error(`Exporter '${name}' not registered`);
-        }
-        this.defaultExporter = name;
-    }
-    /**
-     * Get the default exporter
-     */
-    getDefault() {
-        if (!this.defaultExporter) {
-            return undefined;
-        }
-        return this.exporters.get(this.defaultExporter);
-    }
-    /**
-     * Set the sampler for the registry
-     */
-    setSampler(sampler) {
-        this.sampler = sampler;
-    }
-    /**
-     * Get the current sampler
-     */
-    getSampler() {
-        return this.sampler;
-    }
-    /**
-     * Configure the circuit breaker settings
-     * @param config - Partial circuit breaker configuration
-     */
-    configureCircuitBreaker(config) {
-        Object.assign(this.circuitBreakerConfig, config);
-    }
-    /**
-     * Check if circuit is open for an exporter
-     * @param exporterName - Name of the exporter
-     * @returns true if circuit is open (exporter should be skipped)
-     */
-    isCircuitOpen(exporterName) {
-        const breaker = this.circuitBreakers.get(exporterName);
-        if (!breaker) {
-            return false;
-        }
-        if (breaker.state === "open") {
-            // Check if we should try half-open
-            if (Date.now() - breaker.lastFailure >
-                this.circuitBreakerConfig.resetTimeout) {
-                breaker.state = "half-open";
-                logger.info(`[ExporterRegistry] Circuit half-open for ${exporterName}, attempting recovery`);
-                return false;
-            }
-            return true;
-        }
-        return false;
-    }
-    /**
-     * Record a failure for an exporter's circuit breaker
-     * @param exporterName - Name of the exporter
-     */
-    recordFailure(exporterName) {
-        let breaker = this.circuitBreakers.get(exporterName);
-        if (!breaker) {
-            breaker = { failures: 0, lastFailure: 0, state: "closed" };
-            this.circuitBreakers.set(exporterName, breaker);
-        }
-        breaker.failures++;
-        breaker.lastFailure = Date.now();
-        if (breaker.failures >= this.circuitBreakerConfig.failureThreshold) {
-            breaker.state = "open";
-            logger.warn(`[ExporterRegistry] Circuit opened for ${exporterName} after ${breaker.failures} failures`);
-        }
-    }
-    /**
-     * Record a success for an exporter's circuit breaker
-     * Resets the circuit to closed state
-     * @param exporterName - Name of the exporter
-     */
-    recordSuccess(exporterName) {
-        const breaker = this.circuitBreakers.get(exporterName);
-        if (breaker) {
-            if (breaker.state === "half-open") {
-                logger.info(`[ExporterRegistry] Circuit closed for ${exporterName} after successful recovery`);
-            }
-            breaker.failures = 0;
-            breaker.state = "closed";
-        }
-    }
-    /**
-     * Get circuit breaker status for an exporter
-     * @param exporterName - Name of the exporter
-     * @returns Circuit breaker state or undefined if not tracked
-     */
-    getCircuitBreakerStatus(exporterName) {
-        return this.circuitBreakers.get(exporterName);
-    }
-    /**
-     * Reset circuit breaker for an exporter
-     * @param exporterName - Name of the exporter
-     */
-    resetCircuitBreaker(exporterName) {
-        this.circuitBreakers.delete(exporterName);
-    }
-    /**
-     * Export span to all registered exporters
-     * Applies sampling and circuit breaker protection before export
-     */
-    async exportToAll(span) {
-        const results = new Map();
-        // Apply sampling
-        if (!this.sampler.shouldSample(span)) {
-            // Return empty results if not sampled
-            return results;
-        }
-        const exportPromises = Array.from(this.exporters.entries()).map(async ([name, exporter]) => {
-            // Check circuit breaker before attempting export
-            if (this.isCircuitOpen(name)) {
-                results.set(name, {
-                    success: false,
-                    exportedCount: 0,
-                    failedCount: 1,
-                    errors: [
-                        {
-                            spanId: span.spanId,
-                            error: "Circuit breaker open - exporter temporarily disabled",
-                            retryable: true,
-                        },
-                    ],
-                    durationMs: 0,
-                });
-                return;
-            }
-            if (exporter.isInitialized()) {
-                try {
-                    const result = await withExportTimeout(exporter.exportSpan(span), DEFAULT_EXPORT_TIMEOUT_MS, name);
-                    results.set(name, result);
-                    // Record success or failure based on result
-                    if (result.success) {
-                        this.recordSuccess(name);
-                    }
-                    else {
-                        this.recordFailure(name);
-                    }
-                }
-                catch (error) {
-                    this.recordFailure(name);
-                    results.set(name, {
-                        success: false,
-                        exportedCount: 0,
-                        failedCount: 1,
-                        errors: [
-                            {
-                                spanId: span.spanId,
-                                error: error instanceof Error ? error.message : String(error),
-                                retryable: true,
-                            },
-                        ],
-                        durationMs: 0,
-                    });
-                }
-            }
-            else {
-                logger.debug(`[ExporterRegistry] Skipping uninitialized exporter '${name}' for span ${span.spanId}`);
-            }
-        });
-        await Promise.all(exportPromises);
-        return results;
-    }
-    /**
-     * Export span to a specific exporter
-     * Applies sampling and circuit breaker protection
-     */
-    async exportTo(name, span) {
-        const exporter = this.exporters.get(name);
-        if (!exporter) {
-            return null;
-        }
-        if (!exporter.isInitialized()) {
-            logger.debug(`[ExporterRegistry] Skipping uninitialized exporter '${name}' for span ${span.spanId}`);
-            return null;
-        }
-        // Check circuit breaker before attempting export
-        if (this.isCircuitOpen(name)) {
-            return {
-                success: false,
-                exportedCount: 0,
-                failedCount: 1,
-                errors: [
-                    {
-                        spanId: span.spanId,
-                        error: "Circuit breaker open - exporter temporarily disabled",
-                        retryable: true,
-                    },
-                ],
-                durationMs: 0,
-            };
-        }
-        // Apply sampling
-        if (!this.sampler.shouldSample(span)) {
-            return {
-                success: true,
-                exportedCount: 0,
-                failedCount: 0,
-                durationMs: 0,
-            };
-        }
-        try {
-            const result = await withExportTimeout(exporter.exportSpan(span), DEFAULT_EXPORT_TIMEOUT_MS, name);
-            if (result.success) {
-                this.recordSuccess(name);
-            }
-            else {
-                this.recordFailure(name);
-            }
-            return result;
-        }
-        catch (error) {
-            this.recordFailure(name);
-            return {
-                success: false,
-                exportedCount: 0,
-                failedCount: 1,
-                errors: [
-                    {
-                        spanId: span.spanId,
-                        error: error instanceof Error ? error.message : String(error),
-                        retryable: true,
-                    },
-                ],
-                durationMs: 0,
-            };
-        }
-    }
-    /**
-     * Initialize all exporters
-     */
-    async initializeAll() {
-        const results = await Promise.allSettled(Array.from(this.exporters.entries()).map(([name, e]) => e.initialize().catch((err) => {
-            logger.error(`[ExporterRegistry] Failed to initialize exporter '${name}':`, err);
-            throw err;
-        })));
-        for (const result of results) {
-            if (result.status === "rejected") {
-                logger.warn(`[ExporterRegistry] One or more exporters failed to initialize`);
-                break;
-            }
-        }
-    }
-    /**
-     * Shutdown all exporters
-     */
-    async shutdownAll() {
-        const results = await Promise.allSettled(Array.from(this.exporters.entries()).map(([name, e]) => e.shutdown().catch((err) => {
-            logger.error(`[ExporterRegistry] Failed to shutdown exporter '${name}':`, err);
-            throw err;
-        })));
-        for (const result of results) {
-            if (result.status === "rejected") {
-                logger.warn(`[ExporterRegistry] One or more exporters failed to shutdown`);
-                break;
-            }
-        }
-    }
-    /**
-     * Flush all exporters
-     */
-    async flushAll() {
-        const results = await Promise.allSettled(Array.from(this.exporters.entries()).map(([name, e]) => e.flush().catch((err) => {
-            logger.error(`[ExporterRegistry] Failed to flush exporter '${name}':`, err);
-            throw err;
-        })));
-        for (const result of results) {
-            if (result.status === "rejected") {
-                logger.warn(`[ExporterRegistry] One or more exporters failed to flush`);
-                break;
-            }
-        }
-    }
-    /**
-     * Get health status of all exporters
-     */
-    async healthCheckAll() {
-        const results = new Map();
-        const healthPromises = Array.from(this.exporters.entries()).map(async ([name, exporter]) => {
-            const status = await exporter.healthCheck();
-            results.set(name, status);
-        });
-        await Promise.all(healthPromises);
-        return results;
-    }
-    /**
-     * Check if all exporters are healthy
-     */
-    async isHealthy() {
-        const statuses = await this.healthCheckAll();
-        return Array.from(statuses.values()).every((s) => s.healthy);
-    }
-    /**
-     * Get total pending spans across all exporters
-     */
-    getTotalPendingSpans() {
-        let total = 0;
-        const exporterArray = Array.from(this.exporters.values());
-        for (const exporter of exporterArray) {
-            total += exporter.getPendingCount();
-        }
-        return total;
-    }
-    /**
-     * Clear all registered exporters and reset state
-     * (For testing and cleanup)
-     */
-    clear() {
-        this.exporters.clear();
-        this.defaultExporter = null;
-        this.circuitBreakers.clear();
-        this.sampler = new AlwaysSampler();
-    }
-}
-/**
- * Global exporter registry instance (singleton pattern from main)
- */
-let globalRegistry = null;
-/**
- * Get the global exporter registry instance
- */
-export function getExporterRegistry() {
-    if (!globalRegistry) {
-        globalRegistry = new ExporterRegistry();
-    }
-    return globalRegistry;
-}
-/**
- * Reset the global exporter registry (for testing)
- */
-export function resetExporterRegistry() {
-    if (globalRegistry) {
-        globalRegistry.clear();
-    }
-    globalRegistry = null;
-}

package/dist/client/observability/exporters/arizeExporter.js

@@ -1,138 +0,0 @@
-/**
- * Arize Exporter
- * Exports spans to Arize ML monitoring platform
- */
-import { BaseExporter } from "./baseExporter.js";
-/**
- * Arize exporter for ML monitoring and prediction logs
- * Supports feature tracking and model performance monitoring
- */
-export class ArizeExporter extends BaseExporter {
-    spaceKey;
-    apiKey;
-    modelId;
-    modelVersion;
-    endpoint = "https://api.arize.com/v1";
-    constructor(config) {
-        super("arize", config);
-        this.spaceKey = config.spaceKey;
-        this.apiKey = config.apiKey;
-        this.modelId = config.modelId ?? "neurolink-ai";
-        this.modelVersion = config.modelVersion ?? "1.0.0";
-    }
-    async initialize() {
-        if (this.initialized) {
-            return;
-        }
-        this.initialized = true;
-        this.startFlushInterval(this.config.flushIntervalMs ?? 10000);
-    }
-    async exportSpan(span) {
-        const startTime = Date.now();
-        try {
-            const prediction = this.convertToArizePrediction(span);
-            const response = await fetch(`${this.endpoint}/log`, {
-                method: "POST",
-                headers: {
-                    "Content-Type": "application/json",
-                    Authorization: `Bearer ${this.apiKey}`,
-                    "space-key": this.spaceKey,
-                },
-                body: JSON.stringify(prediction),
-            });
-            if (!response.ok) {
-                throw new Error(`Export failed: ${response.statusText}`);
-            }
-            return this.createSuccessResult(1, Date.now() - startTime);
-        }
-        catch (error) {
-            return this.createFailureResult([span.spanId], error instanceof Error ? error.message : String(error), Date.now() - startTime);
-        }
-    }
-    async exportBatch(spans) {
-        // Arize's /v1/log endpoint does not support batch payloads, so we send
-        // individual requests in parallel. This is intentional — not a missed
-        // optimization.
-        const results = await Promise.all(spans.map((s) => this.exportSpan(s)));
-        const successful = results.filter((r) => r.success).length;
-        const failed = spans.length - successful;
-        return {
-            success: failed === 0,
-            exportedCount: successful,
-            failedCount: failed,
-            errors: results.flatMap((r) => r.errors ?? []),
-            durationMs: results.reduce((sum, r) => sum + r.durationMs, 0),
-        };
-    }
-    async flush() {
-        if (!this.initialized || this.buffer.length === 0) {
-            return;
-        }
-        const spans = [...this.buffer];
-        this.buffer = [];
-        await this.exportBatch(spans);
-    }
-    async shutdown() {
-        await this.flush();
-        this.stopFlushInterval();
-        this.initialized = false;
-    }
-    async healthCheck() {
-        try {
-            await this.withRetry(() => this.ping(), "health check");
-            return this.createHealthStatus(true);
-        }
-        catch {
-            return this.createHealthStatus(false, ["Health check failed"]);
-        }
-    }
-    /**
-     * Verify connectivity to Arize API
-     */
-    async ping() {
-        const response = await fetch(`${this.endpoint}/health`, {
-            method: "GET",
-            headers: {
-                Authorization: `Bearer ${this.apiKey}`,
-                "space-key": this.spaceKey,
-            },
-        });
-        if (!response.ok && response.status !== 404) {
-            // 404 is acceptable as health endpoint may not exist
-            throw new Error(`Arize API unreachable: ${response.status}`);
-        }
-    }
-    /**
-     * Convert span to Arize prediction log format
-     */
-    convertToArizePrediction(span) {
-        return {
-            space_key: this.spaceKey,
-            model_id: span.attributes["ai.model"] ?? this.modelId,
-            model_version: this.modelVersion,
-            prediction_id: span.spanId,
-            prediction_timestamp: new Date(span.startTime).getTime(),
-            features: {
-                provider: span.attributes["ai.provider"],
-                temperature: span.attributes["ai.temperature"],
-                max_tokens: span.attributes["ai.max_tokens"],
-                user_id: span.attributes["user.id"],
-                session_id: span.attributes["session.id"],
-            },
-            prediction: {
-                input: span.attributes["input"],
-                output: span.attributes["output"],
-            },
-            tags: {
-                span_type: span.type,
-                environment: this.config.environment,
-            },
-            latency_ms: span.durationMs,
-            token_count: {
-                prompt: span.attributes["ai.tokens.input"],
-                completion: span.attributes["ai.tokens.output"],
-                total: span.attributes["ai.tokens.total"],
-            },
-        };
-    }
-}