@revenium/anthropic 1.0.9 → 1.1.1

package/dist/esm/config.js

@@ -1,31 +1,31 @@
-import { validateReveniumConfig } from './utils/validation.js';
-import { DEFAULT_CONFIG, ENV_VARS, LOGGING_CONFIG } from './constants.js';
+import { validateReveniumConfig } from "./utils/validation.js";
+import { DEFAULT_CONFIG, ENV_VARS, LOGGING_CONFIG } from "./constants.js";
 /**
  * Simple console logger implementation for Anthropic middleware
  */
 class ConsoleLogger {
     isDebugEnabled() {
-        return process.env[ENV_VARS.DEBUG] === 'true';
+        return process.env[ENV_VARS.DEBUG] === "true";
     }
     formatMessage(level, message, context) {
         const timestamp = new Date().toISOString();
-        const prefix = `[${LOGGING_CONFIG.MIDDLEWARE_NAME}${level === 'DEBUG' ? ' Debug' : ''}]`;
-        const contextStr = context ? ` ${JSON.stringify(context)}` : '';
+        const prefix = `[${LOGGING_CONFIG.MIDDLEWARE_NAME}${level === "DEBUG" ? " Debug" : ""}]`;
+        const contextStr = context ? ` ${JSON.stringify(context)}` : "";
         return `${timestamp} ${prefix} ${message}${contextStr}`;
     }
     debug(message, context) {
         if (this.isDebugEnabled()) {
-            console.debug(this.formatMessage('DEBUG', message, context));
+            console.debug(this.formatMessage("DEBUG", message, context));
         }
     }
     info(message, context) {
-        console.info(this.formatMessage('INFO', message, context));
+        console.info(this.formatMessage("INFO", message, context));
     }
     warn(message, context) {
-        console.warn(this.formatMessage('WARN', message, context));
+        console.warn(this.formatMessage("WARN", message, context));
     }
     error(message, context) {
-        console.error(this.formatMessage('ERROR', message, context));
+        console.error(this.formatMessage("ERROR", message, context));
     }
 }
 /**
@@ -40,13 +40,14 @@ function loadConfigFromEnvironment() {
         reveniumApiKey: process.env[ENV_VARS.REVENIUM_API_KEY],
         reveniumBaseUrl: process.env[ENV_VARS.REVENIUM_BASE_URL],
         anthropicApiKey: process.env[ENV_VARS.ANTHROPIC_API_KEY],
-        debug: process.env[ENV_VARS.DEBUG] === 'true',
+        debug: process.env[ENV_VARS.DEBUG] === "true",
         logLevel: process.env[ENV_VARS.LOG_LEVEL],
         apiTimeout: process.env[ENV_VARS.API_TIMEOUT],
         failSilent: process.env[ENV_VARS.FAIL_SILENT],
         maxRetries: process.env[ENV_VARS.MAX_RETRIES],
         printSummary: process.env[ENV_VARS.PRINT_SUMMARY],
-        teamId: process.env[ENV_VARS.TEAM_ID]
+        teamId: process.env[ENV_VARS.TEAM_ID],
+        capturePrompts: process.env[ENV_VARS.CAPTURE_PROMPTS],
     };
     return env;
 }
@@ -57,11 +58,11 @@ function parsePrintSummary(value) {
     if (!value)
         return undefined;
     const lowerValue = value.toLowerCase();
-    if (lowerValue === 'true' || lowerValue === 'human')
-        return 'human';
-    if (lowerValue === 'json')
-        return 'json';
-    if (lowerValue === 'false')
+    if (lowerValue === "true" || lowerValue === "human")
+        return "human";
+    if (lowerValue === "json")
+        return "json";
+    if (lowerValue === "false")
         return false;
     return undefined;
 }
@@ -73,9 +74,10 @@ function createConfigFromEnvironment(env) {
         return null;
     }
     const apiTimeout = env.apiTimeout ? parseInt(env.apiTimeout, 10) : undefined;
-    const failSilent = env.failSilent !== 'false'; // Default to true
+    const failSilent = env.failSilent !== "false"; // Default to true
     const maxRetries = env.maxRetries ? parseInt(env.maxRetries, 10) : undefined;
     const printSummary = parsePrintSummary(env.printSummary);
+    const capturePrompts = env.capturePrompts === "true";
     return {
         reveniumApiKey: env.reveniumApiKey,
         reveniumBaseUrl: env.reveniumBaseUrl || DEFAULT_CONFIG.REVENIUM_BASE_URL,
@@ -84,7 +86,8 @@ function createConfigFromEnvironment(env) {
         failSilent,
         maxRetries,
         printSummary,
-        teamId: env.teamId?.trim()
+        teamId: env.teamId?.trim(),
+        capturePrompts,
     };
 }
 /**
@@ -94,18 +97,18 @@ export function validateConfig(config) {
     const validation = validateReveniumConfig(config);
     if (!validation.isValid) {
         // Log detailed validation errors
-        getLogger().error('Configuration validation failed', {
+        getLogger().error("Configuration validation failed", {
             errors: validation.errors,
             warnings: validation.warnings,
-            suggestions: validation.suggestions
+            suggestions: validation.suggestions,
         });
         // Create detailed error message
-        let errorMessage = 'Configuration validation failed:\n';
+        let errorMessage = "Configuration validation failed:\n";
         validation.errors.forEach((error, index) => {
             errorMessage += `  ${index + 1}. ${error}\n`;
         });
         if (validation.suggestions && validation.suggestions.length > 0) {
-            errorMessage += '\nSuggestions:\n';
+            errorMessage += "\nSuggestions:\n";
             validation.suggestions.forEach((suggestion) => {
                 errorMessage += `  • ${suggestion}\n`;
             });
@@ -114,8 +117,8 @@ export function validateConfig(config) {
     }
     // Log warnings if any
     if (validation.warnings && validation.warnings.length > 0) {
-        getLogger().warn('Configuration warnings', {
-            warnings: validation.warnings
+        getLogger().warn("Configuration warnings", {
+            warnings: validation.warnings,
         });
     }
 }
@@ -138,18 +141,18 @@ export function setConfig(config) {
     const validation = validateReveniumConfig(config);
     if (!validation.isValid) {
         // Log detailed validation errors
-        getLogger().error('Configuration validation failed', {
+        getLogger().error("Configuration validation failed", {
             errors: validation.errors,
             warnings: validation.warnings,
-            suggestions: validation.suggestions
+            suggestions: validation.suggestions,
         });
         // Create detailed error message
-        let errorMessage = 'Configuration validation failed:\n';
+        let errorMessage = "Configuration validation failed:\n";
         validation.errors.forEach((error, index) => {
             errorMessage += `  ${index + 1}. ${error}\n`;
         });
         if (validation.suggestions && validation.suggestions.length > 0) {
-            errorMessage += '\nSuggestions:\n';
+            errorMessage += "\nSuggestions:\n";
             validation.suggestions.forEach((suggestion) => {
                 errorMessage += `  • ${suggestion}\n`;
             });
@@ -158,16 +161,16 @@ export function setConfig(config) {
     }
     // Log warnings if any
     if (validation.warnings && validation.warnings.length > 0) {
-        getLogger().warn('Configuration warnings', {
-            warnings: validation.warnings
+        getLogger().warn("Configuration warnings", {
+            warnings: validation.warnings,
         });
     }
     // Use the normalized config from validation (with defaults applied and fields trimmed)
     globalConfig = validation.config;
-    globalLogger.debug('Revenium configuration updated', {
+    globalLogger.debug("Revenium configuration updated", {
         baseUrl: globalConfig.reveniumBaseUrl,
         hasApiKey: !!globalConfig.reveniumApiKey,
-        hasAnthropicKey: !!globalConfig.anthropicApiKey
+        hasAnthropicKey: !!globalConfig.anthropicApiKey,
     });
 }
 /**
@@ -181,7 +184,7 @@ export function getLogger() {
  */
 export function setLogger(logger) {
     globalLogger = logger;
-    globalLogger.debug('Custom logger set for Revenium middleware');
+    globalLogger.debug("Custom logger set for Revenium middleware");
 }
 /**
  * Initialize configuration from environment variables
@@ -192,12 +195,12 @@ export function initializeConfig() {
     if (config) {
         try {
             setConfig(config);
-            globalLogger.debug('Revenium middleware initialized from environment variables');
+            globalLogger.debug("Revenium middleware initialized from environment variables");
             return true;
         }
         catch (error) {
-            globalLogger.error('Failed to initialize Revenium configuration', {
-                error: error instanceof Error ? error.message : String(error)
+            globalLogger.error("Failed to initialize Revenium configuration", {
+                error: error instanceof Error ? error.message : String(error),
             });
             return false;
         }
@@ -218,14 +221,14 @@ export function getConfigStatus() {
             hasConfig: false,
             hasApiKey: false,
             hasAnthropicKey: false,
-            baseUrl: ''
+            baseUrl: "",
         };
     }
     return {
         hasConfig: true,
         hasApiKey: !!globalConfig.reveniumApiKey,
         hasAnthropicKey: !!globalConfig.anthropicApiKey,
-        baseUrl: globalConfig.reveniumBaseUrl
+        baseUrl: globalConfig.reveniumBaseUrl,
     };
 }
 /**

package/dist/esm/constants.js

@@ -22,6 +22,10 @@ export const DEFAULT_CONFIG = {
     MAX_RETRY_ATTEMPTS: 10,
     /** Warning threshold for low API timeout */
     LOW_TIMEOUT_WARNING_THRESHOLD: 3000,
