agent-inspect 2.4.0 → 2.5.0

package/packages/cli/dist/index.mjs

@@ -4,7 +4,7 @@ import path13 from 'path';
 import { fileURLToPath, pathToFileURL } from 'url';
 import { Command, Option } from 'commander';
 import { AsyncLocalStorage } from 'async_hooks';
-import crypto, { webcrypto } from 'crypto';
+import crypto, { webcrypto, createHash } from 'crypto';
 import { unlink, stat, mkdir, writeFile, appendFile, readdir, readFile, access, open } from 'fs/promises';
 import os from 'os';
 import process2, { stdin, stdout } from 'process';
@@ -12,7 +12,7 @@ import tty from 'tty';
 import { createInterface } from 'readline';
 
 // package.json
-var version = "2.4.0";
+var version = "2.5.0";
 
 // packages/core/src/correlation-metadata.ts
 var TRACE_CORRELATION_KEYS = [
@@ -1429,18 +1429,18 @@ var proto = Object.defineProperties(() => {
     }
   }
 });
-var createStyler = (open2, close, parent) => {
+var createStyler = (open3, close, parent) => {
   let openAll;
   let closeAll;
   if (parent === void 0) {
-    openAll = open2;
+    openAll = open3;
     closeAll = close;
   } else {
-    openAll = parent.openAll + open2;
+    openAll = parent.openAll + open3;
     closeAll = close + parent.closeAll;
   }
   return {
-    open: open2,
+    open: open3,
     close,
     openAll,
     closeAll,
@@ -1505,15 +1505,15 @@ function getStatusIcon(status) {
   if (status === "error") return source_default.red("\u2716");
   return source_default.yellow("\u23F3");
 }
-function renderStepLine(name, durationMs, status, depth) {
+function renderStepLine(name, durationMs2, status, depth) {
   try {
     const nm = formatTerminalName(name);
     const ind = getIndent(depth ?? 0);
-    if (status === "running" && durationMs === void 0) {
+    if (status === "running" && durationMs2 === void 0) {
       return `${ind}${source_default.yellow("\u23F3")} ${nm}`;
     }
-    const hasDur = durationMs !== void 0 && Number.isFinite(durationMs);
-    const dur = hasDur ? formatDuration2(durationMs) : void 0;
+    const hasDur = durationMs2 !== void 0 && Number.isFinite(durationMs2);
+    const dur = hasDur ? formatDuration2(durationMs2) : void 0;
     if (status === "running") {
       return dur !== void 0 ? `${ind}${source_default.yellow("\u23F3")} ${nm} (${dur})` : `${ind}${source_default.yellow("\u23F3")} ${nm}`;
     }
@@ -1666,14 +1666,14 @@ async function extractMetadata(filePath, _quickScan) {
   } else {
     status = "unknown";
   }
-  const durationMs = explicitDurationMs ?? (startedAt !== void 0 && endedAt !== void 0 && Number.isFinite(startedAt) && Number.isFinite(endedAt) && endedAt >= startedAt ? endedAt - startedAt : void 0);
+  const durationMs2 = explicitDurationMs ?? (startedAt !== void 0 && endedAt !== void 0 && Number.isFinite(startedAt) && Number.isFinite(endedAt) && endedAt >= startedAt ? endedAt - startedAt : void 0);
   return {
     runId: resolvedRunId,
     name,
     status,
     startedAt,
     endedAt,
-    durationMs,
+    durationMs: durationMs2,
     eventCount: parsedTrace.sourceEventCount,
     filePath,
     fileSize: stats.size,
@@ -1694,7 +1694,7 @@ function buildRunSummary(events) {
   const runId = started?.runId ?? events.find((e) => typeof e.runId === "string")?.runId ?? "unknown-run";
   const name = typeof started?.name === "string" && started.name.trim() !== "" ? started.name : void 0;
   const status = lastCompleted ? lastCompleted.status : started ? "running" : "unknown";
-  const durationMs = lastCompleted && isFiniteNumber(lastCompleted.durationMs) ? lastCompleted.durationMs : void 0;
+  const durationMs2 = lastCompleted && isFiniteNumber(lastCompleted.durationMs) ? lastCompleted.durationMs : void 0;
   started && isFiniteNumber(started.startTime) ? started.startTime : void 0;
   const steps = /* @__PURE__ */ new Map();
   for (const e of events) {
@@ -1784,7 +1784,7 @@ function buildRunSummary(events) {
     runId,
     name,
     status,
-    durationMs,
+    durationMs: durationMs2,
     totalSteps,
     llmSteps,
     toolSteps,
@@ -3215,10 +3215,10 @@ function statusIcon(status) {
   if (status === "running") return "\u23F3";
   return "?";
 }
-function durationCell(status, durationMs) {
+function durationCell(status, durationMs2) {
   if (status === "running" || status === "unknown") return "-";
-  if (durationMs !== void 0 && Number.isFinite(durationMs)) {
-    return formatDuration2(durationMs);
+  if (durationMs2 !== void 0 && Number.isFinite(durationMs2)) {
+    return formatDuration2(durationMs2);
   }
   return "-";
 }
@@ -3875,7 +3875,7 @@ var TreeBuilder = class {
       const startedAt = sorted.length > 0 ? sorted[0].timestamp : void 0;
       const endedAt = sorted.length > 0 ? sorted[sorted.length - 1].timestamp : void 0;
       const status = computeRunStatus(sorted);
-      const durationMs = startedAt !== void 0 && endedAt !== void 0 && Number.isFinite(startedAt) && Number.isFinite(endedAt) && endedAt >= startedAt && status !== "running" ? endedAt - startedAt : void 0;
+      const durationMs2 = startedAt !== void 0 && endedAt !== void 0 && Number.isFinite(startedAt) && Number.isFinite(endedAt) && endedAt >= startedAt && status !== "running" ? endedAt - startedAt : void 0;
       const name = sorted.find((e) => e.kind === "RUN")?.name;
       out.push({
         runId,
@@ -3883,7 +3883,7 @@ var TreeBuilder = class {
         status,
         startedAt,
         endedAt: status === "running" ? void 0 : endedAt,
-        durationMs,
+        durationMs: durationMs2,
         children: roots,
         metadata: {
           totalEvents: sorted.length,
@@ -4578,9 +4578,9 @@ function mapOpenInferenceSpan(span, index, version2) {
   if (endedAt !== void 0) {
     event.endedAt = endedAt;
   }
-  const durationMs = durationBetweenIso(startedAt, endedAt);
-  if (durationMs !== void 0) {
-    event.durationMs = durationMs;
+  const durationMs2 = durationBetweenIso(startedAt, endedAt);
+  if (durationMs2 !== void 0) {
+    event.durationMs = durationMs2;
   }
   if (tokenUsage !== void 0) {
     event.tokenUsage = tokenUsage;
@@ -5165,9 +5165,9 @@ function mapOtlpSpan(context) {
   if (endedAt !== void 0) {
     event.endedAt = endedAt;
   }
-  const durationMs = durationBetweenIso(startedAt, endedAt);
-  if (durationMs !== void 0) {
-    event.durationMs = durationMs;
+  const durationMs2 = durationBetweenIso(startedAt, endedAt);
+  if (durationMs2 !== void 0) {
+    event.durationMs = durationMs2;
   }
   if (tokenUsage !== void 0) {
     event.tokenUsage = tokenUsage;
@@ -6172,13 +6172,13 @@ var EventNormalizer = class {
     const timestampMissing = parsedTs === void 0;
     const parentIdKey = cfg.parentIdKey;
     const parentId = parentIdKey ? safeString(raw[parentIdKey]) : void 0;
-    let durationMs;
+    let durationMs2;
     const durationKey = cfg.durationKey;
     if (durationKey) {
       const v = raw[durationKey];
-      if (isFiniteNumber2(v)) durationMs = v;
+      if (isFiniteNumber2(v)) durationMs2 = v;
     } else if (isFiniteNumber2(raw.durationMs)) {
-      durationMs = raw.durationMs;
+      durationMs2 = raw.durationMs;
     }
     let status;
     const statusKey = cfg.statusKey;
@@ -6225,7 +6225,7 @@ var EventNormalizer = class {
       kind,
       timestamp,
       ...status ? { status } : {},
-      ...durationMs !== void 0 ? { durationMs } : {},
+      ...durationMs2 !== void 0 ? { durationMs: durationMs2 } : {},
       ...Object.keys(attributes).length > 0 ? { attributes } : {},
       confidence,
       source: {
@@ -7813,7 +7813,7 @@ function manualTraceEventsToRunTree(events) {
   }
   const startedAt = started.startTime;
   const endedAt = lastCompleted !== void 0 && runStatus !== "running" ? lastCompleted.endTime : void 0;
-  const durationMs = lastCompleted !== void 0 && Number.isFinite(lastCompleted.durationMs) ? lastCompleted.durationMs : void 0;
+  const durationMs2 = lastCompleted !== void 0 && Number.isFinite(lastCompleted.durationMs) ? lastCompleted.durationMs : void 0;
   const steps = /* @__PURE__ */ new Map();
   for (const e of events) {
     if (e.event !== "step_started") continue;
@@ -7902,7 +7902,7 @@ function manualTraceEventsToRunTree(events) {
     status: runStatus,
     startedAt,
     endedAt,
-    durationMs,
+    durationMs: durationMs2,
     children: roots,
     metadata: {
       totalEvents: inspectNodes.size,
@@ -8215,7 +8215,7 @@ function manualTraceEventsToComparableRun(events) {
   let runStatus;
   if (lastCompleted === void 0) runStatus = "running";
   else runStatus = lastCompleted.status;
-  const durationMs = lastCompleted !== void 0 && Number.isFinite(lastCompleted.durationMs) ? lastCompleted.durationMs : void 0;
+  const durationMs2 = lastCompleted !== void 0 && Number.isFinite(lastCompleted.durationMs) ? lastCompleted.durationMs : void 0;
   const steps = /* @__PURE__ */ new Map();
   let order = 0;
   for (const e of events) {
@@ -8286,7 +8286,7 @@ function manualTraceEventsToComparableRun(events) {
     runId,
     name: rs.name,
     status: runStatus,
-    durationMs,
+    durationMs: durationMs2,
     steps: roots
   };
 }
@@ -11180,7 +11180,7 @@ function createSafetyRawContentRule(options = {}) {
 }
 function createSafetySecretPatternRule(options = {}) {
   const patterns = options.patterns ?? DEFAULT_SECRET_PATTERNS;
-  const maxStringLength = options.maxStringLength ?? 4096;
+  const maxStringLength2 = options.maxStringLength ?? 4096;
   return {
     id: "safety.secretPattern",
     category: "safety",
@@ -11190,7 +11190,7 @@ function createSafetySecretPatternRule(options = {}) {
       for (const event of context.events) {
         for (const entry of eventValueEntries(event, { includeSummaries: true, includeError: true })) {
           if (typeof entry.value !== "string") continue;
-          const sample = entry.value.slice(0, maxStringLength);
+          const sample = entry.value.slice(0, maxStringLength2);
           for (const pattern of patterns) {
             pattern.pattern.lastIndex = 0;
             if (!pattern.pattern.test(sample)) continue;
@@ -11446,6 +11446,621 @@ function runTraceChecks(input3, options = {}) {
     diagnostics
   };
 }
+var ALL_RULES = [
+  "circuit.same-tool-repetition",
+  "circuit.same-args-repetition",
+  "circuit.max-loop-iterations",
+  "circuit.max-retries",
+  "circuit.tool-timeout",
+  "circuit.runaway-llm-loop",
+  "circuit.excessive-branch-width"
+];
+function closed(ruleId, message) {
+  return { ruleId, status: "closed", severity: "info", message, evidence: [] };
+}
+function open2(ruleId, message, evidence, severity = "error") {
+  return {
+    ruleId,
+    status: severity === "warning" ? "warn" : "open",
+    severity,
+    message,
+    evidence
+  };
+}
+function isToolEvent(event) {
+  const name = event.name.toLowerCase();
+  return event.kind === "tool" || name.startsWith("tool:") || name.startsWith("function:") || name.includes(".tool.") || name.startsWith("mcp:");
+}
+function isLlmEvent(event) {
+  const name = event.name.toLowerCase();
+  return event.kind === "llm" || name.startsWith("llm:") || name.includes(".llm.") || name.includes("generation");
+}
+function toolLabel(event) {
+  const attrs = event.attributes ?? {};
+  const fromAttr = attrs.toolName ?? attrs.tool ?? attrs.function;
+  if (typeof fromAttr === "string" && fromAttr.length > 0) return fromAttr;
+  return event.name.replace(/^(tool:|function:|mcp:)/i, "");
+}
+function argsHash(toolName2, args) {
+  return createHash("sha256").update(`${toolName2}:${JSON.stringify(args ?? null)}`).digest("hex").slice(0, 16);
+}
+function toolArgs(event) {
+  const attrs = event.attributes ?? {};
+  return attrs.arguments ?? attrs.args ?? attrs.input ?? attrs.parameters;
+}
+function durationMs(event) {
+  if (typeof event.durationMs === "number") return event.durationMs;
+  const attrs = event.attributes ?? {};
+  const fromAttr = attrs.durationMs ?? attrs.duration;
+  return typeof fromAttr === "number" ? fromAttr : void 0;
+}
+function attemptNumber(event) {
+  const attrs = event.attributes ?? {};
+  const value = attrs.attempt ?? attrs.retryAttempt ?? attrs.retryCount;
+  return typeof value === "number" ? value : void 0;
+}
+function evaluateSameToolRepetition(events, maxRepeats) {
+  const ruleId = "circuit.same-tool-repetition";
+  const counts = /* @__PURE__ */ new Map();
+  for (const event of events.filter(isToolEvent)) {
+    const label = toolLabel(event);
+    counts.set(label, (counts.get(label) ?? 0) + 1);
+  }
+  const evidence = [];
+  for (const [toolName2, count] of counts) {
+    if (count > maxRepeats) {
+      evidence.push({ ruleId, toolName: toolName2, count, threshold: maxRepeats });
+    }
+  }
+  if (evidence.length === 0) {
+    return closed(ruleId, "Tool repetition within threshold.");
+  }
+  return open2(ruleId, "Same tool repeated beyond threshold.", evidence);
+}
+function evaluateSameArgsRepetition(events, maxRepeats) {
+  const ruleId = "circuit.same-args-repetition";
+  const counts = /* @__PURE__ */ new Map();
+  for (const event of events.filter(isToolEvent)) {
+    const label = toolLabel(event);
+    const hash = argsHash(label, toolArgs(event));
+    const key = `${label}:${hash}`;
+    const current = counts.get(key) ?? { toolName: label, count: 0 };
+    current.count += 1;
+    counts.set(key, current);
+  }
+  const evidence = [];
+  for (const entry of counts.values()) {
+    if (entry.count > maxRepeats) {
+      evidence.push({ ruleId, toolName: entry.toolName, count: entry.count, threshold: maxRepeats });
+    }
+  }
+  if (evidence.length === 0) {
+    return closed(ruleId, "Tool argument repetition within threshold.");
+  }
+  return open2(ruleId, "Same tool arguments repeated beyond threshold.", evidence);
+}
+function evaluateMaxLoopIterations(events, maxIterations) {
+  const ruleId = "circuit.max-loop-iterations";
+  const iterationEvents = events.filter((event) => {
+    const attrs = event.attributes ?? {};
+    return typeof attrs.iteration === "number" || event.name.toLowerCase().includes("loop");
+  });
+  const maxSeen = iterationEvents.reduce((max, event) => {
+    const attrs = event.attributes ?? {};
+    const iteration = typeof attrs.iteration === "number" ? attrs.iteration : max;
+    return Math.max(max, iteration);
+  }, iterationEvents.length);
+  if (maxSeen <= maxIterations) {
+    return closed(ruleId, "Loop iterations within threshold.");
+  }
+  return open2(ruleId, "Loop iterations exceeded threshold.", [
+    { ruleId, count: maxSeen, threshold: maxIterations }
+  ]);
+}
+function evaluateMaxRetries(events, maxRetries) {
+  const ruleId = "circuit.max-retries";
+  const attempts = events.map(attemptNumber).filter((value) => value !== void 0);
+  const maxAttempt = attempts.length > 0 ? Math.max(...attempts) : 0;
+  if (maxAttempt <= maxRetries) {
+    return closed(ruleId, "Retry count within threshold.");
+  }
+  return open2(ruleId, "Retry count exceeded threshold.", [
+    { ruleId, count: maxAttempt, threshold: maxRetries }
+  ]);
+}
+function evaluateToolTimeout(events, maxDurationMs) {
+  const ruleId = "circuit.tool-timeout";
+  const evidence = [];
+  for (const event of events.filter(isToolEvent)) {
+    const duration = durationMs(event);
+    if (duration !== void 0 && duration > maxDurationMs) {
+      evidence.push({
+        ruleId,
+        toolName: toolLabel(event),
+        count: duration,
+        threshold: maxDurationMs,
+        eventId: event.eventId
+      });
+    }
+  }
+  if (evidence.length === 0) {
+    return closed(ruleId, "Tool durations within timeout.");
+  }
+  return open2(ruleId, "Tool call exceeded configured timeout.", evidence, "warning");
+}
+function evaluateRunawayLlmLoop(events, maxLlmCalls) {
+  const ruleId = "circuit.runaway-llm-loop";
+  const llmCount = events.filter(isLlmEvent).length;
+  const hasTerminal = events.some((event) => {
+    const status = (event.status ?? event.attributes?.status ?? "").toString().toLowerCase();
+    return status === "ok" || status === "success" || status === "completed";
+  });
+  if (llmCount <= maxLlmCalls || hasTerminal) {
+    return closed(ruleId, "LLM call count within threshold or run completed.");
+  }
+  return open2(ruleId, "Runaway LLM loop detected.", [
+    { ruleId, count: llmCount, threshold: maxLlmCalls }
+  ]);
+}
+function evaluateExcessiveBranchWidth(events, maxWidth) {
+  const ruleId = "circuit.excessive-branch-width";
+  const children = /* @__PURE__ */ new Map();
+  for (const event of events) {
+    const parentId = event.parentId;
+    if (!parentId) continue;
+    children.set(parentId, (children.get(parentId) ?? 0) + 1);
+  }
+  const evidence = [];
+  for (const [parentId, count] of children) {
+    if (count > maxWidth) {
+      evidence.push({ ruleId, path: parentId, count, threshold: maxWidth });
+    }
+  }
+  if (evidence.length === 0) {
+    return closed(ruleId, "Branch width within threshold.");
+  }
+  return open2(ruleId, "Excessive parallel branch width detected.", evidence, "warning");
+}
+function runRule(ruleId, events, options) {
+  switch (ruleId) {
+    case "circuit.same-tool-repetition":
+      if (options.sameToolRepetition === void 0) return void 0;
+      return evaluateSameToolRepetition(events, options.sameToolRepetition.maxRepeats);
+    case "circuit.same-args-repetition":
+      if (options.sameArgsRepetition === void 0) return void 0;
+      return evaluateSameArgsRepetition(events, options.sameArgsRepetition.maxRepeats);
+    case "circuit.max-loop-iterations":
+      if (options.maxLoopIterations === void 0) return void 0;
+      return evaluateMaxLoopIterations(events, options.maxLoopIterations.maxIterations);
+    case "circuit.max-retries":
+      if (options.maxRetries === void 0) return void 0;
+      return evaluateMaxRetries(events, options.maxRetries.maxRetries);
+    case "circuit.tool-timeout":
+      if (options.toolTimeout === void 0) return void 0;
+      return evaluateToolTimeout(events, options.toolTimeout.maxDurationMs);
+    case "circuit.runaway-llm-loop":
+      if (options.runawayLlmLoop === void 0) return void 0;
+      return evaluateRunawayLlmLoop(events, options.runawayLlmLoop.maxLlmCalls);
+    case "circuit.excessive-branch-width":
+      if (options.excessiveBranchWidth === void 0) return void 0;
+      return evaluateExcessiveBranchWidth(events, options.excessiveBranchWidth.maxWidth);
+    default:
+      return void 0;
+  }
+}
+function runCircuits(events, options = {}) {
+  const selected = options.rules ?? ALL_RULES;
+  const results = [];
+  for (const ruleId of selected) {
+    const result = runRule(ruleId, events, options);
+    if (result) results.push(result);
+  }
+  const ok = !results.some((result) => result.status === "open" && result.severity === "error");
+  return { ok, results };
+}
+
+// packages/guardrails/src/rules.ts
+var DEFAULT_INJECTION_PATTERNS = [
+  "ignore previous instructions",
+  "ignore all prior",
+  "disregard your instructions",
+  "system prompt",
+  "you are now",
+  "jailbreak"
+];
+function pass(ruleId, message) {
+  return { ruleId, status: "pass", severity: "info", message, evidence: [] };
+}
+function fail(ruleId, message, evidence, severity = "error") {
+  return { ruleId, status: severity === "warning" ? "warn" : "fail", severity, message, evidence };
+}
+function boundedPreview(value, max = 80) {
+  if (value.length <= max) return value;
+  return `${value.slice(0, max - 3)}...`;
+}
+function evaluateBannedPhrase(text, options) {
+  const ruleId = "guardrail.banned-phrase";
+  const haystack = options.caseInsensitive !== false ? text.toLowerCase() : text;
+  const evidence = [];
+  for (const phrase of options.phrases) {
+    const needle = options.caseInsensitive !== false ? phrase.toLowerCase() : phrase;
+    if (needle.length > 0 && haystack.includes(needle)) {
+      evidence.push({ ruleId, match: phrase, preview: boundedPreview(text) });
+    }
+  }
+  if (evidence.length === 0) {
+    return pass(ruleId, "No banned phrases matched.");
+  }
+  return fail(ruleId, `Matched ${evidence.length} banned phrase(s).`, evidence);
+}
+var SEVERITY_RANK2 = { info: 0, warning: 1, error: 2 };
+function evaluatePiiLeak(value, options = {}) {
+  const ruleId = "guardrail.pii-leak";
+  const minSeverity = options.minSeverity ?? "warning";
+  const result = redact(value, { profile: options.profile ?? "share", collectFindings: true });
+  const findings = result.findings.filter(
+    (finding) => SEVERITY_RANK2[finding.severity] >= SEVERITY_RANK2[minSeverity]
+  );
+  if (findings.length === 0) {
+    return pass(ruleId, "No PII-style redaction findings.");
+  }
+  const evidence = findings.map((finding) => ({
+    ruleId,
+    path: finding.path,
+    detector: finding.detector,
+    preview: finding.preview
+  }));
+  return fail(ruleId, `Detected ${findings.length} PII-style finding(s).`, evidence);
+}
+function measureDepth(value) {
+  if (value === null || typeof value !== "object") return 0;
+  if (Array.isArray(value)) {
+    return 1 + Math.max(0, ...value.map((item) => measureDepth(item)));
+  }
+  const depths = Object.values(value).map((item) => measureDepth(item));
+  return 1 + (depths.length === 0 ? 0 : Math.max(...depths));
+}
+function maxStringLength(value) {
+  if (typeof value === "string") return value.length;
+  if (value === null || typeof value !== "object") return 0;
+  if (Array.isArray(value)) {
+    return Math.max(0, ...value.map((item) => maxStringLength(item)));
+  }
+  return Math.max(0, ...Object.values(value).map((item) => maxStringLength(item)));
+}
+function evaluateUnsafeToolArgs(toolName2, toolArgs2, options = {}) {
+  const ruleId = "guardrail.unsafe-tool-args";
+  const blocked = new Set((options.blockedTools ?? []).map((name) => name.toLowerCase()));
+  const evidence = [];
+  if (blocked.has(toolName2.toLowerCase())) {
+    evidence.push({ ruleId, preview: toolName2, match: toolName2 });
+  }
+  const maxDepth = options.maxDepth ?? 12;
+  const depth = measureDepth(toolArgs2);
+  if (depth > maxDepth) {
+    evidence.push({ ruleId, path: "args", preview: `depth=${depth}` });
+  }
+  const maxLen = options.maxStringLength ?? 16384;
+  const longest = maxStringLength(toolArgs2);
+  if (longest > maxLen) {
+    evidence.push({ ruleId, path: "args", preview: `maxStringLength=${longest}` });
+  }
+  if (evidence.length === 0) {
+    return pass(ruleId, "Tool arguments within configured bounds.");
+  }
+  return fail(ruleId, "Unsafe or oversized tool arguments detected.", evidence);
+}
+function evaluatePromptInjection(text, options = {}) {
+  const ruleId = "guardrail.prompt-injection";
+  const patterns = options.patterns ?? DEFAULT_INJECTION_PATTERNS;
+  const haystack = text.toLowerCase();
+  const evidence = [];
+  for (const pattern of patterns) {
+    const needle = pattern.toLowerCase();
+    if (needle.length > 0 && haystack.includes(needle)) {
+      evidence.push({ ruleId, match: pattern, preview: boundedPreview(text) });
+    }
+  }
+  if (evidence.length === 0) {
+    return pass(ruleId, "No prompt-injection patterns matched.");
+  }
+  return fail(ruleId, `Matched ${evidence.length} injection pattern(s).`, evidence, "warning");
+}
+function validateSchemaField(value, field, path15, evidence) {
+  const ruleId = "guardrail.structured-output";
+  if (field.type) {
+    const actual = value === null ? "null" : Array.isArray(value) ? "array" : typeof value;
+    if (actual !== field.type) {
+      evidence.push({ ruleId, path: path15, preview: `expected ${field.type}, got ${actual}` });
+      return;
+    }
+  }
+  if (field.enum && !field.enum.some((item) => Object.is(item, value))) {
+    evidence.push({ ruleId, path: path15, preview: "value not in enum" });
+  }
+  if (field.type === "object" && field.required && value && typeof value === "object" && !Array.isArray(value)) {
+    const record = value;
+    for (const key of field.required) {
+      if (!(key in record)) {
+        evidence.push({ ruleId, path: `${path15}.${key}`, preview: "missing required key" });
+      }
+    }
+  }
+}
+function evaluateStructuredOutput(value, options) {
+  const ruleId = "guardrail.structured-output";
+  if (!value || typeof value !== "object" || Array.isArray(value)) {
+    return fail(ruleId, "Structured output must be an object.", [
+      { ruleId, preview: typeof value }
+    ]);
+  }
+  const record = value;
+  const evidence = [];
+  for (const [key, field] of Object.entries(options.schema)) {
+    validateSchemaField(record[key], field, key, evidence);
+  }
+  if (evidence.length === 0) {
+    return pass(ruleId, "Structured output matches schema subset.");
+  }
+  return fail(ruleId, "Structured output schema violation.", evidence);
+}
+function evaluateOversizeOutput(value, options = {}) {
+  const ruleId = "guardrail.oversize-output";
+  const text = typeof value === "string" ? value : JSON.stringify(value);
+  const maxLength = options.maxLength ?? options.maxSerializedLength ?? 32768;
+  if (text.length <= maxLength) {
+    return pass(ruleId, "Output within size limits.");
+  }
+  return fail(ruleId, `Output exceeds max length (${text.length} > ${maxLength}).`, [
+    { ruleId, preview: `length=${text.length}` }
+  ]);
+}
+function evaluateRequiredJsonShape(value, options) {
+  const ruleId = "guardrail.required-json-shape";
+  let parsed = value;
+  if (typeof value === "string") {
+    try {
+      parsed = JSON.parse(value);
+    } catch {
+      return fail(ruleId, "Value is not valid JSON.", [{ ruleId, preview: boundedPreview(value) }]);
+    }
+  }
+  if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
+    return fail(ruleId, "JSON value must be an object.", [{ ruleId, preview: typeof parsed }]);
+  }
+  const record = parsed;
+  const evidence = [];
+  for (const key of options.requiredKeys) {
+    if (!(key in record)) {
+      evidence.push({ ruleId, path: key, preview: "missing required key" });
+    }
+  }
+  if (evidence.length === 0) {
+    return pass(ruleId, "Required JSON keys present.");
+  }
+  return fail(ruleId, "Missing required JSON keys.", evidence);
+}
+
+// packages/guardrails/src/run.ts
+var ALL_RULES2 = [
+  "guardrail.banned-phrase",
+  "guardrail.pii-leak",
+  "guardrail.unsafe-tool-args",
+  "guardrail.prompt-injection",
+  "guardrail.structured-output",
+  "guardrail.oversize-output",
+  "guardrail.required-json-shape"
+];
+function isErrorFailure(result) {
+  return result.status === "fail" && result.severity === "error";
+}
+function runRule2(ruleId, input3, options) {
+  switch (ruleId) {
+    case "guardrail.banned-phrase": {
+      if (!options.bannedPhrase || input3.text === void 0) return void 0;
+      return evaluateBannedPhrase(input3.text, options.bannedPhrase);
+    }
+    case "guardrail.pii-leak": {
+      if (input3.value === void 0 && input3.text === void 0) return void 0;
+      return evaluatePiiLeak(input3.value ?? input3.text, options.piiLeak);
+    }
+    case "guardrail.unsafe-tool-args": {
+      if (!input3.toolName) return void 0;
+      return evaluateUnsafeToolArgs(input3.toolName, input3.toolArgs ?? {}, options.unsafeToolArgs);
+    }
+    case "guardrail.prompt-injection": {
+      if (input3.text === void 0) return void 0;
+      return evaluatePromptInjection(input3.text, options.promptInjection);
+    }
+    case "guardrail.structured-output": {
+      if (!options.structuredOutput || input3.value === void 0) return void 0;
+      return evaluateStructuredOutput(input3.value, options.structuredOutput);
+    }
+    case "guardrail.oversize-output": {
+      if (input3.value === void 0 && input3.text === void 0) return void 0;
+      return evaluateOversizeOutput(input3.value ?? input3.text, options.oversizeOutput);
+    }
+    case "guardrail.required-json-shape": {
+      if (!options.requiredJsonShape || input3.value === void 0 && input3.text === void 0) return void 0;
+      return evaluateRequiredJsonShape(input3.value ?? input3.text, options.requiredJsonShape);
+    }
+    default:
+      return void 0;
+  }
+}
+function runGuardrails(input3, options = {}) {
+  const selected = options.rules ?? ALL_RULES2;
+  const results = [];
+  for (const ruleId of selected) {
+    const result = runRule2(ruleId, input3, options);
+    if (result) results.push(result);
+  }
+  const ok = !results.some(isErrorFailure);
+  return { ok, results };
+}
+
+// packages/cli/src/safety-extensions.ts
+var GUARDRAIL_ALIASES = {
+  "banned-phrase": "guardrail.banned-phrase",
+  "pii-leak": "guardrail.pii-leak",
+  "unsafe-tool-args": "guardrail.unsafe-tool-args",
+  "prompt-injection": "guardrail.prompt-injection",
+  "structured-output": "guardrail.structured-output",
+  "oversize-output": "guardrail.oversize-output",
+  "required-json-shape": "guardrail.required-json-shape"
+};
+var CIRCUIT_ALIASES = {
+  "same-tool-repetition": "circuit.same-tool-repetition",
+  "same-args-repetition": "circuit.same-args-repetition",
+  "max-loop-iterations": "circuit.max-loop-iterations",
+  "max-retries": "circuit.max-retries",
+  "tool-timeout": "circuit.tool-timeout",
+  "runaway-llm-loop": "circuit.runaway-llm-loop",
+  "excessive-branch-width": "circuit.excessive-branch-width"
+};
+function parseGuardrailRules(values) {
+  if (!values?.length) return void 0;
+  return values.map((value) => {
+    const rule = GUARDRAIL_ALIASES[value] ?? value;
+    return rule;
+  });
+}
+function parseCircuitRules(values) {
+  if (!values?.length) return void 0;
+  return values.map((value) => {
+    const rule = CIRCUIT_ALIASES[value] ?? value;
+    return rule;
+  });
+}
+function toSeverity(severity) {
+  return severity;
+}
+function guardrailFinding(result) {
+  return {
+    ruleId: result.ruleId,
+    severity: toSeverity(result.severity),
+    status: result.status === "pass" ? "pass" : result.status === "warn" ? "warning" : "fail",
+    message: result.message,
+    evidence: result.evidence.map(
+      (item) => ({
+        path: item.preview ? `${item.path ?? "value"} (${item.preview})` : item.path
+      })
+    )
+  };
+}
+function circuitFinding(result) {
+  return {
+    ruleId: result.ruleId,
+    severity: toSeverity(result.severity),
+    status: result.status === "closed" ? "pass" : result.status === "warn" ? "warning" : "fail",
+    message: result.message,
+    evidence: result.evidence.map(
+      (item) => ({
+        runId: item.runId,
+        eventId: item.eventId,
+        path: item.path,
+        name: item.toolName
+      })
+    ),
+    actual: result.evidence[0]?.count,
+    expected: result.evidence[0]?.threshold
+  };
+}
+function eventToCircuit(event) {
+  return {
+    eventId: event.eventId,
+    runId: event.runId,
+    name: event.name,
+    kind: event.kind,
+    parentId: event.parentId,
+    startedAt: event.startedAt,
+    endedAt: event.endedAt,
+    durationMs: event.durationMs,
+    attributes: event.attributes,
+    status: event.status
+  };
+}
+function collectGuardrailInputs(read) {
+  const inputs = [];
+  for (const event of read.events) {
+    const attrs = event.attributes ?? {};
+    for (const key of ["output", "answer", "text", "content", "message"]) {
+      const value = attrs[key];
+      if (typeof value === "string") inputs.push({ text: value });
+      else if (value !== void 0) inputs.push({ value });
+    }
+    if (event.kind === "TOOL" || event.name.startsWith("tool:")) {
+      inputs.push({
+        toolName: String(attrs.toolName ?? attrs.tool ?? event.name),
+        toolArgs: attrs.arguments ?? attrs.args ?? attrs.input
+      });
+    }
+  }
+  return inputs;
+}
+var DEFAULT_GUARDRAIL_OPTIONS = {
+  bannedPhrase: { phrases: ["delete all data", "ignore all instructions"] },
+  promptInjection: {},
+  piiLeak: { profile: "share" }
+};
+var DEFAULT_CIRCUIT_OPTIONS = {
+  sameToolRepetition: { maxRepeats: 3 },
+  sameArgsRepetition: { maxRepeats: 2 },
+  maxLoopIterations: { maxIterations: 20 },
+  maxRetries: { maxRetries: 3 },
+  toolTimeout: { maxDurationMs: 6e4 },
+  runawayLlmLoop: { maxLlmCalls: 12 },
+  excessiveBranchWidth: { maxWidth: 8 }
+};
+function mergeSafetyExtensions(result, read, options) {
+  const findings = [...result.findings];
+  let failed = result.summary.failed;
+  let warnings = result.summary.warnings;
+  let passed = result.summary.passed;
+  const guardrailRules = parseGuardrailRules(options.guardrails);
+  if (guardrailRules) {
+    for (const input3 of collectGuardrailInputs(read)) {
+      const run = runGuardrails(input3, {
+        ...DEFAULT_GUARDRAIL_OPTIONS,
+        rules: guardrailRules
+      });
+      for (const item of run.results) {
+        const finding = guardrailFinding(item);
+        findings.push(finding);
+        if (finding.status === "fail") failed += 1;
+        else if (finding.status === "warning") warnings += 1;
+        else passed += 1;
+      }
+    }
+  }
+  const circuitRules = parseCircuitRules(options.circuits);
+  if (circuitRules) {
+    const circuitRun = runCircuits(
+      read.events.map(eventToCircuit),
+      { ...DEFAULT_CIRCUIT_OPTIONS, rules: circuitRules }
+    );
+    for (const item of circuitRun.results) {
+      const finding = circuitFinding(item);
+      findings.push(finding);
+      if (finding.status === "fail") failed += 1;
+      else if (finding.status === "warning") warnings += 1;
+      else passed += 1;
+    }
+  }
+  const ok = failed === 0 && result.summary.errors === 0;
+  return {
+    ...result,
+    ok,
+    status: failed > 0 ? "fail" : result.status,
+    summary: {
+      passed,
+      failed,
+      warnings,
+      errors: result.summary.errors
+    },
+    findings
+  };
+}
 
 // packages/cli/src/check.ts
 var DEFAULT_SELECT = ["run.status"];
@@ -11685,12 +12300,19 @@ async function checkCommand(target, options = {}, stdin = process.stdin) {
           }
         );
         perRun.push(
-          runTraceChecks(
-            { read },
+          mergeSafetyExtensions(
+            runTraceChecks(
+              { read },
+              {
+                rules: built.rules,
+                select: built.select,
+                runId: meta.runId
+              }
+            ),
+            read,
             {
-              rules: built.rules,
-              select: built.select,
-              runId: meta.runId
+              ...options.guardrails ? { guardrails: options.guardrails } : {},
+              ...options.circuit ? { circuits: options.circuit } : {}
             }
           )
         );
@@ -11709,12 +12331,19 @@ async function checkCommand(target, options = {}, stdin = process.stdin) {
       const read = await openTrace(input3, {
         ...options.format !== void 0 ? { format: options.format } : {}
       });
-      result = runTraceChecks(
-        { read },
+      result = mergeSafetyExtensions(
+        runTraceChecks(
+          { read },
+          {
+            rules: built.rules,
+            select: built.select,
+            ...options.run !== void 0 ? { runId: options.run } : {}
+          }
+        ),
+        read,
         {
-          rules: built.rules,
-          select: built.select,
-          ...options.run !== void 0 ? { runId: options.run } : {}
+          ...options.guardrails ? { guardrails: options.guardrails } : {},
+          ...options.circuit ? { circuits: options.circuit } : {}
         }
       );
     }
@@ -11918,7 +12547,7 @@ function evidenceForEvent(event, path15) {
     }
   ];
 }
-function fail(ruleId, message, evidence, expected, actual) {
+function fail2(ruleId, message, evidence, expected, actual) {
   return {
     ruleId,
     status: "fail",
@@ -12114,7 +12743,7 @@ var checks = {
       "eval.requireSuccess",
       "run",
       (context) => context.run.status === "ok" ? [] : [
-        fail(
+        fail2(
           "eval.requireSuccess",
           "Run did not complete successfully.",
           evidenceForRun(context.run, "status"),
@@ -12129,7 +12758,7 @@ var checks = {
     return createRule("eval.requiredTools", "tool", (context) => {
       const tools = new Set(nodeNames(context.nodes, "TOOL"));
       return expected.filter((name) => !tools.has(name)).map(
-        (name) => fail(
+        (name) => fail2(
           "eval.requiredTools",
           `Required tool ${name} did not appear.`,
           evidenceForRun(context.run, "children"),
@@ -12145,7 +12774,7 @@ var checks = {
       "eval.forbiddenTools",
       "tool",
       (context) => context.nodes.filter((node) => node.event.kind === "TOOL" && blocked.includes(node.event.name)).map(
-        (node) => fail(
+        (node) => fail2(
           "eval.forbiddenTools",
           `Forbidden tool ${node.event.name} appeared.`,
           evidenceForEvent(node.event, "name"),
@@ -12160,7 +12789,7 @@ var checks = {
       "eval.maxDurationMs",
       "run",
       (context) => context.run.durationMs !== void 0 && context.run.durationMs > maxDurationMs ? [
-        fail(
+        fail2(
           "eval.maxDurationMs",
           `Run duration exceeded ${maxDurationMs}ms.`,
           evidenceForRun(context.run, "durationMs"),
@@ -12177,7 +12806,7 @@ var checks = {
         void 0
       );
       return deepest !== void 0 && deepest.depth > maxDepth ? [
-        fail(
+        fail2(
           "eval.maxDepth",
           `Run tree depth exceeded ${maxDepth}.`,
           evidenceForEvent(deepest.event, "depth"),
@@ -12194,7 +12823,7 @@ var checks = {
       (context) => context.nodes.flatMap((node) => {
         const retries = numericAttribute(node, ["retryCount", "retries", "attempt"]);
         return retries !== void 0 && retries > maxRetries ? [
-          fail(
+          fail2(
             "eval.maxRetries",
             `Retry count exceeded ${maxRetries}.`,
             evidenceForEvent(node.event, "attributes.retryCount"),
@@ -12209,7 +12838,7 @@ var checks = {
     return createRule("eval.maxTotalTokens", "llm", (context) => {
       const total = totalTokenCount(context.events);
       return total > maxTotalTokens ? [
-        fail(
+        fail2(
           "eval.maxTotalTokens",
           `Total token usage exceeded ${maxTotalTokens}.`,
           evidenceForRun(context.run, "tokenUsage.total"),
@@ -12224,7 +12853,7 @@ var checks = {
       "eval.noFailedSteps",
       "run",
       (context) => context.nodes.filter((node) => node.event.status === "error" || node.event.kind === "ERROR").map(
-        (node) => fail(
+        (node) => fail2(
           "eval.noFailedSteps",
           "Run contains a failed step or error node.",
           evidenceForEvent(node.event, "status"),
@@ -12242,7 +12871,7 @@ var checks = {
         (node, index) => index < firstLlmIndex && node.event.kind === "RETRIEVER"
       );
       return retrievalIndex === -1 ? [
-        fail(
+        fail2(
           "eval.requiredRetrievalBeforeGeneration",
           "No retrieval step appeared before the first LLM generation.",
           evidenceForEvent(context.nodes[firstLlmIndex].event, "kind"),
@@ -12258,7 +12887,7 @@ var checks = {
       const decisions = context.nodes.filter((node) => node.event.kind === "DECISION");
       if (decisions.length === 0) {
         return [
-          fail(
+          fail2(
             "eval.requiredDecisionMetadata",
             "No decision node is available for required metadata.",
             evidenceForRun(context.run, "children"),
@@ -12269,7 +12898,7 @@ var checks = {
       }
       return decisions.flatMap(
         (node) => required.filter((key) => !hasAttribute(node, key)).map(
-          (key) => fail(
+          (key) => fail2(
             "eval.requiredDecisionMetadata",
             `Decision metadata ${key} is missing.`,
             evidenceForEvent(node.event, `attributes.${key}`),
@@ -12290,7 +12919,7 @@ var checks = {
       const contexts = collectTextFields(context.nodes, contextKeys, ["RETRIEVER", "TOOL"]);
       if (answers.length === 0 || contexts.length === 0) {
         return [
-          fail(
+          fail2(
             "eval.contextOverlap",
             "Answer and context text are required for overlap evaluation.",
             firstEvidence(answers.length > 0 ? answers : contexts, context.run, "children"),
@@ -12304,7 +12933,7 @@ var checks = {
       const sharedTerms = [...answerTerms].filter((term) => contextTerms.has(term)).length;
       const overlap = answerTerms.size === 0 ? 0 : sharedTerms / answerTerms.size;
       return sharedTerms < minSharedTerms || overlap < minOverlap ? [
-        fail(
+        fail2(
           "eval.contextOverlap",
           "Answer text did not sufficiently overlap retrieved context.",
           firstEvidence(answers, context.run, "attributes.answer"),
@@ -12332,7 +12961,7 @@ var checks = {
       const quotes = quotedSnippets(answerText, minQuoteLength);
       if (quotes.length === 0) {
         return requireQuote ? [
-          fail(
+          fail2(
             "eval.quoteOverlap",
             "Answer did not contain a quote for overlap evaluation.",
             firstEvidence(answers, context.run, "attributes.answer"),
@@ -12343,7 +12972,7 @@ var checks = {
       }
       const missing = quotes.filter((quote) => !contextText.includes(quote.toLowerCase()));
       return missing.length > 0 ? [
-        fail(
+        fail2(
           "eval.quoteOverlap",
           "Quoted answer text did not appear in retrieved context.",
           firstEvidence(answers, context.run, "attributes.answer"),
@@ -12361,7 +12990,7 @@ var checks = {
       const citations = collectTextFields(context.nodes, citationKeys);
       const count = citationCount(answers.map((field) => field.text).join(" "), citations);
       return count === 0 ? [
-        fail(
+        fail2(
           "eval.citationPresence",
           "Answer did not include citations or source references.",
           firstEvidence(answers, context.run, "attributes.answer"),
@@ -12379,7 +13008,7 @@ var checks = {
       const availableSet = new Set(available);
       const missing = expected.filter((id) => !availableSet.has(id));
       return missing.length > 0 ? [
-        fail(
+        fail2(
           "eval.requiredSourceIds",
           "Required source IDs were not present in trace context or citations.",
           evidenceForRun(context.run, "children"),
@@ -12399,7 +13028,7 @@ var checks = {
       const tooShort = options.minCharacters !== void 0 && characters < options.minCharacters || options.minWords !== void 0 && words < options.minWords;
       const tooLong = options.maxCharacters !== void 0 && characters > options.maxCharacters || options.maxWords !== void 0 && words > options.maxWords;
       return answer.length === 0 || tooShort || tooLong ? [
-        fail(
+        fail2(
           "eval.answerLengthBounds",
           "Answer length fell outside required bounds.",
           firstEvidence(answers, context.run, "attributes.answer"),
@@ -12422,7 +13051,7 @@ var checks = {
       const answer = answers.map((field) => field.text).join(" ").toLowerCase();
       const matches = banned.filter((phrase) => answer.includes(phrase));
       return matches.length > 0 ? [
-        fail(
+        fail2(
           "eval.bannedUnsupportedPhrases",
           "Answer contained banned unsupported-answer phrasing.",
           firstEvidence(answers, context.run, "attributes.answer"),
@@ -12800,7 +13429,7 @@ function invalidArgumentResult(command, error) {
   });
 }
 function buildSafetyRules(options) {
-  const maxStringLength = parseLimit3(options.maxStringLength, "--max-string-length") ?? DEFAULT_MAX_STRING_LENGTH;
+  const maxStringLength2 = parseLimit3(options.maxStringLength, "--max-string-length") ?? DEFAULT_MAX_STRING_LENGTH;
   const maxArrayLength = parseLimit3(options.maxArrayLength, "--max-array-length") ?? DEFAULT_MAX_ARRAY_LENGTH;
   const maxObjectKeys = parseLimit3(options.maxObjectKeys, "--max-object-keys") ?? DEFAULT_MAX_OBJECT_KEYS;
   const maxSerializedBytes = parseLimit3(options.maxSerializedBytes, "--max-serialized-bytes") ?? DEFAULT_MAX_SERIALIZED_BYTES;
@@ -12809,7 +13438,7 @@ function buildSafetyRules(options) {
     createSafetyRedactionRule(),
     createSafetySecretPatternRule(),
     createSafetyOversizedAttributeRule({
-      maxStringLength,
+      maxStringLength: maxStringLength2,
       maxArrayLength,
       maxObjectKeys,
       maxSerializedBytes
@@ -13693,6 +14322,14 @@ function createCliProgram() {
   ]).option("--max-total-tokens <number>", "add llm.usage with a max total-token budget").option("--session <id>", "check all runs in a workflow session (requires --dir)").option("--group <id>", "check all runs sharing a groupId (requires --dir)").option(
     "--correlate-group",
     "when using --session, also match synthetic group: session keys"
+  ).option(
+    "--guardrails <rule>",
+    "run optional guardrail rules (repeatable): banned-phrase, pii-leak, prompt-injection, ...",
+    (value, previous = []) => [...previous, value]
+  ).option(
+    "--circuit <rule>",
+    "run optional circuit rules (repeatable): same-tool-repetition, max-retries, ...",
+    (value, previous = []) => [...previous, value]
   ).action((target, opts) => {
     runCommand(() => checkCommand(target, opts));
   });