agentid-sdk 0.1.34 → 0.1.36

package/dist/langchain.mjs

@@ -1,9 +1,11 @@
 import {
+  PIIManager,
   SecurityBlockError
-} from "./chunk-FCOSLPMF.mjs";
+} from "./chunk-TY4AXWGS.mjs";
 
 // src/langchain.ts
 import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
+var piiManager = new PIIManager();
 function safeString(val) {
   return typeof val === "string" ? val : "";
 }
@@ -205,6 +207,83 @@ function extractOutputText(output) {
   const text = first?.text ?? first?.message?.content;
   return typeof text === "string" ? text : "";
 }
+function normalizePiiMapping(value) {
+  if (!value || typeof value !== "object" || Array.isArray(value)) {
+    return void 0;
+  }
+  const entries = Object.entries(value).filter(
+    ([placeholder, replacement]) => placeholder.length > 0 && typeof replacement === "string"
+  );
+  if (entries.length === 0) {
+    return void 0;
+  }
+  return Object.fromEntries(entries);
+}
+function deanonymizeText(text, mapping) {
+  if (!text || !mapping) {
+    return text;
+  }
+  return piiManager.deanonymize(text, mapping);
+}
+function deanonymizeContent(content, mapping) {
+  if (!mapping) {
+    return content;
+  }
+  if (typeof content === "string") {
+    return deanonymizeText(content, mapping);
+  }
+  if (!Array.isArray(content)) {
+    return content;
+  }
+  return content.map((item) => {
+    if (typeof item === "string") {
+      return deanonymizeText(item, mapping);
+    }
+    if (!item || typeof item !== "object") {
+      return item;
+    }
+    const record = { ...item };
+    if (typeof record.text === "string") {
+      record.text = deanonymizeText(record.text, mapping);
+    }
+    if (typeof record.content === "string") {
+      record.content = deanonymizeText(record.content, mapping);
+    }
+    return record;
+  });
+}
+function deanonymizeLangChainOutputForClient(output, mapping) {
+  if (!mapping) {
+    return;
+  }
+  const generations = output?.generations;
+  if (!Array.isArray(generations)) {
+    return;
+  }
+  for (const generationGroup of generations) {
+    if (!Array.isArray(generationGroup)) {
+      continue;
+    }
+    for (const generation of generationGroup) {
+      if (!generation || typeof generation !== "object") {
+        continue;
+      }
+      if (typeof generation.text === "string") {
+        generation.text = deanonymizeText(generation.text, mapping);
+      }
+      const message = generation.message;
+      if (message && typeof message === "object") {
+        const typedMessage = message;
+        if ("content" in typedMessage) {
+          typedMessage.content = deanonymizeContent(typedMessage.content, mapping);
+        }
+        if (typeof typedMessage.text === "string") {
+          typedMessage.text = deanonymizeText(typedMessage.text, mapping);
+        }
+      }
+    }
+  }
+}
 function extractTokenUsage(output) {
   const llmOutput = output?.llmOutput ?? output?.llm_output;
   const usage = llmOutput?.tokenUsage ?? llmOutput?.token_usage ?? llmOutput?.usage ?? void 0;
@@ -269,16 +348,23 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
   get requestOptions() {
     return this.apiKeyOverride ? { apiKey: this.apiKeyOverride } : void 0;
   }
-  getLangchainCapabilities() {
-    const piiMaskingEnabled = this.agent.getEffectivePiiMasking(this.requestOptions);
+  getLangchainCapabilities(piiMaskingEnabled) {
+    const resolvedPiiMaskingEnabled = typeof piiMaskingEnabled === "boolean" ? piiMaskingEnabled : this.agent.getEffectivePiiMasking(this.requestOptions);
     return {
       capabilities: {
         has_feedback_handler: true,
-        pii_masking_enabled: piiMaskingEnabled,
+        pii_masking_enabled: resolvedPiiMaskingEnabled,
         framework: "langchain"
       }
     };
   }
+  resolvePreparedPiiMaskingEnabled(prepared) {
+    const agentWithResolvedConfig = this.agent;
+    if (typeof agentWithResolvedConfig.getEffectivePiiMaskingForConfig === "function") {
+      return agentWithResolvedConfig.getEffectivePiiMaskingForConfig(prepared.capabilityConfig);
+    }
+    return this.agent.getEffectivePiiMasking(this.requestOptions);
+  }
   async preflight(input, stream, clientEventId) {
     const prepared = await this.agent.prepareInputForDispatch({
       input,
@@ -299,6 +385,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
     const stream = extractStreamFlag(serialized, extraParams);
     const prepared = await this.preflight(input, stream, requestedClientEventId);
     const sanitizedInput = prepared.sanitizedInput;
+    const piiMaskingEnabled = this.resolvePreparedPiiMaskingEnabled(prepared);
     if (sanitizedInput !== input) {
       const mutated = setPromptInPrompts(prompts, sanitizedInput);
       if (!mutated) {
@@ -315,7 +402,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
       model: modelName,
       client_event_id: requestedClientEventId,
       expected_languages: this.expectedLanguages,
-      client_capabilities: this.getLangchainCapabilities()
+      client_capabilities: this.getLangchainCapabilities(piiMaskingEnabled)
     }, this.requestOptions);
     let transformedForRun = sanitizedInput;
     let sdkLocalScanMs = prepared.sdkLocalScanMs;
@@ -369,7 +456,12 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
       model: modelName,
       clientEventId: canonicalClientEventId,
       guardEventId,
-      transparency
+      transparency,
+      piiMapping: normalizePiiMapping(prepared.piiMapping),
+      shouldDeanonymize: prepared.shouldDeanonymize === true,
+      responseStreamed: stream,
+      sdkConfigVersion: prepared.capabilityConfig?.version ?? null,
+      piiMaskingEnabled
     });
     logCallbackDebug("handleLLMStart state_set", {
       runId: id,
@@ -388,6 +480,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
     const stream = extractStreamFlag(serialized, extraParams);
     const prepared = await this.preflight(input, stream, requestedClientEventId);
     const sanitizedInput = prepared.sanitizedInput;
+    const piiMaskingEnabled = this.resolvePreparedPiiMaskingEnabled(prepared);
     if (sanitizedInput !== input) {
       const mutated = setPromptInMessages(messages, sanitizedInput);
       if (!mutated) {
@@ -404,7 +497,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
       model: modelName,
       client_event_id: requestedClientEventId,
       expected_languages: this.expectedLanguages,
-      client_capabilities: this.getLangchainCapabilities()
+      client_capabilities: this.getLangchainCapabilities(piiMaskingEnabled)
     }, this.requestOptions);
     let transformedForRun = sanitizedInput;
     let sdkLocalScanMs = prepared.sdkLocalScanMs;
@@ -458,7 +551,12 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
       model: modelName,
       clientEventId: canonicalClientEventId,
       guardEventId,
-      transparency
+      transparency,
+      piiMapping: normalizePiiMapping(prepared.piiMapping),
+      shouldDeanonymize: prepared.shouldDeanonymize === true,
+      responseStreamed: stream,
+      sdkConfigVersion: prepared.capabilityConfig?.version ?? null,
+      piiMaskingEnabled
     });
     logCallbackDebug("handleChatModelStart state_set", {
       runId: id,
@@ -480,7 +578,11 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
       0,
       Date.now() - (state.pipelineStartedAtMs ?? state.startedAtMs)
     );
-    const outText = extractOutputText(output);
+    const maskedOutputText = extractOutputText(output);
+    if (state.shouldDeanonymize && state.piiMapping) {
+      deanonymizeLangChainOutputForClient(output, state.piiMapping);
+    }
+    const clientOutputText = extractOutputText(output);
     const usage = extractTokenUsage(output);
     const metadata = {};
     if (state.clientEventId) {
@@ -500,19 +602,25 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
     if (state.transparency) {
       metadata.transparency = state.transparency;
     }
+    if (typeof state.sdkConfigVersion === "number" && Number.isFinite(state.sdkConfigVersion)) {
+      metadata.sdk_config_version = Math.trunc(state.sdkConfigVersion);
+    }
+    metadata.response_streamed = state.responseStreamed === true;
+    metadata.transformed_output = maskedOutputText;
+    metadata.output_masked = maskedOutputText !== clientOutputText;
     metadata.model_latency_ms = modelLatencyMs;
     metadata.total_pipeline_latency_ms = totalPipelineLatencyMs;
     const resolvedModel = state.model ?? extractModelFromOutput(output) ?? "unknown";
     await this.agent.log({
       system_id: this.systemId,
       input: state.input,
-      output: outText,
+      output: maskedOutputText,
       event_id: state.clientEventId,
       model: resolvedModel,
       usage,
       latency: modelLatencyMs,
       metadata: Object.keys(metadata).length > 0 ? metadata : void 0,
-      client_capabilities: this.getLangchainCapabilities()
+      client_capabilities: this.getLangchainCapabilities(state.piiMaskingEnabled)
     }, this.requestOptions);
     logCallbackDebug("handleLLMEnd logged", {
       runId: id,
@@ -554,7 +662,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
       event_type: "error",
       severity: "error",
       metadata,
-      client_capabilities: this.getLangchainCapabilities()
+      client_capabilities: this.getLangchainCapabilities(state?.piiMaskingEnabled)
     }, this.requestOptions);
   }
 };

package/dist/transparency-badge.d.mts

@@ -1,5 +1,5 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
-import { T as TransparencyMetadata } from './agentid-M5I7-YqI.mjs';
+import { T as TransparencyMetadata } from './agentid-JQx2Iy7B.mjs';
 
 type AgentIDTransparencyBadgeTelemetry = {
     systemId: string;

package/dist/transparency-badge.d.ts

@@ -1,5 +1,5 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
-import { T as TransparencyMetadata } from './agentid-M5I7-YqI.js';
+import { T as TransparencyMetadata } from './agentid-JQx2Iy7B.js';
 
 type AgentIDTransparencyBadgeTelemetry = {
     systemId: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentid-sdk",
-  "version": "0.1.34",
+  "version": "0.1.36",
   "description": "AgentID JavaScript/TypeScript SDK for guard, ingest, tracing, and analytics.",
   "license": "MIT",
   "homepage": "https://agentid.ai",