@revenium/perplexity 1.0.5 → 1.0.7

package/README.md

@@ -31,9 +31,15 @@ import { createPerplexityClient } from "@revenium/perplexity";
 config();
 
 async function example() {
-  // Create client instance
+  // Create client instance (uses environment variables by default)
   const client = createPerplexityClient();
 
+  // Or create with custom API key and base URL
+  const customClient = createPerplexityClient(
+    "your-api-key",
+    "https://api.perplexity.ai"
+  );
+
   // Make a chat completion
   const response = await client.createChatCompletion({
     model: "sonar-pro",
@@ -97,8 +103,11 @@ REVENIUM_METERING_BASE_URL=https://api.revenium.io/meter/v2
 # Optional: Perplexity base URL (defaults to https://api.perplexity.ai)
 PERPLEXITY_BASE_URL=https://api.perplexity.ai
 
-# Optional: Enable verbose logging
+# Optional: Enable verbose logging for debugging
 REVENIUM_VERBOSE_STARTUP=true
+
+# Optional: Set log level (DEBUG, INFO, WARNING, ERROR)
+REVENIUM_LOG_LEVEL=INFO
 ```
 
 ## 📚 Examples
@@ -152,14 +161,22 @@ const stream = await client.createStreamingChatCompletion({
       content: "Write a creative story about AI",
     },
   ],
+  // Optional: Include usage metadata for tracking
+  usageMetadata: {
+    traceId: "story-generation-001",
+    taskType: "creative-writing",
+    subscriberEmail: "user@example.com",
+  },
 });
 
+console.log("📝 Streaming response:");
 for await (const chunk of stream) {
   const content = chunk.choices[0]?.delta?.content;
   if (content) {
     process.stdout.write(content);
   }
 }
+console.log("\n✅ Streaming completed!");
 ```
 
 ## 📊 Usage Metadata
@@ -247,14 +264,33 @@ npm run run-all-examples
 
 ### PerplexityClient
 
-#### `createPerplexityClient(apiKey?: string)`
+#### `createPerplexityClient(apiKey?: string, baseUrl?: string)`
 
 Creates a new Perplexity client instance.
 
+**Parameters:**
+
+- `apiKey` (optional): Your Perplexity API key. If not provided, uses `PERPLEXITY_API_KEY` environment variable.
+- `baseUrl` (optional): Custom base URL for Perplexity API. If not provided, uses `PERPLEXITY_BASE_URL` environment variable or defaults to `https://api.perplexity.ai`.
+
 ```typescript
+// Using environment variables
 const client = createPerplexityClient();
-// or
+
+// With custom API key
 const client = createPerplexityClient("your-api-key");
+
+// With custom API key and base URL
+const client = createPerplexityClient(
+  "your-api-key",
+  "https://api.perplexity.ai"
+);
+
+// With custom base URL only (uses env var for API key)
+const client = createPerplexityClient(
+  undefined,
+  "https://custom-perplexity-endpoint.com"
+);
 ```
 
 #### `client.createChatCompletion(params)`
@@ -310,16 +346,74 @@ interface UsageMetadata {
 }
 ```
 
+## 🔧 Configuration Options
+
+### Client Configuration
+
+The `createPerplexityClient` function accepts two optional parameters:
+
+```typescript
+// All default configuration (uses environment variables)
+const client = createPerplexityClient();
+
+// Custom API key only
+const client = createPerplexityClient("your-custom-api-key");
+
+// Custom API key and base URL
+const client = createPerplexityClient(
+  "your-custom-api-key",
+  "https://custom-perplexity-endpoint.com"
+);
+
+// Custom base URL only (API key from environment)
+const client = createPerplexityClient(
+  undefined,
+  "https://custom-perplexity-endpoint.com"
+);
+```
+
+### Environment Variable Priority
+
+The client follows this priority order for configuration:
+
+1. **API Key**: Function parameter → `PERPLEXITY_API_KEY` environment variable
+2. **Base URL**: Function parameter → `PERPLEXITY_BASE_URL` environment variable → Default (`https://api.perplexity.ai`)
+
 ## 🔧 Integration Tips
 
 ### Express.js
 
 Load environment variables first, create a separate Perplexity module, and use the patched instance everywhere.
 
+```typescript
+// perplexity.js
+import { config } from "dotenv";
+import { createPerplexityClient } from "@revenium/perplexity";
+
+config();
+export const perplexityClient = createPerplexityClient();
+```
+
 ### Next.js
 
 Use global variables to prevent re-initialization, works with both Pages and App Router.
 
+```typescript
+// lib/perplexity.ts
+import { createPerplexityClient } from "@revenium/perplexity";
+
+declare global {
+  var perplexityClient: ReturnType<typeof createPerplexityClient> | undefined;
+}
+
+export const perplexityClient =
+  globalThis.perplexityClient ?? createPerplexityClient();
+
+if (process.env.NODE_ENV !== "production") {
+  globalThis.perplexityClient = perplexityClient;
+}
+```
+
 ### Serverless
 
 Implement singleton pattern for cold starts, handle CORS for browser clients.
@@ -335,6 +429,8 @@ Validate environment variables at startup, add health checks for orchestration.
 1. **Middleware not activating**: Ensure you're importing the package before creating client instances
 2. **Environment variables**: Verify all required environment variables are set
 3. **API keys**: Check that both Perplexity and Revenium API keys are valid
+4. **Custom base URL**: If using a custom base URL, ensure it's accessible and compatible with Perplexity API
+5. **Module imports**: Make sure you're using the correct import: `import { createPerplexityClient } from "@revenium/perplexity"`
 
 ### Debug Mode
 

package/dist/perplexity-ai/middleware.js

@@ -97,11 +97,43 @@ async function processPerplexityResponse(response, transactionId, model, startTi
     try {
         const endTime = new Date();
         const duration = (0, utils_1.calculateDurationMs)(startTime, endTime);
-        // Parse response body
-        const responseData = await response.json();
-        // Extract token counts and other data
-        const tokenCounts = (0, utils_1.extractPerplexityTokenCounts)(responseData);
-        const stopReason = (0, utils_1.extractStopReason)(responseData);
+        // Check if this is a streaming response
+        const contentType = response.headers.get("content-type") || "";
+        const isStreamingResponse = contentType.includes("text/plain") ||
+            contentType.includes("text/event-stream");
+        let responseData;
+        let tokenCounts = { inputTokens: 0, outputTokens: 0, totalTokens: 0 };
+        let stopReason = "END";
+        if (isStreamingResponse) {
+            // For streaming responses, we can't easily parse the full response
+            // We'll extract what we can from the stream or use defaults
+            const responseText = await response.text();
+            // Try to extract token information from the last complete JSON chunk
+            const jsonChunks = responseText
+                .split("\n")
+                .filter((line) => line.startsWith("data: ") && line !== "data: [DONE]")
+                .map((line) => line.substring(6)); // Remove 'data: ' prefix
+            if (jsonChunks.length > 0) {
+                try {
+                    // Parse the last chunk to get usage information
+                    const lastChunk = JSON.parse(jsonChunks[jsonChunks.length - 1]);
+                    if (lastChunk.usage) {
+                        tokenCounts = (0, utils_1.extractPerplexityTokenCounts)(lastChunk);
+                    }
+                }
+                catch (e) {
+                    // If we can't parse, use defaults
+                    models_1.Logger.warning("Could not parse streaming response for token counts, using defaults");
+                }
+            }
+            responseData = { usage: tokenCounts };
+        }
+        else {
+            // Parse response body for non-streaming responses
+            responseData = await response.json();
+            tokenCounts = (0, utils_1.extractPerplexityTokenCounts)(responseData);
+            stopReason = (0, utils_1.extractStopReason)(responseData);
+        }
         const modelName = (0, utils_1.extractModelName)(responseData, model);
         // Send metering data (only if API key is available)
         if ((0, getEnv_1.getReveniumApiKey)()) {

package/dist/perplexity-ai/middleware.js.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/perplexity-ai/middleware.ts"],"names":[],"mappings":";;AAAA,sCAA2C;AAC3C,oCAKkB;AAClB,oCAUkB;AAClB,sCAAoE;AACpE,0DAcqC;AACrC,4DAIsC;AAEtC,4CAIyB;AAEzB,0BAA0B;AAC1B,MAAM,cAAc,GAA+B,IAAI,GAAG,EAGvD,CAAC;AAEJ,0EAA0E;AAC1E,IAAI,qBAAqB,GAAY,KAAK,CAAC;AAE3C,sCAAsC;AACtC,SAAS,8BAA8B;IACrC,IAAI,qBAAqB,EAAE,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,uDAA4C,CAAC,CAAC;QAC1D,OAAO;IACT,CAAC;IAED,qDAAqD;IACrD,IAAI,CAAC,IAAA,0BAAiB,GAAE,EAAE,CAAC;QACzB,eAAM,CAAC,OAAO,CAAC,8DAAmD,CAAC,CAAC;IACtE,CAAC;IAED,IAAI,CAAC;QACH,4CAA4C;QAC5C,+BAA+B,EAAE,CAAC;QAElC,sBAAsB;QACtB,qBAAqB,GAAG,IAAI,CAAC;QAC7B,OAAO,CAAC,GAAG,CAAC,+CAAoC,CAAC,CAAC;IACpD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,wDAA6C,EAAE,KAAK,CAAC,CAAC;QACpE,MAAM,IAAI,2BAAkB,CAC1B,GAAG,6DAAkD,KAAK,KAAK,EAAE,CAClE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,2CAA2C;AAC3C,SAAS,+BAA+B;IACtC,gCAAgC;IAChC,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC;IAEnC,mDAAmD;IACnD,MAAM,CAAC,KAAK,GAAG,KAAK,WAClB,KAA6B,EAC7B,IAAkB;QAElB,MAAM,GAAG,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QACjE,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;QAEpC,4CAA4C;QAC5C,IACE,GAAG,CAAC,QAAQ,CAAC,IAAA,6BAAoB,GAAE,CAAC;YACpC,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EACjC,CAAC;YACD,OAAO,MAAM,uBAAuB,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;QACjE,CAAC;QAED,kDAAkD;QAClD,OAAO,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,oEAAyD,CAAC,CAAC;AACzE,CAAC;AAED,gCAAgC;AAChC,KAAK,UAAU,uBAAuB,CACpC,GAAW,EACX,IAA6B,EAC7B,aAA2B;IAE3B,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;IAC7B,MAAM,aAAa,GAAG,IAAA,6BAAqB,GAAE,CAAC;IAE9C,oCAAoC;IACpC,IAAI,WAAW,GAAQ,EAAE,CAAC;IAC1B,IAAI,IAAI,EAAE,IAAI,EAAE,CAAC;QACf,IAAI,CAAC;YACH,WAAW;gBACT,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QACtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAM,CAAC,OAAO,CACZ,uEAA4D,CAC7D,CAAC;QACJ,CAAC;IACH,CAAC;IAED,yBAAyB;IACzB,MAAM,aAAa,GAAG,IAAA,4BAAoB,EAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC9D,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,IAAI,oCAAwB,CAAC;IAE5D,eAAM,CAAC,KAAK,CAAC,sEAA2D,EAAE;QACxE,aAAa;QACb,KAAK;QACL,GAAG;KACJ,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,4BAA4B;QAC5B,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAEhD,2DAA2D;QAC3D,MAAM,aAAa,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;QAEvC,0DAA0D;QAC1D,yBAAyB,CACvB,aAAa,EACb,aAAa,EACb,KAAK,EACL,SAAS,EACT,aAAa,CACd,CAAC;QAEF,2CAA2C;QAC3C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CAAC,sDAA2C,EAAE;YACxD,aAAa;YACb,KAAK;SACN,CAAC,CAAC;QACH,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,iCAAiC;AACjC,KAAK,UAAU,yBAAyB,CACtC,QAAkB,EAClB,aAAqB,EACrB,KAAa,EACb,SAAe,EACf,aAA6B,EAC7B,UAAuB,EACvB,QAAiB,EACjB,QAAiB,EACjB,aAAsB,EACtB,cAAuB,EACvB,SAAkB,EAClB,UAAoB;IAEpB,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,IAAA,2BAAmB,EAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAEzD,sBAAsB;QACtB,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAE3C,sCAAsC;QACtC,MAAM,WAAW,GAAG,IAAA,oCAA4B,EAAC,YAAY,CAAC,CAAC;QAC/D,MAAM,UAAU,GAAG,IAAA,yBAAiB,EAAC,YAAY,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,IAAA,wBAAgB,EAAC,YAAY,EAAE,KAAK,CAAC,CAAC;QAExD,oDAAoD;QACpD,IAAI,IAAA,0BAAiB,GAAE,EAAE,CAAC;YACxB,IAAI,CAAC;gBACH,MAAM,IAAA,wBAAgB,EACpB;oBACE,UAAU,EAAE,KAAK;oBACjB,QAAQ,EAAE,QAAQ,IAAI,IAAI;oBAC1B,UAAU,EAAE,UAAU,IAAI,IAAI;oBAC9B,QAAQ,EAAE,QAAQ,IAAI,MAAM;oBAC5B,KAAK,EAAE,eAAe;oBACtB,aAAa,EAAE,aAAa,IAAI,MAAM;oBACtC,eAAe,EAAE,WAAW,CAAC,WAAW;oBACxC,gBAAgB,EAAE,WAAW,CAAC,YAAY;oBAC1C,mBAAmB,EAAE,CAAC;oBACtB,uBAAuB,EAAE,CAAC;oBAC1B,mBAAmB,EAAE,CAAC;oBACtB,eAAe,EAAE,WAAW,CAAC,WAAW;oBACxC,cAAc,EAAE,cAAc,IAAI,mBAAmB;oBACrD,SAAS,EAAE,SAAS,IAAI,YAAY;oBACpC,UAAU,EAAE,UAAU,IAAI;wBACxB,EAAE,EAAE,QAAQ,IAAA,6BAAqB,GAAE,EAAE;wBACrC,KAAK,EAAE,oBAAoB;wBAC3B,UAAU,EAAE;4BACV,IAAI,EAAE,SAAS;4BACf,KAAK,EAAE,UAAU;yBAClB;qBACF;oBACD,KAAK,EAAE,SAAS;oBAChB,aAAa;oBACb,YAAY,EAAE,IAAA,uBAAe,EAAC,OAAO,CAAC;oBACtC,eAAe,EAAE,QAAQ;oBACzB,QAAQ,EAAE,gBAAQ,CAAC,UAAU;oBAC7B,WAAW,EAAE,IAAA,uBAAe,EAAC,SAAS,CAAC;oBACvC,mBAAmB,EAAE,IAAA,uBAAe,EAAC,OAAO,CAAC;oBAC7C,gBAAgB,EAAE,CAAC;oBACnB,iBAAiB,EAAE,MAAM;iBAC1B,EACD,IAAA,0BAAiB,GAAE,EACnB,IAAA,2BAAkB,GAAE,CACrB,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,eAAM,CAAC,OAAO,CAAC,uDAA4C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1E,CAAC;QACH,CAAC;aAAM,CAAC;YACN,eAAM,CAAC,KAAK,CAAC,oDAAyC,CAAC,CAAC;QAC1D,CAAC;QAED,eAAM,CAAC,KAAK,CACV,8EAAmE,EACnE;YACE,aAAa;SACd,CACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CAAC,8DAAmD,EAAE;YAChE,aAAa;YACb,KAAK;SACN,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,qDAAqD;AACrD,8BAA8B,EAAE,CAAC"}
+{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/perplexity-ai/middleware.ts"],"names":[],"mappings":";;AAAA,sCAA2C;AAC3C,oCAKkB;AAClB,oCAUkB;AAClB,sCAAoE;AACpE,0DAcqC;AACrC,4DAIsC;AAEtC,4CAIyB;AAEzB,0BAA0B;AAC1B,MAAM,cAAc,GAA+B,IAAI,GAAG,EAGvD,CAAC;AAEJ,0EAA0E;AAC1E,IAAI,qBAAqB,GAAY,KAAK,CAAC;AAE3C,sCAAsC;AACtC,SAAS,8BAA8B;IACrC,IAAI,qBAAqB,EAAE,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,uDAA4C,CAAC,CAAC;QAC1D,OAAO;IACT,CAAC;IAED,qDAAqD;IACrD,IAAI,CAAC,IAAA,0BAAiB,GAAE,EAAE,CAAC;QACzB,eAAM,CAAC,OAAO,CAAC,8DAAmD,CAAC,CAAC;IACtE,CAAC;IAED,IAAI,CAAC;QACH,4CAA4C;QAC5C,+BAA+B,EAAE,CAAC;QAElC,sBAAsB;QACtB,qBAAqB,GAAG,IAAI,CAAC;QAC7B,OAAO,CAAC,GAAG,CAAC,+CAAoC,CAAC,CAAC;IACpD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,wDAA6C,EAAE,KAAK,CAAC,CAAC;QACpE,MAAM,IAAI,2BAAkB,CAC1B,GAAG,6DAAkD,KAAK,KAAK,EAAE,CAClE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,2CAA2C;AAC3C,SAAS,+BAA+B;IACtC,gCAAgC;IAChC,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC;IAEnC,mDAAmD;IACnD,MAAM,CAAC,KAAK,GAAG,KAAK,WAClB,KAA6B,EAC7B,IAAkB;QAElB,MAAM,GAAG,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QACjE,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;QAEpC,4CAA4C;QAC5C,IACE,GAAG,CAAC,QAAQ,CAAC,IAAA,6BAAoB,GAAE,CAAC;YACpC,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EACjC,CAAC;YACD,OAAO,MAAM,uBAAuB,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;QACjE,CAAC;QAED,kDAAkD;QAClD,OAAO,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,oEAAyD,CAAC,CAAC;AACzE,CAAC;AAED,gCAAgC;AAChC,KAAK,UAAU,uBAAuB,CACpC,GAAW,EACX,IAA6B,EAC7B,aAA2B;IAE3B,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;IAC7B,MAAM,aAAa,GAAG,IAAA,6BAAqB,GAAE,CAAC;IAE9C,oCAAoC;IACpC,IAAI,WAAW,GAAQ,EAAE,CAAC;IAC1B,IAAI,IAAI,EAAE,IAAI,EAAE,CAAC;QACf,IAAI,CAAC;YACH,WAAW;gBACT,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QACtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAM,CAAC,OAAO,CACZ,uEAA4D,CAC7D,CAAC;QACJ,CAAC;IACH,CAAC;IAED,yBAAyB;IACzB,MAAM,aAAa,GAAG,IAAA,4BAAoB,EAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC9D,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,IAAI,oCAAwB,CAAC;IAE5D,eAAM,CAAC,KAAK,CAAC,sEAA2D,EAAE;QACxE,aAAa;QACb,KAAK;QACL,GAAG;KACJ,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,4BAA4B;QAC5B,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAEhD,2DAA2D;QAC3D,MAAM,aAAa,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;QAEvC,0DAA0D;QAC1D,yBAAyB,CACvB,aAAa,EACb,aAAa,EACb,KAAK,EACL,SAAS,EACT,aAAa,CACd,CAAC;QAEF,2CAA2C;QAC3C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CAAC,sDAA2C,EAAE;YACxD,aAAa;YACb,KAAK;SACN,CAAC,CAAC;QACH,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,iCAAiC;AACjC,KAAK,UAAU,yBAAyB,CACtC,QAAkB,EAClB,aAAqB,EACrB,KAAa,EACb,SAAe,EACf,aAA6B,EAC7B,UAAuB,EACvB,QAAiB,EACjB,QAAiB,EACjB,aAAsB,EACtB,cAAuB,EACvB,SAAkB,EAClB,UAAoB;IAEpB,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,IAAA,2BAAmB,EAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAEzD,wCAAwC;QACxC,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;QAC/D,MAAM,mBAAmB,GACvB,WAAW,CAAC,QAAQ,CAAC,YAAY,CAAC;YAClC,WAAW,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;QAE5C,IAAI,YAAiB,CAAC;QACtB,IAAI,WAAW,GAAG,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;QACtE,IAAI,UAAU,GAAG,KAAK,CAAC;QAEvB,IAAI,mBAAmB,EAAE,CAAC;YACxB,mEAAmE;YACnE,4DAA4D;YAC5D,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAE3C,qEAAqE;YACrE,MAAM,UAAU,GAAG,YAAY;iBAC5B,KAAK,CAAC,IAAI,CAAC;iBACX,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,IAAI,KAAK,cAAc,CAAC;iBACtE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,yBAAyB;YAE9D,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACH,gDAAgD;oBAChD,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;oBAChE,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC;wBACpB,WAAW,GAAG,IAAA,oCAA4B,EAAC,SAAS,CAAC,CAAC;oBACxD,CAAC;gBACH,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,kCAAkC;oBAClC,eAAM,CAAC,OAAO,CACZ,qEAAqE,CACtE,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,YAAY,GAAG,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,kDAAkD;YAClD,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACrC,WAAW,GAAG,IAAA,oCAA4B,EAAC,YAAY,CAAC,CAAC;YACzD,UAAU,GAAG,IAAA,yBAAiB,EAAC,YAAY,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,SAAS,GAAG,IAAA,wBAAgB,EAAC,YAAY,EAAE,KAAK,CAAC,CAAC;QAExD,oDAAoD;QACpD,IAAI,IAAA,0BAAiB,GAAE,EAAE,CAAC;YACxB,IAAI,CAAC;gBACH,MAAM,IAAA,wBAAgB,EACpB;oBACE,UAAU,EAAE,KAAK;oBACjB,QAAQ,EAAE,QAAQ,IAAI,IAAI;oBAC1B,UAAU,EAAE,UAAU,IAAI,IAAI;oBAC9B,QAAQ,EAAE,QAAQ,IAAI,MAAM;oBAC5B,KAAK,EAAE,eAAe;oBACtB,aAAa,EAAE,aAAa,IAAI,MAAM;oBACtC,eAAe,EAAE,WAAW,CAAC,WAAW;oBACxC,gBAAgB,EAAE,WAAW,CAAC,YAAY;oBAC1C,mBAAmB,EAAE,CAAC;oBACtB,uBAAuB,EAAE,CAAC;oBAC1B,mBAAmB,EAAE,CAAC;oBACtB,eAAe,EAAE,WAAW,CAAC,WAAW;oBACxC,cAAc,EAAE,cAAc,IAAI,mBAAmB;oBACrD,SAAS,EAAE,SAAS,IAAI,YAAY;oBACpC,UAAU,EAAE,UAAU,IAAI;wBACxB,EAAE,EAAE,QAAQ,IAAA,6BAAqB,GAAE,EAAE;wBACrC,KAAK,EAAE,oBAAoB;wBAC3B,UAAU,EAAE;4BACV,IAAI,EAAE,SAAS;4BACf,KAAK,EAAE,UAAU;yBAClB;qBACF;oBACD,KAAK,EAAE,SAAS;oBAChB,aAAa;oBACb,YAAY,EAAE,IAAA,uBAAe,EAAC,OAAO,CAAC;oBACtC,eAAe,EAAE,QAAQ;oBACzB,QAAQ,EAAE,gBAAQ,CAAC,UAAU;oBAC7B,WAAW,EAAE,IAAA,uBAAe,EAAC,SAAS,CAAC;oBACvC,mBAAmB,EAAE,IAAA,uBAAe,EAAC,OAAO,CAAC;oBAC7C,gBAAgB,EAAE,CAAC;oBACnB,iBAAiB,EAAE,MAAM;iBAC1B,EACD,IAAA,0BAAiB,GAAE,EACnB,IAAA,2BAAkB,GAAE,CACrB,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,eAAM,CAAC,OAAO,CAAC,uDAA4C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1E,CAAC;QACH,CAAC;aAAM,CAAC;YACN,eAAM,CAAC,KAAK,CAAC,oDAAyC,CAAC,CAAC;QAC1D,CAAC;QAED,eAAM,CAAC,KAAK,CACV,8EAAmE,EACnE;YACE,aAAa;SACd,CACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CAAC,8DAAmD,EAAE;YAChE,aAAa;YACb,KAAK;SACN,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,qDAAqD;AACrD,8BAA8B,EAAE,CAAC"}

package/examples/metadata-example.ts

@@ -30,7 +30,10 @@ async function metadataExample() {
       },
     });
 
-    console.log("✅ Response:", basicResponse.choices[0]?.message?.content?.substring(0, 100) + "...");
+    console.log(
+      "✅ Response:",
+      basicResponse.choices[0]?.message?.content?.substring(0, 100) + "..."
+    );
 
     // Example 2: Advanced metadata
     console.log("\n2️⃣ Advanced metadata tracking:");
@@ -53,7 +56,10 @@ async function metadataExample() {
       },
     });
 
-    console.log("✅ Response:", advancedResponse.choices[0]?.message?.content?.substring(0, 100) + "...");
+    console.log(
+      "✅ Response:",
+      advancedResponse.choices[0]?.message?.content?.substring(0, 100) + "..."
+    );
 
     // Example 3: Streaming with metadata
     console.log("\n3️⃣ Streaming with metadata:");
@@ -83,7 +89,6 @@ async function metadataExample() {
 
     console.log("\n\n🎉 All metadata examples completed successfully!");
     console.log("📊 All usage tracked with custom metadata");
-
   } catch (error) {
     console.error("❌ Error:", error);
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@revenium/perplexity",
-  "version": "1.0.5",
+  "version": "1.0.7",
   "description": "NodeJS middleware for perplexity's AI API",
   "homepage": "https://github.com/revenium/revenium-middleware-perplexity-node#readme",
   "bugs": {

package/playground/basic.js

@@ -0,0 +1,31 @@
+import { createPerplexityClient } from "@revenium/perplexity";
+
+const basicExample = async () => {
+  console.log("\n🤖 Perplexity AI - Basic Client Example");
+  console.log("=".repeat(50));
+
+  try {
+    // Create client instance
+    const client = createPerplexityClient();
+
+    // Make a simple chat completion
+    const response = await client.createChatCompletion({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "What is the meaning of life, the universe and everything?",
+        },
+      ],
+    });
+
+    console.log("✅ Response received:");
+    console.log(response.choices[0].message.content);
+    console.log("\n📊 Token usage automatically tracked by middleware");
+    console.log("🎉 Basic client example successful!");
+  } catch (error) {
+    console.error("❌ Error:", error);
+  }
+};
+
+basicExample();

package/playground/custom-config.js

@@ -0,0 +1,82 @@
+import { createPerplexityClient } from "@revenium/perplexity";
+
+async function customConfigExample() {
+  console.log("\n⚙️ Perplexity AI - Custom Configuration Example");
+  console.log("=".repeat(50));
+
+  try {
+    // Example 1: Using environment variables (default)
+    console.log("\n1️⃣ Using environment variables:");
+    const defaultClient = createPerplexityClient();
+    
+    const response1 = await defaultClient.createChatCompletion({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "Hello from default configuration!",
+        },
+      ],
+    });
+    console.log("✅ Default config response:", response1.choices[0]?.message?.content?.substring(0, 50) + "...");
+
+    // Example 2: Custom API key only
+    console.log("\n2️⃣ Using custom API key:");
+    const customKeyClient = createPerplexityClient(process.env.PERPLEXITY_API_KEY);
+    
+    const response2 = await customKeyClient.createChatCompletion({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "Hello from custom API key configuration!",
+        },
+      ],
+    });
+    console.log("✅ Custom key response:", response2.choices[0]?.message?.content?.substring(0, 50) + "...");
+
+    // Example 3: Custom API key and base URL
+    console.log("\n3️⃣ Using custom API key and base URL:");
+    const fullCustomClient = createPerplexityClient(
+      process.env.PERPLEXITY_API_KEY,
+      "https://api.perplexity.ai"
+    );
+    
+    const response3 = await fullCustomClient.createChatCompletion({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "Hello from fully custom configuration!",
+        },
+      ],
+    });
+    console.log("✅ Full custom response:", response3.choices[0]?.message?.content?.substring(0, 50) + "...");
+
+    // Example 4: Custom base URL only (API key from env)
+    console.log("\n4️⃣ Using custom base URL only:");
+    const customUrlClient = createPerplexityClient(
+      undefined,
+      "https://api.perplexity.ai"
+    );
+    
+    const response4 = await customUrlClient.createChatCompletion({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "Hello from custom URL configuration!",
+        },
+      ],
+    });
+    console.log("✅ Custom URL response:", response4.choices[0]?.message?.content?.substring(0, 50) + "...");
+
+    console.log("\n🎉 All configuration examples completed successfully!");
+    console.log("📊 All requests tracked by middleware regardless of configuration");
+
+  } catch (error) {
+    console.error("❌ Error:", error);
+  }
+}
+
+customConfigExample();

