plugin-sensitive-filter-xr 0.0.1 → 0.0.4

package/dist/lib/sensitiveFilter.js

@@ -1,13 +1,17 @@
-import { __decorate } from "tslib";
+import { __decorate, __metadata } from "tslib";
 import { z as z4 } from 'zod/v4';
 import { AIMessage, HumanMessage } from '@langchain/core/messages';
-import { Injectable } from '@nestjs/common';
-import { AgentMiddlewareStrategy, } from '@xpert-ai/plugin-sdk';
-import { SensitiveFilterIcon, resolveGeneralPackRules, sensitiveFilterConfigSchema, } from './types.js';
+import { Inject, Injectable } from '@nestjs/common';
+import { CommandBus } from '@nestjs/cqrs';
+import { AgentMiddlewareStrategy, CreateModelClientCommand, WrapWorkflowNodeExecutionCommand, } from '@xpert-ai/plugin-sdk';
+import { SensitiveFilterIcon, llmDecisionSchema, resolveGeneralPackRules, sensitiveFilterConfigSchema, } from './types.js';
 const SENSITIVE_FILTER_MIDDLEWARE_NAME = 'SensitiveFilterMiddleware';
 const DEFAULT_INPUT_BLOCK_MESSAGE = '输入内容触发敏感策略，已拦截。';
 const DEFAULT_OUTPUT_BLOCK_MESSAGE = '输出内容触发敏感策略，已拦截。';
 const DEFAULT_REWRITE_TEXT = '[已过滤]';
+const CONFIG_PARSE_ERROR = '敏感词过滤配置格式不正确，请检查填写内容。';
+const BUSINESS_RULES_VALIDATION_ERROR = '请至少配置 1 条有效业务规则（pattern/type/action/scope/severity），或启用通用规则包。';
+const LLM_MODE_VALIDATION_ERROR = '请完善 LLM 过滤配置：需填写过滤模型、生效范围、判定提示词、失败处理策略。';
 function isRecord(value) {
     return typeof value === 'object' && value !== null;
 }
@@ -62,9 +66,6 @@ function extractInputText(state, runtime) {
     }
     return '';
 }
-function escapeRegExp(value) {
-    return value.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
-}
 function getSeverityWeight(severity) {
     return severity === 'high' ? 2 : 1;
 }
