agentid-sdk 0.1.35 → 0.1.37
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +16 -14
- package/dist/{agentid-IonlG0NB.d.mts → agentid-JQx2Iy7B.d.mts} +1 -0
- package/dist/{agentid-IonlG0NB.d.ts → agentid-JQx2Iy7B.d.ts} +1 -0
- package/dist/{chunk-XFSIAH3F.mjs → chunk-HWES3LI2.mjs} +8 -2
- package/dist/index.d.mts +2 -2
- package/dist/index.d.ts +2 -2
- package/dist/index.js +8 -2
- package/dist/index.mjs +1 -1
- package/dist/langchain.d.mts +2 -1
- package/dist/langchain.d.ts +2 -1
- package/dist/langchain.js +21 -10
- package/dist/langchain.mjs +21 -10
- package/dist/transparency-badge.d.mts +1 -1
- package/dist/transparency-badge.d.ts +1 -1
- package/package.json +1 -1
package/README.md
CHANGED
|
@@ -87,18 +87,16 @@ await agent.log({
|
|
|
87
87
|
npm install agentid-sdk openai
|
|
88
88
|
```
|
|
89
89
|
|
|
90
|
-
```ts
|
|
91
|
-
import OpenAI from "openai";
|
|
92
|
-
import { AgentID } from "agentid-sdk";
|
|
93
|
-
|
|
94
|
-
const agent = new AgentID(
|
|
95
|
-
|
|
96
|
-
});
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
system_id: process.env.AGENTID_SYSTEM_ID!,
|
|
101
|
-
user_id: "customer-123",
|
|
90
|
+
```ts
|
|
91
|
+
import OpenAI from "openai";
|
|
92
|
+
import { AgentID } from "agentid-sdk";
|
|
93
|
+
|
|
94
|
+
const agent = new AgentID();
|
|
95
|
+
|
|
96
|
+
const openai = new OpenAI({ apiKey: process.env.OPENAI_API_KEY! });
|
|
97
|
+
const secured = agent.wrapOpenAI(openai, {
|
|
98
|
+
system_id: process.env.AGENTID_SYSTEM_ID!,
|
|
99
|
+
user_id: "customer-123",
|
|
102
100
|
expected_languages: ["en"],
|
|
103
101
|
});
|
|
104
102
|
|
|
@@ -107,8 +105,12 @@ const response = await secured.chat.completions.create({
|
|
|
107
105
|
messages: [{ role: "user", content: "What is the capital of the Czech Republic?" }],
|
|
108
106
|
});
|
|
109
107
|
|
|
110
|
-
console.log(response.choices[0]?.message?.content ?? "");
|
|
111
|
-
```
|
|
108
|
+
console.log(response.choices[0]?.message?.content ?? "");
|
|
109
|
+
```
|
|
110
|
+
|
|
111
|
+
By default, official AgentID SDK integrations inherit `enable_sdk_pii_masking`
|
|
112
|
+
from the dashboard/runtime config. You only need to set `piiMasking: true` in
|
|
113
|
+
code if you want to force local masking on even when the dashboard policy is off.
|
|
112
114
|
|
|
113
115
|
Wrapped OpenAI calls persist telemetry for both regular and streamed completions. For `stream: true`, logging happens when the stream finishes.
|
|
114
116
|
|
|
@@ -157,6 +157,7 @@ declare class AgentID {
|
|
|
157
157
|
get piiMasking(): boolean | undefined;
|
|
158
158
|
private resolveEffectivePiiMasking;
|
|
159
159
|
getEffectivePiiMasking(options?: RequestOptions): boolean;
|
|
160
|
+
getEffectivePiiMaskingForConfig(capabilityConfig?: CapabilityConfig): boolean;
|
|
160
161
|
private buildClientCapabilities;
|
|
161
162
|
private resolveApiKey;
|
|
162
163
|
private resolveClientEventId;
|
|
@@ -157,6 +157,7 @@ declare class AgentID {
|
|
|
157
157
|
get piiMasking(): boolean | undefined;
|
|
158
158
|
private resolveEffectivePiiMasking;
|
|
159
159
|
getEffectivePiiMasking(options?: RequestOptions): boolean;
|
|
160
|
+
getEffectivePiiMaskingForConfig(capabilityConfig?: CapabilityConfig): boolean;
|
|
160
161
|
private buildClientCapabilities;
|
|
161
162
|
private resolveApiKey;
|
|
162
163
|
private resolveClientEventId;
|
|
@@ -1810,7 +1810,7 @@ function getInjectionScanner() {
|
|
|
1810
1810
|
|
|
1811
1811
|
// src/sdk-version.ts
|
|
1812
1812
|
var FALLBACK_SDK_VERSION = "js-0.0.0-dev";
|
|
1813
|
-
var AGENTID_SDK_VERSION_HEADER = "js-0.1.
|
|
1813
|
+
var AGENTID_SDK_VERSION_HEADER = "js-0.1.37".trim().length > 0 ? "js-0.1.37" : FALLBACK_SDK_VERSION;
|
|
1814
1814
|
|
|
1815
1815
|
// src/local-security-enforcer.ts
|
|
1816
1816
|
var DEFAULT_FAIL_OPEN_CONFIG = {
|
|
@@ -2604,14 +2604,20 @@ var AgentID = class {
|
|
|
2604
2604
|
return this.configuredPiiMasking ?? void 0;
|
|
2605
2605
|
}
|
|
2606
2606
|
resolveEffectivePiiMasking(config) {
|
|
2607
|
+
if (config?.enable_sdk_pii_masking === true) {
|
|
2608
|
+
return true;
|
|
2609
|
+
}
|
|
2607
2610
|
if (this.configuredPiiMasking !== null) {
|
|
2608
2611
|
return this.configuredPiiMasking;
|
|
2609
2612
|
}
|
|
2610
|
-
return
|
|
2613
|
+
return false;
|
|
2611
2614
|
}
|
|
2612
2615
|
getEffectivePiiMasking(options) {
|
|
2613
2616
|
return this.resolveEffectivePiiMasking(this.getCachedCapabilityConfig(options));
|
|
2614
2617
|
}
|
|
2618
|
+
getEffectivePiiMaskingForConfig(capabilityConfig) {
|
|
2619
|
+
return this.resolveEffectivePiiMasking(capabilityConfig);
|
|
2620
|
+
}
|
|
2615
2621
|
buildClientCapabilities(framework = "js_sdk", hasFeedbackHandler = false, capabilityConfig) {
|
|
2616
2622
|
return {
|
|
2617
2623
|
capabilities: {
|
package/dist/index.d.mts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { P as PIIManager } from './agentid-
|
|
2
|
-
export { A as AgentID, D as DependencyError, G as GuardParams, a as GuardResponse, L as LogParams, b as PIIMapping, c as PreparedInput, R as RequestOptions, S as SecurityBlockError, T as TransparencyMetadata } from './agentid-
|
|
1
|
+
import { P as PIIManager } from './agentid-JQx2Iy7B.mjs';
|
|
2
|
+
export { A as AgentID, D as DependencyError, G as GuardParams, a as GuardResponse, L as LogParams, b as PIIMapping, c as PreparedInput, R as RequestOptions, S as SecurityBlockError, T as TransparencyMetadata } from './agentid-JQx2Iy7B.mjs';
|
|
3
3
|
|
|
4
4
|
type TokenUsage = Record<string, unknown>;
|
|
5
5
|
interface LLMAdapter {
|
package/dist/index.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { P as PIIManager } from './agentid-
|
|
2
|
-
export { A as AgentID, D as DependencyError, G as GuardParams, a as GuardResponse, L as LogParams, b as PIIMapping, c as PreparedInput, R as RequestOptions, S as SecurityBlockError, T as TransparencyMetadata } from './agentid-
|
|
1
|
+
import { P as PIIManager } from './agentid-JQx2Iy7B.js';
|
|
2
|
+
export { A as AgentID, D as DependencyError, G as GuardParams, a as GuardResponse, L as LogParams, b as PIIMapping, c as PreparedInput, R as RequestOptions, S as SecurityBlockError, T as TransparencyMetadata } from './agentid-JQx2Iy7B.js';
|
|
3
3
|
|
|
4
4
|
type TokenUsage = Record<string, unknown>;
|
|
5
5
|
interface LLMAdapter {
|
package/dist/index.js
CHANGED
|
@@ -75,7 +75,7 @@ var OpenAIAdapter = class {
|
|
|
75
75
|
|
|
76
76
|
// src/sdk-version.ts
|
|
77
77
|
var FALLBACK_SDK_VERSION = "js-0.0.0-dev";
|
|
78
|
-
var AGENTID_SDK_VERSION_HEADER = "js-0.1.
|
|
78
|
+
var AGENTID_SDK_VERSION_HEADER = "js-0.1.37".trim().length > 0 ? "js-0.1.37" : FALLBACK_SDK_VERSION;
|
|
79
79
|
|
|
80
80
|
// src/pii-national-identifiers.ts
|
|
81
81
|
var MAX_CANDIDATES_PER_RULE = 256;
|
|
@@ -2637,14 +2637,20 @@ var AgentID = class {
|
|
|
2637
2637
|
return this.configuredPiiMasking ?? void 0;
|
|
2638
2638
|
}
|
|
2639
2639
|
resolveEffectivePiiMasking(config) {
|
|
2640
|
+
if (config?.enable_sdk_pii_masking === true) {
|
|
2641
|
+
return true;
|
|
2642
|
+
}
|
|
2640
2643
|
if (this.configuredPiiMasking !== null) {
|
|
2641
2644
|
return this.configuredPiiMasking;
|
|
2642
2645
|
}
|
|
2643
|
-
return
|
|
2646
|
+
return false;
|
|
2644
2647
|
}
|
|
2645
2648
|
getEffectivePiiMasking(options) {
|
|
2646
2649
|
return this.resolveEffectivePiiMasking(this.getCachedCapabilityConfig(options));
|
|
2647
2650
|
}
|
|
2651
|
+
getEffectivePiiMaskingForConfig(capabilityConfig) {
|
|
2652
|
+
return this.resolveEffectivePiiMasking(capabilityConfig);
|
|
2653
|
+
}
|
|
2648
2654
|
buildClientCapabilities(framework = "js_sdk", hasFeedbackHandler = false, capabilityConfig) {
|
|
2649
2655
|
return {
|
|
2650
2656
|
capabilities: {
|
package/dist/index.mjs
CHANGED
package/dist/langchain.d.mts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
|
|
2
|
-
import { A as AgentID } from './agentid-
|
|
2
|
+
import { A as AgentID } from './agentid-JQx2Iy7B.mjs';
|
|
3
3
|
|
|
4
4
|
/**
|
|
5
5
|
* LangChainJS callback handler (dependency-free shape).
|
|
@@ -23,6 +23,7 @@ declare class AgentIDCallbackHandler extends BaseCallbackHandler {
|
|
|
23
23
|
});
|
|
24
24
|
private get requestOptions();
|
|
25
25
|
private getLangchainCapabilities;
|
|
26
|
+
private resolvePreparedPiiMaskingEnabled;
|
|
26
27
|
private preflight;
|
|
27
28
|
handleLLMStart(serialized: unknown, prompts: unknown, runId?: string, _parentRunId?: string, extraParams?: unknown): Promise<void>;
|
|
28
29
|
handleChatModelStart(serialized: unknown, messages: unknown, runId?: string, _parentRunId?: string, extraParams?: unknown): Promise<void>;
|
package/dist/langchain.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
|
|
2
|
-
import { A as AgentID } from './agentid-
|
|
2
|
+
import { A as AgentID } from './agentid-JQx2Iy7B.js';
|
|
3
3
|
|
|
4
4
|
/**
|
|
5
5
|
* LangChainJS callback handler (dependency-free shape).
|
|
@@ -23,6 +23,7 @@ declare class AgentIDCallbackHandler extends BaseCallbackHandler {
|
|
|
23
23
|
});
|
|
24
24
|
private get requestOptions();
|
|
25
25
|
private getLangchainCapabilities;
|
|
26
|
+
private resolvePreparedPiiMaskingEnabled;
|
|
26
27
|
private preflight;
|
|
27
28
|
handleLLMStart(serialized: unknown, prompts: unknown, runId?: string, _parentRunId?: string, extraParams?: unknown): Promise<void>;
|
|
28
29
|
handleChatModelStart(serialized: unknown, messages: unknown, runId?: string, _parentRunId?: string, extraParams?: unknown): Promise<void>;
|
package/dist/langchain.js
CHANGED
|
@@ -27,7 +27,7 @@ var import_base = require("@langchain/core/callbacks/base");
|
|
|
27
27
|
|
|
28
28
|
// src/sdk-version.ts
|
|
29
29
|
var FALLBACK_SDK_VERSION = "js-0.0.0-dev";
|
|
30
|
-
var AGENTID_SDK_VERSION_HEADER = "js-0.1.
|
|
30
|
+
var AGENTID_SDK_VERSION_HEADER = "js-0.1.37".trim().length > 0 ? "js-0.1.37" : FALLBACK_SDK_VERSION;
|
|
31
31
|
|
|
32
32
|
// src/pii-national-identifiers.ts
|
|
33
33
|
var MAX_CANDIDATES_PER_RULE = 256;
|
|
@@ -1495,16 +1495,23 @@ var AgentIDCallbackHandler = class extends import_base.BaseCallbackHandler {
|
|
|
1495
1495
|
get requestOptions() {
|
|
1496
1496
|
return this.apiKeyOverride ? { apiKey: this.apiKeyOverride } : void 0;
|
|
1497
1497
|
}
|
|
1498
|
-
getLangchainCapabilities() {
|
|
1499
|
-
const
|
|
1498
|
+
getLangchainCapabilities(piiMaskingEnabled) {
|
|
1499
|
+
const resolvedPiiMaskingEnabled = typeof piiMaskingEnabled === "boolean" ? piiMaskingEnabled : this.agent.getEffectivePiiMasking(this.requestOptions);
|
|
1500
1500
|
return {
|
|
1501
1501
|
capabilities: {
|
|
1502
1502
|
has_feedback_handler: true,
|
|
1503
|
-
pii_masking_enabled:
|
|
1503
|
+
pii_masking_enabled: resolvedPiiMaskingEnabled,
|
|
1504
1504
|
framework: "langchain"
|
|
1505
1505
|
}
|
|
1506
1506
|
};
|
|
1507
1507
|
}
|
|
1508
|
+
resolvePreparedPiiMaskingEnabled(prepared) {
|
|
1509
|
+
const agentWithResolvedConfig = this.agent;
|
|
1510
|
+
if (typeof agentWithResolvedConfig.getEffectivePiiMaskingForConfig === "function") {
|
|
1511
|
+
return agentWithResolvedConfig.getEffectivePiiMaskingForConfig(prepared.capabilityConfig);
|
|
1512
|
+
}
|
|
1513
|
+
return this.agent.getEffectivePiiMasking(this.requestOptions);
|
|
1514
|
+
}
|
|
1508
1515
|
async preflight(input, stream, clientEventId) {
|
|
1509
1516
|
const prepared = await this.agent.prepareInputForDispatch({
|
|
1510
1517
|
input,
|
|
@@ -1525,6 +1532,7 @@ var AgentIDCallbackHandler = class extends import_base.BaseCallbackHandler {
|
|
|
1525
1532
|
const stream = extractStreamFlag(serialized, extraParams);
|
|
1526
1533
|
const prepared = await this.preflight(input, stream, requestedClientEventId);
|
|
1527
1534
|
const sanitizedInput = prepared.sanitizedInput;
|
|
1535
|
+
const piiMaskingEnabled = this.resolvePreparedPiiMaskingEnabled(prepared);
|
|
1528
1536
|
if (sanitizedInput !== input) {
|
|
1529
1537
|
const mutated = setPromptInPrompts(prompts, sanitizedInput);
|
|
1530
1538
|
if (!mutated) {
|
|
@@ -1541,7 +1549,7 @@ var AgentIDCallbackHandler = class extends import_base.BaseCallbackHandler {
|
|
|
1541
1549
|
model: modelName,
|
|
1542
1550
|
client_event_id: requestedClientEventId,
|
|
1543
1551
|
expected_languages: this.expectedLanguages,
|
|
1544
|
-
client_capabilities: this.getLangchainCapabilities()
|
|
1552
|
+
client_capabilities: this.getLangchainCapabilities(piiMaskingEnabled)
|
|
1545
1553
|
}, this.requestOptions);
|
|
1546
1554
|
let transformedForRun = sanitizedInput;
|
|
1547
1555
|
let sdkLocalScanMs = prepared.sdkLocalScanMs;
|
|
@@ -1599,7 +1607,8 @@ var AgentIDCallbackHandler = class extends import_base.BaseCallbackHandler {
|
|
|
1599
1607
|
piiMapping: normalizePiiMapping(prepared.piiMapping),
|
|
1600
1608
|
shouldDeanonymize: prepared.shouldDeanonymize === true,
|
|
1601
1609
|
responseStreamed: stream,
|
|
1602
|
-
sdkConfigVersion: prepared.capabilityConfig?.version ?? null
|
|
1610
|
+
sdkConfigVersion: prepared.capabilityConfig?.version ?? null,
|
|
1611
|
+
piiMaskingEnabled
|
|
1603
1612
|
});
|
|
1604
1613
|
logCallbackDebug("handleLLMStart state_set", {
|
|
1605
1614
|
runId: id,
|
|
@@ -1618,6 +1627,7 @@ var AgentIDCallbackHandler = class extends import_base.BaseCallbackHandler {
|
|
|
1618
1627
|
const stream = extractStreamFlag(serialized, extraParams);
|
|
1619
1628
|
const prepared = await this.preflight(input, stream, requestedClientEventId);
|
|
1620
1629
|
const sanitizedInput = prepared.sanitizedInput;
|
|
1630
|
+
const piiMaskingEnabled = this.resolvePreparedPiiMaskingEnabled(prepared);
|
|
1621
1631
|
if (sanitizedInput !== input) {
|
|
1622
1632
|
const mutated = setPromptInMessages(messages, sanitizedInput);
|
|
1623
1633
|
if (!mutated) {
|
|
@@ -1634,7 +1644,7 @@ var AgentIDCallbackHandler = class extends import_base.BaseCallbackHandler {
|
|
|
1634
1644
|
model: modelName,
|
|
1635
1645
|
client_event_id: requestedClientEventId,
|
|
1636
1646
|
expected_languages: this.expectedLanguages,
|
|
1637
|
-
client_capabilities: this.getLangchainCapabilities()
|
|
1647
|
+
client_capabilities: this.getLangchainCapabilities(piiMaskingEnabled)
|
|
1638
1648
|
}, this.requestOptions);
|
|
1639
1649
|
let transformedForRun = sanitizedInput;
|
|
1640
1650
|
let sdkLocalScanMs = prepared.sdkLocalScanMs;
|
|
@@ -1692,7 +1702,8 @@ var AgentIDCallbackHandler = class extends import_base.BaseCallbackHandler {
|
|
|
1692
1702
|
piiMapping: normalizePiiMapping(prepared.piiMapping),
|
|
1693
1703
|
shouldDeanonymize: prepared.shouldDeanonymize === true,
|
|
1694
1704
|
responseStreamed: stream,
|
|
1695
|
-
sdkConfigVersion: prepared.capabilityConfig?.version ?? null
|
|
1705
|
+
sdkConfigVersion: prepared.capabilityConfig?.version ?? null,
|
|
1706
|
+
piiMaskingEnabled
|
|
1696
1707
|
});
|
|
1697
1708
|
logCallbackDebug("handleChatModelStart state_set", {
|
|
1698
1709
|
runId: id,
|
|
@@ -1756,7 +1767,7 @@ var AgentIDCallbackHandler = class extends import_base.BaseCallbackHandler {
|
|
|
1756
1767
|
usage,
|
|
1757
1768
|
latency: modelLatencyMs,
|
|
1758
1769
|
metadata: Object.keys(metadata).length > 0 ? metadata : void 0,
|
|
1759
|
-
client_capabilities: this.getLangchainCapabilities()
|
|
1770
|
+
client_capabilities: this.getLangchainCapabilities(state.piiMaskingEnabled)
|
|
1760
1771
|
}, this.requestOptions);
|
|
1761
1772
|
logCallbackDebug("handleLLMEnd logged", {
|
|
1762
1773
|
runId: id,
|
|
@@ -1798,7 +1809,7 @@ var AgentIDCallbackHandler = class extends import_base.BaseCallbackHandler {
|
|
|
1798
1809
|
event_type: "error",
|
|
1799
1810
|
severity: "error",
|
|
1800
1811
|
metadata,
|
|
1801
|
-
client_capabilities: this.getLangchainCapabilities()
|
|
1812
|
+
client_capabilities: this.getLangchainCapabilities(state?.piiMaskingEnabled)
|
|
1802
1813
|
}, this.requestOptions);
|
|
1803
1814
|
}
|
|
1804
1815
|
};
|
package/dist/langchain.mjs
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import {
|
|
2
2
|
PIIManager,
|
|
3
3
|
SecurityBlockError
|
|
4
|
-
} from "./chunk-
|
|
4
|
+
} from "./chunk-HWES3LI2.mjs";
|
|
5
5
|
|
|
6
6
|
// src/langchain.ts
|
|
7
7
|
import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
|
|
@@ -348,16 +348,23 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
|
|
|
348
348
|
get requestOptions() {
|
|
349
349
|
return this.apiKeyOverride ? { apiKey: this.apiKeyOverride } : void 0;
|
|
350
350
|
}
|
|
351
|
-
getLangchainCapabilities() {
|
|
352
|
-
const
|
|
351
|
+
getLangchainCapabilities(piiMaskingEnabled) {
|
|
352
|
+
const resolvedPiiMaskingEnabled = typeof piiMaskingEnabled === "boolean" ? piiMaskingEnabled : this.agent.getEffectivePiiMasking(this.requestOptions);
|
|
353
353
|
return {
|
|
354
354
|
capabilities: {
|
|
355
355
|
has_feedback_handler: true,
|
|
356
|
-
pii_masking_enabled:
|
|
356
|
+
pii_masking_enabled: resolvedPiiMaskingEnabled,
|
|
357
357
|
framework: "langchain"
|
|
358
358
|
}
|
|
359
359
|
};
|
|
360
360
|
}
|
|
361
|
+
resolvePreparedPiiMaskingEnabled(prepared) {
|
|
362
|
+
const agentWithResolvedConfig = this.agent;
|
|
363
|
+
if (typeof agentWithResolvedConfig.getEffectivePiiMaskingForConfig === "function") {
|
|
364
|
+
return agentWithResolvedConfig.getEffectivePiiMaskingForConfig(prepared.capabilityConfig);
|
|
365
|
+
}
|
|
366
|
+
return this.agent.getEffectivePiiMasking(this.requestOptions);
|
|
367
|
+
}
|
|
361
368
|
async preflight(input, stream, clientEventId) {
|
|
362
369
|
const prepared = await this.agent.prepareInputForDispatch({
|
|
363
370
|
input,
|
|
@@ -378,6 +385,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
|
|
|
378
385
|
const stream = extractStreamFlag(serialized, extraParams);
|
|
379
386
|
const prepared = await this.preflight(input, stream, requestedClientEventId);
|
|
380
387
|
const sanitizedInput = prepared.sanitizedInput;
|
|
388
|
+
const piiMaskingEnabled = this.resolvePreparedPiiMaskingEnabled(prepared);
|
|
381
389
|
if (sanitizedInput !== input) {
|
|
382
390
|
const mutated = setPromptInPrompts(prompts, sanitizedInput);
|
|
383
391
|
if (!mutated) {
|
|
@@ -394,7 +402,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
|
|
|
394
402
|
model: modelName,
|
|
395
403
|
client_event_id: requestedClientEventId,
|
|
396
404
|
expected_languages: this.expectedLanguages,
|
|
397
|
-
client_capabilities: this.getLangchainCapabilities()
|
|
405
|
+
client_capabilities: this.getLangchainCapabilities(piiMaskingEnabled)
|
|
398
406
|
}, this.requestOptions);
|
|
399
407
|
let transformedForRun = sanitizedInput;
|
|
400
408
|
let sdkLocalScanMs = prepared.sdkLocalScanMs;
|
|
@@ -452,7 +460,8 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
|
|
|
452
460
|
piiMapping: normalizePiiMapping(prepared.piiMapping),
|
|
453
461
|
shouldDeanonymize: prepared.shouldDeanonymize === true,
|
|
454
462
|
responseStreamed: stream,
|
|
455
|
-
sdkConfigVersion: prepared.capabilityConfig?.version ?? null
|
|
463
|
+
sdkConfigVersion: prepared.capabilityConfig?.version ?? null,
|
|
464
|
+
piiMaskingEnabled
|
|
456
465
|
});
|
|
457
466
|
logCallbackDebug("handleLLMStart state_set", {
|
|
458
467
|
runId: id,
|
|
@@ -471,6 +480,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
|
|
|
471
480
|
const stream = extractStreamFlag(serialized, extraParams);
|
|
472
481
|
const prepared = await this.preflight(input, stream, requestedClientEventId);
|
|
473
482
|
const sanitizedInput = prepared.sanitizedInput;
|
|
483
|
+
const piiMaskingEnabled = this.resolvePreparedPiiMaskingEnabled(prepared);
|
|
474
484
|
if (sanitizedInput !== input) {
|
|
475
485
|
const mutated = setPromptInMessages(messages, sanitizedInput);
|
|
476
486
|
if (!mutated) {
|
|
@@ -487,7 +497,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
|
|
|
487
497
|
model: modelName,
|
|
488
498
|
client_event_id: requestedClientEventId,
|
|
489
499
|
expected_languages: this.expectedLanguages,
|
|
490
|
-
client_capabilities: this.getLangchainCapabilities()
|
|
500
|
+
client_capabilities: this.getLangchainCapabilities(piiMaskingEnabled)
|
|
491
501
|
}, this.requestOptions);
|
|
492
502
|
let transformedForRun = sanitizedInput;
|
|
493
503
|
let sdkLocalScanMs = prepared.sdkLocalScanMs;
|
|
@@ -545,7 +555,8 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
|
|
|
545
555
|
piiMapping: normalizePiiMapping(prepared.piiMapping),
|
|
546
556
|
shouldDeanonymize: prepared.shouldDeanonymize === true,
|
|
547
557
|
responseStreamed: stream,
|
|
548
|
-
sdkConfigVersion: prepared.capabilityConfig?.version ?? null
|
|
558
|
+
sdkConfigVersion: prepared.capabilityConfig?.version ?? null,
|
|
559
|
+
piiMaskingEnabled
|
|
549
560
|
});
|
|
550
561
|
logCallbackDebug("handleChatModelStart state_set", {
|
|
551
562
|
runId: id,
|
|
@@ -609,7 +620,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
|
|
|
609
620
|
usage,
|
|
610
621
|
latency: modelLatencyMs,
|
|
611
622
|
metadata: Object.keys(metadata).length > 0 ? metadata : void 0,
|
|
612
|
-
client_capabilities: this.getLangchainCapabilities()
|
|
623
|
+
client_capabilities: this.getLangchainCapabilities(state.piiMaskingEnabled)
|
|
613
624
|
}, this.requestOptions);
|
|
614
625
|
logCallbackDebug("handleLLMEnd logged", {
|
|
615
626
|
runId: id,
|
|
@@ -651,7 +662,7 @@ var AgentIDCallbackHandler = class extends BaseCallbackHandler {
|
|
|
651
662
|
event_type: "error",
|
|
652
663
|
severity: "error",
|
|
653
664
|
metadata,
|
|
654
|
-
client_capabilities: this.getLangchainCapabilities()
|
|
665
|
+
client_capabilities: this.getLangchainCapabilities(state?.piiMaskingEnabled)
|
|
655
666
|
}, this.requestOptions);
|
|
656
667
|
}
|
|
657
668
|
};
|