@revenium/anthropic 1.1.0 → 1.1.1

package/dist/esm/wrapper.js

@@ -6,6 +6,8 @@ import { getLogger, getConfig } from "./config.js";
 import { trackUsageAsync, extractUsageFromResponse, extractUsageFromStream, } from "./tracking.js";
 import { validateAnthropicMessageParams, validateUsageMetadata, } from "./utils/validation.js";
 import { AnthropicPatchingError, RequestProcessingError, StreamProcessingError, createErrorContext, handleError, } from "./utils/error-handling.js";
+import { shouldCapturePrompts } from "./utils/prompt-extraction.js";
+import { detectVisionContent } from "./utils/trace-fields.js";
 import { randomUUID } from "crypto";
 // Global logger
 const logger = getLogger();
@@ -128,6 +130,65 @@ export function unpatchAnthropic() {
 export function isAnthropicPatched() {
     return patchingContext.isPatched;
 }
+/**
+ * Reconstruct a response object from streaming chunks for prompt capture
+ */
+function reconstructResponseFromChunks(chunks, model) {
+    const contentBlocks = [];
+    let stopReason;
+    let stopSequence;
+    const usage = {};
+    for (const chunk of chunks) {
+        if (chunk.type === "content_block_start" && chunk.content_block) {
+            contentBlocks.push({ ...chunk.content_block });
+        }
+        else if (chunk.type === "content_block_delta" && chunk.delta) {
+            const lastBlock = contentBlocks[contentBlocks.length - 1];
+            if (lastBlock && chunk.delta.type === "text_delta") {
+                if (lastBlock.type === "text") {
+                    lastBlock.text = (lastBlock.text || "") + (chunk.delta.text || "");
+                }
+            }
+            else if (lastBlock && chunk.delta.type === "input_json_delta") {
+                if (lastBlock.type === "tool_use") {
+                    lastBlock.input = lastBlock.input || "";
+                    lastBlock.input +=
+                        chunk.delta.partial_json || "";
+                }
+            }
+        }
+        else if (chunk.type === "message_delta" && chunk.delta) {
+            const delta = chunk.delta;
+            if (delta.stop_reason) {
+                stopReason = delta.stop_reason;
+            }
+            if (delta.stop_sequence) {
+                stopSequence = delta.stop_sequence;
+            }
+        }
+        else if (chunk.type === "message_start" && chunk.message?.usage) {
+            Object.assign(usage, chunk.message.usage);
+        }
+        else if (chunk.usage) {
+            Object.assign(usage, chunk.usage);
+        }
+    }
+    return {
+        id: `reconstructed-${Date.now()}`,
+        type: "message",
+        role: "assistant",
+        content: contentBlocks,
+        model,
+        stop_reason: stopReason || "end_turn",
+        stop_sequence: stopSequence,
+        usage: {
+            input_tokens: usage.input_tokens || 0,
+            output_tokens: usage.output_tokens || 0,
+            cache_creation_input_tokens: usage.cache_creation_input_tokens,
+            cache_read_input_tokens: usage.cache_read_input_tokens,
+        },
+    };
+}
 /**
  * Handle streaming response by collecting chunks and extracting usage data
  */
@@ -160,6 +221,10 @@ async function handleStreamingResponse(stream, context) {
                 timeToFirstToken,
             });
             const usage = extractUsageFromStream(chunks);