package/playground/enhanced.js

@@ -0,0 +1,36 @@
+import { createPerplexityClient } from "@revenium/perplexity";
+
+const enhancedExample = async () => {
+  console.log("\n1️⃣ Enhanced Example:");
+
+  try {
+    const client = createPerplexityClient();
+    const advancedResponse = await client.createChatCompletion({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "What are the latest developments in AI?",
+        },
+      ],
+      usageMetadata: {
+        traceId: "conv-advanced-123",
+        taskType: "research",
+        subscriberId: "user-12345",
+        subscriberCredentialName: "api-key-1",
+        productId: "business-intelligence",
+        agent: "research-assistant-v2",
+        responseQualityScore: 0.95,
+      },
+    });
+
+    console.log(
+      "✅ Response:",
+      advancedResponse.choices[0]?.message?.content?.substring(0, 100) + "..."
+    );
+  } catch (error) {
+    console.error("❌ Error:", error);
+  }
+};
+
+enhancedExample();

package/playground/metadata.js

@@ -0,0 +1,94 @@
+import { createPerplexityClient } from "@revenium/perplexity";
+
+async function metadataExample() {
+  console.log("\n📊 Perplexity AI - Metadata Tracking Example");
+  console.log("=".repeat(50));
+
+  try {
+    // Create client instance
+    const client = createPerplexityClient();
+
+    // Example 1: Basic metadata
+    console.log("\n1️⃣ Basic metadata tracking:");
+    const basicResponse = await client.createChatCompletion({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "Analyze this quarterly report for key insights",
+        },
+      ],
+      usageMetadata: {
+        traceId: "conv-28a7e9d4",
+        taskType: "document-analysis",
+        subscriberEmail: "user@example.com",
+        organizationId: "acme-corp",
+        subscriptionId: "premium-plan",
+      },
+    });
+
+    console.log(
+      "✅ Response:",
+      basicResponse.choices[0]?.message?.content?.substring(0, 100) + "..."
+    );
+
+    // Example 2: Advanced metadata
+    console.log("\n2️⃣ Advanced metadata tracking:");
+    const advancedResponse = await client.createChatCompletion({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "What are the latest developments in AI?",
+        },
+      ],
+      usageMetadata: {
+        traceId: "conv-advanced-123",
+        taskType: "research",
+        subscriberId: "user-12345",
+        subscriberCredentialName: "api-key-1",
+        productId: "business-intelligence",
+        agent: "research-assistant-v2",
+        responseQualityScore: 0.95,
+      },
+    });
+
+    console.log(
+      "✅ Response:",
+      advancedResponse.choices[0]?.message?.content?.substring(0, 100) + "..."
+    );
+
+    // Example 3: Streaming with metadata
+    console.log("\n3️⃣ Streaming with metadata:");
+    const stream = await client.createStreamingChatCompletion({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "Write a creative story about AI",
+        },
+      ],
+      usageMetadata: {
+        traceId: "conv-streaming-456",
+        taskType: "creative-writing",
+        organizationId: "creative-studio",
+        agent: "story-generator",
+      },
+    });
+
+    console.log("📝 Streaming response with metadata:");
+    for await (const chunk of stream) {
+      const content = chunk.choices[0]?.delta?.content;
+      if (content) {
+        process.stdout.write(content);
+      }
+    }
+
+    console.log("\n\n🎉 All metadata examples completed successfully!");
+    console.log("📊 All usage tracked with custom metadata");
+  } catch (error) {
+    console.error("❌ Error:", error);
+  }
+}
+
+metadataExample();

