fraim-framework 2.0.124 → 2.0.127

package/dist/src/local-mcp-server/prometheus-scraper.js

@@ -1,152 +0,0 @@
-"use strict";
-/**
- * Prometheus Scraper for Claude Code OTel Token Usage Metrics
- *
- * Scrapes Claude Code's local Prometheus endpoint to capture cumulative
- * token usage counters. Returns raw snapshot values — delta computation
- * happens server-side per architecture constraint 15.2.
- */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.parsePrometheusText = parsePrometheusText;
-exports.extractTokenSnapshot = extractTokenSnapshot;
-exports.scrapeTokenSnapshot = scrapeTokenSnapshot;
-/** Candidate Prometheus endpoint ports (OTel SDK defaults) */
-const CANDIDATE_PORTS = [9464, 8888];
-/** Candidate metric names for token usage (OTel dots→underscores, optional suffixes) */
-const TOKEN_METRIC_CANDIDATES = [
-    'claude_code_token_usage_total',
-    'claude_code_token_usage',
-    'claude_code_token_usage_tokens_total',
-    'claude_code_token_usage_tokens',
-];
-/** Candidate metric names for cost usage */
-const COST_METRIC_CANDIDATES = [
-    'claude_code_cost_usage_total',
-    'claude_code_cost_usage',
-    'claude_code_cost_usage_usd_total',
-    'claude_code_cost_usage_usd',
-];
-/**
- * Parse Prometheus text exposition format into structured metrics.
- *
- * Format per line:  metric_name{label1="val1",label2="val2"} value [timestamp]
- * Lines starting with # are comments (HELP, TYPE). Empty lines are skipped.
- */
-function parsePrometheusText(text) {
-    const results = [];
-    for (const line of text.split('\n')) {
-        if (line.startsWith('#') || line.trim() === '')
-            continue;
-        // Match: metric_name{labels} value
-        const labeledMatch = line.match(/^([a-zA-Z_:][a-zA-Z0-9_:]*)\{([^}]*)\}\s+([\d.eE+-]+)/);
-        if (labeledMatch) {
-            const [, name, labelsStr, valueStr] = labeledMatch;
-            const labels = {};
-            for (const pair of labelsStr.matchAll(/(\w+)="([^"]*)"/g)) {
-                labels[pair[1]] = pair[2];
-            }
-            results.push({ name, labels, value: parseFloat(valueStr) });
-            continue;
-        }
-        // Match: metric_name value (no labels)
-        const noLabelMatch = line.match(/^([a-zA-Z_:][a-zA-Z0-9_:]*)\s+([\d.eE+-]+)/);
-        if (noLabelMatch) {
-            const [, name, valueStr] = noLabelMatch;
-            results.push({ name, labels: {}, value: parseFloat(valueStr) });
-        }
-    }
-    return results;
-}
-/**
- * Extract a TokenSnapshot from parsed Prometheus metrics.
- * Tries multiple candidate metric names for resilience to SDK version changes.
- */
-function extractTokenSnapshot(metrics) {
-    const snapshot = {
-        inputTokens: 0,
-        outputTokens: 0,
-        cacheReadTokens: 0,
-        cacheCreationTokens: 0,
-        costUsd: 0,
-        claudeSessionId: null,
-        model: null,
-        capturedAt: new Date(),
-    };
-    for (const m of metrics) {
-        if (TOKEN_METRIC_CANDIDATES.includes(m.name)) {
-            const type = m.labels['type'];
-            switch (type) {
-                case 'input':
-                    snapshot.inputTokens = m.value;
-                    break;
-                case 'output':
-                    snapshot.outputTokens = m.value;
-                    break;
-                case 'cacheRead':
-                    snapshot.cacheReadTokens = m.value;
-                    break;
-                case 'cacheCreation':
-                    snapshot.cacheCreationTokens = m.value;
-                    break;
-            }
-            snapshot.claudeSessionId = m.labels['session_id'] || snapshot.claudeSessionId;
-            snapshot.model = m.labels['model'] || snapshot.model;
-        }
-        if (COST_METRIC_CANDIDATES.includes(m.name)) {
-            snapshot.costUsd += m.value;
-            snapshot.claudeSessionId = m.labels['session_id'] || snapshot.claudeSessionId;
-        }
-    }
-    return snapshot;
-}
-/**
- * Attempt to fetch Prometheus metrics from a candidate endpoint.
- * Returns response text or null on failure.
- */
-async function tryFetch(port, timeoutMs) {
-    try {
-        const controller = new AbortController();
-        const timeout = setTimeout(() => controller.abort(), timeoutMs);
-        const resp = await fetch(`http://localhost:${port}/metrics`, {
-            signal: controller.signal,
-        });
-        clearTimeout(timeout);
-        if (resp.ok) {
-            return await resp.text();
-        }
-        return null;
-    }
-    catch {
-        return null;
-    }
-}
-/**
- * Scrape Claude Code's Prometheus endpoint and return a TokenSnapshot.
- *
- * Returns null if:
- * - Prometheus endpoint is not available (OTel not enabled)
- * - Scrape times out (>2s by default)
- * - No matching token metrics found
- *
- * @param log Optional logging function for debug output
- */
-async function scrapeTokenSnapshot(log) {
-    const configuredPort = process.env.FRAIM_PROMETHEUS_PORT;
-    const ports = configuredPort ? [parseInt(configuredPort, 10)] : CANDIDATE_PORTS;
-    const timeoutMs = 2000;
-    for (const port of ports) {
-        const text = await tryFetch(port, timeoutMs);
-        if (text && text.includes('#')) {
-            const metrics = parsePrometheusText(text);
-            const snapshot = extractTokenSnapshot(metrics);
-            // Only return if we actually found token metrics
-            if (snapshot.inputTokens > 0 || snapshot.outputTokens > 0) {
-                log?.(`Captured token snapshot from localhost:${port} (input=${snapshot.inputTokens}, output=${snapshot.outputTokens})`);
-                return snapshot;
-            }
-            log?.(`Prometheus endpoint found at localhost:${port} but no token metrics matched`);
-        }
-    }
-    log?.('No Prometheus endpoint available — token snapshot skipped');
-    return null;
-}