agentid-sdk 0.1.34 → 0.1.35

package/dist/langchain.mjs

@@ -1,9 +1,11 @@
 import {
+  PIIManager,
   SecurityBlockError
-} from "./chunk-FCOSLPMF.mjs";
+} from "./chunk-XFSIAH3F.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;
@@ -369,7 +448,11 @@ 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
     });
     logCallbackDebug("handleLLMStart state_set", {
       runId: id,
@@ -458,7 +541,11 @@ 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
     });
     logCallbackDebug("handleChatModelStart state_set", {
       runId: id,
@@ -480,7 +567,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,13 +591,19 @@ 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,

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-IonlG0NB.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-IonlG0NB.js';
 
 type AgentIDTransparencyBadgeTelemetry = {
     systemId: string;

package/package.json

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