package/playground/multiple-models.js

@@ -0,0 +1,62 @@
+import { createPerplexityClient } from "@revenium/perplexity";
+
+async function multipleModelsExample() {
+  console.log("\n🔧 Perplexity AI - Multiple Models Example");
+  console.log("=".repeat(50));
+
+  try {
+    // Create client instance
+    const client = createPerplexityClient();
+
+    // Test different prompts with the same model
+    const testCases = [
+      {
+        name: "sonar-pro (Quantum Computing)",
+        description: "Latest and most capable model",
+        prompt: "Explain quantum computing in simple terms",
+      },
+      {
+        name: "sonar-pro (AI Developments)",
+        description: "Latest and most capable model",
+        prompt: "What are the latest developments in AI?",
+      },
+      {
+        name: "sonar-pro (Weather)",
+        description: "Latest and most capable model",
+        prompt: "What is the weather like today?",
+      },
+    ];
+
+    for (const testCase of testCases) {
+      console.log(`\n🤖 Testing: ${testCase.name}`);
+      console.log(`📝 Description: ${testCase.description}`);
+      console.log("-".repeat(40));
+
+      try {
+        const response = await client.createChatCompletion({
+          model: "sonar-pro",
+          messages: [
+            {
+              role: "user",
+              content: testCase.prompt,
+            },
+          ],
+        });
+
+        console.log("✅ Response:");
+        console.log(response.choices[0]?.message?.content?.substring(0, 200) + "...");
+        console.log("📊 Usage tracked by middleware");
+
+      } catch (error) {
+        console.log(`❌ Error with ${testCase.name}:`, error);
+      }
+    }
+
+    console.log("\n🎉 Multiple models example completed!");
+
+  } catch (error) {
+    console.error("❌ Error:", error);
+  }
+}
+
+multipleModelsExample();

