@revenium/perplexity 2.0.8 → 2.1.0

package/.env.example

@@ -6,5 +6,22 @@ REVENIUM_METERING_BASE_URL=https://api.revenium.ai
 PERPLEXITY_API_KEY=pplx_your_perplexity_api_key_here
 PERPLEXITY_API_BASE_URL=https://api.perplexity.ai
 
+# Trace Visualization Fields (Optional)
+REVENIUM_ENVIRONMENT=production
+REVENIUM_REGION=us-east-1
+REVENIUM_CREDENTIAL_ALIAS=Perplexity Production Key
+REVENIUM_TRACE_TYPE=customer_support
+REVENIUM_TRACE_NAME=Support Ticket #12345
+REVENIUM_PARENT_TRANSACTION_ID=parent-txn-123
+REVENIUM_TRANSACTION_NAME=Answer Customer Question
+REVENIUM_RETRY_NUMBER=0
+
 # Debug Configuration (Optional)
-REVENIUM_DEBUG=false                        # Set to true to enable debug logging
+REVENIUM_DEBUG=false
+
+# Prompt Capture Configuration (Optional)
+# REVENIUM_CAPTURE_PROMPTS=false
+# REVENIUM_MAX_PROMPT_SIZE=50000
+
+# AWS Region Detection (Optional - auto-detected if not set)
+AWS_REGION=us-east-1

package/CHANGELOG.md

@@ -5,6 +5,48 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [2.1.0] - 2026-01-21
+
+### Added
+
+- Prompt capture functionality with automatic credential sanitization
+- New prompt capture fields in ReveniumPayload: systemPrompt, inputMessages, outputResponse, promptsTruncated
+- Configuration options: capturePrompts (boolean), maxPromptSize (number, default: 50000)
+- Environment variables: REVENIUM_CAPTURE_PROMPTS, REVENIUM_MAX_PROMPT_SIZE
+- Per-call override via usageMetadata.capturePrompts
+- Comprehensive credential sanitization with 13 patterns:
+  - Perplexity API keys (pplx-\*)
+  - OpenAI keys (sk-_, sk-proj-_, sk-ant-\*)
+  - AWS access keys (AKIA\*)
+  - GitHub tokens (ghp*\*, ghs*\*)
+  - JWT tokens (eyJ*.eyJ*.\*)
+  - Bearer tokens
+  - Generic API keys, tokens, passwords, secrets
+- Automatic truncation of prompts exceeding maxPromptSize with truncation flag
+
+### Security
+
+- All captured prompts are automatically sanitized to remove sensitive credentials
+- Credential patterns are redacted before transmission to Revenium API
+- Prompt capture is opt-in (disabled by default) for privacy
+
+## [2.0.9] - 2026-01-06
+
+### Added
+
+- Trace visualization fields support for distributed tracing and analytics
+- New trace fields: environment, region, credentialAlias, traceType, traceName, parentTransactionId, transactionName, retryNumber, operationSubtype
+- Automatic region detection from AWS/Azure/GCP environment variables with fallback to AWS EC2 metadata service
+- Comprehensive unit tests for all trace field functions
+- Jest testing infrastructure with full test coverage
+- Updated .env.example with trace visualization field examples
+- Documentation for trace visualization fields in README.md
+
+### Changed
+
+- buildPayload function is now async to support region detection
+- Updated ReveniumPayload interface with trace visualization fields
+
 ## [2.0.8] - 2025-11-14
 
 ### Changed

package/README.md

@@ -84,6 +84,10 @@ REVENIUM_METERING_API_KEY=hak_your_revenium_api_key
 
 # Optional: Enable debug logging
 # REVENIUM_DEBUG=false