+            let reconstructedResponse = undefined;
+            if (shouldCapturePrompts(metadata)) {
+                reconstructedResponse = reconstructResponseFromChunks(chunks, model);
+            }
             // Create tracking data
             const trackingData = {
                 requestId,
@@ -176,6 +241,8 @@ async function handleStreamingResponse(stream, context) {
                 responseTime,
                 timeToFirstToken,
                 requestBody: requestBody,
+                response: reconstructedResponse,
+                hasVisionContent: detectVisionContent(requestBody),
             };
             // Track usage asynchronously
             trackUsageAsync(trackingData);
@@ -237,6 +304,8 @@ async function patchedCreateMethod(params, options) {
             const responseTime = new Date();
             // Extract usage information
             const usage = extractUsageFromResponse(response);
+            // Detect vision content
+            const hasVisionContent = detectVisionContent(params);
             // Create tracking data
             const trackingData = {
                 requestId,
@@ -251,7 +320,9 @@ async function patchedCreateMethod(params, options) {
                 metadata,
                 requestTime,
                 responseTime,
+                hasVisionContent,
                 requestBody: params,
+                response,
             };
             // Track usage asynchronously
             trackUsageAsync(trackingData);
@@ -336,6 +407,12 @@ async function* patchedStreamMethod(params, options) {
             : undefined;
         // Extract usage information from all chunks
         const usage = extractUsageFromStream(chunks);
+        // Detect vision content
+        const hasVisionContent = detectVisionContent(params);
+        let reconstructedResponse = undefined;
+        if (shouldCapturePrompts(metadata)) {
+            reconstructedResponse = reconstructResponseFromChunks(chunks, params.model);
+        }
         // Create tracking data
         const trackingData = {
             requestId,
@@ -351,7 +428,9 @@ async function* patchedStreamMethod(params, options) {
             requestTime,
             responseTime,
             timeToFirstToken,
+            hasVisionContent,
             requestBody: params,
+            response: reconstructedResponse,
         };
         // Track usage asynchronously
         trackUsageAsync(trackingData);

package/dist/types/types.d.ts

@@ -47,7 +47,7 @@ export interface ReveniumConfig {
     printSummary?: boolean | SummaryFormat;
     /** Revenium team ID for fetching cost metrics from the API. If not provided, the summary will still be printed but without cost information. */
     teamId?: string;
-    /** Whether to capture prompts and responses for analysis (default: false) */
+    /** Whether to capture and send prompts to Revenium API (default: false) */
     capturePrompts?: boolean;
 }
 /**
@@ -60,17 +60,27 @@ export interface UsageMetadata {
     traceId?: string;
     /** Classification of AI operation (e.g., 'customer-support', 'content-generation', 'code-review') */
     taskType?: string;
-    /** Customer organization identifier for multi-tenant applications */
+    /** Customer organization name for multi-tenant applications (used for lookup/auto-creation) */
+    organizationName?: string;
+    /**
+     * @deprecated Use organizationName instead. This field will be removed in a future version.
+     * Customer organization identifier for multi-tenant applications
+     */
     organizationId?: string;
     /** Reference to billing plan or subscription tier */
     subscriptionId?: string;
-    /** Product or feature identifier that is using AI services */
+    /** Product or feature name that is using AI services (used for lookup/auto-creation) */
+    productName?: string;
+    /**
+     * @deprecated Use productName instead. This field will be removed in a future version.
+     * Product or feature identifier that is using AI services
+     */
     productId?: string;
     /** Agent or bot identifier for automated systems */
     agent?: string;
     /** Quality score of AI response (0.0-1.0) for performance tracking */
     responseQualityScore?: number;
-    /** Whether to capture prompts and responses for this request (overrides global config) */
+    /** Per-call override for prompt capture (overrides environment variable and config) */
     capturePrompts?: boolean;
     /** Allow additional custom fields for extensibility */
     [key: string]: unknown;
@@ -133,8 +143,12 @@ export interface TrackingData {
     responseTime: Date;
     /** Time to first token in milliseconds (for streaming responses) */
     timeToFirstToken?: number;
+    /** Whether the request contains vision/image content */
+    hasVisionContent?: boolean;
     /** Request body containing Anthropic message parameters */
     requestBody?: AnthropicMessageParams;
+    /** Response data from Anthropic API */
+    response?: AnthropicResponse;
 }
 /**
  * Internal payload structure for Revenium API
@@ -165,10 +179,10 @@ export interface ReveniumPayload {
     cacheReadTokenCount: number;
     /** Total token count (sum of all token types) */
     totalTokenCount: number;
-    /** Organization identifier for multi-tenant tracking */
-    organizationId?: string;
-    /** Product identifier */
-    productId?: string;
+    /** Organization name for multi-tenant tracking (used for lookup/auto-creation) */
+    organizationName?: string;
+    /** Product name (used for lookup/auto-creation) */
+    productName?: string;
     /** Subscriber information */
     subscriber?: Subscriber;
     /** Subscription identifier */
@@ -201,9 +215,14 @@ export interface ReveniumPayload {
     parentTransactionId?: string;
     transactionName?: string;
     region?: string;
+    hasVisionContent?: boolean;
     credentialAlias?: string;
     traceType?: string;
     traceName?: string;
+    systemPrompt?: string;
+    inputMessages?: string;
+    outputResponse?: string;
+    promptsTruncated?: boolean;
 }
 /**
  * Anthropic content block types for message validation

package/dist/types/utils/trace-fields.d.ts

@@ -4,6 +4,7 @@ export declare function getCredentialAlias(): string | null;
 export declare function getTraceType(): string | null;
 export declare function getTraceName(): string | null;
 export declare function detectOperationSubtype(requestBody?: any): string | null;
+export declare function detectVisionContent(params?: any): boolean;
 export declare function getParentTransactionId(): string | null;
 export declare function getTransactionName(): string | null;
 export declare function getRetryNumber(): number;

package/dist/types/utils/validation.d.ts

@@ -2,7 +2,7 @@
  * Validation utilities for Anthropic middleware
  * Provides type-safe validation with detailed error reporting
  */
-import { ReveniumConfig, UsageMetadata } from '../types';
+import { ReveniumConfig, UsageMetadata } from "../types";
 /**
  * Type guard for checking if a value is a non-null object
  */

package/examples/advanced.ts

@@ -17,8 +17,8 @@ const streamingWithMetadata = async () => {
   try {
     const metadata: UsageMetadata = {
       subscriber: { id: "user-123", email: "user@example.com" },
-      organizationId: "my-org",
-      productId: "my-product",
+      organizationName: "my-org",
+      productName: "my-product",
       taskType: "streaming-demo",
       traceId: `session_${Date.now()}`,
     };
@@ -101,7 +101,7 @@ const manualTracking = async () => {
       responseTime: now,
       metadata: {
         subscriber: { id: "manual-user" },
-        organizationId: "manual-org",
+        organizationName: "manual-org",
         taskType: "manual-tracking",
       },
     };
@@ -111,7 +111,7 @@ const manualTracking = async () => {
     console.log(
       `Tracked: ${
         trackingData.inputTokens + trackingData.outputTokens
-      } tokens\n`
+      } tokens\n`,
     );
 
     const status = getStatus();

package/examples/getting_started.ts

@@ -27,11 +27,11 @@ async function main() {
       },
 
       // Organization & billing
-      organizationId: 'my-customers-name',
+      organizationName: 'my-customers-name',
       subscriptionId: 'plan-enterprise-2024',
 
       // Product & task tracking
-      productId: 'my-product',
+      productName: 'my-product',
       taskType: 'doc-summary',
       agent: 'customer-support',
 

package/examples/metadata.ts

@@ -15,9 +15,9 @@ async function main() {
       },
       traceId: "trace-123",
       taskType: "task-123",
-      organizationId: "org-123",
+      organizationName: "AcmeCorp",
       subscriptionId: "sub-123",
-      productId: "prod-123",
+      productName: "ai-assistant",
       agent: "agent-123",
       responseQualityScore: 0.95,
     };
@@ -48,7 +48,7 @@ async function main() {
         : "Non-text response";
     console.log("RESPONSE: \n", textResponse);
     console.log(
-      `Tokens: ${response.usage?.input_tokens} input + ${response.usage?.output_tokens} output\n`
+      `Tokens: ${response.usage?.input_tokens} input + ${response.usage?.output_tokens} output\n`,
     );
   } catch (error) {
     console.error("Error: ", error);

package/examples/prompt-capture-example.ts

@@ -0,0 +1,105 @@
+import Anthropic from "@anthropic-ai/sdk";
+import { configure } from "../src/index";
+
+async function main() {
+  console.log("=== Anthropic Prompt Capture Example ===\n");
+
+  if (!process.env.ANTHROPIC_API_KEY) {
+    console.error("Error: ANTHROPIC_API_KEY environment variable is required");
+    console.error("Please set it before running this example:");
+    console.error("  export ANTHROPIC_API_KEY=your_api_key_here");
+    process.exit(1);
+  }
+
+  const config = {
+    reveniumApiKey: process.env.REVENIUM_METERING_API_KEY || "hak_test_key",
+    reveniumBaseUrl:
+      process.env.REVENIUM_METERING_BASE_URL || "https://api.revenium.ai",
+    anthropicApiKey: process.env.ANTHROPIC_API_KEY,
+    capturePrompts: true,
+  };
+
+  configure(config);
+
+  const anthropic = new Anthropic({
+    apiKey: config.anthropicApiKey,
+  });
+
+  console.log("Example 1: Prompt capture enabled via config");
+  console.log("Making request with prompt capture enabled...\n");
+
+  try {
+    const response = await anthropic.messages.create({
+      model: "claude-3-5-sonnet-20241022",
+      max_tokens: 100,
+      system: "You are a helpful assistant that provides concise answers.",
+      messages: [
+        {
+          role: "user",
+          content: "What is the capital of France?",
+        },
+      ],
+      usageMetadata: {
+        organizationName: "org-prompt-capture-demo",
+        productName: "prod-anthropic-prompt-capture",
+      },
+    });
+
+    console.log(
+      "Response:",
+      response.content[0].type === "text" ? response.content[0].text : "",
+    );
+    console.log("\nPrompts captured and sent to Revenium API!");
+  } catch (error) {
+    console.error(
+      "Error:",
+      error instanceof Error ? error.message : String(error),
+    );
+  }
+
+  console.log("\n" + "=".repeat(50) + "\n");
+
+  console.log("Example 2: Per-call override to disable prompt capture");
+  console.log("Making request with prompt capture disabled via metadata...\n");
+
+  try {
+    const response2 = await anthropic.messages.create({
+      model: "claude-3-5-sonnet-20241022",
+      max_tokens: 100,
+      system: "You are a helpful assistant.",
+      messages: [
+        {
+          role: "user",
+          content: "What is 2+2?",
+        },
+      ],
+      usageMetadata: {
+        organizationName: "org-prompt-capture-demo",
+        productName: "prod-anthropic-prompt-capture",
+        capturePrompts: false,
+      },
+    });
+
+    console.log(
+      "Response:",
+      response2.content[0].type === "text" ? response2.content[0].text : "",
+    );
+    console.log("\nPrompts NOT captured (overridden via metadata)!");
+  } catch (error) {
+    console.error(
+      "Error:",
+      error instanceof Error ? error.message : String(error),
+    );
+  }
+
+  console.log("\n" + "=".repeat(50) + "\n");
+  console.log("Examples completed!");
+  console.log(
+    "\nNote: Set REVENIUM_CAPTURE_PROMPTS=true in .env to enable globally",
+  );
+  console.log(
+    "Or use capturePrompts: true in config or metadata for per-call control",
+  );
+}
+
+main().catch(console.error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@revenium/anthropic",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "description": "Transparent TypeScript middleware for automatic Revenium usage tracking with Anthropic Claude AI",
   "main": "dist/cjs/index.js",
   "module": "dist/esm/index.js",