package/playground/openai-client.js

@@ -0,0 +1,73 @@
+import { OpenAI } from "openai";
+
+// Import middleware to activate automatic tracking
+import "@revenium/perplexity";
+
+async function openaiClientExample() {
+  console.log("\n🤖 Perplexity AI - OpenAI Client Example");
+  console.log("=".repeat(50));
+
+  try {
+    // Create OpenAI client configured for Perplexity (like in basic example)
+    const client = new OpenAI({
+      apiKey: process.env.PERPLEXITY_API_KEY,
+      baseURL: "https://api.perplexity.ai",
+    });
+
+    console.log("✅ OpenAI client configured for Perplexity AI");
+    console.log("📊 Middleware automatically tracking all requests");
+
+    // Test different types of requests
+    console.log("\n1️⃣ Basic chat completion:");
+    const basicResponse = await client.chat.completions.create({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "What is the meaning of life, the universe and everything?",
+        },
+      ],
+    });
+    console.log("✅ Response:", basicResponse.choices[0]?.message?.content?.substring(0, 100) + "...");
+
+    console.log("\n2️⃣ Streaming chat completion:");
+    const stream = await client.chat.completions.create({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "Write a short poem about AI",
+        },
+      ],
+      stream: true,
+    });
+
+    console.log("📝 Streaming response:");
+    for await (const chunk of stream) {
+      const content = chunk.choices[0]?.delta?.content;
+      if (content) {
+        process.stdout.write(content);
+      }
+    }
+
+    console.log("\n\n3️⃣ Different prompt:");
+    const differentResponse = await client.chat.completions.create({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "Explain machine learning in one sentence",
+        },
+      ],
+    });
+    console.log("✅ Response:", differentResponse.choices[0]?.message?.content);
+
+    console.log("\n🎉 All examples completed successfully!");
+    console.log("📊 All usage automatically tracked by middleware");
+
+  } catch (error) {
+    console.error("❌ Error:", error);
+  }
+}
+
+openaiClientExample();

