@polpo-ai/core 0.15.43 → 0.15.45
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 +163 -0
- package/dist/__tests__/guardrail-detectors.test.d.ts +2 -0
- package/dist/__tests__/guardrail-detectors.test.d.ts.map +1 -0
- package/dist/__tests__/guardrail-detectors.test.js +316 -0
- package/dist/__tests__/guardrail-detectors.test.js.map +1 -0
- package/dist/__tests__/guardrail-engine.test.d.ts +2 -0
- package/dist/__tests__/guardrail-engine.test.d.ts.map +1 -0
- package/dist/__tests__/guardrail-engine.test.js +455 -0
- package/dist/__tests__/guardrail-engine.test.js.map +1 -0
- package/dist/__tests__/loop-context-trust.test.d.ts +2 -0
- package/dist/__tests__/loop-context-trust.test.d.ts.map +1 -0
- package/dist/__tests__/loop-context-trust.test.js +41 -0
- package/dist/__tests__/loop-context-trust.test.js.map +1 -0
- package/dist/__tests__/output-guardrails.test.d.ts +2 -0
- package/dist/__tests__/output-guardrails.test.d.ts.map +1 -0
- package/dist/__tests__/output-guardrails.test.js +151 -0
- package/dist/__tests__/output-guardrails.test.js.map +1 -0
- package/dist/__tests__/runtime-context.test.d.ts +2 -0
- package/dist/__tests__/runtime-context.test.d.ts.map +1 -0
- package/dist/__tests__/runtime-context.test.js +204 -0
- package/dist/__tests__/runtime-context.test.js.map +1 -0
- package/dist/__tests__/runtime-inspection.test.d.ts +2 -0
- package/dist/__tests__/runtime-inspection.test.d.ts.map +1 -0
- package/dist/__tests__/runtime-inspection.test.js +189 -0
- package/dist/__tests__/runtime-inspection.test.js.map +1 -0
- package/dist/__tests__/task-execution-router.test.js +62 -0
- package/dist/__tests__/task-execution-router.test.js.map +1 -1
- package/dist/__tests__/task-runner-guardrail-events.test.d.ts +2 -0
- package/dist/__tests__/task-runner-guardrail-events.test.d.ts.map +1 -0
- package/dist/__tests__/task-runner-guardrail-events.test.js +130 -0
- package/dist/__tests__/task-runner-guardrail-events.test.js.map +1 -0
- package/dist/adapter.d.ts +24 -2
- package/dist/adapter.d.ts.map +1 -1
- package/dist/events.d.ts +7 -0
- package/dist/events.d.ts.map +1 -1
- package/dist/guardrails/detectors.d.ts +24 -0
- package/dist/guardrails/detectors.d.ts.map +1 -0
- package/dist/guardrails/detectors.js +416 -0
- package/dist/guardrails/detectors.js.map +1 -0
- package/dist/guardrails/engine.d.ts +12 -0
- package/dist/guardrails/engine.d.ts.map +1 -0
- package/dist/guardrails/engine.js +189 -0
- package/dist/guardrails/engine.js.map +1 -0
- package/dist/guardrails/errors.d.ts +17 -0
- package/dist/guardrails/errors.d.ts.map +1 -0
- package/dist/guardrails/errors.js +23 -0
- package/dist/guardrails/errors.js.map +1 -0
- package/dist/guardrails/index.d.ts +8 -0
- package/dist/guardrails/index.d.ts.map +1 -0
- package/dist/guardrails/index.js +7 -0
- package/dist/guardrails/index.js.map +1 -0
- package/dist/guardrails/output-policy.d.ts +4 -0
- package/dist/guardrails/output-policy.d.ts.map +1 -0
- package/dist/guardrails/output-policy.js +64 -0
- package/dist/guardrails/output-policy.js.map +1 -0
- package/dist/guardrails/settings.d.ts +12 -0
- package/dist/guardrails/settings.d.ts.map +1 -0
- package/dist/guardrails/settings.js +108 -0
- package/dist/guardrails/settings.js.map +1 -0
- package/dist/guardrails/tool-middleware.d.ts +15 -0
- package/dist/guardrails/tool-middleware.d.ts.map +1 -0
- package/dist/guardrails/tool-middleware.js +158 -0
- package/dist/guardrails/tool-middleware.js.map +1 -0
- package/dist/guardrails/types.d.ts +148 -0
- package/dist/guardrails/types.d.ts.map +1 -0
- package/dist/guardrails/types.js +2 -0
- package/dist/guardrails/types.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +4 -0
- package/dist/index.js.map +1 -1
- package/dist/loop/step-helpers.d.ts +2 -1
- package/dist/loop/step-helpers.d.ts.map +1 -1
- package/dist/loop/step-helpers.js +37 -4
- package/dist/loop/step-helpers.js.map +1 -1
- package/dist/runtime-context/context.d.ts +17 -0
- package/dist/runtime-context/context.d.ts.map +1 -0
- package/dist/runtime-context/context.js +338 -0
- package/dist/runtime-context/context.js.map +1 -0
- package/dist/runtime-context/index.d.ts +2 -1
- package/dist/runtime-context/index.d.ts.map +1 -1
- package/dist/runtime-context/index.js +1 -0
- package/dist/runtime-context/index.js.map +1 -1
- package/dist/runtime-context/types.d.ts +30 -0
- package/dist/runtime-context/types.d.ts.map +1 -1
- package/dist/runtime-context/types.js.map +1 -1
- package/dist/runtime-inspection/accounting.d.ts +4 -0
- package/dist/runtime-inspection/accounting.d.ts.map +1 -0
- package/dist/runtime-inspection/accounting.js +145 -0
- package/dist/runtime-inspection/accounting.js.map +1 -0
- package/dist/runtime-inspection/index.d.ts +4 -0
- package/dist/runtime-inspection/index.d.ts.map +1 -0
- package/dist/runtime-inspection/index.js +3 -0
- package/dist/runtime-inspection/index.js.map +1 -0
- package/dist/runtime-inspection/types.d.ts +40 -0
- package/dist/runtime-inspection/types.d.ts.map +1 -0
- package/dist/runtime-inspection/types.js +22 -0
- package/dist/runtime-inspection/types.js.map +1 -0
- package/dist/task-runner.d.ts +1 -0
- package/dist/task-runner.d.ts.map +1 -1
- package/dist/task-runner.js +97 -15
- package/dist/task-runner.js.map +1 -1
- package/dist/types/agent.d.ts +8 -0
- package/dist/types/agent.d.ts.map +1 -1
- package/dist/types/agent.js.map +1 -1
- package/dist/types/config.d.ts +17 -1
- package/dist/types/config.d.ts.map +1 -1
- package/dist/types/config.js.map +1 -1
- package/package.json +10 -1
|
@@ -0,0 +1,416 @@
|
|
|
1
|
+
import { Ajv } from "ajv";
|
|
2
|
+
const SECRET_PATTERNS = [
|
|
3
|
+
/\bgh[pousr]_[A-Za-z0-9]{30,255}\b/g,
|
|
4
|
+
/\bAKIA[A-Z0-9]{16}\b/g,
|
|
5
|
+
/\bxox[baprs]-[A-Za-z0-9-]{10,255}\b/g,
|
|
6
|
+
/\bsk-(?:(?:proj|svcacct)-)?[A-Za-z0-9_-]{32,255}\b/g,
|
|
7
|
+
/\bsk_[a-f0-9]{40,255}\b/gi,
|
|
8
|
+
/\beyJ[A-Za-z0-9_-]{5,}\.[A-Za-z0-9_-]{5,}\.[A-Za-z0-9_-]{5,}\b/g,
|
|
9
|
+
/-----BEGIN (?:RSA |EC |OPENSSH )?PRIVATE KEY-----[\s\S]*?-----END (?:RSA |EC |OPENSSH )?PRIVATE KEY-----/g,
|
|
10
|
+
];
|
|
11
|
+
function redactString(value) {
|
|
12
|
+
let redacted = value;
|
|
13
|
+
let matches = 0;
|
|
14
|
+
for (const pattern of SECRET_PATTERNS) {
|
|
15
|
+
redacted = redacted.replace(pattern, () => {
|
|
16
|
+
matches++;
|
|
17
|
+
return "[REDACTED]";
|
|
18
|
+
});
|
|
19
|
+
}
|
|
20
|
+
return { value: redacted, matches };
|
|
21
|
+
}
|
|
22
|
+
function mapStrings(value, mapper, seen = new WeakMap()) {
|
|
23
|
+
if (typeof value === "string")
|
|
24
|
+
return mapper(value);
|
|
25
|
+
if (value === null || typeof value !== "object")
|
|
26
|
+
return { value, matches: 0 };
|
|
27
|
+
const existing = seen.get(value);
|
|
28
|
+
if (existing !== undefined)
|
|
29
|
+
return { value: existing, matches: 0 };
|
|
30
|
+
if (Array.isArray(value)) {
|
|
31
|
+
const output = [];
|
|
32
|
+
seen.set(value, output);
|
|
33
|
+
let matches = 0;
|
|
34
|
+
for (const item of value) {
|
|
35
|
+
const mapped = mapStrings(item, mapper, seen);
|
|
36
|
+
output.push(mapped.value);
|
|
37
|
+
matches += mapped.matches;
|
|
38
|
+
}
|
|
39
|
+
return { value: output, matches };
|
|
40
|
+
}
|
|
41
|
+
const output = {};
|
|
42
|
+
seen.set(value, output);
|
|
43
|
+
let matches = 0;
|
|
44
|
+
for (const [key, item] of Object.entries(value)) {
|
|
45
|
+
const mapped = mapStrings(item, mapper, seen);
|
|
46
|
+
output[key] = mapped.value;
|
|
47
|
+
matches += mapped.matches;
|
|
48
|
+
}
|
|
49
|
+
return { value: output, matches };
|
|
50
|
+
}
|
|
51
|
+
function collectStrings(value, output = [], seen = new WeakSet()) {
|
|
52
|
+
if (typeof value === "string")
|
|
53
|
+
output.push(value);
|
|
54
|
+
else if (value && typeof value === "object" && !seen.has(value)) {
|
|
55
|
+
seen.add(value);
|
|
56
|
+
for (const item of Array.isArray(value)
|
|
57
|
+
? value
|
|
58
|
+
: Object.values(value)) {
|
|
59
|
+
collectStrings(item, output, seen);
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
return output;
|
|
63
|
+
}
|
|
64
|
+
export function createSecretPatternPolicy(options = {}) {
|
|
65
|
+
const action = options.action ?? "redact";
|
|
66
|
+
return {
|
|
67
|
+
id: options.id ?? "secrets.common-patterns",
|
|
68
|
+
version: options.version ?? "1",
|
|
69
|
+
priority: options.priority ?? 500,
|
|
70
|
+
phases: options.phases ?? ["input", "context", "tool.before", "tool.after", "output"],
|
|
71
|
+
evaluate(input) {
|
|
72
|
+
const redacted = mapStrings(input.value, redactString);
|
|
73
|
+
if (redacted.matches === 0)
|
|
74
|
+
return null;
|
|
75
|
+
return {
|
|
76
|
+
action,
|
|
77
|
+
risk: options.risk ?? "high",
|
|
78
|
+
reason: `Detected ${redacted.matches} common secret pattern${redacted.matches === 1 ? "" : "s"}`,
|
|
79
|
+
...(action === "redact" || action === "rewrite" ? { value: redacted.value } : {}),
|
|
80
|
+
};
|
|
81
|
+
},
|
|
82
|
+
};
|
|
83
|
+
}
|
|
84
|
+
const DESTRUCTIVE_PATTERNS = [
|
|
85
|
+
/\brm\s+(?:-[^\s]*[rf][^\s]*\s+){1,}(?:\/(?:\s|$)|~(?:\/|\s|$)|\$HOME(?:\/|\s|$))/i,
|
|
86
|
+
/\b(?:mkfs(?:\.[a-z0-9]+)?|wipefs)\b/i,
|
|
87
|
+
/\bdd\s+if=.*\bof=\/dev\//i,
|
|
88
|
+
/:\(\)\s*\{\s*:\|:&\s*\};:/,
|
|
89
|
+
/\bDROP\s+(?:DATABASE|SCHEMA|TABLE)\b/i,
|
|
90
|
+
/\bTRUNCATE\s+(?:TABLE\s+)?\S+/i,
|
|
91
|
+
/\bDELETE\s+FROM\s+\S+(?![\s\S]*\bWHERE\b)/i,
|
|
92
|
+
/\bUPDATE\s+\S+\s+SET\b(?![\s\S]*\bWHERE\b)/i,
|
|
93
|
+
];
|
|
94
|
+
export function createDestructiveOperationPolicy(options = {}) {
|
|
95
|
+
return {
|
|
96
|
+
id: options.id ?? "tools.destructive-operation",
|
|
97
|
+
version: options.version ?? "1",
|
|
98
|
+
priority: options.priority ?? 200,
|
|
99
|
+
phases: ["tool.before"],
|
|
100
|
+
evaluate(input) {
|
|
101
|
+
const toolName = input.tool?.name.toLowerCase() ?? "";
|
|
102
|
+
if (!/(bash|shell|command|exec|sql|query)/.test(toolName))
|
|
103
|
+
return null;
|
|
104
|
+
const matched = collectStrings(input.value).some((value) => DESTRUCTIVE_PATTERNS.some((pattern) => pattern.test(value)));
|
|
105
|
+
if (!matched)
|
|
106
|
+
return null;
|
|
107
|
+
return {
|
|
108
|
+
action: options.action ?? "approval",
|
|
109
|
+
risk: options.risk ?? "critical",
|
|
110
|
+
reason: "Potentially destructive shell or SQL operation",
|
|
111
|
+
};
|
|
112
|
+
},
|
|
113
|
+
};
|
|
114
|
+
}
|
|
115
|
+
function isPrivateIpv4(hostname) {
|
|
116
|
+
const parts = hostname.split(".");
|
|
117
|
+
if (parts.length !== 4 || parts.some((part) => !/^\d{1,3}$/.test(part)))
|
|
118
|
+
return false;
|
|
119
|
+
const values = parts.map(Number);
|
|
120
|
+
if (values.some((value) => value > 255))
|
|
121
|
+
return false;
|
|
122
|
+
return values[0] === 10
|
|
123
|
+
|| values[0] === 127
|
|
124
|
+
|| (values[0] === 169 && values[1] === 254)
|
|
125
|
+
|| (values[0] === 172 && values[1] >= 16 && values[1] <= 31)
|
|
126
|
+
|| (values[0] === 192 && values[1] === 168)
|
|
127
|
+
|| (values[0] === 100 && values[1] >= 64 && values[1] <= 127)
|
|
128
|
+
|| (values[0] === 198 && (values[1] === 18 || values[1] === 19))
|
|
129
|
+
|| values[0] >= 224
|
|
130
|
+
|| values[0] === 0;
|
|
131
|
+
}
|
|
132
|
+
function mappedIpv4(hostname) {
|
|
133
|
+
const match = /^::ffff:([0-9a-f]{1,4}):([0-9a-f]{1,4})$/i.exec(hostname);
|
|
134
|
+
if (!match)
|
|
135
|
+
return undefined;
|
|
136
|
+
const high = Number.parseInt(match[1], 16);
|
|
137
|
+
const low = Number.parseInt(match[2], 16);
|
|
138
|
+
return [
|
|
139
|
+
high >>> 8,
|
|
140
|
+
high & 0xff,
|
|
141
|
+
low >>> 8,
|
|
142
|
+
low & 0xff,
|
|
143
|
+
].join(".");
|
|
144
|
+
}
|
|
145
|
+
function isPrivateHostname(rawHostname) {
|
|
146
|
+
const hostname = rawHostname.toLowerCase().replace(/^\[|\]$/g, "").replace(/\.$/, "");
|
|
147
|
+
const ipv4 = mappedIpv4(hostname);
|
|
148
|
+
return hostname === "localhost"
|
|
149
|
+
|| hostname === "::"
|
|
150
|
+
|| hostname === "::1"
|
|
151
|
+
|| hostname === "0:0:0:0:0:0:0:1"
|
|
152
|
+
|| /^f[cd][0-9a-f]{2}(?::|$)/.test(hostname)
|
|
153
|
+
|| /^fe[89ab][0-9a-f](?::|$)/.test(hostname)
|
|
154
|
+
|| /^ff[0-9a-f]{2}(?::|$)/.test(hostname)
|
|
155
|
+
|| hostname.endsWith(".localhost")
|
|
156
|
+
|| hostname.endsWith(".local")
|
|
157
|
+
|| hostname.endsWith(".internal")
|
|
158
|
+
|| hostname.endsWith(".home.arpa")
|
|
159
|
+
|| hostname === "metadata.google.internal"
|
|
160
|
+
|| (ipv4 !== undefined && isPrivateIpv4(ipv4))
|
|
161
|
+
|| isPrivateIpv4(hostname);
|
|
162
|
+
}
|
|
163
|
+
function findPrivateTarget(value, seen = new WeakSet()) {
|
|
164
|
+
if (value && typeof value === "object") {
|
|
165
|
+
if (seen.has(value))
|
|
166
|
+
return undefined;
|
|
167
|
+
seen.add(value);
|
|
168
|
+
for (const [key, item] of Object.entries(value)) {
|
|
169
|
+
if (typeof item === "string" &&
|
|
170
|
+
/^(?:host|hostname|address)$/i.test(key) &&
|
|
171
|
+
isPrivateHostname(item)) {
|
|
172
|
+
return item.toLowerCase();
|
|
173
|
+
}
|
|
174
|
+
const nested = findPrivateTarget(item, seen);
|
|
175
|
+
if (nested)
|
|
176
|
+
return nested;
|
|
177
|
+
}
|
|
178
|
+
return undefined;
|
|
179
|
+
}
|
|
180
|
+
if (typeof value !== "string")
|
|
181
|
+
return undefined;
|
|
182
|
+
const candidates = value.match(/https?:\/\/[^\s"'<>]+/gi) ?? [];
|
|
183
|
+
for (const candidate of candidates) {
|
|
184
|
+
try {
|
|
185
|
+
const hostname = new URL(candidate).hostname.replace(/^\[|\]$/g, "");
|
|
186
|
+
if (isPrivateHostname(hostname))
|
|
187
|
+
return hostname.toLowerCase();
|
|
188
|
+
}
|
|
189
|
+
catch {
|
|
190
|
+
// Malformed URLs belong to argument-schema validation, not SSRF policy.
|
|
191
|
+
}
|
|
192
|
+
}
|
|
193
|
+
return undefined;
|
|
194
|
+
}
|
|
195
|
+
export function createPrivateNetworkPolicy(options = {}) {
|
|
196
|
+
return {
|
|
197
|
+
id: options.id ?? "network.private-target",
|
|
198
|
+
version: options.version ?? "1",
|
|
199
|
+
priority: options.priority ?? 220,
|
|
200
|
+
phases: ["tool.before"],
|
|
201
|
+
evaluate(input) {
|
|
202
|
+
const toolName = input.tool?.name.toLowerCase() ?? "";
|
|
203
|
+
if (!/(http|fetch|request|browser|navigate|webhook|url|download|connect|network|api)/.test(toolName))
|
|
204
|
+
return null;
|
|
205
|
+
const target = findPrivateTarget(input.value);
|
|
206
|
+
if (!target)
|
|
207
|
+
return null;
|
|
208
|
+
return {
|
|
209
|
+
action: options.action ?? "block",
|
|
210
|
+
risk: options.risk ?? "critical",
|
|
211
|
+
reason: `Outbound request targets a non-public network address: ${target}`,
|
|
212
|
+
};
|
|
213
|
+
},
|
|
214
|
+
};
|
|
215
|
+
}
|
|
216
|
+
const toolSchemaAjv = new Ajv({
|
|
217
|
+
allErrors: false,
|
|
218
|
+
allowUnionTypes: true,
|
|
219
|
+
strict: false,
|
|
220
|
+
validateFormats: false,
|
|
221
|
+
});
|
|
222
|
+
const toolSchemaValidators = new WeakMap();
|
|
223
|
+
async function resolveToolArgumentSchema(value, seen = new WeakSet()) {
|
|
224
|
+
let current = await value;
|
|
225
|
+
for (let depth = 0; depth < 8; depth++) {
|
|
226
|
+
if (!current || typeof current !== "object" || Array.isArray(current)) {
|
|
227
|
+
return undefined;
|
|
228
|
+
}
|
|
229
|
+
if (seen.has(current))
|
|
230
|
+
return undefined;
|
|
231
|
+
seen.add(current);
|
|
232
|
+
const candidate = current;
|
|
233
|
+
if (candidate.type !== undefined
|
|
234
|
+
|| candidate.properties !== undefined
|
|
235
|
+
|| candidate.required !== undefined
|
|
236
|
+
|| candidate.$ref !== undefined
|
|
237
|
+
|| candidate.allOf !== undefined
|
|
238
|
+
|| candidate.anyOf !== undefined
|
|
239
|
+
|| candidate.oneOf !== undefined) {
|
|
240
|
+
return candidate;
|
|
241
|
+
}
|
|
242
|
+
if (!("jsonSchema" in candidate))
|
|
243
|
+
return candidate;
|
|
244
|
+
current = await candidate.jsonSchema;
|
|
245
|
+
}
|
|
246
|
+
return undefined;
|
|
247
|
+
}
|
|
248
|
+
function validateToolArguments(schema, value) {
|
|
249
|
+
let validate = toolSchemaValidators.get(schema);
|
|
250
|
+
if (!validate) {
|
|
251
|
+
const compiled = toolSchemaAjv.compile(schema);
|
|
252
|
+
toolSchemaValidators.set(schema, compiled);
|
|
253
|
+
validate = compiled;
|
|
254
|
+
}
|
|
255
|
+
if (validate(value))
|
|
256
|
+
return undefined;
|
|
257
|
+
const error = validate.errors?.[0];
|
|
258
|
+
const path = error?.instancePath || "$";
|
|
259
|
+
return `${path} ${error?.message ?? "does not match the tool schema"}`;
|
|
260
|
+
}
|
|
261
|
+
export function createToolArgumentsPolicy(options = {}) {
|
|
262
|
+
return {
|
|
263
|
+
id: options.id ?? "tools.arguments-schema",
|
|
264
|
+
version: options.version ?? "1",
|
|
265
|
+
priority: options.priority ?? 100,
|
|
266
|
+
phases: ["tool.before"],
|
|
267
|
+
async evaluate(input) {
|
|
268
|
+
if (input.tool?.schema === undefined)
|
|
269
|
+
return null;
|
|
270
|
+
let schema;
|
|
271
|
+
try {
|
|
272
|
+
schema = await resolveToolArgumentSchema(input.tool.schema);
|
|
273
|
+
}
|
|
274
|
+
catch {
|
|
275
|
+
schema = undefined;
|
|
276
|
+
}
|
|
277
|
+
if (!schema) {
|
|
278
|
+
return {
|
|
279
|
+
action: options.action ?? "block",
|
|
280
|
+
risk: options.risk ?? "high",
|
|
281
|
+
reason: "Tool parameter schema is malformed",
|
|
282
|
+
};
|
|
283
|
+
}
|
|
284
|
+
if (schema.type !== "object") {
|
|
285
|
+
return {
|
|
286
|
+
action: options.action ?? "block",
|
|
287
|
+
risk: options.risk ?? "high",
|
|
288
|
+
reason: "Tool parameter root must be an object schema",
|
|
289
|
+
};
|
|
290
|
+
}
|
|
291
|
+
let error;
|
|
292
|
+
try {
|
|
293
|
+
error = validateToolArguments(schema, input.value);
|
|
294
|
+
}
|
|
295
|
+
catch {
|
|
296
|
+
return {
|
|
297
|
+
action: options.action ?? "block",
|
|
298
|
+
risk: options.risk ?? "high",
|
|
299
|
+
reason: "Tool parameter schema could not be validated",
|
|
300
|
+
};
|
|
301
|
+
}
|
|
302
|
+
if (!error)
|
|
303
|
+
return null;
|
|
304
|
+
return {
|
|
305
|
+
action: options.action ?? "block",
|
|
306
|
+
risk: options.risk ?? "high",
|
|
307
|
+
reason: `Invalid tool arguments: ${error}`,
|
|
308
|
+
};
|
|
309
|
+
},
|
|
310
|
+
};
|
|
311
|
+
}
|
|
312
|
+
export function createCrossScopePolicy(options = {}) {
|
|
313
|
+
return {
|
|
314
|
+
id: options.id ?? "context.cross-scope",
|
|
315
|
+
version: options.version ?? "1",
|
|
316
|
+
priority: options.priority ?? 50,
|
|
317
|
+
phases: ["input", "context", "model.preflight", "tool.before", "tool.after", "output"],
|
|
318
|
+
evaluate(input) {
|
|
319
|
+
const expected = input.context.scope?.expected;
|
|
320
|
+
const actual = input.context.scope?.actual;
|
|
321
|
+
if (!expected || !actual)
|
|
322
|
+
return null;
|
|
323
|
+
const mismatch = Object.entries(expected).find(([key, expectedValue]) => expectedValue !== undefined && actual[key] !== expectedValue);
|
|
324
|
+
if (!mismatch)
|
|
325
|
+
return null;
|
|
326
|
+
return {
|
|
327
|
+
action: options.action ?? "block",
|
|
328
|
+
risk: options.risk ?? "critical",
|
|
329
|
+
reason: `Runtime context scope mismatch for "${mismatch[0]}"`,
|
|
330
|
+
};
|
|
331
|
+
},
|
|
332
|
+
};
|
|
333
|
+
}
|
|
334
|
+
function countCharacters(value, limit, seen = new WeakSet()) {
|
|
335
|
+
if (typeof value === "string")
|
|
336
|
+
return value.length;
|
|
337
|
+
if (value === null || value === undefined)
|
|
338
|
+
return String(value).length;
|
|
339
|
+
if (typeof value !== "object")
|
|
340
|
+
return String(value).length;
|
|
341
|
+
if (seen.has(value))
|
|
342
|
+
return limit + 1;
|
|
343
|
+
seen.add(value);
|
|
344
|
+
let total = Array.isArray(value) ? 2 : 2;
|
|
345
|
+
const entries = Array.isArray(value)
|
|
346
|
+
? value
|
|
347
|
+
: Object.entries(value);
|
|
348
|
+
for (const entry of entries) {
|
|
349
|
+
if (Array.isArray(entry) && !Array.isArray(value)) {
|
|
350
|
+
total += entry[0].length + countCharacters(entry[1], limit - total, seen);
|
|
351
|
+
}
|
|
352
|
+
else {
|
|
353
|
+
total += countCharacters(entry, limit - total, seen);
|
|
354
|
+
}
|
|
355
|
+
if (total > limit)
|
|
356
|
+
return total;
|
|
357
|
+
}
|
|
358
|
+
return total;
|
|
359
|
+
}
|
|
360
|
+
export function createBoundedValuePolicy(options) {
|
|
361
|
+
if (!Number.isSafeInteger(options.maxCharacters) || options.maxCharacters < 1) {
|
|
362
|
+
throw new TypeError("maxCharacters must be a positive safe integer");
|
|
363
|
+
}
|
|
364
|
+
return {
|
|
365
|
+
id: options.id ?? "context.bounded-value",
|
|
366
|
+
version: options.version ?? "1",
|
|
367
|
+
priority: options.priority ?? 80,
|
|
368
|
+
phases: options.phases,
|
|
369
|
+
evaluate(input) {
|
|
370
|
+
if (countCharacters(input.value, options.maxCharacters) <= options.maxCharacters)
|
|
371
|
+
return null;
|
|
372
|
+
return {
|
|
373
|
+
action: options.action ?? "block",
|
|
374
|
+
risk: options.risk ?? "high",
|
|
375
|
+
reason: `Value exceeds the ${options.maxCharacters}-character guardrail limit`,
|
|
376
|
+
};
|
|
377
|
+
},
|
|
378
|
+
};
|
|
379
|
+
}
|
|
380
|
+
export function createDefaultToolGuardrailPolicies() {
|
|
381
|
+
return Object.freeze([
|
|
382
|
+
createCrossScopePolicy(),
|
|
383
|
+
createToolArgumentsPolicy(),
|
|
384
|
+
createDestructiveOperationPolicy(),
|
|
385
|
+
createPrivateNetworkPolicy(),
|
|
386
|
+
createSecretPatternPolicy({
|
|
387
|
+
id: "secrets.tool-input",
|
|
388
|
+
phases: ["tool.before"],
|
|
389
|
+
action: "redact",
|
|
390
|
+
}),
|
|
391
|
+
createSecretPatternPolicy({
|
|
392
|
+
id: "secrets.tool-output",
|
|
393
|
+
phases: ["tool.after"],
|
|
394
|
+
action: "redact",
|
|
395
|
+
}),
|
|
396
|
+
]);
|
|
397
|
+
}
|
|
398
|
+
export function createDefaultOutputGuardrailPolicies(maxCharacters) {
|
|
399
|
+
return Object.freeze([
|
|
400
|
+
createCrossScopePolicy(),
|
|
401
|
+
...(maxCharacters !== undefined
|
|
402
|
+
? [createBoundedValuePolicy({
|
|
403
|
+
id: "output.bounded-value",
|
|
404
|
+
phases: ["output"],
|
|
405
|
+
maxCharacters,
|
|
406
|
+
action: "block",
|
|
407
|
+
})]
|
|
408
|
+
: []),
|
|
409
|
+
createSecretPatternPolicy({
|
|
410
|
+
id: "secrets.output",
|
|
411
|
+
phases: ["output"],
|
|
412
|
+
action: "redact",
|
|
413
|
+
}),
|
|
414
|
+
]);
|
|
415
|
+
}
|
|
416
|
+
//# sourceMappingURL=detectors.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"detectors.js","sourceRoot":"","sources":["../../src/guardrails/detectors.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAyB,MAAM,KAAK,CAAC;AAmBjD,MAAM,eAAe,GAAsB;IACzC,oCAAoC;IACpC,uBAAuB;IACvB,sCAAsC;IACtC,qDAAqD;IACrD,2BAA2B;IAC3B,iEAAiE;IACjE,2GAA2G;CAC5G,CAAC;AAEF,SAAS,YAAY,CAAC,KAAa;IACjC,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;QACtC,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,EAAE;YACxC,OAAO,EAAE,CAAC;YACV,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;AACtC,CAAC;AAED,SAAS,UAAU,CACjB,KAAc,EACd,MAA6D,EAC7D,OAAO,IAAI,OAAO,EAAmB;IAErC,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACpD,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IAC9E,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,KAAe,CAAC,CAAC;IAC3C,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IACnE,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,MAAM,GAAc,EAAE,CAAC;QAC7B,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACxB,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC1B,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC;QAC5B,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;IACpC,CAAC;IACD,MAAM,MAAM,GAA4B,EAAE,CAAC;IAC3C,IAAI,CAAC,GAAG,CAAC,KAAe,EAAE,MAAM,CAAC,CAAC;IAClC,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;QAC3E,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;QAC3B,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC;IAC5B,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,cAAc,CACrB,KAAc,EACd,SAAmB,EAAE,EACrB,OAAO,IAAI,OAAO,EAAU;IAE5B,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;SAC7C,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAe,CAAC,EAAE,CAAC;QAC1E,IAAI,CAAC,GAAG,CAAC,KAAe,CAAC,CAAC;QAC1B,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YACrC,CAAC,CAAC,KAAK;YACP,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAgC,CAAC,EAAE,CAAC;YACpD,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,UAEtC,EAAE;IACJ,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,QAAQ,CAAC;IAC1C,OAAO;QACL,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,yBAAyB;QAC3C,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,GAAG;QAC/B,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,GAAG;QACjC,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,YAAY,EAAE,QAAQ,CAAC;QACrF,QAAQ,CAAC,KAAK;YACZ,MAAM,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;YACvD,IAAI,QAAQ,CAAC,OAAO,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC;YACxC,OAAO;gBACL,MAAM;gBACN,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,MAAM;gBAC5B,MAAM,EAAE,YAAY,QAAQ,CAAC,OAAO,yBAAyB,QAAQ,CAAC,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE;gBAChG,GAAG,CAAC,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAClD,CAAC;QACpC,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,oBAAoB,GAAsB;IAC9C,mFAAmF;IACnF,sCAAsC;IACtC,2BAA2B;IAC3B,2BAA2B;IAC3B,uCAAuC;IACvC,gCAAgC;IAChC,4CAA4C;IAC5C,6CAA6C;CAC9C,CAAC;AAEF,MAAM,UAAU,gCAAgC,CAC9C,UAA2B,EAAE;IAE7B,OAAO;QACL,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,6BAA6B;QAC/C,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,GAAG;QAC/B,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,GAAG;QACjC,MAAM,EAAE,CAAC,aAAa,CAAC;QACvB,QAAQ,CAAC,KAAK;YACZ,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC;YACtD,IAAI,CAAC,qCAAqC,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAAE,OAAO,IAAI,CAAC;YACvE,MAAM,OAAO,GAAG,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CACzD,oBAAoB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAC5D,CAAC;YACF,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC;YAC1B,OAAO;gBACL,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,UAAU;gBACpC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,UAAU;gBAChC,MAAM,EAAE,gDAAgD;aACzD,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB;IACrC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAClC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAAE,OAAO,KAAK,CAAC;IACtF,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACjC,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,GAAG,GAAG,CAAC;QAAE,OAAO,KAAK,CAAC;IACtD,OAAO,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE;WAClB,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG;WACjB,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC;WACxC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;WACzD,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC;WACxC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC;WAC1D,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;WAC7D,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG;WAChB,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED,SAAS,UAAU,CAAC,QAAgB;IAClC,MAAM,KAAK,GAAG,2CAA2C,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACzE,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,EAAE,CAAC,CAAC;IAC5C,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,EAAE,CAAC,CAAC;IAC3C,OAAO;QACL,IAAI,KAAK,CAAC;QACV,IAAI,GAAG,IAAI;QACX,GAAG,KAAK,CAAC;QACT,GAAG,GAAG,IAAI;KACX,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACd,CAAC;AAED,SAAS,iBAAiB,CAAC,WAAmB;IAC5C,MAAM,QAAQ,GAAG,WAAW,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACtF,MAAM,IAAI,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IAClC,OAAO,QAAQ,KAAK,WAAW;WAC1B,QAAQ,KAAK,IAAI;WACjB,QAAQ,KAAK,KAAK;WAClB,QAAQ,KAAK,iBAAiB;WAC9B,0BAA0B,CAAC,IAAI,CAAC,QAAQ,CAAC;WACzC,0BAA0B,CAAC,IAAI,CAAC,QAAQ,CAAC;WACzC,uBAAuB,CAAC,IAAI,CAAC,QAAQ,CAAC;WACtC,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC;WAC/B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;WAC3B,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;WAC9B,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC;WAC/B,QAAQ,KAAK,0BAA0B;WACvC,CAAC,IAAI,KAAK,SAAS,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC;WAC3C,aAAa,CAAC,QAAQ,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,iBAAiB,CACxB,KAAc,EACd,OAAO,IAAI,OAAO,EAAU;IAE5B,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACvC,IAAI,IAAI,CAAC,GAAG,CAAC,KAAe,CAAC;YAAE,OAAO,SAAS,CAAC;QAChD,IAAI,CAAC,GAAG,CAAC,KAAe,CAAC,CAAC;QAC1B,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;YAC3E,IACE,OAAO,IAAI,KAAK,QAAQ;gBACxB,8BAA8B,CAAC,IAAI,CAAC,GAAG,CAAC;gBACxC,iBAAiB,CAAC,IAAI,CAAC,EACvB,CAAC;gBACD,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;YAC5B,CAAC;YACD,MAAM,MAAM,GAAG,iBAAiB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC7C,IAAI,MAAM;gBAAE,OAAO,MAAM,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAEhD,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,yBAAyB,CAAC,IAAI,EAAE,CAAC;IAChE,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACrE,IAAI,iBAAiB,CAAC,QAAQ,CAAC;gBAAE,OAAO,QAAQ,CAAC,WAAW,EAAE,CAAC;QACjE,CAAC;QAAC,MAAM,CAAC;YACP,wEAAwE;QAC1E,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,UAA2B,EAAE;IAE7B,OAAO;QACL,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,wBAAwB;QAC1C,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,GAAG;QAC/B,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,GAAG;QACjC,MAAM,EAAE,CAAC,aAAa,CAAC;QACvB,QAAQ,CAAC,KAAK;YACZ,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC;YACtD,IAAI,CAAC,gFAAgF,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAAE,OAAO,IAAI,CAAC;YAClH,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC9C,IAAI,CAAC,MAAM;gBAAE,OAAO,IAAI,CAAC;YACzB,OAAO;gBACL,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO;gBACjC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,UAAU;gBAChC,MAAM,EAAE,0DAA0D,MAAM,EAAE;aAC3E,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC;IAC5B,SAAS,EAAE,KAAK;IAChB,eAAe,EAAE,IAAI;IACrB,MAAM,EAAE,KAAK;IACb,eAAe,EAAE,KAAK;CACvB,CAAC,CAAC;AACH,MAAM,oBAAoB,GAAG,IAAI,OAAO,EAA4B,CAAC;AAErE,KAAK,UAAU,yBAAyB,CACtC,KAAc,EACd,OAAO,IAAI,OAAO,EAAU;IAE5B,IAAI,OAAO,GAAG,MAAM,KAAK,CAAC;IAC1B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;QACvC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACtE,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,IAAI,CAAC,GAAG,CAAC,OAAiB,CAAC;YAAE,OAAO,SAAS,CAAC;QAClD,IAAI,CAAC,GAAG,CAAC,OAAiB,CAAC,CAAC;QAE5B,MAAM,SAAS,GAAG,OAAkC,CAAC;QACrD,IACE,SAAS,CAAC,IAAI,KAAK,SAAS;eACzB,SAAS,CAAC,UAAU,KAAK,SAAS;eAClC,SAAS,CAAC,QAAQ,KAAK,SAAS;eAChC,SAAS,CAAC,IAAI,KAAK,SAAS;eAC5B,SAAS,CAAC,KAAK,KAAK,SAAS;eAC7B,SAAS,CAAC,KAAK,KAAK,SAAS;eAC7B,SAAS,CAAC,KAAK,KAAK,SAAS,EAChC,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,CAAC,YAAY,IAAI,SAAS,CAAC;YAAE,OAAO,SAAS,CAAC;QACnD,OAAO,GAAG,MAAM,SAAS,CAAC,UAAU,CAAC;IACvC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,qBAAqB,CAC5B,MAA+B,EAC/B,KAAc;IAEd,IAAI,QAAQ,GAAG,oBAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAChD,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,QAAQ,GAAG,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC/C,oBAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC3C,QAAQ,GAAG,QAAQ,CAAC;IACtB,CAAC;IACD,IAAI,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACtC,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,IAAI,GAAG,KAAK,EAAE,YAAY,IAAI,GAAG,CAAC;IACxC,OAAO,GAAG,IAAI,IAAI,KAAK,EAAE,OAAO,IAAI,gCAAgC,EAAE,CAAC;AACzE,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,UAA2B,EAAE;IAE7B,OAAO;QACL,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,wBAAwB;QAC1C,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,GAAG;QAC/B,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,GAAG;QACjC,MAAM,EAAE,CAAC,aAAa,CAAC;QACvB,KAAK,CAAC,QAAQ,CAAC,KAAK;YAClB,IAAI,KAAK,CAAC,IAAI,EAAE,MAAM,KAAK,SAAS;gBAAE,OAAO,IAAI,CAAC;YAClD,IAAI,MAA2C,CAAC;YAChD,IAAI,CAAC;gBACH,MAAM,GAAG,MAAM,yBAAyB,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC9D,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,GAAG,SAAS,CAAC;YACrB,CAAC;YACD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO;oBACL,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO;oBACjC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,MAAM;oBAC5B,MAAM,EAAE,oCAAoC;iBAC7C,CAAC;YACJ,CAAC;YACD,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC7B,OAAO;oBACL,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO;oBACjC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,MAAM;oBAC5B,MAAM,EAAE,8CAA8C;iBACvD,CAAC;YACJ,CAAC;YACD,IAAI,KAAyB,CAAC;YAC9B,IAAI,CAAC;gBACH,KAAK,GAAG,qBAAqB,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACrD,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO;oBACL,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO;oBACjC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,MAAM;oBAC5B,MAAM,EAAE,8CAA8C;iBACvD,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,KAAK;gBAAE,OAAO,IAAI,CAAC;YACxB,OAAO;gBACL,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO;gBACjC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,MAAM;gBAC5B,MAAM,EAAE,2BAA2B,KAAK,EAAE;aAC3C,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,UAA2B,EAAE;IAE7B,OAAO;QACL,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,qBAAqB;QACvC,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,GAAG;QAC/B,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,EAAE;QAChC,MAAM,EAAE,CAAC,OAAO,EAAE,SAAS,EAAE,iBAAiB,EAAE,aAAa,EAAE,YAAY,EAAE,QAAQ,CAAC;QACtF,QAAQ,CAAC,KAAK;YACZ,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC;YAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC;YAC3C,IAAI,CAAC,QAAQ,IAAI,CAAC,MAAM;gBAAE,OAAO,IAAI,CAAC;YACtC,MAAM,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAC5C,CAAC,CAAC,GAAG,EAAE,aAAa,CAAC,EAAE,EAAE,CACvB,aAAa,KAAK,SAAS,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,aAAa,CAC/D,CAAC;YACF,IAAI,CAAC,QAAQ;gBAAE,OAAO,IAAI,CAAC;YAC3B,OAAO;gBACL,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO;gBACjC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,UAAU;gBAChC,MAAM,EAAE,uCAAuC,QAAQ,CAAC,CAAC,CAAC,GAAG;aAC9D,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,KAAc,EAAE,KAAa,EAAE,OAAO,IAAI,OAAO,EAAU;IAClF,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC,MAAM,CAAC;IACnD,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;IACvE,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;IAC3D,IAAI,IAAI,CAAC,GAAG,CAAC,KAAe,CAAC;QAAE,OAAO,KAAK,GAAG,CAAC,CAAC;IAChD,IAAI,CAAC,GAAG,CAAC,KAAe,CAAC,CAAC;IAC1B,IAAI,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACzC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAClC,CAAC,CAAC,KAAK;QACP,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,CAAC;IACrD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAClD,KAAK,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,GAAG,KAAK,EAAE,IAAI,CAAC,CAAC;QAC5E,CAAC;aAAM,CAAC;YACN,KAAK,IAAI,eAAe,CAAC,KAAK,EAAE,KAAK,GAAG,KAAK,EAAE,IAAI,CAAC,CAAC;QACvD,CAAC;QACD,IAAI,KAAK,GAAG,KAAK;YAAE,OAAO,KAAK,CAAC;IAClC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,OAGxC;IACC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,OAAO,CAAC,aAAa,GAAG,CAAC,EAAE,CAAC;QAC9E,MAAM,IAAI,SAAS,CAAC,+CAA+C,CAAC,CAAC;IACvE,CAAC;IACD,OAAO;QACL,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,uBAAuB;QACzC,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,GAAG;QAC/B,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,EAAE;QAChC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,QAAQ,CAAC,KAAK;YACZ,IAAI,eAAe,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,aAAa,CAAC,IAAI,OAAO,CAAC,aAAa;gBAAE,OAAO,IAAI,CAAC;YAC9F,OAAO;gBACL,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO;gBACjC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,MAAM;gBAC5B,MAAM,EAAE,qBAAqB,OAAO,CAAC,aAAa,4BAA4B;aAC/E,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kCAAkC;IAChD,OAAO,MAAM,CAAC,MAAM,CAAC;QACnB,sBAAsB,EAAE;QACxB,yBAAyB,EAAE;QAC3B,gCAAgC,EAAE;QAClC,0BAA0B,EAAE;QAC5B,yBAAyB,CAAC;YACxB,EAAE,EAAE,oBAAoB;YACxB,MAAM,EAAE,CAAC,aAAa,CAAC;YACvB,MAAM,EAAE,QAAQ;SACjB,CAAC;QACF,yBAAyB,CAAC;YACxB,EAAE,EAAE,qBAAqB;YACzB,MAAM,EAAE,CAAC,YAAY,CAAC;YACtB,MAAM,EAAE,QAAQ;SACjB,CAAC;KACH,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,oCAAoC,CAClD,aAAsB;IAEtB,OAAO,MAAM,CAAC,MAAM,CAAC;QACnB,sBAAsB,EAAE;QACxB,GAAG,CAAC,aAAa,KAAK,SAAS;YAC7B,CAAC,CAAC,CAAC,wBAAwB,CAAC;oBACxB,EAAE,EAAE,sBAAsB;oBAC1B,MAAM,EAAE,CAAC,QAAQ,CAAC;oBAClB,aAAa;oBACb,MAAM,EAAE,OAAO;iBAChB,CAAC,CAAC;YACL,CAAC,CAAC,EAAE,CAAC;QACP,yBAAyB,CAAC;YACxB,EAAE,EAAE,gBAAgB;YACpB,MAAM,EAAE,CAAC,QAAQ,CAAC;YAClB,MAAM,EAAE,QAAQ;SACjB,CAAC;KACH,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { RuntimeGuardrailEngineOptions, RuntimeGuardrailEvaluation, RuntimeGuardrailEvaluationInput, RuntimeGuardrailPolicy } from "./types.js";
|
|
2
|
+
export declare class RuntimeGuardrailEngine {
|
|
3
|
+
private readonly policies;
|
|
4
|
+
private readonly options;
|
|
5
|
+
private sequence;
|
|
6
|
+
constructor(policies: readonly RuntimeGuardrailPolicy[], options?: RuntimeGuardrailEngineOptions);
|
|
7
|
+
evaluate<T>(input: RuntimeGuardrailEvaluationInput<T>): Promise<RuntimeGuardrailEvaluation<T>>;
|
|
8
|
+
private createId;
|
|
9
|
+
private throwIfAborted;
|
|
10
|
+
private emitDecision;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=engine.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"engine.d.ts","sourceRoot":"","sources":["../../src/guardrails/engine.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EACV,6BAA6B,EAC7B,0BAA0B,EAC1B,+BAA+B,EAC/B,sBAAsB,EAEvB,MAAM,YAAY,CAAC;AA0FpB,qBAAa,sBAAsB;IACjC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA2B;IACpD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAgC;IACxD,OAAO,CAAC,QAAQ,CAAK;gBAGnB,QAAQ,EAAE,SAAS,sBAAsB,EAAE,EAC3C,OAAO,GAAE,6BAAkC;IASvC,QAAQ,CAAC,CAAC,EACd,KAAK,EAAE,+BAA+B,CAAC,CAAC,CAAC,GACxC,OAAO,CAAC,0BAA0B,CAAC,CAAC,CAAC,CAAC;IAsEzC,OAAO,CAAC,QAAQ;IAIhB,OAAO,CAAC,cAAc;YAOR,YAAY;CAyB3B"}
|
|
@@ -0,0 +1,189 @@
|
|
|
1
|
+
import { GuardrailAbortedError } from "./errors.js";
|
|
2
|
+
const VALID_ACTIONS = new Set([
|
|
3
|
+
"allow",
|
|
4
|
+
"audit",
|
|
5
|
+
"taint",
|
|
6
|
+
"redact",
|
|
7
|
+
"rewrite",
|
|
8
|
+
"block",
|
|
9
|
+
"approval",
|
|
10
|
+
]);
|
|
11
|
+
const VALID_RISKS = new Set([
|
|
12
|
+
"none",
|
|
13
|
+
"low",
|
|
14
|
+
"medium",
|
|
15
|
+
"high",
|
|
16
|
+
"critical",
|
|
17
|
+
]);
|
|
18
|
+
function cloneValue(value, seen = new WeakMap()) {
|
|
19
|
+
if (value === null || typeof value !== "object")
|
|
20
|
+
return value;
|
|
21
|
+
const existing = seen.get(value);
|
|
22
|
+
if (existing !== undefined)
|
|
23
|
+
return existing;
|
|
24
|
+
if (Array.isArray(value)) {
|
|
25
|
+
const clone = [];
|
|
26
|
+
seen.set(value, clone);
|
|
27
|
+
for (const item of value)
|
|
28
|
+
clone.push(cloneValue(item, seen));
|
|
29
|
+
return clone;
|
|
30
|
+
}
|
|
31
|
+
if (value instanceof Date)
|
|
32
|
+
return new Date(value.getTime());
|
|
33
|
+
const clone = {};
|
|
34
|
+
seen.set(value, clone);
|
|
35
|
+
for (const [key, entry] of Object.entries(value)) {
|
|
36
|
+
clone[key] = cloneValue(entry, seen);
|
|
37
|
+
}
|
|
38
|
+
return clone;
|
|
39
|
+
}
|
|
40
|
+
function freezeValue(value, seen = new WeakSet()) {
|
|
41
|
+
if (value === null || typeof value !== "object" || seen.has(value))
|
|
42
|
+
return value;
|
|
43
|
+
seen.add(value);
|
|
44
|
+
for (const entry of Object.values(value))
|
|
45
|
+
freezeValue(entry, seen);
|
|
46
|
+
return Object.freeze(value);
|
|
47
|
+
}
|
|
48
|
+
function freezeEvaluation(value, decisions, terminalAction) {
|
|
49
|
+
const frozenDecisions = Object.freeze(decisions.map((decision) => Object.freeze(decision)));
|
|
50
|
+
return Object.freeze({
|
|
51
|
+
value: freezeValue(value),
|
|
52
|
+
decisions: frozenDecisions,
|
|
53
|
+
...(terminalAction ? { terminalAction } : {}),
|
|
54
|
+
});
|
|
55
|
+
}
|
|
56
|
+
function assertPolicyResult(policy, result) {
|
|
57
|
+
if (!VALID_ACTIONS.has(result.action)) {
|
|
58
|
+
throw new Error(`Policy "${policy.id}" returned an invalid action`);
|
|
59
|
+
}
|
|
60
|
+
if (!VALID_RISKS.has(result.risk)) {
|
|
61
|
+
throw new Error(`Policy "${policy.id}" returned an invalid risk`);
|
|
62
|
+
}
|
|
63
|
+
if (typeof result.reason !== "string" || result.reason.trim().length === 0) {
|
|
64
|
+
throw new Error(`Policy "${policy.id}" returned an empty reason`);
|
|
65
|
+
}
|
|
66
|
+
if ((result.action === "redact" || result.action === "rewrite") &&
|
|
67
|
+
!Object.prototype.hasOwnProperty.call(result, "value")) {
|
|
68
|
+
throw new Error(`Policy "${policy.id}" must return a value for ${result.action}`);
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
function defaultPolicyFailure(input, options) {
|
|
72
|
+
if (input.tool?.sideEffect === "read" && options.readOnlyPolicyFailure !== "block") {
|
|
73
|
+
return { action: "audit", risk: "medium" };
|
|
74
|
+
}
|
|
75
|
+
return { action: "block", risk: "high" };
|
|
76
|
+
}
|
|
77
|
+
export class RuntimeGuardrailEngine {
|
|
78
|
+
policies;
|
|
79
|
+
options;
|
|
80
|
+
sequence = 0;
|
|
81
|
+
constructor(policies, options = {}) {
|
|
82
|
+
this.policies = policies
|
|
83
|
+
.map((policy, index) => ({ policy, index }))
|
|
84
|
+
.sort((a, b) => (a.policy.priority ?? 100) - (b.policy.priority ?? 100) || a.index - b.index)
|
|
85
|
+
.map(({ policy }) => policy);
|
|
86
|
+
this.options = options;
|
|
87
|
+
}
|
|
88
|
+
async evaluate(input) {
|
|
89
|
+
this.throwIfAborted(input.signal);
|
|
90
|
+
let value = freezeValue(cloneValue(input.value));
|
|
91
|
+
const decisions = [];
|
|
92
|
+
for (const policy of this.policies) {
|
|
93
|
+
if (!policy.phases.includes(input.phase))
|
|
94
|
+
continue;
|
|
95
|
+
this.throwIfAborted(input.signal, decisions);
|
|
96
|
+
const startedAt = (this.options.now ?? Date.now)();
|
|
97
|
+
let result;
|
|
98
|
+
let failure;
|
|
99
|
+
try {
|
|
100
|
+
result = await policy.evaluate(Object.freeze({
|
|
101
|
+
...input,
|
|
102
|
+
value,
|
|
103
|
+
}));
|
|
104
|
+
this.throwIfAborted(input.signal, decisions);
|
|
105
|
+
if (result)
|
|
106
|
+
assertPolicyResult(policy, result);
|
|
107
|
+
}
|
|
108
|
+
catch (error) {
|
|
109
|
+
if (error instanceof GuardrailAbortedError)
|
|
110
|
+
throw error;
|
|
111
|
+
failure = error;
|
|
112
|
+
result = null;
|
|
113
|
+
}
|
|
114
|
+
const latencyMs = Math.max(0, (this.options.now ?? Date.now)() - startedAt);
|
|
115
|
+
let decision;
|
|
116
|
+
if (failure) {
|
|
117
|
+
const fallback = defaultPolicyFailure(input, this.options);
|
|
118
|
+
decision = {
|
|
119
|
+
id: this.createId(),
|
|
120
|
+
policyId: policy.id,
|
|
121
|
+
...(policy.version ? { policyVersion: policy.version } : {}),
|
|
122
|
+
phase: input.phase,
|
|
123
|
+
action: fallback.action,
|
|
124
|
+
risk: fallback.risk,
|
|
125
|
+
reason: "Policy evaluation failed",
|
|
126
|
+
latencyMs,
|
|
127
|
+
fallbackUsed: true,
|
|
128
|
+
};
|
|
129
|
+
}
|
|
130
|
+
else if (result) {
|
|
131
|
+
decision = {
|
|
132
|
+
id: this.createId(),
|
|
133
|
+
policyId: policy.id,
|
|
134
|
+
...(policy.version ? { policyVersion: policy.version } : {}),
|
|
135
|
+
phase: input.phase,
|
|
136
|
+
action: result.action,
|
|
137
|
+
risk: result.risk,
|
|
138
|
+
reason: result.reason.trim(),
|
|
139
|
+
latencyMs,
|
|
140
|
+
fallbackUsed: result.fallbackUsed ?? false,
|
|
141
|
+
};
|
|
142
|
+
if (result.action === "redact" || result.action === "rewrite") {
|
|
143
|
+
value = freezeValue(cloneValue(result.value));
|
|
144
|
+
}
|
|
145
|
+
}
|
|
146
|
+
if (!decision)
|
|
147
|
+
continue;
|
|
148
|
+
const frozenDecision = Object.freeze(decision);
|
|
149
|
+
decisions.push(frozenDecision);
|
|
150
|
+
await this.emitDecision(frozenDecision, input);
|
|
151
|
+
this.throwIfAborted(input.signal, decisions);
|
|
152
|
+
if (decision.action === "block" || decision.action === "approval") {
|
|
153
|
+
return freezeEvaluation(value, decisions, decision.action);
|
|
154
|
+
}
|
|
155
|
+
}
|
|
156
|
+
return freezeEvaluation(value, decisions);
|
|
157
|
+
}
|
|
158
|
+
createId() {
|
|
159
|
+
return this.options.createId?.() ?? `guardrail-${++this.sequence}`;
|
|
160
|
+
}
|
|
161
|
+
throwIfAborted(signal, decisions = []) {
|
|
162
|
+
if (signal?.aborted)
|
|
163
|
+
throw new GuardrailAbortedError(undefined, decisions);
|
|
164
|
+
}
|
|
165
|
+
async emitDecision(decision, input) {
|
|
166
|
+
try {
|
|
167
|
+
await this.options.onDecision?.(Object.freeze({
|
|
168
|
+
decision,
|
|
169
|
+
context: input.context,
|
|
170
|
+
...(input.tool
|
|
171
|
+
? {
|
|
172
|
+
tool: Object.freeze({
|
|
173
|
+
name: input.tool.name,
|
|
174
|
+
callId: input.tool.callId,
|
|
175
|
+
sideEffect: input.tool.sideEffect,
|
|
176
|
+
}),
|
|
177
|
+
}
|
|
178
|
+
: {}),
|
|
179
|
+
...(input.outputTruncated !== undefined
|
|
180
|
+
? { outputTruncated: input.outputTruncated }
|
|
181
|
+
: {}),
|
|
182
|
+
}));
|
|
183
|
+
}
|
|
184
|
+
catch {
|
|
185
|
+
// Audit sinks are observational and cannot change enforcement.
|
|
186
|
+
}
|
|
187
|
+
}
|
|
188
|
+
}
|
|
189
|
+
//# sourceMappingURL=engine.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"engine.js","sourceRoot":"","sources":["../../src/guardrails/engine.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AASpD,MAAM,aAAa,GAAG,IAAI,GAAG,CAAyB;IACpD,OAAO;IACP,OAAO;IACP,OAAO;IACP,QAAQ;IACR,SAAS;IACT,OAAO;IACP,UAAU;CACX,CAAC,CAAC;AAEH,MAAM,WAAW,GAAG,IAAI,GAAG,CAAuB;IAChD,MAAM;IACN,KAAK;IACL,QAAQ;IACR,MAAM;IACN,UAAU;CACX,CAAC,CAAC;AAEH,SAAS,UAAU,CAAI,KAAQ,EAAE,OAAO,IAAI,OAAO,EAAmB;IACpE,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC9D,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,KAAe,CAAC,CAAC;IAC3C,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,QAAa,CAAC;IACjD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,KAAK,GAAc,EAAE,CAAC;QAC5B,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACvB,KAAK,MAAM,IAAI,IAAI,KAAK;YAAE,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QAC7D,OAAO,KAAU,CAAC;IACpB,CAAC;IACD,IAAI,KAAK,YAAY,IAAI;QAAE,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAM,CAAC;IACjE,MAAM,KAAK,GAA4B,EAAE,CAAC;IAC1C,IAAI,CAAC,GAAG,CAAC,KAAe,EAAE,KAAK,CAAC,CAAC;IACjC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;QAC5E,KAAK,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,KAAU,CAAC;AACpB,CAAC;AAED,SAAS,WAAW,CAAI,KAAQ,EAAE,OAAO,IAAI,OAAO,EAAU;IAC5D,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,KAAe,CAAC;QAAE,OAAO,KAAK,CAAC;IAC3F,IAAI,CAAC,GAAG,CAAC,KAAe,CAAC,CAAC;IAC1B,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,KAAgC,CAAC;QAAE,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC9F,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,gBAAgB,CACvB,KAAQ,EACR,SAAqC,EACrC,cAAqC;IAErC,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAC5F,OAAO,MAAM,CAAC,MAAM,CAAC;QACnB,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC;QACzB,SAAS,EAAE,eAAe;QAC1B,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC9C,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kBAAkB,CACzB,MAAiC,EACjC,MAAuC;IAEvC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,WAAW,MAAM,CAAC,EAAE,8BAA8B,CAAC,CAAC;IACtE,CAAC;IACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,WAAW,MAAM,CAAC,EAAE,4BAA4B,CAAC,CAAC;IACpE,CAAC;IACD,IAAI,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3E,MAAM,IAAI,KAAK,CAAC,WAAW,MAAM,CAAC,EAAE,4BAA4B,CAAC,CAAC;IACpE,CAAC;IACD,IACE,CAAC,MAAM,CAAC,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,CAAC;QAC3D,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,EACtD,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,WAAW,MAAM,CAAC,EAAE,6BAA6B,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IACpF,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAC3B,KAAsC,EACtC,OAAsC;IAEtC,IAAI,KAAK,CAAC,IAAI,EAAE,UAAU,KAAK,MAAM,IAAI,OAAO,CAAC,qBAAqB,KAAK,OAAO,EAAE,CAAC;QACnF,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;IAC7C,CAAC;IACD,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;AAC3C,CAAC;AAED,MAAM,OAAO,sBAAsB;IAChB,QAAQ,CAA2B;IACnC,OAAO,CAAgC;IAChD,QAAQ,GAAG,CAAC,CAAC;IAErB,YACE,QAA2C,EAC3C,UAAyC,EAAE;QAE3C,IAAI,CAAC,QAAQ,GAAG,QAAQ;aACrB,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;aAC3C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;aAC5F,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,KAAyC;QAEzC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAClC,IAAI,KAAK,GAAG,WAAW,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;QACjD,MAAM,SAAS,GAA+B,EAAE,CAAC;QAEjD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC;gBAAE,SAAS;YACnD,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YACnD,IAAI,MAA8C,CAAC;YACnD,IAAI,OAAgB,CAAC;YAErB,IAAI,CAAC;gBACH,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC;oBAC3C,GAAG,KAAK;oBACR,KAAK;iBACN,CAAC,CAA2C,CAAC;gBAC9C,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;gBAC7C,IAAI,MAAM;oBAAE,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,qBAAqB;oBAAE,MAAM,KAAK,CAAC;gBACxD,OAAO,GAAG,KAAK,CAAC;gBAChB,MAAM,GAAG,IAAI,CAAC;YAChB,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,SAAS,CAAC,CAAC;YAC5E,IAAI,QAA8C,CAAC;YACnD,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,QAAQ,GAAG,oBAAoB,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC3D,QAAQ,GAAG;oBACT,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE;oBACnB,QAAQ,EAAE,MAAM,CAAC,EAAE;oBACnB,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5D,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,MAAM,EAAE,QAAQ,CAAC,MAAM;oBACvB,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,MAAM,EAAE,0BAA0B;oBAClC,SAAS;oBACT,YAAY,EAAE,IAAI;iBACnB,CAAC;YACJ,CAAC;iBAAM,IAAI,MAAM,EAAE,CAAC;gBAClB,QAAQ,GAAG;oBACT,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE;oBACnB,QAAQ,EAAE,MAAM,CAAC,EAAE;oBACnB,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5D,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE;oBAC5B,SAAS;oBACT,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,KAAK;iBAC3C,CAAC;gBACF,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;oBAC9D,KAAK,GAAG,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC,KAAU,CAAC,CAAC,CAAC;gBACrD,CAAC;YACH,CAAC;YAED,IAAI,CAAC,QAAQ;gBAAE,SAAS;YACxB,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC/C,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC/B,MAAM,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;YAC/C,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAC7C,IAAI,QAAQ,CAAC,MAAM,KAAK,OAAO,IAAI,QAAQ,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBAClE,OAAO,gBAAgB,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAED,OAAO,gBAAgB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAC5C,CAAC;IAEO,QAAQ;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,IAAI,aAAa,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;IACrE,CAAC;IAEO,cAAc,CACpB,MAAoB,EACpB,YAAiD,EAAE;QAEnD,IAAI,MAAM,EAAE,OAAO;YAAE,MAAM,IAAI,qBAAqB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IAC7E,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,QAAkC,EAClC,KAAsC;QAEtC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC;gBAC5C,QAAQ;gBACR,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,GAAG,CAAC,KAAK,CAAC,IAAI;oBACZ,CAAC,CAAC;wBACE,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC;4BAClB,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI;4BACrB,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM;4BACzB,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU;yBAClC,CAAC;qBACH;oBACH,CAAC,CAAC,EAAE,CAAC;gBACP,GAAG,CAAC,KAAK,CAAC,eAAe,KAAK,SAAS;oBACrC,CAAC,CAAC,EAAE,eAAe,EAAE,KAAK,CAAC,eAAe,EAAE;oBAC5C,CAAC,CAAC,EAAE,CAAC;aACR,CAAC,CAAC,CAAC;QACN,CAAC;QAAC,MAAM,CAAC;YACP,+DAA+D;QACjE,CAAC;IACH,CAAC;CACF"}
|