agentid-sdk 0.1.37 → 0.1.40

package/dist/index.mjs

@@ -1,20 +1,85 @@
 import {
   AgentID,
+  AgentIDWorkflowStep,
+  AgentIDWorkflowTrail,
   DependencyError,
   InjectionScanner,
   OpenAIAdapter,
   PIIManager,
   SecurityBlockError,
+  createAgentIdCorrelationId,
+  createAgentIdOperationLog,
+  createAgentIdTelemetryContext,
+  createAgentIdWorkflowTrail,
   getInjectionScanner,
   scanWithRegex
-} from "./chunk-HWES3LI2.mjs";
+} from "./chunk-25SZBEYX.mjs";
+
+// src/message-history.ts
+function isPlainRecord(value) {
+  if (!value || typeof value !== "object") return false;
+  const proto = Object.getPrototypeOf(value);
+  return proto === Object.prototype || proto === null;
+}
+function protectMessageHistory(messages, options = { pii: true, secrets: true }) {
+  const piiManager = new PIIManager();
+  let textPartsCount = 0;
+  let transformedTextPartsCount = 0;
+  const protectString = (value) => {
+    textPartsCount += 1;
+    const masked = piiManager.anonymize(value, options).maskedText;
+    if (masked !== value) {
+      transformedTextPartsCount += 1;
+    }
+    return masked;
+  };
+  const visit = (value, key) => {
+    if (typeof value === "string") {
+      return key === "content" || key === "text" ? protectString(value) : value;
+    }
+    if (Array.isArray(value)) {
+      let changed = false;
+      const next = value.map((item) => {
+        const protectedItem = visit(item);
+        if (protectedItem !== item) changed = true;
+        return protectedItem;
+      });
+      return changed ? next : value;
+    }
+    if (isPlainRecord(value)) {
+      let changed = false;
+      const next = {};
+      for (const [entryKey, entryValue] of Object.entries(value)) {
+        const protectedValue = visit(entryValue, entryKey);
+        next[entryKey] = protectedValue;
+        if (protectedValue !== entryValue) changed = true;
+      }
+      return changed ? next : value;
+    }
+    return value;
+  };
+  const protectedMessages = visit(messages);
+  return {
+    messages: protectedMessages,
+    transformed: protectedMessages !== messages,
+    textPartsCount,
+    transformedTextPartsCount
+  };
+}
 export {
   AgentID,
+  AgentIDWorkflowStep,
+  AgentIDWorkflowTrail,
   DependencyError,
   InjectionScanner,
   OpenAIAdapter,
   PIIManager,
   SecurityBlockError,
+  createAgentIdCorrelationId,
+  createAgentIdOperationLog,
+  createAgentIdTelemetryContext,
+  createAgentIdWorkflowTrail,
   getInjectionScanner,
+  protectMessageHistory,
   scanWithRegex
 };

package/dist/langchain.d.mts

@@ -1,5 +1,5 @@
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
-import { A as AgentID } from './agentid-JQx2Iy7B.mjs';
+import { b as AgentID, j as AgentTelemetryContext } from './agentid-BWlN5KCq.mjs';
 
 /**
  * LangChainJS callback handler (dependency-free shape).
@@ -13,22 +13,36 @@ declare class AgentIDCallbackHandler extends BaseCallbackHandler {
     private systemId;
     private apiKeyOverride?;
     private expectedLanguages?;
+    private telemetry?;
     private runs;
+    private toolRuns;
+    private chainRuns;
     constructor(agent: AgentID, options: {
         system_id: string;
         expected_languages?: string[];
         expectedLanguages?: string[];
         apiKey?: string;
         api_key?: string;
+        telemetry?: AgentTelemetryContext;
     });
     private get requestOptions();
     private getLangchainCapabilities;
     private resolvePreparedPiiMaskingEnabled;
+    private resolvePreparedSecretMaskingEnabled;
     private preflight;
+    private resolveTelemetry;
+    private resolveParentEventContext;
+    private logWorkflowOperation;
     handleLLMStart(serialized: unknown, prompts: unknown, runId?: string, _parentRunId?: string, extraParams?: unknown): Promise<void>;
     handleChatModelStart(serialized: unknown, messages: unknown, runId?: string, _parentRunId?: string, extraParams?: unknown): Promise<void>;
     handleLLMEnd(output: unknown, runId?: string): Promise<void>;
     handleLLMError(err: unknown, runId?: string): Promise<void>;
+    handleToolStart(serialized: unknown, _input: string, runId?: string, parentRunId?: string, ...rest: unknown[]): Promise<void>;
+    handleToolEnd(_output: unknown, runId?: string): Promise<void>;
+    handleToolError(err: unknown, runId?: string): Promise<void>;
+    handleChainStart(serialized: unknown, _inputs: Record<string, unknown>, runId?: string, parentRunId?: string, ...rest: unknown[]): Promise<void>;
+    handleChainEnd(_outputs: Record<string, unknown>, runId?: string): Promise<void>;
+    handleChainError(err: unknown, runId?: string): Promise<void>;
 }
 
 export { AgentIDCallbackHandler };

package/dist/langchain.d.ts

@@ -1,5 +1,5 @@
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
-import { A as AgentID } from './agentid-JQx2Iy7B.js';
+import { b as AgentID, j as AgentTelemetryContext } from './agentid-BWlN5KCq.js';
 
 /**
  * LangChainJS callback handler (dependency-free shape).
@@ -13,22 +13,36 @@ declare class AgentIDCallbackHandler extends BaseCallbackHandler {
     private systemId;
     private apiKeyOverride?;
     private expectedLanguages?;
+    private telemetry?;
     private runs;
+    private toolRuns;
+    private chainRuns;
     constructor(agent: AgentID, options: {
         system_id: string;
         expected_languages?: string[];
         expectedLanguages?: string[];
         apiKey?: string;
         api_key?: string;
+        telemetry?: AgentTelemetryContext;
     });
     private get requestOptions();
     private getLangchainCapabilities;
     private resolvePreparedPiiMaskingEnabled;
+    private resolvePreparedSecretMaskingEnabled;
     private preflight;
+    private resolveTelemetry;
+    private resolveParentEventContext;
+    private logWorkflowOperation;
     handleLLMStart(serialized: unknown, prompts: unknown, runId?: string, _parentRunId?: string, extraParams?: unknown): Promise<void>;
     handleChatModelStart(serialized: unknown, messages: unknown, runId?: string, _parentRunId?: string, extraParams?: unknown): Promise<void>;
     handleLLMEnd(output: unknown, runId?: string): Promise<void>;
     handleLLMError(err: unknown, runId?: string): Promise<void>;
+    handleToolStart(serialized: unknown, _input: string, runId?: string, parentRunId?: string, ...rest: unknown[]): Promise<void>;
+    handleToolEnd(_output: unknown, runId?: string): Promise<void>;
+    handleToolError(err: unknown, runId?: string): Promise<void>;
+    handleChainStart(serialized: unknown, _inputs: Record<string, unknown>, runId?: string, parentRunId?: string, ...rest: unknown[]): Promise<void>;
+    handleChainEnd(_outputs: Record<string, unknown>, runId?: string): Promise<void>;
+    handleChainError(err: unknown, runId?: string): Promise<void>;
 }
 
 export { AgentIDCallbackHandler };