+
+# Optional: Terminal cost/metrics summary
+# REVENIUM_PRINT_SUMMARY=true  # or 'human' or 'json'
+# REVENIUM_TEAM_ID=your_team_id  # Required for cost retrieval
 ```
 
 **Replace the placeholder values with your actual keys!**
@@ -200,53 +204,235 @@ The middleware automatically captures comprehensive usage data:
 
 The following table shows all fields this middleware sends to the Revenium API:
 
-| Field | Type | Required | Description |
-|-------|------|----------|-------------|
-| **Core Fields** | | | |
-| `model` | string | Yes | Perplexity model name (e.g., "sonar-pro") |
-| `provider` | string | Yes | Always "Perplexity" |
-| `inputTokenCount` | integer | Yes | Number of input tokens consumed |
-| `outputTokenCount` | integer | Yes | Number of output tokens generated |
-| `totalTokenCount` | integer | Yes | Total tokens (input + output) |
-| `requestDuration` | integer | Yes | Request duration in milliseconds |
-| **Timing** | | | |
-| `requestTime` | string | Auto | ISO 8601 timestamp when request started |
-| `responseTime` | string | Auto | ISO 8601 timestamp when response completed |
-| `completionStartTime` | string | Auto | ISO 8601 timestamp when completion started |
-| `timeToFirstToken` | integer | Streaming | Time to first token in ms (streaming only) |
-| **Request Details** | | | |
-| `transactionId` | string | Auto | Unique transaction identifier |
-| `operationType` | string | Auto | Always "CHAT" for chat completions |
-| `stopReason` | string | Auto | Completion finish reason ("END", "STOP", etc.) |
-| `isStreamed` | boolean | Auto | Whether response was streamed |
-| `costType` | string | Auto | Always "AI" |
-| `modelSource` | string | Auto | Always "PERPLEXITY" |
-| `middlewareSource` | string | Auto | Always "revenium-perplexity-node" |
-| **Cost Information** | | | |
-| `inputTokenCost` | number | Optional | Cost for input tokens (if provided by Perplexity) |
-| `outputTokenCost` | number | Optional | Cost for output tokens (if provided by Perplexity) |
-| `totalCost` | number | Optional | Total cost (if provided by Perplexity) |
-| **Business Context** | | | |
-| `organizationId` | string | Optional | Your organization identifier |
-| `productId` | string | Optional | Your product identifier |
-| `subscriptionId` | string | Optional | Your subscription identifier |
-| `taskType` | string | Optional | Type of AI task (e.g., "chat", "research") |
-| `traceId` | string | Optional | Session or conversation tracking ID |
-| `agent` | string | Optional | AI agent or bot identifier |
-| `responseQualityScore` | number | Optional | Custom quality rating (0.0-1.0) |
-| `subscriber.id` | string | Optional | User identifier |
-| `subscriber.email` | string | Optional | User email address |
-| `subscriber.credential` | object | Optional | Authentication credential (name, value) |
+| Field                   | Type    | Required  | Description                                        |
+| ----------------------- | ------- | --------- | -------------------------------------------------- |
+| **Core Fields**         |         |           |                                                    |
+| `model`                 | string  | Yes       | Perplexity model name (e.g., "sonar-pro")          |
+| `provider`              | string  | Yes       | Always "Perplexity"                                |
+| `inputTokenCount`       | integer | Yes       | Number of input tokens consumed                    |
+| `outputTokenCount`      | integer | Yes       | Number of output tokens generated                  |
+| `totalTokenCount`       | integer | Yes       | Total tokens (input + output)                      |
+| `requestDuration`       | integer | Yes       | Request duration in milliseconds                   |
+| **Timing**              |         |           |                                                    |
+| `requestTime`           | string  | Auto      | ISO 8601 timestamp when request started            |
+| `responseTime`          | string  | Auto      | ISO 8601 timestamp when response completed         |
+| `completionStartTime`   | string  | Auto      | ISO 8601 timestamp when completion started         |
+| `timeToFirstToken`      | integer | Streaming | Time to first token in ms (streaming only)         |
+| **Request Details**     |         |           |                                                    |
+| `transactionId`         | string  | Auto      | Unique transaction identifier                      |
+| `operationType`         | string  | Auto      | Always "CHAT" for chat completions                 |
+| `stopReason`            | string  | Auto      | Completion finish reason ("END", "STOP", etc.)     |
+| `isStreamed`            | boolean | Auto      | Whether response was streamed                      |
+| `costType`              | string  | Auto      | Always "AI"                                        |
+| `modelSource`           | string  | Auto      | Always "PERPLEXITY"                                |
+| `middlewareSource`      | string  | Auto      | Always "revenium-perplexity-node"                  |
+| **Cost Information**    |         |           |                                                    |
+| `inputTokenCost`        | number  | Optional  | Cost for input tokens (if provided by Perplexity)  |
+| `outputTokenCost`       | number  | Optional  | Cost for output tokens (if provided by Perplexity) |
+| `totalCost`             | number  | Optional  | Total cost (if provided by Perplexity)             |
+| **Business Context**    |         |           |                                                    |
+| `organizationId`        | string  | Optional  | Your organization identifier                       |
+| `productId`             | string  | Optional  | Your product identifier                            |
+| `subscriptionId`        | string  | Optional  | Your subscription identifier                       |
+| `taskType`              | string  | Optional  | Type of AI task (e.g., "chat", "research")         |
+| `traceId`               | string  | Optional  | Session or conversation tracking ID                |
+| `agent`                 | string  | Optional  | AI agent or bot identifier                         |
+| `responseQualityScore`  | number  | Optional  | Custom quality rating (0.0-1.0)                    |
+| `subscriber.id`         | string  | Optional  | User identifier                                    |
+| `subscriber.email`      | string  | Optional  | User email address                                 |
+| `subscriber.credential` | object  | Optional  | Authentication credential (name, value)            |
 
 **Notes:**
+
 - **Required** fields are always sent with every request
 - **Auto** fields are automatically populated by the middleware
 - **Optional** fields are only sent if you provide them via `usageMetadata`
 - **Streaming** fields are only sent for streaming requests
 
 **Reference:**
+
 - [API Reference](https://revenium.readme.io/reference/meter_ai_completion) - Complete metadata field documentation
 
+## Trace Visualization Fields
+
+The middleware automatically captures trace visualization fields for distributed tracing and analytics:
+
+| Field                 | Type   | Description                                                                     | Environment Variable               |
+| --------------------- | ------ | ------------------------------------------------------------------------------- | ---------------------------------- |
+| `environment`         | string | Deployment environment (production, staging, development)                       | `REVENIUM_ENVIRONMENT`, `NODE_ENV` |
+| `operationType`       | string | Operation classification (CHAT, EMBED, etc.) - automatically detected           | N/A (auto-detected)                |
+| `operationSubtype`    | string | Additional detail (function_call, etc.) - automatically detected                | N/A (auto-detected)                |
+| `retryNumber`         | number | Retry attempt number (0 for first attempt, 1+ for retries)                      | `REVENIUM_RETRY_NUMBER`            |
+| `parentTransactionId` | string | Parent transaction reference for distributed tracing                            | `REVENIUM_PARENT_TRANSACTION_ID`   |
+| `transactionName`     | string | Human-friendly operation label                                                  | `REVENIUM_TRANSACTION_NAME`        |
+| `region`              | string | Cloud region (us-east-1, etc.) - auto-detected from AWS/Azure/GCP               | `AWS_REGION`, `REVENIUM_REGION`    |
+| `credentialAlias`     | string | Human-readable credential name                                                  | `REVENIUM_CREDENTIAL_ALIAS`        |
+| `traceType`           | string | Categorical identifier (alphanumeric, hyphens, underscores only, max 128 chars) | `REVENIUM_TRACE_TYPE`              |
+| `traceName`           | string | Human-readable label for trace instances (max 256 chars)                        | `REVENIUM_TRACE_NAME`              |
+
+**All trace visualization fields are optional.** The middleware will automatically detect and populate these fields when possible.
+
+### Example Configuration
+
+```env
+REVENIUM_ENVIRONMENT=production
+REVENIUM_REGION=us-east-1
+REVENIUM_CREDENTIAL_ALIAS=Perplexity Production Key
+REVENIUM_TRACE_TYPE=customer_support
+REVENIUM_TRACE_NAME=Support Ticket #12345
+REVENIUM_PARENT_TRANSACTION_ID=parent-txn-123
+REVENIUM_TRANSACTION_NAME=Answer Customer Question
+REVENIUM_RETRY_NUMBER=0
+```
+
+For a complete example, see [`.env.example`](https://github.com/revenium/revenium-middleware-perplexity-node/blob/HEAD/.env.example).
+
+## Prompt Capture
+
+The middleware can capture prompts and responses for analysis. This feature is **disabled by default** for privacy.
+
+### Configuration
+
+Enable prompt capture using environment variables, programmatic configuration, or per-call metadata:
+
+**Environment Variables:**
+
+```bash
+export REVENIUM_CAPTURE_PROMPTS=true
+export REVENIUM_MAX_PROMPT_SIZE=50000
+```
+
+**Programmatic Configuration:**
+
+```typescript
+import { Initialize } from "@revenium/perplexity";
+
+Initialize({
+  reveniumApiKey: "hak_your_key",
+  reveniumBaseUrl: "https://api.revenium.ai",
+  perplexityApiKey: "pplx_your_key",
+  capturePrompts: true,
+  maxPromptSize: 50000,
+});
+```
+
+**Per-Call Override:**
+
+```typescript
+const response = await client.chat.completions.create(
+  {
+    model: "llama-3.1-sonar-small-128k-online",
+    messages: [{ role: "user", content: "Hello" }],
+  },
+  {
+    usageMetadata: { capturePrompts: true },
+  },
+);
+```
+
+### Configuration Priority
+
+The middleware uses the following priority order (highest to lowest):
+
+1. Per-call `usageMetadata.capturePrompts`
+2. Programmatic `config.capturePrompts`
+3. Environment variable `REVENIUM_CAPTURE_PROMPTS`
+4. Default: `false`
+
+### Security
+
+All captured prompts are automatically sanitized to remove sensitive credentials:
+
+- Perplexity API keys (pplx-\*)
+- OpenAI keys (sk-\*, sk-proj-\*, sk-ant-\*)
+- AWS access keys (AKIA\*)
+- GitHub tokens (ghp*\*, ghs*\*)
+- JWT tokens
+- Bearer tokens
+- Generic API keys, tokens, passwords, secrets
+
+Prompts exceeding `maxPromptSize` (default: 50000 characters) are automatically truncated with a flag indicating truncation.
+
+## Terminal Cost/Metrics Summary
+
+The middleware can print a cost and metrics summary to your terminal after each API request. This is useful for development and debugging.
+
+### Configuration
+
+Enable terminal summary output using environment variables or programmatic configuration:
+
+**Environment Variables:**
+
+```bash
+# Enable human-readable summary (default format)
+export REVENIUM_PRINT_SUMMARY=true
+
+# Or specify format explicitly
+export REVENIUM_PRINT_SUMMARY=human  # Human-readable format
+export REVENIUM_PRINT_SUMMARY=json   # JSON format for log parsing
+
+# Optional: Set team ID to fetch cost data
+export REVENIUM_TEAM_ID=your_team_id
+```
+
+**Programmatic Configuration:**
+
+```typescript
+import { Initialize } from "@revenium/perplexity";
+
+Initialize({
+  reveniumApiKey: "hak_your_api_key",
+  reveniumBaseUrl: "https://api.revenium.ai",
+  perplexityApiKey: "pplx_your_api_key",
+  printSummary: true, // or 'human' or 'json'
+  teamId: "your_team_id", // Optional: for cost retrieval
+});
+```
+
+### Output Formats
+
+**Human-readable format** (`printSummary: true` or `printSummary: 'human'`):
+
+```
+============================================================
+📊 REVENIUM USAGE SUMMARY
+============================================================
+🤖 Model: sonar-pro
+🏢 Provider: Perplexity
+⏱️  Duration: 1.23s
+
+💬 Token Usage:
+   📥 Input Tokens:  150
+   📤 Output Tokens: 75
+   📊 Total Tokens:  225
+
+💰 Cost: $0.004500
+🔖 Trace ID: trace-abc-123
+============================================================
+```
+
+**JSON format** (`printSummary: 'json'`):
+
+```json
+{
+  "model": "sonar-pro",
+  "provider": "Perplexity",
+  "durationSeconds": 1.23,
+  "inputTokenCount": 150,
+  "outputTokenCount": 75,
+  "totalTokenCount": 225,
+  "cost": 0.0045,
+  "traceId": "trace-abc-123"
+}
+```
+
+### Cost Retrieval
+
+- **Without `teamId`**: Shows token counts and duration, displays hint to set `REVENIUM_TEAM_ID`
+- **With `teamId`**: Fetches actual cost from Revenium API with automatic retry logic
+- **Cost pending**: Shows "(pending aggregation)" if cost data isn't available yet
+- **Fire-and-forget**: Never blocks your application, even if cost fetch fails
+
 ## API Overview
 
 - **`Initialize(config?)`** - Initialize the middleware (from environment or explicit config)

package/dist/cjs/core/config/loader.js

@@ -33,16 +33,37 @@ function loadConfigFromEnv() {
     const perplexityApiKey = process.env.PERPLEXITY_API_KEY;
     const perplexityBaseUrl = process.env.PERPLEXITY_API_BASE_URL || DEFAULT_PERPLEXITY_BASE_URL;
     const debug = process.env.REVENIUM_DEBUG === "true";
+    let printSummary = undefined;
+    const printSummaryEnv = process.env.REVENIUM_PRINT_SUMMARY;
+    if (printSummaryEnv) {
+        if (printSummaryEnv === "true") {
+            printSummary = true;
+        }
+        else if (printSummaryEnv === "false") {
+            printSummary = false;
+        }
+        else if (printSummaryEnv === "human" || printSummaryEnv === "json") {
+            printSummary = printSummaryEnv;
+        }
+    }
+    const teamId = process.env.REVENIUM_TEAM_ID;
+    const capturePromptsEnv = process.env.REVENIUM_CAPTURE_PROMPTS;
     if (!reveniumApiKey)
         return null;
     if (!perplexityApiKey)
         return null;
-    return {
+    const config = {
         reveniumApiKey,
         reveniumBaseUrl,
         perplexityApiKey,
         perplexityBaseUrl,
         debug,
+        printSummary,
+        teamId,
     };
+    if (capturePromptsEnv !== undefined) {
+        config.capturePrompts = capturePromptsEnv.toLowerCase() === "true";
+    }
+    return config;
 }
 //# sourceMappingURL=loader.js.map

package/dist/cjs/core/config/loader.js.map

@@ -1 +1 @@
-{"version":3,"file":"loader.js","sourceRoot":"","sources":["../../../../src/core/config/loader.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;AAoBH,8CA2BC;AA5CD,mCAA8C;AAE9C;;GAEG;AACH,MAAM,yBAAyB,GAAG,yBAAyB,CAAC;AAC5D,MAAM,2BAA2B,GAAG,2BAA2B,CAAC;AAEhE;;GAEG;AACH,IAAI,aAAa,GAAG,KAAK,CAAC;AAE1B;;;GAGG;AACH,SAAgB,iBAAiB;IAC/B,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,IAAA,eAAU,GAAE,CAAC;QACb,aAAa,GAAG,IAAI,CAAC;IACvB,CAAC;IAED,MAAM,cAAc,GAClB,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IACxE,MAAM,eAAe,GACnB,OAAO,CAAC,GAAG,CAAC,0BAA0B;QACtC,OAAO,CAAC,GAAG,CAAC,iBAAiB;QAC7B,yBAAyB,CAAC;IAC5B,MAAM,gBAAgB,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IACxD,MAAM,iBAAiB,GACrB,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,2BAA2B,CAAC;IACrE,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,KAAK,MAAM,CAAC;IAEpD,IAAI,CAAC,cAAc;QAAE,OAAO,IAAI,CAAC;IACjC,IAAI,CAAC,gBAAgB;QAAE,OAAO,IAAI,CAAC;IAEnC,OAAO;QACL,cAAc;QACd,eAAe;QACf,gBAAgB;QAChB,iBAAiB;QACjB,KAAK;KACN,CAAC;AACJ,CAAC"}
+{"version":3,"file":"loader.js","sourceRoot":"","sources":["../../../../src/core/config/loader.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;AAoBH,8CAkDC;AAnED,mCAA8C;AAE9C;;GAEG;AACH,MAAM,yBAAyB,GAAG,yBAAyB,CAAC;AAC5D,MAAM,2BAA2B,GAAG,2BAA2B,CAAC;AAEhE;;GAEG;AACH,IAAI,aAAa,GAAG,KAAK,CAAC;AAE1B;;;GAGG;AACH,SAAgB,iBAAiB;IAC/B,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,IAAA,eAAU,GAAE,CAAC;QACb,aAAa,GAAG,IAAI,CAAC;IACvB,CAAC;IAED,MAAM,cAAc,GAClB,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IACxE,MAAM,eAAe,GACnB,OAAO,CAAC,GAAG,CAAC,0BAA0B;QACtC,OAAO,CAAC,GAAG,CAAC,iBAAiB;QAC7B,yBAAyB,CAAC;IAC5B,MAAM,gBAAgB,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IACxD,MAAM,iBAAiB,GACrB,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,2BAA2B,CAAC;IACrE,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,KAAK,MAAM,CAAC;IAEpD,IAAI,YAAY,GAA2C,SAAS,CAAC;IACrE,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC;IAC3D,IAAI,eAAe,EAAE,CAAC;QACpB,IAAI,eAAe,KAAK,MAAM,EAAE,CAAC;YAC/B,YAAY,GAAG,IAAI,CAAC;QACtB,CAAC;aAAM,IAAI,eAAe,KAAK,OAAO,EAAE,CAAC;YACvC,YAAY,GAAG,KAAK,CAAC;QACvB,CAAC;aAAM,IAAI,eAAe,KAAK,OAAO,IAAI,eAAe,KAAK,MAAM,EAAE,CAAC;YACrE,YAAY,GAAG,eAAe,CAAC;QACjC,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC5C,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC;IAE/D,IAAI,CAAC,cAAc;QAAE,OAAO,IAAI,CAAC;IACjC,IAAI,CAAC,gBAAgB;QAAE,OAAO,IAAI,CAAC;IAEnC,MAAM,MAAM,GAAQ;QAClB,cAAc;QACd,eAAe;QACf,gBAAgB;QAChB,iBAAiB;QACjB,KAAK;QACL,YAAY;QACZ,MAAM;KACP,CAAC;IAEF,IAAI,iBAAiB,KAAK,SAAS,EAAE,CAAC;QACpC,MAAM,CAAC,cAAc,GAAG,iBAAiB,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC;IACrE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/cjs/core/config/manager.js

@@ -12,6 +12,7 @@ exports.getLogger = getLogger;
 exports.initializeConfig = initializeConfig;
 const loader_js_1 = require("./loader.js");
 const validator_js_1 = require("./validator.js");
+const summary_printer_js_1 = require("../../utils/summary-printer.js");
 /**
  * Global configuration instance
  */
@@ -52,6 +53,14 @@ function setConfig(config) {
         baseUrl: config.reveniumBaseUrl,
         hasReveniumKey: !!config.reveniumApiKey,
         hasPerplexityKey: !!config.perplexityApiKey,
+        printSummary: config.printSummary,
+        teamId: config.teamId,
+    });
+    (0, summary_printer_js_1.setConfig)({
+        reveniumApiKey: config.reveniumApiKey,
+        reveniumBaseUrl: config.reveniumBaseUrl,
+        teamId: config.teamId,
+        printSummary: config.printSummary,
     });
 }
 /**

package/dist/cjs/core/config/manager.js.map

@@ -1 +1 @@
-{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../../../src/core/config/manager.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAoCH,8BAEC;AAKD,8BAQC;AAKD,8BAEC;AAKD,4CAkBC;AA9ED,2CAAgD;AAChD,iDAAgD;AAEhD;;GAEG;AACH,IAAI,YAAY,GAA0B,IAAI,CAAC;AAE/C;;GAEG;AACU,QAAA,aAAa,GAAW;IACnC,KAAK,EAAE,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QAC7C,IAAI,YAAY,EAAE,KAAK,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,KAAK,MAAM,EAAE,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,oBAAoB,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IACD,IAAI,EAAE,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QAC5C,OAAO,CAAC,IAAI,CAAC,cAAc,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACjD,CAAC;IACD,IAAI,EAAE,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QAC5C,OAAO,CAAC,IAAI,CAAC,sBAAsB,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACzD,CAAC;IACD,KAAK,EAAE,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QAC7C,OAAO,CAAC,KAAK,CAAC,oBAAoB,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACxD,CAAC;CACF,CAAC;AAEF,IAAI,YAAY,GAAW,qBAAa,CAAC;AAEzC;;GAEG;AACH,SAAgB,SAAS;IACvB,OAAO,YAAY,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,SAAgB,SAAS,CAAC,MAAsB;IAC9C,IAAA,6BAAc,EAAC,MAAM,CAAC,CAAC;IACvB,YAAY,GAAG,MAAM,CAAC;IACtB,YAAY,CAAC,KAAK,CAAC,gCAAgC,EAAE;QACnD,OAAO,EAAE,MAAM,CAAC,eAAe;QAC/B,cAAc,EAAE,CAAC,CAAC,MAAM,CAAC,cAAc;QACvC,gBAAgB,EAAE,CAAC,CAAC,MAAM,CAAC,gBAAgB;KAC5C,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAgB,SAAS;IACvB,OAAO,YAAY,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,SAAgB,gBAAgB;IAC9B,MAAM,SAAS,GAAG,IAAA,6BAAiB,GAAE,CAAC;IAEtC,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,YAAY,CAAC,KAAK,CAChB,iHAAiH,CAClH,CAAC;QACF,MAAM,IAAI,KAAK,CACb,iHAAiH,CAClH,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,SAAS,CAAC,CAAC;IACrB,YAAY,CAAC,KAAK,CAChB,4DAA4D,CAC7D,CAAC;IAEF,OAAO,SAAS,CAAC;AACnB,CAAC"}
+{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../../../src/core/config/manager.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAqCH,8BAEC;AAKD,8BAiBC;AAKD,8BAEC;AAKD,4CAkBC;AAxFD,2CAAgD;AAChD,iDAAgD;AAChD,uEAAsF;AAEtF;;GAEG;AACH,IAAI,YAAY,GAA0B,IAAI,CAAC;AAE/C;;GAEG;AACU,QAAA,aAAa,GAAW;IACnC,KAAK,EAAE,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QAC7C,IAAI,YAAY,EAAE,KAAK,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,KAAK,MAAM,EAAE,CAAC;YACjE,OAAO,CAAC,KAAK,CAAC,oBAAoB,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IACD,IAAI,EAAE,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QAC5C,OAAO,CAAC,IAAI,CAAC,cAAc,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACjD,CAAC;IACD,IAAI,EAAE,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QAC5C,OAAO,CAAC,IAAI,CAAC,sBAAsB,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACzD,CAAC;IACD,KAAK,EAAE,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QAC7C,OAAO,CAAC,KAAK,CAAC,oBAAoB,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACxD,CAAC;CACF,CAAC;AAEF,IAAI,YAAY,GAAW,qBAAa,CAAC;AAEzC;;GAEG;AACH,SAAgB,SAAS;IACvB,OAAO,YAAY,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,SAAgB,SAAS,CAAC,MAAsB;IAC9C,IAAA,6BAAc,EAAC,MAAM,CAAC,CAAC;IACvB,YAAY,GAAG,MAAM,CAAC;IACtB,YAAY,CAAC,KAAK,CAAC,gCAAgC,EAAE;QACnD,OAAO,EAAE,MAAM,CAAC,eAAe;QAC/B,cAAc,EAAE,CAAC,CAAC,MAAM,CAAC,cAAc;QACvC,gBAAgB,EAAE,CAAC,CAAC,MAAM,CAAC,gBAAgB;QAC3C,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC,CAAC;IAEH,IAAA,8BAAuB,EAAC;QACtB,cAAc,EAAE,MAAM,CAAC,cAAc;QACrC,eAAe,EAAE,MAAM,CAAC,eAAe;QACvC,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,YAAY,EAAE,MAAM,CAAC,YAAY;KAClC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAgB,SAAS;IACvB,OAAO,YAAY,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,SAAgB,gBAAgB;IAC9B,MAAM,SAAS,GAAG,IAAA,6BAAiB,GAAE,CAAC;IAEtC,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,YAAY,CAAC,KAAK,CAChB,iHAAiH,CAClH,CAAC;QACF,MAAM,IAAI,KAAK,CACb,iHAAiH,CAClH,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,SAAS,CAAC,CAAC;IACrB,YAAY,CAAC,KAAK,CAChB,4DAA4D,CAC7D,CAAC;IAEF,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/cjs/core/middleware/interfaces.js

@@ -8,6 +8,7 @@ const config_1 = require("../config");
 const transaction_id_js_1 = require("../../utils/transaction-id.js");
 const index_js_1 = require("../tracking/index.js");
 const streaming_wrapper_js_1 = require("./streaming-wrapper.js");
+const prompt_extraction_js_1 = require("../../utils/prompt-extraction.js");
 const logger = (0, config_1.getLogger)();
 /**
  * Chat interface - provides access to chat completions
@@ -48,6 +49,9 @@ class CompletionsInterface {
             if ("usage" in response && response.usage) {
                 // Extract cost if available (Perplexity-specific)
                 const usage = response.usage;
+                const responseContent = "choices" in response
+                    ? response.choices[0]?.message?.content || undefined
+                    : undefined;
                 (0, index_js_1.trackUsageAsync)({
                     requestId: ("id" in response ? response.id : null) || transactionId,
                     model: params.model,
@@ -61,12 +65,31 @@ class CompletionsInterface {
                     usageMetadata: metadata,
                     isStreamed: false,
                     cost: usage.cost,
+                    responseFormat: params.response_format,
+                    messages: params.messages,
+                    responseContent: responseContent
+                        ? (0, prompt_extraction_js_1.sanitizeCredentials)(responseContent)
+                        : undefined,
                 });
             }
             return response;
         }
         catch (error) {
             logger.error(`[Revenium] Error in chat completion: ${error.message}`);
+            const endTime = Date.now();
+            const duration = endTime - startTime.getTime();
+            (0, index_js_1.trackUsageAsync)({
+                requestId: transactionId,
+                model: params.model,
+                promptTokens: 0,
+                completionTokens: 0,
+                totalTokens: 0,
+                duration,
+                finishReason: "error",
+                usageMetadata: metadata,
+                isStreamed: false,
+                messages: params.messages,
+            });
             throw error;
         }
     }
@@ -82,10 +105,24 @@ class CompletionsInterface {
                 ...params,
                 stream: true,
             });
-            return new streaming_wrapper_js_1.StreamingWrapper(stream, params.model, startTime, transactionId, metadata, this.config);
+            return new streaming_wrapper_js_1.StreamingWrapper(stream, params.model, startTime, transactionId, metadata, this.config, params.response_format, params.messages);
         }
         catch (error) {
             logger.error(`[Revenium] Error in streaming chat completion: ${error.message}`);
+            const endTime = Date.now();
+            const duration = endTime - startTime.getTime();
+            (0, index_js_1.trackUsageAsync)({
+                requestId: transactionId,
+                model: params.model,
+                promptTokens: 0,
+                completionTokens: 0,
+                totalTokens: 0,
+                duration,
+                finishReason: "error",
+                usageMetadata: metadata,
+                isStreamed: true,
+                messages: params.messages,
+            });
             throw error;
         }
     }

package/dist/cjs/core/middleware/interfaces.js.map

@@ -1 +1 @@
-{"version":3,"file":"interfaces.js","sourceRoot":"","sources":["../../../../src/core/middleware/interfaces.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAIH,sCAAsC;AACtC,qEAAsE;AACtE,mDAAuD;AACvD,iEAA0D;AAE1D,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;AAE3B;;GAEG;AACH,MAAa,aAAa;IACxB,YAAoB,MAAc,EAAU,MAAW;QAAnC,WAAM,GAAN,MAAM,CAAQ;QAAU,WAAM,GAAN,MAAM,CAAK;IAAG,CAAC;IAE3D;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IAC5D,CAAC;CACF;AATD,sCASC;AAED;;GAEG;AACH,MAAa,oBAAoB;IAC/B,YAAoB,MAAc,EAAU,MAAW;QAAnC,WAAM,GAAN,MAAM,CAAQ;QAAU,WAAM,GAAN,MAAM,CAAK;IAAG,CAAC;IAE3D;;OAEG;IACH,KAAK,CAAC,MAAM,CACV,MAA8C,EAC9C,QAAwB;QAExB,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,MAAM,aAAa,GAAG,IAAA,yCAAqB,GAAE,CAAC;QAE9C,IAAI,CAAC;YACH,MAAM,CAAC,KAAK,CACV,mDAAmD,MAAM,CAAC,KAAK,EAAE,CAClE,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAEnE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,OAAO,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;YAE/C,2EAA2E;YAC3E,IAAI,OAAO,IAAI,QAAQ,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAC1C,kDAAkD;gBAClD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAY,CAAC;gBAEpC,IAAA,0BAAe,EAAC;oBACd,SAAS,EAAE,CAAC,IAAI,IAAI,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,aAAa;oBACnE,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;oBAC1C,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,iBAAiB;oBAClD,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;oBACxC,QAAQ;oBACR,YAAY,EACV,SAAS,IAAI,QAAQ;wBACnB,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,IAAI;wBAC5C,CAAC,CAAC,IAAI;oBACV,aAAa,EAAE,QAAQ;oBACvB,UAAU,EAAE,KAAK;oBACjB,IAAI,EAAE,KAAK,CAAC,IAAI;iBACjB,CAAC,CAAC;YACL,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,CAAC,wCAAwC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACtE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CACnB,MAA8C,EAC9C,QAAwB;QAExB,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,MAAM,aAAa,GAAG,IAAA,yCAAqB,GAAE,CAAC;QAE9C,IAAI,CAAC;YACH,MAAM,CAAC,KAAK,CACV,6DAA6D,MAAM,CAAC,KAAK,EAAE,CAC5E,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACvD,GAAG,MAAM;gBACT,MAAM,EAAE,IAAI;aACb,CAAC,CAAC;YAEH,OAAO,IAAI,uCAAgB,CACzB,MAAa,EACb,MAAM,CAAC,KAAK,EACZ,SAAS,EACT,aAAa,EACb,QAAQ,EACR,IAAI,CAAC,MAAM,CACZ,CAAC;QACJ,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,CACV,kDAAkD,KAAK,CAAC,OAAO,EAAE,CAClE,CAAC;YACF,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAvFD,oDAuFC"}
+{"version":3,"file":"interfaces.js","sourceRoot":"","sources":["../../../../src/core/middleware/interfaces.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAIH,sCAAsC;AACtC,qEAAsE;AACtE,mDAAuD;AACvD,iEAA0D;AAC1D,2EAAuE;AAEvE,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;AAE3B;;GAEG;AACH,MAAa,aAAa;IACxB,YACU,MAAc,EACd,MAAW;QADX,WAAM,GAAN,MAAM,CAAQ;QACd,WAAM,GAAN,MAAM,CAAK;IAClB,CAAC;IAEJ;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IAC5D,CAAC;CACF;AAZD,sCAYC;AAED;;GAEG;AACH,MAAa,oBAAoB;IAC/B,YACU,MAAc,EACd,MAAW;QADX,WAAM,GAAN,MAAM,CAAQ;QACd,WAAM,GAAN,MAAM,CAAK;IAClB,CAAC;IAEJ;;OAEG;IACH,KAAK,CAAC,MAAM,CACV,MAA8C,EAC9C,QAAwB;QAExB,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,MAAM,aAAa,GAAG,IAAA,yCAAqB,GAAE,CAAC;QAE9C,IAAI,CAAC;YACH,MAAM,CAAC,KAAK,CACV,mDAAmD,MAAM,CAAC,KAAK,EAAE,CAClE,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAEnE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,OAAO,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;YAE/C,2EAA2E;YAC3E,IAAI,OAAO,IAAI,QAAQ,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAC1C,kDAAkD;gBAClD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAY,CAAC;gBACpC,MAAM,eAAe,GACnB,SAAS,IAAI,QAAQ;oBACnB,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,SAAS;oBACpD,CAAC,CAAC,SAAS,CAAC;gBAEhB,IAAA,0BAAe,EAAC;oBACd,SAAS,EAAE,CAAC,IAAI,IAAI,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,aAAa;oBACnE,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;oBAC1C,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,iBAAiB;oBAClD,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;oBACxC,QAAQ;oBACR,YAAY,EACV,SAAS,IAAI,QAAQ;wBACnB,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,IAAI;wBAC5C,CAAC,CAAC,IAAI;oBACV,aAAa,EAAE,QAAQ;oBACvB,UAAU,EAAE,KAAK;oBACjB,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,cAAc,EAAE,MAAM,CAAC,eAAe;oBACtC,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,eAAe,EAAE,eAAe;wBAC9B,CAAC,CAAC,IAAA,0CAAmB,EAAC,eAAe,CAAC;wBACtC,CAAC,CAAC,SAAS;iBACd,CAAC,CAAC;YACL,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,CAAC,wCAAwC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAEtE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,OAAO,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;YAE/C,IAAA,0BAAe,EAAC;gBACd,SAAS,EAAE,aAAa;gBACxB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,YAAY,EAAE,CAAC;gBACf,gBAAgB,EAAE,CAAC;gBACnB,WAAW,EAAE,CAAC;gBACd,QAAQ;gBACR,YAAY,EAAE,OAAO;gBACrB,aAAa,EAAE,QAAQ;gBACvB,UAAU,EAAE,KAAK;gBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;aAC1B,CAAC,CAAC;YAEH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CACnB,MAA8C,EAC9C,QAAwB;QAExB,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,MAAM,aAAa,GAAG,IAAA,yCAAqB,GAAE,CAAC;QAE9C,IAAI,CAAC;YACH,MAAM,CAAC,KAAK,CACV,6DAA6D,MAAM,CAAC,KAAK,EAAE,CAC5E,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACvD,GAAG,MAAM;gBACT,MAAM,EAAE,IAAI;aACb,CAAC,CAAC;YAEH,OAAO,IAAI,uCAAgB,CACzB,MAAa,EACb,MAAM,CAAC,KAAK,EACZ,SAAS,EACT,aAAa,EACb,QAAQ,EACR,IAAI,CAAC,MAAM,EACX,MAAM,CAAC,eAAe,EACtB,MAAM,CAAC,QAAQ,CAChB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,CACV,kDAAkD,KAAK,CAAC,OAAO,EAAE,CAClE,CAAC;YAEF,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,OAAO,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;YAE/C,IAAA,0BAAe,EAAC;gBACd,SAAS,EAAE,aAAa;gBACxB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,YAAY,EAAE,CAAC;gBACf,gBAAgB,EAAE,CAAC;gBACnB,WAAW,EAAE,CAAC;gBACd,QAAQ;gBACR,YAAY,EAAE,OAAO;gBACrB,aAAa,EAAE,QAAQ;gBACvB,UAAU,EAAE,IAAI;gBAChB,QAAQ,EAAE,MAAM,CAAC,QAAQ;aAC1B,CAAC,CAAC;YAEH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAvID,oDAuIC"}

package/dist/cjs/core/middleware/streaming-wrapper.js

@@ -6,18 +6,22 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.StreamingWrapper = void 0;
 const config_1 = require("../config");
 const tracking_1 = require("../tracking");
+const prompt_extraction_js_1 = require("../../utils/prompt-extraction.js");
 const logger = (0, config_1.getLogger)();
 /**
  * Wrapper for streaming responses that tracks usage
  */
 class StreamingWrapper {
-    constructor(stream, model, startTime, transactionId, metadata, config) {
+    constructor(stream, model, startTime, transactionId, metadata, config, responseFormat, messages) {
+        this.accumulatedContent = "";
         this.stream = stream;
         this.model = model;
         this.startTime = startTime;
         this.transactionId = transactionId;
         this.metadata = metadata;
         this.config = config || {};
+        this.responseFormat = responseFormat;
+        this.messages = messages || [];
     }
     /**
      * Iterate over stream chunks
@@ -30,6 +34,7 @@ class StreamingWrapper {
         let costData = undefined;
         let firstChunkTime = null;
         let timeToFirstToken = 0;
+        let completed = false;
         try {
             for await (const chunk of this.stream) {
                 // Capture time of first chunk
@@ -39,6 +44,14 @@ class StreamingWrapper {
                         firstChunkTime.getTime() - this.startTime.getTime();
                 }
                 lastChunk = chunk;
+                if (chunk.choices?.[0]?.delta?.content &&
+                    (0, prompt_extraction_js_1.shouldCapturePrompts)(this.metadata)) {
+                    const maxSize = (0, prompt_extraction_js_1.getMaxPromptSize)();
+                    const remaining = maxSize - this.accumulatedContent.length;
+                    if (remaining > 0) {
+                        this.accumulatedContent += chunk.choices[0].delta.content.slice(0, remaining);
+                    }
+                }
                 // Track usage if available in chunk
                 if (chunk.usage) {
                     inputTokens = chunk.usage.prompt_tokens || 0;
@@ -51,6 +64,7 @@ class StreamingWrapper {
                 }
                 yield chunk;
             }
+            completed = true;
             // Send metering data when stream completes (fire-and-forget)
             const endTime = Date.now();
             const duration = endTime - this.startTime.getTime();
@@ -67,13 +81,64 @@ class StreamingWrapper {
                 isStreamed: true,
                 timeToFirstToken,
                 cost: costData,
+                responseFormat: this.responseFormat,
+                messages: this.messages,
+                responseContent: this.accumulatedContent
+                    ? (0, prompt_extraction_js_1.sanitizeCredentials)(this.accumulatedContent)
+                    : undefined,
             });
         }
         catch (error) {
-            // Log error
+            completed = true;
             logger.error("[Revenium] Error in stream processing:", error.message);
+            const endTime = Date.now();
+            const duration = endTime - this.startTime.getTime();
+            (0, tracking_1.trackUsageAsync)({
+                requestId: this.transactionId,
+                model: this.model,
+                promptTokens: inputTokens,
+                completionTokens: outputTokens,
+                totalTokens,
+                duration,
+                finishReason: "error",
+                usageMetadata: this.metadata,
+                isStreamed: true,
+                messages: this.messages,
+                responseContent: this.accumulatedContent
+                    ? (0, prompt_extraction_js_1.sanitizeCredentials)(this.accumulatedContent)
+                    : undefined,
+            });
             throw error;
         }
+        finally {
+            if (!completed) {
+                const endTime = Date.now();
+                const duration = endTime - this.startTime.getTime();
+                (0, tracking_1.trackUsageAsync)({
+                    requestId: this.transactionId,
+                    model: this.model,
+                    promptTokens: inputTokens,
+                    completionTokens: outputTokens,
+                    totalTokens,
+                    duration,
+                    finishReason: "cancelled",
+                    usageMetadata: this.metadata,
+                    isStreamed: true,
+                    timeToFirstToken,
+                    cost: costData,
+                    responseFormat: this.responseFormat,
+                    messages: this.messages,
+                    responseContent: this.accumulatedContent
+                        ? (0, prompt_extraction_js_1.sanitizeCredentials)(this.accumulatedContent)
+                        : undefined,
+                });
+                logger.debug("[Revenium] Streaming cancelled", {
+                    requestId: this.transactionId,
+                    model: this.model,
+                    duration,
+                });
+            }
+        }
     }
 }
 exports.StreamingWrapper = StreamingWrapper;