@revenium/anthropic 1.0.8 → 1.1.0

package/dist/cjs/utils/summary-printer.js

@@ -0,0 +1,189 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.printUsageSummary = printUsageSummary;
+const config_1 = require("../config");
+const constants_1 = require("../constants");
+const logger = (0, config_1.getLogger)();
+function delayWithUnref(ms) {
+    return new Promise((resolve) => {
+        const timer = setTimeout(resolve, ms);
+        if (typeof timer.unref === "function") {
+            timer.unref();
+        }
+    });
+}
+async function fetchCompletionMetrics(transactionId, maxRetries = constants_1.SUMMARY_PRINTER_CONFIG.MAX_RETRIES, retryDelay = constants_1.SUMMARY_PRINTER_CONFIG.RETRY_DELAY) {
+    const config = (0, config_1.getConfig)();
+    if (!config) {
+        logger.debug("No config available for summary printing");
+        return null;
+    }
+    if (!config.teamId) {
+        logger.debug("Team ID not configured, skipping cost retrieval for summary");
+        return null;
+    }
+    const baseUrl = (config.reveniumBaseUrl || constants_1.DEFAULT_CONFIG.REVENIUM_BASE_URL).replace(/\/+$/, "");
+    const url = `${baseUrl}/profitstream/v2/api/sources/metrics/ai/completions`;
+    const teamId = config.teamId.trim();
+    const urlWithParams = `${url}?teamId=${encodeURIComponent(teamId)}&transactionId=${encodeURIComponent(transactionId)}`;
+    logger.debug("Fetching completion metrics", { url: urlWithParams });
+    for (let attempt = 0; attempt < maxRetries; attempt++) {
+        // Create an AbortController with timeout to prevent hung requests from keeping Node process alive
+        const controller = new AbortController();
+        const timeoutId = setTimeout(() => controller.abort(), constants_1.SUMMARY_PRINTER_CONFIG.FETCH_TIMEOUT);
+        // Unref the timer so it doesn't keep the process alive
+        if (typeof timeoutId.unref === "function") {
+            timeoutId.unref();
+        }
+        try {
+            const response = await fetch(urlWithParams, {
+                method: "GET",
+                headers: {
+                    Accept: "application/json",
+                    "x-api-key": config.reveniumApiKey,
+                },
+                signal: controller.signal,
+            });
+            if (!response.ok) {
+                try {
+                    await response.text();
+                }
+                catch { }
+                logger.debug(`Completions metrics API returned ${response.status}`, {
+                    attempt: attempt + 1,
+                });
+                if (attempt < maxRetries - 1) {
+                    await delayWithUnref(retryDelay);
+                    continue;
+                }
+                return null;
+            }
+            const data = (await response.json());
+            const completions = data._embedded?.aICompletionMetricResourceList;
+            if (completions && completions.length > 0) {
+                return completions[0];
+            }
+            if (attempt < maxRetries - 1) {
+                logger.debug(`Waiting for metrics to aggregate (attempt ${attempt + 1}/${maxRetries})...`);
+                await delayWithUnref(retryDelay);
+            }
+        }
+        catch (error) {
+            logger.debug("Failed to fetch completion metrics", {
+                error: error instanceof Error ? error.message : String(error),
+                attempt: attempt + 1,
+            });
+            if (attempt < maxRetries - 1) {
+                await delayWithUnref(retryDelay);
+            }
+        }
+        finally {
+            clearTimeout(timeoutId);
+        }
+    }
+    return null;
+}
+function isSummaryFormat(value) {
+    return value === "human" || value === "json";
+}
+function formatAndPrintJsonSummary(payload, metrics) {
+    const config = (0, config_1.getConfig)();
+    const summary = {
+        model: payload.model,
+        provider: payload.provider,
+        durationSeconds: payload.requestDuration / 1000,
+        inputTokenCount: payload.inputTokenCount,
+        outputTokenCount: payload.outputTokenCount,
+        totalTokenCount: payload.totalTokenCount,
+        cost: typeof metrics?.totalCost === "number" ? metrics.totalCost : null,
+    };
+    if (summary.cost === null) {
+        summary.costStatus = config?.teamId ? "pending" : "unavailable";
+    }
+    if (payload.traceId) {
+        summary.traceId = payload.traceId;
+    }
+    console.log(JSON.stringify(summary));
+}
+function formatAndPrintHumanSummary(payload, metrics) {
+    console.log("\n" + "=".repeat(60));
+    console.log("📊 REVENIUM USAGE SUMMARY");
+    console.log("=".repeat(60));
+    console.log(`🤖 Model: ${payload.model}`);
+    console.log(`🏢 Provider: ${payload.provider}`);
+    console.log(`⏱️  Duration: ${(payload.requestDuration / 1000).toFixed(2)}s`);
+    console.log("\n💬 Token Usage:");
+    console.log(`   📥 Input Tokens:  ${(payload.inputTokenCount ?? 0).toLocaleString()}`);
+    console.log(`   📤 Output Tokens: ${(payload.outputTokenCount ?? 0).toLocaleString()}`);
+    console.log(`   📊 Total Tokens:  ${(payload.totalTokenCount ?? 0).toLocaleString()}`);
+    if (typeof metrics?.totalCost === "number") {
+        console.log(`\n💰 Cost: $${metrics.totalCost.toFixed(6)}`);
+    }
+    else {
+        const config = (0, config_1.getConfig)();
+        if (!config?.teamId) {
+            console.log(`\n💰 Cost: Set REVENIUM_TEAM_ID environment variable to see pricing`);
+        }
+        else {
+            console.log(`\n💰 Cost: (pending aggregation)`);
+        }
+    }
+    if (payload.traceId) {
+        console.log(`\n🔖 Trace ID: ${payload.traceId}`);
+    }
+    console.log("=".repeat(60) + "\n");
+}
+function formatAndPrintSummary(payload, metrics, format) {
+    if (format === "json") {
+        formatAndPrintJsonSummary(payload, metrics);
+    }
+    else {
+        formatAndPrintHumanSummary(payload, metrics);
+    }
+}
+function safeFormatAndPrintSummary(payload, metrics, format) {
+    try {
+        formatAndPrintSummary(payload, metrics, format);
+    }
+    catch (error) {
+        logger.debug("Failed to format and print summary", {
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+}
+function getSummaryFormat(value) {
+    if (!value)
+        return null;
+    if (value === true)
+        return "human";
+    if (isSummaryFormat(value)) {
+        return value;
+    }
+    return null;
+}
+function printUsageSummary(payload) {
+    const config = (0, config_1.getConfig)();
+    const format = getSummaryFormat(config?.printSummary);
+    if (!format) {
+        return;
+    }
+    if (config?.teamId && payload.transactionId) {
+        fetchCompletionMetrics(payload.transactionId)
+            .then((metrics) => {
+            safeFormatAndPrintSummary(payload, metrics, format);
+        })
+            .catch((error) => {
+            logger.debug("Failed to print usage summary with metrics", {
+                error: error instanceof Error ? error.message : String(error),
+            });
+            safeFormatAndPrintSummary(payload, null, format);
+        })
+            .catch(() => {
+            // Final safety catch to prevent unhandled rejections
+        });
+    }
+    else {
+        safeFormatAndPrintSummary(payload, null, format);
+    }
+}
+//# sourceMappingURL=summary-printer.js.map

package/dist/cjs/utils/trace-fields.js

@@ -0,0 +1,117 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getEnvironment = getEnvironment;
+exports.getRegion = getRegion;
+exports.getCredentialAlias = getCredentialAlias;
+exports.getTraceType = getTraceType;
+exports.getTraceName = getTraceName;
+exports.detectOperationSubtype = detectOperationSubtype;
+exports.getParentTransactionId = getParentTransactionId;
+exports.getTransactionName = getTransactionName;
+exports.getRetryNumber = getRetryNumber;
+const config_1 = require("../config");
+const logger = (0, config_1.getLogger)();
+let cachedRegion = null;
+let regionCached = false;
+function getEnvironment() {
+    const env = process.env.REVENIUM_ENVIRONMENT ||
+        process.env.NODE_ENV ||
+        process.env.DEPLOYMENT_ENV ||
+        null;
+    if (env && env.length > 255) {
+        logger.warn(`environment exceeds max length of 255 characters. Truncating.`);
+        return env.substring(0, 255).trim();
+    }
+    return env ? env.trim() : null;
+}
+async function getRegion() {
+    if (regionCached) {
+        return cachedRegion;
+    }
+    const envRegion = process.env.AWS_REGION ||
+        process.env.AZURE_REGION ||
+        process.env.GCP_REGION ||
+        process.env.REVENIUM_REGION;
+    if (envRegion) {
+        cachedRegion = envRegion.trim();
+        regionCached = true;
+        return cachedRegion;
+    }
+    try {
+        const controller = new AbortController();
+        const timeoutId = setTimeout(() => controller.abort(), 1000);
+        const response = await fetch("http://169.254.169.254/latest/meta-data/placement/region", {
+            signal: controller.signal,
+        });
+        clearTimeout(timeoutId);
+        if (!response.ok) {
+            cachedRegion = null;
+            regionCached = true;
+            return null;
+        }
+        const text = await response.text();
+        cachedRegion = text.trim();
+        regionCached = true;
+        return cachedRegion;
+    }
+    catch (error) {
+        cachedRegion = null;
+        regionCached = true;
+        return null;
+    }
+}
+function getCredentialAlias() {
+    const alias = process.env.REVENIUM_CREDENTIAL_ALIAS || null;
+    if (alias && alias.length > 255) {
+        logger.warn(`credentialAlias exceeds max length of 255 characters. Truncating.`);
+        return alias.substring(0, 255).trim();
+    }
+    return alias ? alias.trim() : null;
+}
+function getTraceType() {
+    const traceType = process.env.REVENIUM_TRACE_TYPE;
+    if (!traceType) {
+        return null;
+    }
+    if (!/^[a-zA-Z0-9_-]+$/.test(traceType)) {
+        logger.warn(`Invalid trace_type format: ${traceType}. Must be alphanumeric with hyphens/underscores only.`);
+        return null;
+    }
+    if (traceType.length > 128) {
+        logger.warn(`trace_type exceeds max length of 128 characters: ${traceType}. Truncating.`);
+        return traceType.substring(0, 128);
+    }
+    return traceType;
+}
+function getTraceName() {
+    const traceName = process.env.REVENIUM_TRACE_NAME;
+    if (!traceName) {
+        return null;
+    }
+    if (traceName.length > 256) {
+        logger.warn(`trace_name exceeds max length of 256 characters. Truncating.`);
+        return traceName.substring(0, 256);
+    }
+    return traceName;
+}
+function detectOperationSubtype(requestBody) {
+    if (requestBody && requestBody.tools && requestBody.tools.length > 0) {
+        return "function_call";
+    }
+    return null;
+}
+function getParentTransactionId() {
+    return process.env.REVENIUM_PARENT_TRANSACTION_ID || null;
+}
+function getTransactionName() {
+    return process.env.REVENIUM_TRANSACTION_NAME || null;
+}
+function getRetryNumber() {
+    const retryNum = process.env.REVENIUM_RETRY_NUMBER;
+    if (retryNum) {
+        const parsed = parseInt(retryNum, 10);
+        return isNaN(parsed) ? 0 : parsed;
+    }
+    return 0;
+}
+//# sourceMappingURL=trace-fields.js.map

package/dist/cjs/utils/validation.js

@@ -139,29 +139,31 @@ function validateReveniumConfig(config) {
     else if (cfg?.reveniumApiKey?.length < constants_1.VALIDATION_CONFIG.MIN_API_KEY_LENGTH) {
         warnings.push('reveniumApiKey appears to be too short - verify it is correct');
     }
-    // Validate Revenium base URL
-    if (!isString(cfg?.reveniumBaseUrl)) {
-        errors.push('reveniumBaseUrl is required and must be a string');
-    }
-    else if (!cfg?.reveniumBaseUrl?.trim()) {
-        errors.push('reveniumBaseUrl cannot be empty');
-    }
-    else {
-        try {
-            const url = new URL(cfg?.reveniumBaseUrl);
-            if (!url.protocol.startsWith('http')) {
-                errors.push('reveniumBaseUrl must use HTTP or HTTPS protocol');
+    // Validate Revenium base URL (optional - defaults to https://api.revenium.ai)
+    if (cfg?.reveniumBaseUrl !== undefined) {
+        if (!isString(cfg?.reveniumBaseUrl)) {
+            errors.push('reveniumBaseUrl must be a string if provided');
+        }
+        else if (!cfg?.reveniumBaseUrl?.trim()) {
+            errors.push('reveniumBaseUrl cannot be empty if provided');
+        }
+        else {
+            try {
+                const url = new URL(cfg?.reveniumBaseUrl);
+                if (!url.protocol.startsWith('http')) {
+                    errors.push('reveniumBaseUrl must use HTTP or HTTPS protocol');
+                }
+                // Check for localhost/development hostnames (IPv4, IPv6, and named)
+                const localhostHostnames = ['localhost', '127.0.0.1', '::1', '[::1]'];
+                if (localhostHostnames.includes(url.hostname)) {
+                    warnings.push('Using localhost for Revenium API - ensure this is intended for development');
+                }
             }
-            // Check for localhost/development hostnames (IPv4, IPv6, and named)
-            const localhostHostnames = ['localhost', '127.0.0.1', '::1', '[::1]'];
-            if (localhostHostnames.includes(url.hostname)) {
-                warnings.push('Using localhost for Revenium API - ensure this is intended for development');
+            catch {
+                errors.push('reveniumBaseUrl must be a valid URL');
+                suggestions.push('Use format: https://api.revenium.ai');
             }
         }
-        catch {
-            errors.push('reveniumBaseUrl must be a valid URL');
-            suggestions.push('Use format: https://api.revenium.ai');
-        }
     }
     // Validate optional Anthropic API key
     if (cfg?.anthropicApiKey !== undefined && !isString(cfg?.anthropicApiKey)) {
@@ -203,6 +205,23 @@ function validateReveniumConfig(config) {
     else if (cfg?.maxRetries !== undefined && cfg?.maxRetries === 0) {
         warnings.push('maxRetries is 0 - no retry attempts will be made');
     }
+    // Validate optional printSummary
+    if (cfg?.printSummary !== undefined) {
+        const validPrintSummaryValues = [true, false, 'human', 'json'];
+        if (!validPrintSummaryValues.includes(cfg?.printSummary)) {
+            errors.push("printSummary must be a boolean or one of: 'human', 'json'");
+            suggestions.push("Use printSummary: true, 'human', or 'json' to enable summary output");
+        }
+    }
+    // Validate optional teamId
+    if (cfg?.teamId !== undefined) {
+        if (!isString(cfg?.teamId)) {
+            errors.push('teamId must be a string if provided');
+        }
+        else if (cfg?.teamId?.trim()?.length === 0) {
+            errors.push('teamId cannot be empty if provided');
+        }
+    }
     if (errors.length > 0) {
         return {
             isValid: false,
@@ -211,14 +230,27 @@ function validateReveniumConfig(config) {
             suggestions
         };
     }
-    // Build validated config
+    // Determine validated printSummary value
+    let validatedPrintSummary;
+    if (cfg?.printSummary === true || cfg?.printSummary === 'human') {
+        validatedPrintSummary = 'human';
+    }
+    else if (cfg?.printSummary === 'json') {
+        validatedPrintSummary = 'json';
+    }
+    else if (cfg?.printSummary === false) {
+        validatedPrintSummary = false;
+    }
+    // Build validated config (apply default for reveniumBaseUrl if not provided)
     const validatedConfig = {
         reveniumApiKey: cfg?.reveniumApiKey,
-        reveniumBaseUrl: cfg?.reveniumBaseUrl,
+        reveniumBaseUrl: isString(cfg?.reveniumBaseUrl) ? cfg?.reveniumBaseUrl : constants_1.DEFAULT_CONFIG.REVENIUM_BASE_URL,
         anthropicApiKey: isString(cfg?.anthropicApiKey) ? cfg?.anthropicApiKey : undefined,
         apiTimeout: isNumber(cfg?.apiTimeout) ? cfg?.apiTimeout : undefined,
         failSilent: isBoolean(cfg?.failSilent) ? cfg?.failSilent : undefined,
-        maxRetries: isNumber(cfg?.maxRetries) ? cfg?.maxRetries : undefined
+        maxRetries: isNumber(cfg?.maxRetries) ? cfg?.maxRetries : undefined,
+        printSummary: validatedPrintSummary,
+        teamId: isString(cfg?.teamId) ? cfg?.teamId?.trim() : undefined
     };
     return {
         isValid: true,