package/playground/streaming.js

@@ -0,0 +1,40 @@
+import { createPerplexityClient } from "@revenium/perplexity";
+
+async function streamingExample() {
+  console.log("\n📊 Perplexity AI - Streaming Example");
+  console.log("=".repeat(50));
+
+  try {
+    const client = createPerplexityClient();
+    console.log("\n3️⃣ Streaming with metadata:");
+    const stream = await client.createStreamingChatCompletion({
+      model: "sonar-pro",
+      messages: [
+        {
+          role: "user",
+          content: "Write a creative story about AI",
+        },
+      ],
+      usageMetadata: {
+        traceId: "conv-streaming-456",
+        taskType: "creative-writing",
+        organizationId: "creative-studio",
+        agent: "story-generator",
+      },
+    });
+
+    console.log("📝 Streaming response with metadata:");
+    for await (const chunk of stream) {
+      const content = chunk.choices[0]?.delta?.content;
+      if (content) {
+        process.stdout.write(content);
+      }
+    }
+
+    console.log("\n\n🎉 Streaming completed!");
+  } catch (error) {
+    console.error("❌ Error:", error);
+  }
+}
+
+streamingExample();

package/src/perplexity-ai/middleware.ts

@@ -186,12 +186,50 @@ async function processPerplexityResponse(
     const endTime = new Date();
     const duration = calculateDurationMs(startTime, endTime);
 
-    // Parse response body
-    const responseData = await response.json();
+    // Check if this is a streaming response
+    const contentType = response.headers.get("content-type") || "";
+    const isStreamingResponse =
+      contentType.includes("text/plain") ||
+      contentType.includes("text/event-stream");
+
+    let responseData: any;
+    let tokenCounts = { inputTokens: 0, outputTokens: 0, totalTokens: 0 };
+    let stopReason = "END";
+
+    if (isStreamingResponse) {
+      // For streaming responses, we can't easily parse the full response
+      // We'll extract what we can from the stream or use defaults
+      const responseText = await response.text();
+
+      // Try to extract token information from the last complete JSON chunk
+      const jsonChunks = responseText
+        .split("\n")
+        .filter((line) => line.startsWith("data: ") && line !== "data: [DONE]")
+        .map((line) => line.substring(6)); // Remove 'data: ' prefix
+
+      if (jsonChunks.length > 0) {
+        try {
+          // Parse the last chunk to get usage information
+          const lastChunk = JSON.parse(jsonChunks[jsonChunks.length - 1]);
+          if (lastChunk.usage) {
+            tokenCounts = extractPerplexityTokenCounts(lastChunk);
+          }
+        } catch (e) {
+          // If we can't parse, use defaults
+          Logger.warning(
+            "Could not parse streaming response for token counts, using defaults"
+          );
+        }
+      }
+
+      responseData = { usage: tokenCounts };
+    } else {
+      // Parse response body for non-streaming responses
+      responseData = await response.json();
+      tokenCounts = extractPerplexityTokenCounts(responseData);
+      stopReason = extractStopReason(responseData);
+    }
 
-    // Extract token counts and other data
-    const tokenCounts = extractPerplexityTokenCounts(responseData);
-    const stopReason = extractStopReason(responseData);
     const modelName = extractModelName(responseData, model);
 
     // Send metering data (only if API key is available)

package/playground/basic-client-example.js

@@ -1,22 +0,0 @@
-import { createPerplexityClient } from "@revenium/perplexity";
-
-async function basicExample() {
-  const client = createPerplexityClient();
-
-  const response = await client.createChatCompletion({
-    model: "sonar-pro",
-    messages: [
-      {
-        role: "user",
-        content: "What is the meaning of life, the universe and everything?",
-      },
-    ],
-  });
-
-  console.log("✅ Response received:");
-  console.log(response.choices[0].message.content);
-  console.log("\n📊 Token usage automatically tracked by middleware");
-  console.log("🎉 Basic client example successful!");
-}
-
-basicExample();