agentid-sdk 0.1.26 → 0.1.28

package/README.md

@@ -110,6 +110,8 @@ const response = await secured.chat.completions.create({
 console.log(response.choices[0]?.message?.content ?? "");
 ```
 
+Wrapped OpenAI calls persist telemetry for both regular and streamed completions. For `stream: true`, logging happens when the stream finishes.
+
 > Scope note: AgentID compliance/risk controls apply to the specific SDK-wrapped LLM calls (`guard()`, `wrapOpenAI()`, LangChain callback-wrapped flows). They do not automatically classify unrelated code paths in your whole monolithic application.
 
 ### LangChain Integration
@@ -145,6 +147,8 @@ const result = await chain.invoke(
 console.log(result);
 ```
 
+LangChain callbacks log on run completion. Token/cost telemetry for streamed chains depends on the provider exposing usage in the final LangChain result.
+
 ### Raw Ingest API (Telemetry Only)
 
 ```ts

package/dist/{agentid-DviYzyAM.d.mts → agentid-agvYW2vW.d.mts}

@@ -18,6 +18,7 @@ interface GuardParams {
     user_id?: string;
     client_event_id?: string;
     expected_languages?: string[];
+    request_identity?: Record<string, unknown>;
     client_capabilities?: {
         capabilities: {
             has_feedback_handler: boolean;
@@ -67,10 +68,11 @@ interface LogParams {
     input: string;
     output: string;
     model: string;
-    usage?: Record<string, number>;
-    tokens?: Record<string, number>;
+    usage?: Record<string, unknown>;
+    tokens?: Record<string, unknown>;
     latency?: number;
     user_id?: string;
+    request_identity?: Record<string, unknown>;
     metadata?: Record<string, unknown>;
     event_type?: "start" | "complete" | "error" | "human_override" | "security_alert" | "security_block" | "security_policy_violation" | "transparency_badge_rendered";
     severity?: "info" | "warning" | "error" | "high";
@@ -179,6 +181,7 @@ declare class AgentID {
     guard(params: GuardParams, options?: RequestOptions): Promise<GuardResponse>;
     private sendIngest;
     private extractStreamChunkText;
+    private extractStreamChunkUsage;
     private wrapCompletion;
     /**
      * LOG: Sends telemetry after execution.
@@ -197,13 +200,14 @@ declare class AgentID {
      * Wrap an OpenAI client once; AgentID will automatically:
      * - run guard() before chat.completions.create
      * - measure latency
-     * - fire-and-forget ingest logging
+     * - persist ingest telemetry for the wrapped call
      */
     wrapOpenAI<T>(openai: T, options: {
         system_id: string;
         user_id?: string;
         expected_languages?: string[];
         expectedLanguages?: string[];
+        request_identity?: Record<string, unknown>;
         apiKey?: string;
         api_key?: string;
         resolveApiKey?: (request: Record<string, unknown>) => string | undefined;

package/dist/{agentid-DviYzyAM.d.ts → agentid-agvYW2vW.d.ts}

@@ -18,6 +18,7 @@ interface GuardParams {
     user_id?: string;
     client_event_id?: string;
     expected_languages?: string[];
+    request_identity?: Record<string, unknown>;
     client_capabilities?: {
         capabilities: {
             has_feedback_handler: boolean;
@@ -67,10 +68,11 @@ interface LogParams {
     input: string;
     output: string;
     model: string;
-    usage?: Record<string, number>;
-    tokens?: Record<string, number>;
+    usage?: Record<string, unknown>;
+    tokens?: Record<string, unknown>;
     latency?: number;
     user_id?: string;
+    request_identity?: Record<string, unknown>;
     metadata?: Record<string, unknown>;
     event_type?: "start" | "complete" | "error" | "human_override" | "security_alert" | "security_block" | "security_policy_violation" | "transparency_badge_rendered";
     severity?: "info" | "warning" | "error" | "high";
@@ -179,6 +181,7 @@ declare class AgentID {
     guard(params: GuardParams, options?: RequestOptions): Promise<GuardResponse>;
     private sendIngest;
     private extractStreamChunkText;
+    private extractStreamChunkUsage;
     private wrapCompletion;
     /**
      * LOG: Sends telemetry after execution.
@@ -197,13 +200,14 @@ declare class AgentID {
      * Wrap an OpenAI client once; AgentID will automatically:
      * - run guard() before chat.completions.create
      * - measure latency
-     * - fire-and-forget ingest logging
+     * - persist ingest telemetry for the wrapped call
      */
     wrapOpenAI<T>(openai: T, options: {
         system_id: string;
         user_id?: string;
         expected_languages?: string[];
         expectedLanguages?: string[];
+        request_identity?: Record<string, unknown>;
         apiKey?: string;
         api_key?: string;
         resolveApiKey?: (request: Record<string, unknown>) => string | undefined;

package/dist/{chunk-JIQGHFHI.mjs → chunk-W37A4DPR.mjs}

@@ -1683,7 +1683,7 @@ function getInjectionScanner() {
 
 // src/sdk-version.ts
 var FALLBACK_SDK_VERSION = "js-0.0.0-dev";
-var AGENTID_SDK_VERSION_HEADER = "js-0.1.26".trim().length > 0 ? "js-0.1.26" : FALLBACK_SDK_VERSION;
+var AGENTID_SDK_VERSION_HEADER = "js-0.1.28".trim().length > 0 ? "js-0.1.28" : FALLBACK_SDK_VERSION;
 
 // src/local-security-enforcer.ts
 var DEFAULT_FAIL_OPEN_CONFIG = {
@@ -3047,6 +3047,13 @@ var AgentID = class {
     }
     return text;
   }
+  extractStreamChunkUsage(chunk) {
+    if (!chunk || typeof chunk !== "object") {
+      return void 0;
+    }
+    const usage = chunk.usage;
+    return usage && typeof usage === "object" && !Array.isArray(usage) ? usage : void 0;
+  }
   wrapCompletion(completion) {
     if (typeof completion === "string") {
       const masked = this.pii.anonymize(completion);
@@ -3070,7 +3077,9 @@ var AgentID = class {
     const source = completion;
     const collector = createCompletionChunkCollector();
     const extractStreamChunkText = this.extractStreamChunkText.bind(this);
+    const extractStreamChunkUsage = this.extractStreamChunkUsage.bind(this);
     const piiManager = this.pii;
+    let lastUsage;
     let resolveDone = null;
     let rejectDone = null;
     const done = new Promise((resolve, reject) => {
@@ -3085,6 +3094,10 @@ var AgentID = class {
             if (chunkText) {
               await collector.push(chunkText);
             }
+            const chunkUsage = extractStreamChunkUsage(chunk);
+            if (chunkUsage) {
+              lastUsage = chunkUsage;
+            }
             yield chunk;
           }
           await collector.close();
@@ -3094,7 +3107,8 @@ var AgentID = class {
             mode: "static",
             rawOutput,
             transformedOutput: masked.maskedText,
-            outputMasked: masked.maskedText !== rawOutput
+            outputMasked: masked.maskedText !== rawOutput,
+            usage: lastUsage
           });
         } catch (error) {
           await collector.abort(error);
@@ -3144,7 +3158,7 @@ var AgentID = class {
    * Wrap an OpenAI client once; AgentID will automatically:
    * - run guard() before chat.completions.create
    * - measure latency
-   * - fire-and-forget ingest logging
+   * - persist ingest telemetry for the wrapped call
    */
   wrapOpenAI(openai, options) {
     const systemId = options.system_id;
@@ -3240,6 +3254,7 @@ var AgentID = class {
                   user_id: options.user_id,
                   client_event_id: clientEventId,
                   expected_languages: expectedLanguages,
+                  request_identity: options.request_identity,
                   client_capabilities: this.buildClientCapabilities(
                     "openai",
                     false,
@@ -3315,10 +3330,11 @@ var AgentID = class {
                         event_id: canonicalClientEventId,
                         system_id: systemId,
                         user_id: options.user_id,
+                        request_identity: options.request_identity,
                         input: maskedText,
                         output: outputForLog,
                         model: adapter.getModelName(maskedReq),
-                        usage: void 0,
+                        usage: result.usage,
                         latency: modelLatencyMs2,
                         event_type: "complete",
                         metadata: {
@@ -3381,6 +3397,7 @@ var AgentID = class {
                     event_id: canonicalClientEventId,
                     system_id: systemId,
                     user_id: options.user_id,
+                    request_identity: options.request_identity,
                     input: maskedText,
                     output: outputForLog,
                     model,

package/dist/index.d.mts

@@ -1,5 +1,5 @@
-import { T as TransparencyMetadata } from './agentid-DviYzyAM.mjs';
-export { A as AgentID, D as DependencyError, G as GuardParams, a as GuardResponse, L as LogParams, P as PreparedInput, R as RequestOptions, S as SecurityBlockError } from './agentid-DviYzyAM.mjs';
+import { T as TransparencyMetadata } from './agentid-agvYW2vW.mjs';
+export { A as AgentID, D as DependencyError, G as GuardParams, a as GuardResponse, L as LogParams, P as PreparedInput, R as RequestOptions, S as SecurityBlockError } from './agentid-agvYW2vW.mjs';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 
 type PIIMapping = Record<string, string>;
@@ -16,7 +16,7 @@ declare class PIIManager {
     deanonymize(text: string, mapping: PIIMapping): string;
 }
 
-type TokenUsage = Record<string, number>;
+type TokenUsage = Record<string, unknown>;
 interface LLMAdapter {
     extractInput(req: unknown): string | null;
     getModelName(req: unknown, res?: unknown): string;

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { T as TransparencyMetadata } from './agentid-DviYzyAM.js';
-export { A as AgentID, D as DependencyError, G as GuardParams, a as GuardResponse, L as LogParams, P as PreparedInput, R as RequestOptions, S as SecurityBlockError } from './agentid-DviYzyAM.js';
+import { T as TransparencyMetadata } from './agentid-agvYW2vW.js';
+export { A as AgentID, D as DependencyError, G as GuardParams, a as GuardResponse, L as LogParams, P as PreparedInput, R as RequestOptions, S as SecurityBlockError } from './agentid-agvYW2vW.js';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 
 type PIIMapping = Record<string, string>;
@@ -16,7 +16,7 @@ declare class PIIManager {
     deanonymize(text: string, mapping: PIIMapping): string;
 }
 
-type TokenUsage = Record<string, number>;
+type TokenUsage = Record<string, unknown>;
 interface LLMAdapter {
     extractInput(req: unknown): string | null;
     getModelName(req: unknown, res?: unknown): string;

package/dist/index.js

@@ -86,7 +86,7 @@ var OpenAIAdapter = class {
 
 // src/sdk-version.ts
 var FALLBACK_SDK_VERSION = "js-0.0.0-dev";
-var AGENTID_SDK_VERSION_HEADER = "js-0.1.26".trim().length > 0 ? "js-0.1.26" : FALLBACK_SDK_VERSION;
+var AGENTID_SDK_VERSION_HEADER = "js-0.1.28".trim().length > 0 ? "js-0.1.28" : FALLBACK_SDK_VERSION;
 
 // src/pii-national-identifiers.ts
 var MAX_CANDIDATES_PER_RULE = 256;
@@ -3091,6 +3091,13 @@ var AgentID = class {
     }
     return text;
   }
+  extractStreamChunkUsage(chunk) {
+    if (!chunk || typeof chunk !== "object") {
+      return void 0;
+    }
+    const usage = chunk.usage;
+    return usage && typeof usage === "object" && !Array.isArray(usage) ? usage : void 0;
+  }
   wrapCompletion(completion) {
     if (typeof completion === "string") {
       const masked = this.pii.anonymize(completion);
@@ -3114,7 +3121,9 @@ var AgentID = class {
     const source = completion;
     const collector = createCompletionChunkCollector();
     const extractStreamChunkText = this.extractStreamChunkText.bind(this);
+    const extractStreamChunkUsage = this.extractStreamChunkUsage.bind(this);
     const piiManager = this.pii;
+    let lastUsage;
     let resolveDone = null;
     let rejectDone = null;
     const done = new Promise((resolve, reject) => {
@@ -3129,6 +3138,10 @@ var AgentID = class {
             if (chunkText) {
               await collector.push(chunkText);
             }
+            const chunkUsage = extractStreamChunkUsage(chunk);
+            if (chunkUsage) {
+              lastUsage = chunkUsage;
+            }
             yield chunk;
           }
           await collector.close();
@@ -3138,7 +3151,8 @@ var AgentID = class {
             mode: "static",
             rawOutput,
             transformedOutput: masked.maskedText,
-            outputMasked: masked.maskedText !== rawOutput
+            outputMasked: masked.maskedText !== rawOutput,
+            usage: lastUsage
           });
         } catch (error) {
           await collector.abort(error);
@@ -3188,7 +3202,7 @@ var AgentID = class {
    * Wrap an OpenAI client once; AgentID will automatically:
    * - run guard() before chat.completions.create
    * - measure latency
-   * - fire-and-forget ingest logging
+   * - persist ingest telemetry for the wrapped call
    */
   wrapOpenAI(openai, options) {
     const systemId = options.system_id;
@@ -3284,6 +3298,7 @@ var AgentID = class {
                   user_id: options.user_id,
                   client_event_id: clientEventId,
                   expected_languages: expectedLanguages,
+                  request_identity: options.request_identity,
                   client_capabilities: this.buildClientCapabilities(
                     "openai",
                     false,
@@ -3359,10 +3374,11 @@ var AgentID = class {
                         event_id: canonicalClientEventId,
                         system_id: systemId,
                         user_id: options.user_id,
+                        request_identity: options.request_identity,
                         input: maskedText,
                         output: outputForLog,
                         model: adapter.getModelName(maskedReq),
-                        usage: void 0,
+                        usage: result.usage,
                         latency: modelLatencyMs2,
                         event_type: "complete",
                         metadata: {
@@ -3425,6 +3441,7 @@ var AgentID = class {
                     event_id: canonicalClientEventId,
                     system_id: systemId,
                     user_id: options.user_id,
+                    request_identity: options.request_identity,
                     input: maskedText,
                     output: outputForLog,
                     model,

package/dist/index.mjs

@@ -7,7 +7,7 @@ import {
   SecurityBlockError,
   getInjectionScanner,
   scanWithRegex
-} from "./chunk-JIQGHFHI.mjs";
+} from "./chunk-W37A4DPR.mjs";
 
 // src/transparency-badge.tsx
 import * as React from "react";

package/dist/langchain.d.mts

@@ -1,5 +1,5 @@
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
-import { A as AgentID } from './agentid-DviYzyAM.mjs';
+import { A as AgentID } from './agentid-agvYW2vW.mjs';
 
 /**
  * LangChainJS callback handler (dependency-free shape).

package/dist/langchain.d.ts

@@ -1,5 +1,5 @@
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
-import { A as AgentID } from './agentid-DviYzyAM.js';
+import { A as AgentID } from './agentid-agvYW2vW.js';
 
 /**
  * LangChainJS callback handler (dependency-free shape).

package/dist/langchain.js

@@ -27,7 +27,7 @@ var import_base = require("@langchain/core/callbacks/base");
 
 // src/sdk-version.ts
 var FALLBACK_SDK_VERSION = "js-0.0.0-dev";
-var AGENTID_SDK_VERSION_HEADER = "js-0.1.26".trim().length > 0 ? "js-0.1.26" : FALLBACK_SDK_VERSION;
+var AGENTID_SDK_VERSION_HEADER = "js-0.1.28".trim().length > 0 ? "js-0.1.28" : FALLBACK_SDK_VERSION;
 
 // src/pii-national-identifiers.ts
 var REGION_ANCHORS = {

package/dist/langchain.mjs

@@ -1,6 +1,6 @@
 import {
   SecurityBlockError
-} from "./chunk-JIQGHFHI.mjs";
+} from "./chunk-W37A4DPR.mjs";
 
 // src/langchain.ts
 import { BaseCallbackHandler } from "@langchain/core/callbacks/base";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentid-sdk",
-  "version": "0.1.26",
+  "version": "0.1.28",
   "description": "AgentID JavaScript/TypeScript SDK for guard, ingest, tracing, and analytics.",
   "license": "MIT",
   "homepage": "https://agentid.ai",
@@ -59,8 +59,5 @@
     "@types/react": "^19.2.2",
     "tsup": "^8.3.5",
     "typescript": "^5.0.0"
-  },
-  "dependencies": {
-    "agentid-sdk": "^0.1.21"
   }
 }