@@ -166,7 +167,7 @@ function rewriteModelRequestInput(request, rewrittenText) {
 }
 function normalizeRuleDrafts(input) {
     const rules = [];
-    for (const draft of input) {
+    for (const [index, draft] of input.entries()) {
         if (!isRecord(draft)) {
             continue;
         }
@@ -181,7 +182,7 @@ function normalizeRuleDrafts(input) {
         if (!hasAnyValue) {
             continue;
         }
-        if (!id || !pattern || !type || !scope || !severity || !action) {
+        if (!pattern || !type || !action || !scope || !severity) {
             continue;
         }
         if (!['keyword', 'regex'].includes(type)) {
@@ -197,7 +198,7 @@ function normalizeRuleDrafts(input) {
             continue;
         }
         rules.push({
-            id,
+            id: id ?? `rule-${index + 1}`,
             pattern,
             type,
             scope,
@@ -208,6 +209,114 @@ function normalizeRuleDrafts(input) {
     }
     return rules;
 }
+function modeIncludesScope(scope, phase) {
+    return scope === 'both' || scope === phase;
+}
+function parseLlmDecision(raw, rewriteFallbackText) {
+    let payload = raw;
+    if (typeof payload === 'string') {
+        try {
+            payload = JSON.parse(payload);
+        }
+        catch {
+            throw new Error('LLM decision is not valid JSON string');
+        }
+    }
+    if (isRecord(payload) && !('matched' in payload) && 'content' in payload) {
+        const content = extractPrimitiveText(payload['content']).trim();
+        if (!content) {
+            throw new Error('LLM decision content is empty');
+        }
+        try {
+            payload = JSON.parse(content);
+        }
+        catch {
+            throw new Error('LLM decision content is not valid JSON');
+        }
+    }
+    const parsed = llmDecisionSchema.safeParse(payload);
+    if (!parsed.success) {
+        throw new Error(`Invalid LLM decision: ${z4.prettifyError(parsed.error)}`);
+    }
+    const decision = parsed.data;
+    const reason = toNonEmptyString(decision.reason) ?? undefined;
+    const categories = Array.isArray(decision.categories) ? decision.categories.filter(Boolean) : undefined;
+    if (!decision.matched) {
+        return {
+            matched: false,
+            reason,
+            categories,
+        };
+    }
+    if (decision.action === 'block') {
+        return {
+            matched: true,
+            action: 'block',
+            reason,
+            categories,
+        };
+    }
+    return {
+        matched: true,
+        action: 'rewrite',
+        replacementText: toNonEmptyString(decision.replacementText) ?? rewriteFallbackText,
+        reason,
+        categories,
+    };
+}
+function resolveRuntimeLlmConfig(config) {
+    if (!isRecord(config)) {
+        throw new Error(LLM_MODE_VALIDATION_ERROR);
+    }
+    const model = isRecord(config.model) ? config.model : null;
+    const scope = toNonEmptyString(config.scope);
+    const systemPrompt = toNonEmptyString(config.systemPrompt);
+    const onLlmError = toNonEmptyString(config.onLlmError);
+    if (!model || !scope || !systemPrompt || !onLlmError) {
+        throw new Error(LLM_MODE_VALIDATION_ERROR);
+    }
+    const outputMethodRaw = toNonEmptyString(config.outputMethod);
+    const outputMethod = ['functionCalling', 'jsonMode', 'jsonSchema'].includes(outputMethodRaw ?? '')
+        ? outputMethodRaw
+        : 'jsonSchema';
+    const errorRewriteText = toNonEmptyString(config.errorRewriteText) ?? undefined;
+    if (onLlmError === 'rewrite' && !errorRewriteText) {
+        throw new Error('请完善 LLM 过滤配置：当失败处理为改写时，必须填写失败改写文本。');
+    }
+    const timeoutMs = typeof config.timeoutMs === 'number' && Number.isFinite(config.timeoutMs) && config.timeoutMs > 0
+        ? Math.min(Math.floor(config.timeoutMs), 120000)
+        : undefined;
+    return {
+        model,
+        scope,
+        systemPrompt,
+        outputMethod,
+        onLlmError,
+        errorRewriteText,
+        blockMessage: toNonEmptyString(config.blockMessage) ?? undefined,
+        rewriteFallbackText: toNonEmptyString(config.rewriteFallbackText) ?? DEFAULT_REWRITE_TEXT,
+        timeoutMs,
+    };
+}
+function withTimeout(promise, timeoutMs) {
+    if (!timeoutMs || timeoutMs <= 0) {
+        return promise;
+    }
+    return new Promise((resolve, reject) => {
+        const timer = setTimeout(() => {
+            reject(new Error(`LLM filter timeout after ${timeoutMs}ms`));
+        }, timeoutMs);
+        promise
+            .then((result) => {
+            clearTimeout(timer);
+            resolve(result);
+        })
+            .catch((error) => {
+            clearTimeout(timer);
+            reject(error);
+        });
+    });
+}
 let SensitiveFilterMiddleware = class SensitiveFilterMiddleware {
     constructor() {
         this.meta = {
@@ -217,8 +326,8 @@ let SensitiveFilterMiddleware = class SensitiveFilterMiddleware {
                 zh_Hans: '敏感内容过滤中间件',
             },
             description: {
-                en_US: 'Filter sensitive content before input and after output using business rules and optional local general lexicon.',
-                zh_Hans: '基于业务规则进行输入/输出过滤，并可选启用本地开源通用词库（中英双语）作为兜底。',
+                en_US: 'Filter sensitive content before input and after output using rule mode or LLM prompt mode (mutually exclusive).',
+                zh_Hans: '支持规则模式或 LLM 提示词模式（互斥）进行输入/输出敏感内容过滤。',
             },
             icon: {
                 type: 'svg',
@@ -227,15 +336,37 @@ let SensitiveFilterMiddleware = class SensitiveFilterMiddleware {
             configSchema: {
                 type: 'object',
                 properties: {
+                    mode: {
+                        type: 'string',
+                        title: {
+                            en_US: 'Filter Mode',
+                            zh_Hans: '过滤模式',
+                        },
+                        description: {
+                            en_US: 'Choose exactly one mode: Rule or LLM.',
+                            zh_Hans: '二选一：规则模式或 LLM 模式。',
+                        },
+                        enum: ['rule', 'llm'],
+                        default: 'rule',
+                        'x-ui': {
+                            enumLabels: {
+                                rule: { en_US: 'Rule Mode', zh_Hans: '规则模式' },
+                                llm: { en_US: 'LLM Mode', zh_Hans: 'LLM 模式' },
+                            },
+                        },
+                    },
                     rules: {
                         type: 'array',
+                        'x-ui': {
+                            span: 2,
+                        },
                         title: {
                             en_US: 'Business Rules',
                             zh_Hans: '业务规则',
                         },
                         description: {
-                            en_US: 'Add at least one complete business rule. If you only need baseline protection, enable General Pack below.',
-                            zh_Hans: '建议至少添加一条完整业务规则；若仅需基础兜底，可直接启用下方“通用规则包”。',
+                            en_US: 'Used in rule mode. Draft rows are allowed during editing. Runtime requires valid fields.',
+                            zh_Hans: '规则模式使用。编辑阶段允许草稿行，运行阶段要求有效规则字段。',
                         },
                         items: {
                             type: 'object',
@@ -243,6 +374,10 @@ let SensitiveFilterMiddleware = class SensitiveFilterMiddleware {
                                 id: {
                                     type: 'string',
                                     title: { en_US: 'Rule ID', zh_Hans: '规则标识' },
+                                    description: {
+                                        en_US: 'Optional. Auto-generated when empty.',
+                                        zh_Hans: '可选。留空时系统自动生成。',
+                                    },
                                 },
                                 pattern: {
                                     type: 'string',
@@ -298,10 +433,14 @@ let SensitiveFilterMiddleware = class SensitiveFilterMiddleware {
                                     title: { en_US: 'Replacement Text', zh_Hans: '替换文本（可选）' },
                                 },
                             },
+                            required: ['pattern', 'type', 'action', 'scope', 'severity'],
                         },
                     },
                     generalPack: {
                         type: 'object',
+                        'x-ui': {
+                            span: 2,
+                        },
                         title: {
                             en_US: 'General Pack',
                             zh_Hans: '通用规则包（本地开源词库）',
@@ -319,6 +458,10 @@ let SensitiveFilterMiddleware = class SensitiveFilterMiddleware {
                             profile: {
                                 type: 'string',
                                 title: { en_US: 'Profile', zh_Hans: '策略档位' },
+                                description: {
+                                    en_US: 'Strict blocks on hit with broader lexicon; Balanced rewrites on hit with smaller lexicon.',
+                                    zh_Hans: '严格：词库范围更大且命中后拦截；平衡：词库范围较小且命中后改写。',
+                                },
                                 enum: ['strict', 'balanced'],
                                 'x-ui': {
                                     enumLabels: {
@@ -338,80 +481,251 @@ let SensitiveFilterMiddleware = class SensitiveFilterMiddleware {
                         type: 'boolean',
                         default: false,
                         title: { en_US: 'Case Sensitive', zh_Hans: '区分大小写' },
+                        'x-ui': {
+                            span: 2,
+                        },
                     },
                     normalize: {
                         type: 'boolean',
                         default: true,
                         title: { en_US: 'Normalize Text', zh_Hans: '文本标准化' },
+                        'x-ui': {
+                            span: 2,
+                        },
+                    },
+                    llm: {
+                        type: 'object',
+                        'x-ui': {
+                            span: 2,
+                        },
+                        title: {
+                            en_US: 'LLM Filter Config',
+                            zh_Hans: 'LLM 过滤配置',
+                        },
+                        description: {
+                            en_US: 'Used only when mode=llm.',
+                            zh_Hans: '仅在 mode=llm 时生效。',
+                        },
+                        properties: {
+                            model: {
+                                type: 'object',
+                                title: {
+                                    en_US: 'Filter Model',
+                                    zh_Hans: '过滤模型',
+                                },
+                                'x-ui': {
+                                    component: 'ai-model-select',
+                                    span: 2,
+                                    inputs: {
+                                        modelType: 'llm',
+                                        hiddenLabel: true,
+                                    },
+                                },
+                            },
+                            scope: {
+                                type: 'string',
+                                title: { en_US: 'Scope', zh_Hans: '生效范围' },
+                                enum: ['input', 'output', 'both'],
+                                'x-ui': {
+                                    enumLabels: {
+                                        input: { en_US: 'Input', zh_Hans: '仅输入' },
+                                        output: { en_US: 'Output', zh_Hans: '仅输出' },
+                                        both: { en_US: 'Both', zh_Hans: '输入和输出' },
+                                    },
+                                },
+                            },
+                            systemPrompt: {
+                                type: 'string',
+                                title: { en_US: 'System Prompt', zh_Hans: '判定提示词' },
+                                'x-ui': {
+                                    component: 'textarea',
+                                    span: 2,
+                                },
+                            },
+                            outputMethod: {
+                                type: 'string',
+                                title: { en_US: 'Output Method', zh_Hans: '结构化输出方式' },
+                                enum: ['functionCalling', 'jsonMode', 'jsonSchema'],
+                                default: 'jsonSchema',
+                                'x-ui': {
+                                    enumLabels: {
+                                        functionCalling: { en_US: 'Function Calling', zh_Hans: '函数调用' },
+                                        jsonMode: { en_US: 'JSON Mode', zh_Hans: 'JSON模式' },
+                                        jsonSchema: { en_US: 'JSON Schema', zh_Hans: 'JSON架构' },
+                                    },
+                                },
+                            },
+                            onLlmError: {
+                                type: 'string',
+                                title: { en_US: 'On LLM Error', zh_Hans: 'LLM失败处理' },
+                                enum: ['block', 'rewrite'],
+                                'x-ui': {
+                                    enumLabels: {
+                                        block: { en_US: 'Block', zh_Hans: '拦截' },
+                                        rewrite: { en_US: 'Rewrite', zh_Hans: '改写' },
+                                    },
+                                },
+                            },
+                            errorRewriteText: {
+                                type: 'string',
+                                title: { en_US: 'Error Rewrite Text', zh_Hans: '失败改写文本' },
+                            },
+                            blockMessage: {
+                                type: 'string',
+                                title: { en_US: 'Block Message', zh_Hans: '拦截提示文本' },
+                            },
+                            rewriteFallbackText: {
+                                type: 'string',
+                                title: { en_US: 'Rewrite Fallback Text', zh_Hans: '改写兜底文本' },
+                            },
+                            timeoutMs: {
+                                type: 'number',
+                                title: { en_US: 'Timeout (ms)', zh_Hans: '超时毫秒' },
+                            },
+                        },
+                        required: ['model', 'scope', 'systemPrompt', 'onLlmError'],
+                        allOf: [
+                            {
+                                if: {
+                                    properties: {
+                                        onLlmError: {
+                                            const: 'rewrite',
+                                        },
+                                    },
+                                },
+                                then: {
+                                    required: ['errorRewriteText'],
+                                },
+                            },
+                        ],
                     },
                 },
+                required: ['mode'],
+                allOf: [
+                    {
+                        if: {
+                            properties: {
+                                mode: {
+                                    const: 'llm',
+                                },
+                            },
+                        },
+                        then: {
+                            required: ['llm'],
+                        },
+                    },
+                ],
             },
         };
     }
-    createMiddleware(options, _context) {
+    async createMiddleware(options, context) {
         const parsed = sensitiveFilterConfigSchema.safeParse(options ?? {});
         if (!parsed.success) {
-            throw new Error(`Invalid sensitive filter middleware options: ${z4.prettifyError(parsed.error)}`);
+            throw new Error(CONFIG_PARSE_ERROR);
         }
-        const config = parsed.data;
+        if (parsed.data.mode === 'llm') {
+            return this.createLlmModeMiddleware(parsed.data, context);
+        }
+        return this.createRuleModeMiddleware(parsed.data);
+    }
+    createRuleModeMiddleware(config) {
         const caseSensitive = config.caseSensitive ?? false;
         const normalize = config.normalize ?? true;
         const customRules = normalizeRuleDrafts(config.rules ?? []);
         const generalRules = resolveGeneralPackRules(config.generalPack);
         const allRules = [...customRules, ...generalRules];
-        const compiledRules = allRules.map((rule, index) => {
-            const normalizedPattern = rule.type === 'keyword' ? normalizeForMatching(rule.pattern, normalize, caseSensitive) : rule.pattern;
-            if (rule.type === 'regex') {
-                try {
-                    return {
-                        ...rule,
-                        index,
-                        normalizedPattern,
-                        matchRegex: new RegExp(rule.pattern, caseSensitive ? '' : 'i'),
-                        rewriteRegex: new RegExp(rule.pattern, caseSensitive ? 'g' : 'gi'),
-                    };
-                }
-                catch (error) {
-                    const message = error instanceof Error ? error.message : String(error);
-                    throw new Error(`Invalid regex pattern in rule '${rule.id}': ${message}`);
-                }
+        const hasEffectiveRules = allRules.length > 0;
+        let compiledRulesCache = null;
+        const getCompiledRules = () => {
+            if (compiledRulesCache) {
+                return compiledRulesCache;
             }
-            return {
-                ...rule,
-                index,
-                normalizedPattern,
-            };
-        });
+            compiledRulesCache = allRules.map((rule, index) => {
+                const normalizedPattern = rule.type === 'keyword' ? normalizeForMatching(rule.pattern, normalize, caseSensitive) : rule.pattern;
+                if (rule.type === 'regex') {
+                    try {
+                        return {
+                            ...rule,
+                            index,
+                            normalizedPattern,
+                            matchRegex: new RegExp(rule.pattern, caseSensitive ? '' : 'i'),
+                            rewriteRegex: new RegExp(rule.pattern, caseSensitive ? 'g' : 'gi'),
+                        };
+                    }
+                    catch (error) {
+                        const message = error instanceof Error ? error.message : String(error);
+                        throw new Error(`请完善规则配置：规则「${rule.id}」的正则表达式不合法（${message}）。`);
+                    }
+                }
+                return {
+                    ...rule,
+                    index,
+                    normalizedPattern,
+                };
+            });
+            return compiledRulesCache;
+        };
         let inputBlockedMessage = null;
         let pendingInputRewrite = null;
+        let finalAction = 'pass';
+        let auditEntries = [];
+        const resetRunState = () => {
+            inputBlockedMessage = null;
+            pendingInputRewrite = null;
+            finalAction = 'pass';
+            auditEntries = [];
+        };
+        const pushAudit = (entry) => {
+            auditEntries.push({
+                ...entry,
+                timestamp: new Date().toISOString(),
+                mode: 'rule',
+            });
+        };
         return {
             name: SENSITIVE_FILTER_MIDDLEWARE_NAME,
             beforeAgent: async (state, runtime) => {
+                resetRunState();
+                if (!hasEffectiveRules) {
+                    throw new Error(BUSINESS_RULES_VALIDATION_ERROR);
+                }
+                const compiledRules = getCompiledRules();
                 const safeState = state ?? {};
                 const safeRuntime = runtime ?? {};
-                inputBlockedMessage = null;
-                pendingInputRewrite = null;
-                if (compiledRules.length === 0) {
-                    return undefined;
-                }
                 const inputText = extractInputText(safeState, safeRuntime);
                 const inputMatches = findMatches(inputText, 'input', compiledRules, normalize, caseSensitive);
                 const winner = pickWinningRule(inputMatches);
                 if (!winner) {
+                    pushAudit({
+                        phase: 'input',
+                        matched: false,
+                        source: 'rule',
+                        errorPolicyTriggered: false,
+                    });
                     return undefined;
                 }
+                pushAudit({
+                    phase: 'input',
+                    matched: true,
+                    source: 'rule',
+                    action: winner.action,
+                    reason: `rule:${winner.id}`,
+                    errorPolicyTriggered: false,
+                });
                 if (winner.action === 'block') {
+                    finalAction = 'block';
                     inputBlockedMessage = winner.replacementText?.trim() || DEFAULT_INPUT_BLOCK_MESSAGE;
                     return undefined;
                 }
-                const rewrittenInput = rewriteTextByRule(inputText, winner, caseSensitive);
-                pendingInputRewrite = rewrittenInput;
+                finalAction = 'rewrite';
+                pendingInputRewrite = rewriteTextByRule(inputText, winner, caseSensitive);
                 return undefined;
             },
             wrapModelCall: async (request, handler) => {
-                if (compiledRules.length === 0) {
-                    return handler(request);
+                if (!hasEffectiveRules) {
+                    throw new Error(BUSINESS_RULES_VALIDATION_ERROR);
                 }
+                const compiledRules = getCompiledRules();
                 if (inputBlockedMessage) {
                     return new AIMessage(inputBlockedMessage);
                 }
@@ -422,21 +736,292 @@ let SensitiveFilterMiddleware = class SensitiveFilterMiddleware {
                 const outputMatches = findMatches(outputText, 'output', compiledRules, normalize, caseSensitive);
                 const winner = pickWinningRule(outputMatches);
                 if (!winner) {
+                    pushAudit({
+                        phase: 'output',
+                        matched: false,
+                        source: 'rule',
+                        errorPolicyTriggered: false,
+                    });
                     return response;
                 }
+                pushAudit({
+                    phase: 'output',
+                    matched: true,
+                    source: 'rule',
+                    action: winner.action,
+                    reason: `rule:${winner.id}`,
+                    errorPolicyTriggered: false,
+                });
                 if (winner.action === 'block') {
+                    finalAction = 'block';
                     const blockedOutput = winner.replacementText?.trim() || DEFAULT_OUTPUT_BLOCK_MESSAGE;
                     return replaceModelResponseText(response, blockedOutput);
                 }
+                finalAction = 'rewrite';
                 const rewrittenOutput = rewriteTextByRule(outputText, winner, caseSensitive);
                 return replaceModelResponseText(response, rewrittenOutput);
             },
             afterAgent: async () => {
+                console.log('[SensitiveFilterMiddleware][audit]', JSON.stringify({
+                    mode: 'rule',
+                    finalAction,
+                    records: auditEntries,
+                }, null, 2));
+                return undefined;
+            },
+        };
+    }
+    createLlmModeMiddleware(config, context) {
+        const llmDraftConfig = config.llm;
+        let resolvedLlmConfig = null;
+        let modelPromise = null;
+        let structuredModelPromise = null;
+        const getLlmConfig = () => {
+            if (!resolvedLlmConfig) {
+                resolvedLlmConfig = resolveRuntimeLlmConfig(llmDraftConfig);
+            }
+            return resolvedLlmConfig;
+        };
+        const ensureModel = async () => {
+            const llmConfig = getLlmConfig();
+            if (!modelPromise) {
+                modelPromise = this.commandBus.execute(new CreateModelClientCommand(llmConfig.model, {
+                    usageCallback: (event) => {
+                        console.log('[SensitiveFilterMiddleware][LLM] Model Usage:', event);
+                    },
+                }));
+            }
+            return modelPromise;
+        };
+        const ensureStructuredModel = async () => {
+            const llmConfig = getLlmConfig();
+            if (!structuredModelPromise) {
+                structuredModelPromise = (async () => {
+                    const model = await ensureModel();
+                    return model.withStructuredOutput?.(llmDecisionSchema, {
+                        method: llmConfig.outputMethod,
+                    }) ?? null;
+                })();
+            }
+            return structuredModelPromise;
+        };
+        let inputBlockedMessage = null;
+        let pendingInputRewrite = null;
+        let finalAction = 'pass';
+        let auditEntries = [];
+        const resetRunState = () => {
+            inputBlockedMessage = null;
+            pendingInputRewrite = null;
+            finalAction = 'pass';
+            auditEntries = [];
+        };
+        const pushAudit = (entry) => {
+            auditEntries.push({
+                ...entry,
+                timestamp: new Date().toISOString(),
+                mode: 'llm',
+            });
+        };
+        const buildEvaluationMessages = (phase, text, llmConfig) => {
+            return [
+                { role: 'system', content: llmConfig.systemPrompt },
+                {
+                    role: 'user',
+                    content: `phase=${phase}\n` +
+                        '请严格基于给定文本进行敏感判定，并只返回约定结构。\n' +
+                        `text:\n${text}`,
+                },
+            ];
+        };
+        const invokeAndTrack = async (phase, text, runtime, llmConfig) => {
+            const invokeCore = async () => {
+                const messages = buildEvaluationMessages(phase, text, llmConfig);
+                const [model, structuredModel] = await Promise.all([ensureModel(), ensureStructuredModel()]);
+                if (structuredModel) {
+                    return withTimeout(structuredModel.invoke(messages), llmConfig.timeoutMs);
+                }
+                return withTimeout(model.invoke(messages), llmConfig.timeoutMs);
+            };
+            const parseCore = async () => {
+                const raw = await invokeCore();
+                return parseLlmDecision(raw, llmConfig.rewriteFallbackText);
+            };
+            const configurable = (runtime?.configurable ?? {});
+            const { thread_id, checkpoint_ns, checkpoint_id, subscriber, executionId } = configurable;
+            if (!thread_id || !executionId) {
+                return parseCore();
+            }
+            const tracked = await this.commandBus.execute(new WrapWorkflowNodeExecutionCommand(async () => {
+                const decision = await parseCore();
+                return {
+                    state: decision,
+                    output: decision,
+                };
+            }, {
+                execution: {
+                    category: 'workflow',
+                    type: 'middleware',
+                    inputs: {
+                        phase,
+                        text,
+                    },
+                    parentId: executionId,
+                    threadId: thread_id,
+                    checkpointNs: checkpoint_ns,
+                    checkpointId: checkpoint_id,
+                    agentKey: context.node.key,
+                    title: context.node.title,
+                },
+                subscriber,
+            }));
+            return tracked;
+        };
+        const resolveOnErrorDecision = (llmConfig, error) => {
+            const reason = `llm-error:${error instanceof Error ? error.message : String(error)}`;
+            if (llmConfig.onLlmError === 'block') {
+                return {
+                    matched: true,
+                    action: 'block',
+                    reason,
+                };
+            }
+            return {
+                matched: true,
+                action: 'rewrite',
+                replacementText: llmConfig.errorRewriteText ?? llmConfig.rewriteFallbackText,
+                reason,
+            };
+        };
+        const resolveBlockMessage = (llmConfig, phase) => {
+            return llmConfig.blockMessage ??
+                (phase === 'input' ? DEFAULT_INPUT_BLOCK_MESSAGE : DEFAULT_OUTPUT_BLOCK_MESSAGE);
+        };
+        return {
+            name: SENSITIVE_FILTER_MIDDLEWARE_NAME,
+            beforeAgent: async (state, runtime) => {
+                resetRunState();
+                const llmConfig = getLlmConfig();
+                if (!modeIncludesScope(llmConfig.scope, 'input')) {
+                    pushAudit({
+                        phase: 'input',
+                        matched: false,
+                        source: 'llm',
+                        reason: 'scope-skip',
+                        errorPolicyTriggered: false,
+                    });
+                    return undefined;
+                }
+                const inputText = extractInputText(state ?? {}, runtime ?? {});
+                if (!inputText) {
+                    pushAudit({
+                        phase: 'input',
+                        matched: false,
+                        source: 'llm',
+                        reason: 'empty-input',
+                        errorPolicyTriggered: false,
+                    });
+                    return undefined;
+                }
+                let decision;
+                let fromErrorPolicy = false;
+                try {
+                    decision = await invokeAndTrack('input', inputText, runtime, llmConfig);
+                }
+                catch (error) {
+                    decision = resolveOnErrorDecision(llmConfig, error);
+                    fromErrorPolicy = true;
+                }
+                pushAudit({
+                    phase: 'input',
+                    matched: decision.matched,
+                    source: fromErrorPolicy ? 'error-policy' : 'llm',
+                    action: decision.action,
+                    reason: decision.reason,
+                    errorPolicyTriggered: fromErrorPolicy,
+                });
+                if (!decision.matched || !decision.action) {
+                    return undefined;
+                }
+                if (decision.action === 'block') {
+                    finalAction = 'block';
+                    inputBlockedMessage = resolveBlockMessage(llmConfig, 'input');
+                    return undefined;
+                }
+                finalAction = 'rewrite';
+                pendingInputRewrite = toNonEmptyString(decision.replacementText) ?? llmConfig.rewriteFallbackText;
+                return undefined;
+            },
+            wrapModelCall: async (request, handler) => {
+                const llmConfig = getLlmConfig();
+                if (inputBlockedMessage) {
+                    return new AIMessage(inputBlockedMessage);
+                }
+                const modelRequest = pendingInputRewrite ? rewriteModelRequestInput(request, pendingInputRewrite) : request;
+                pendingInputRewrite = null;
+                const response = await handler(modelRequest);
+                if (!modeIncludesScope(llmConfig.scope, 'output')) {
+                    pushAudit({
+                        phase: 'output',
+                        matched: false,
+                        source: 'llm',
+                        reason: 'scope-skip',
+                        errorPolicyTriggered: false,
+                    });
+                    return response;
+                }
+                const outputText = extractModelResponseText(response);
+                if (!outputText) {
+                    pushAudit({
+                        phase: 'output',
+                        matched: false,
+                        source: 'llm',
+                        reason: 'empty-output',
+                        errorPolicyTriggered: false,
+                    });
+                    return response;
+                }
+                let decision;
+                let fromErrorPolicy = false;
+                try {
+                    decision = await invokeAndTrack('output', outputText, request?.runtime, llmConfig);
+                }
+                catch (error) {
+                    decision = resolveOnErrorDecision(llmConfig, error);
+                    fromErrorPolicy = true;
+                }
+                pushAudit({
+                    phase: 'output',
+                    matched: decision.matched,
+                    source: fromErrorPolicy ? 'error-policy' : 'llm',
+                    action: decision.action,
+                    reason: decision.reason,
+                    errorPolicyTriggered: fromErrorPolicy,
+                });
+                if (!decision.matched || !decision.action) {
+                    return response;
+                }
+                if (decision.action === 'block') {
+                    finalAction = 'block';
+                    return replaceModelResponseText(response, resolveBlockMessage(llmConfig, 'output'));
+                }
+                finalAction = 'rewrite';
+                return replaceModelResponseText(response, toNonEmptyString(decision.replacementText) ?? llmConfig.rewriteFallbackText);
+            },
+            afterAgent: async () => {
+                console.log('[SensitiveFilterMiddleware][audit]', JSON.stringify({
+                    mode: 'llm',
+                    finalAction,
+                    records: auditEntries,
+                }, null, 2));
                 return undefined;
             },
         };
     }
 };
+__decorate([
+    Inject(CommandBus),
+    __metadata("design:type", CommandBus)
+], SensitiveFilterMiddleware.prototype, "commandBus", void 0);
 SensitiveFilterMiddleware = __decorate([
     Injectable(),
     AgentMiddlewareStrategy(SENSITIVE_FILTER_MIDDLEWARE_NAME)