+    /** Default prompt capture behavior */
+    CAPTURE_PROMPTS: false,
+    /** Maximum size for each prompt field in characters (50KB) */
+    MAX_PROMPT_SIZE: 50000,
 };
 /**
  * Circuit breaker configuration constants
@@ -113,6 +117,8 @@ export const ENV_VARS = {
     PRINT_SUMMARY: "REVENIUM_PRINT_SUMMARY",
     /** Team ID for cost metrics retrieval */
     TEAM_ID: "REVENIUM_TEAM_ID",
+    /** Prompt capture mode */
+    CAPTURE_PROMPTS: "REVENIUM_CAPTURE_PROMPTS",
 };
 /**
  * Summary printer configuration

package/dist/esm/tracking.js

@@ -8,6 +8,7 @@ import { withRetry, ReveniumApiError, createErrorContext, handleError, } from ".
 import { DEFAULT_CONFIG, API_ENDPOINTS, LOGGING_CONFIG, ANTHROPIC_PATTERNS, } from "./constants.js";
 import { getEnvironment, getRegion, getCredentialAlias, getTraceType, getTraceName, detectOperationSubtype, getParentTransactionId, getTransactionName, getRetryNumber, } from "./utils/trace-fields.js";
 import { printUsageSummary } from "./utils/summary-printer.js";
+import { extractPrompts } from "./utils/prompt-extraction.js";
 // Global logger
 const logger = getLogger();
 if (typeof process !== "undefined") {
@@ -143,7 +144,9 @@ async function buildReveniumPayload(data) {
         "taskType",
         "agent",
         "organizationId",
+        "organizationName",
         "productId",
+        "productName",
         "subscriber",
         "subscriptionId",
         "traceId",
@@ -167,6 +170,10 @@ async function buildReveniumPayload(data) {
     const transactionName = getTransactionName();
     const retryNumber = getRetryNumber();
     const operationSubtype = detectOperationSubtype(data.requestBody);
+    let promptData = null;
+    if (data.requestBody && data.response) {
+        promptData = extractPrompts(data.requestBody, data.response, data.metadata);
+    }
     return {
         stopReason: getStopReason(data.stopReason),
         costType: "AI",
@@ -189,10 +196,12 @@ async function buildReveniumPayload(data) {
         middlewareSource: "nodejs",
         ...(data.metadata?.taskType && { taskType: data.metadata.taskType }),
         ...(data.metadata?.agent && { agent: data.metadata.agent }),
-        ...(data.metadata?.organizationId && {
-            organizationId: data.metadata.organizationId,
+        ...((data.metadata?.organizationName || data.metadata?.organizationId) && {
+            organizationName: data.metadata.organizationName || data.metadata.organizationId,
+        }),
+        ...((data.metadata?.productName || data.metadata?.productId) && {
+            productName: data.metadata.productName || data.metadata.productId,
         }),
-        ...(data.metadata?.productId && { productId: data.metadata.productId }),
         ...(data.metadata?.subscriber && { subscriber: data.metadata.subscriber }),
         ...(data.metadata?.subscriptionId && {
             subscriptionId: data.metadata.subscriptionId,
@@ -210,6 +219,15 @@ async function buildReveniumPayload(data) {
         ...(transactionName && { transactionName }),
         ...(retryNumber !== undefined && { retryNumber }),
         ...(operationSubtype && { operationSubtype }),
+        ...(data.hasVisionContent !== undefined && {
+            hasVisionContent: data.hasVisionContent,
+        }),
+        ...(promptData && {
+            systemPrompt: promptData.systemPrompt,
+            inputMessages: promptData.inputMessages,
+            outputResponse: promptData.outputResponse,
+            promptsTruncated: promptData.promptsTruncated,
+        }),
         ...customFields,
     };
 }

package/dist/esm/utils/prompt-extraction.js

@@ -0,0 +1,154 @@
+import { DEFAULT_CONFIG } from "../constants.js";
+import { getConfig } from "../config.js";
+function sanitizeCredentials(text) {
+    const patterns = [
+        {
+            regex: /sk-proj-[a-zA-Z0-9_-]{48,}/g,
+            replacement: "sk-proj-***REDACTED***",
+        },
+        { regex: /sk-[a-zA-Z0-9_-]{20,}/g, replacement: "sk-***REDACTED***" },
+        {
+            regex: /Bearer\s+[a-zA-Z0-9_\-\.]+/gi,
+            replacement: "Bearer ***REDACTED***",
+        },
+        {
+            regex: /api[_-]?key["\s:=]+[a-zA-Z0-9_\-\.\+\/=]{20,}/gi,
+            replacement: "api_key: ***REDACTED***",
+        },
+        {
+            regex: /token["\s:=]+[a-zA-Z0-9_\-\.]{20,}/gi,
+            replacement: "token: ***REDACTED***",
+        },
+        {
+            regex: /password["\s:=]+[^\s"']{8,}/gi,
+            replacement: "password: ***REDACTED***",
+        },
+    ];
+    let sanitized = text;
+    for (const pattern of patterns) {
+        sanitized = sanitized.replace(pattern.regex, pattern.replacement);
+    }
+    return sanitized;
+}
+function truncateString(str, maxLength) {
+    const sanitized = sanitizeCredentials(str);
+    if (sanitized.length <= maxLength) {
+        return { value: sanitized, truncated: false };
+    }
+    return { value: sanitized.substring(0, maxLength), truncated: true };
+}
+function extractSystemPrompt(params) {
+    if (!params.system) {
+        return "";
+    }
+    if (typeof params.system === "string") {
+        return params.system;
+    }
+    if (Array.isArray(params.system)) {
+        return params.system
+            .map((block) => {
+            if (block.type === "text") {
+                return block.text;
+            }
+            if (block.type === "image") {
+                return "[IMAGE]";
+            }
+            return "";
+        })
+            .filter(Boolean)
+            .join("\n");
+    }
+    return "";
+}
+function extractInputMessages(params) {
+    if (!params.messages || params.messages.length === 0) {
+        return "";
+    }
+    return params.messages
+        .map((message) => {
+        const role = message.role;
+        let content = "";
+        if (typeof message.content === "string") {
+            content = message.content;
+        }
+        else if (Array.isArray(message.content)) {
+            content = message.content
+                .map((block) => {
+                if (block.type === "text") {
+                    return block.text;
+                }
+                if (block.type === "image") {
+                    return "[IMAGE]";
+                }
+                if (block.type === "tool_use") {
+                    const toolName = block.name || "unknown";
+                    return `[TOOL_USE: ${toolName}]`;
+                }
+                if (block.type === "tool_result") {
+                    return "[TOOL_RESULT]";
+                }
+                return "";
+            })
+                .filter(Boolean)
+                .join("\n");
+        }
+        return `[${role}]\n${content}`;
+    })
+        .join("\n\n");
+}
+function extractOutputResponse(response) {
+    if (!response.content || response.content.length === 0) {
+        return "";
+    }
+    return response.content
+        .map((block) => {
+        if (block.type === "text") {
+            return block.text;
+        }
+        if (block.type === "tool_use") {
+            return `[TOOL_USE: ${block.name}]`;
+        }
+        return "";
+    })
+        .filter(Boolean)
+        .join("\n");
+}
+export function shouldCapturePrompts(metadata) {
+    if (metadata?.capturePrompts !== undefined) {
+        return metadata.capturePrompts;
+    }
+    const config = getConfig();
+    if (config?.capturePrompts !== undefined) {
+        return config.capturePrompts;
+    }
+    return DEFAULT_CONFIG.CAPTURE_PROMPTS;
+}
+export function extractPrompts(params, response, metadata) {
+    if (!shouldCapturePrompts(metadata)) {
+        return null;
+    }
+    const maxSize = DEFAULT_CONFIG.MAX_PROMPT_SIZE;
+    let anyTruncated = false;
+    const systemPromptRaw = extractSystemPrompt(params);
+    const systemPromptResult = truncateString(systemPromptRaw, maxSize);
+    anyTruncated = anyTruncated || systemPromptResult.truncated;
+    const inputMessagesRaw = extractInputMessages(params);
+    const inputMessagesResult = truncateString(inputMessagesRaw, maxSize);
+    anyTruncated = anyTruncated || inputMessagesResult.truncated;
+    const outputResponseRaw = extractOutputResponse(response);
+    const outputResponseResult = truncateString(outputResponseRaw, maxSize);
+    anyTruncated = anyTruncated || outputResponseResult.truncated;
+    const hasAnyContent = systemPromptResult.value ||
+        inputMessagesResult.value ||
+        outputResponseResult.value;
+    if (!hasAnyContent) {
+        return null;
+    }
+    return {
+        systemPrompt: systemPromptResult.value || undefined,
+        inputMessages: inputMessagesResult.value || undefined,
+        outputResponse: outputResponseResult.value || undefined,
+        promptsTruncated: anyTruncated,
+    };
+}
+//# sourceMappingURL=prompt-extraction.js.map

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

@@ -89,6 +89,41 @@ export function detectOperationSubtype(requestBody) {
     }
     return null;
 }
+export function detectVisionContent(params) {
+    if (!params) {
+        return false;
+    }
+    try {
+        if (params.messages && Array.isArray(params.messages)) {
+            for (const message of params.messages) {
+                if (!message || typeof message !== "object" || !message.content) {
+                    continue;
+                }
+                if (Array.isArray(message.content)) {
+                    for (const block of message.content) {
+                        if (block && typeof block === "object" && block.type === "image") {
+                            return true;
+                        }
+                    }
+                }
+            }
+        }
+        if (params.system && Array.isArray(params.system)) {
+            for (const block of params.system) {
+                if (block && typeof block === "object" && block.type === "image") {
+                    return true;
+                }
+            }
+        }
+        return false;
+    }
+    catch (error) {
+        logger.debug("Error detecting vision content, defaulting to false", {
+            error: error instanceof Error ? error.message : String(error),
+        });
+        return false;
+    }
+}
 export function getParentTransactionId() {
     return process.env.REVENIUM_PARENT_TRANSACTION_ID || null;
 }