yuangs 2.18.0 → 2.20.0

package/dist/agent/AgentPipelineEnhanced.d.ts

@@ -0,0 +1,33 @@
+import { AgentInput, AgentMode } from './types';
+import { ModelRegistry } from '../policy/model/ModelRegistry';
+export declare class AgentPipeline {
+    private contextBuffer;
+    private modelRegistry;
+    private policy;
+    constructor(modelRegistry?: ModelRegistry);
+    run(input: AgentInput, mode: AgentMode): Promise<void>;
+    /**
+     * 执行带 TokenPolicy 的 pipeline
+     */
+    private runWithTokenPolicy;
+    /**
+     * 处理 Policy 结果
+     */
+    private handlePolicyResult;
+    /**
+     * 应用用户决策
+     */
+    private applyDecision;
+    /**
+     * 执行 LLM Pipeline（原有的流程）
+     */
+    private executeLLMPipeline;
+    /**
+     * 从输入中提取上下文 tokens (@file, #dir)
+     */
+    private extractContextTokens;
+    /**
+     * 确定 mode
+     */
+    private determineMode;
+}

package/dist/agent/AgentPipelineEnhanced.js

@@ -0,0 +1,229 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentPipeline = void 0;
+const contextBuffer_1 = require("../commands/contextBuffer");
+const intent_1 = require("./intent");
+const context_1 = require("./context");
+const prompt_1 = require("./prompt");
+const llm_1 = require("./llm");
+const interpret_1 = require("./interpret");
+const planExecutor_1 = require("./planExecutor");
+const record_1 = require("./record");
+const skills_1 = require("./skills");
+const crypto_1 = require("crypto");
+const renderer_1 = require("../utils/renderer");
+const ora_1 = __importDefault(require("ora"));
+const chalk_1 = __importDefault(require("chalk"));
+const DefaultTokenPolicy_1 = require("../policy/token/DefaultTokenPolicy");
+const ModelRegistry_1 = require("../policy/model/ModelRegistry");
+const syntaxHandler_1 = require("../policy/syntaxHandler");
+const PolicyPresenter_1 = require("../ui/PolicyPresenter");
+const sampler_1 = require("../policy/sampler");
+const MAX_PIPELINE_ITERATIONS = 3;
+class AgentPipeline {
+    contextBuffer = new contextBuffer_1.ContextBuffer();
+    modelRegistry;
+    policy;
+    constructor(modelRegistry) {
+        this.modelRegistry = modelRegistry || new ModelRegistry_1.ModelRegistry([]);
+        this.policy = new DefaultTokenPolicy_1.DefaultTokenPolicy();
+    }
+    async run(input, mode) {
+        const id = (0, crypto_1.randomUUID)();
+        try {
+            await this.runWithTokenPolicy(input, mode, id);
+        }
+        catch (error) {
+            if (error.name === 'MaxIterationsExceeded') {
+                console.log(chalk_1.default.yellow('\n⚠️  已达到最大迭代次数，操作终止'));
+            }
+            else {
+                console.log(chalk_1.default.red(`\n❌ Pipeline 错误: ${error.message}`));
+            }
+        }
+        PolicyPresenter_1.PolicyPresenter.clearSuppressCache();
+    }
+    /**
+     * 执行带 TokenPolicy 的 pipeline
+     */
+    async runWithTokenPolicy(input, mode, executionId) {
+        // 1. 意图解析 (Syntax Phase)
+        const tokens = this.extractContextTokens(input.rawInput);
+        let pendingItems = syntaxHandler_1.SyntaxHandler.parse(tokens);
+        // 2. 治理审计循环 (Governance Loop)
+        // 最多重试 3 次（包括初始评估）以防止无限循环
+        const MAX_ITERATIONS = 3;
+        let passed = false;
+        let iterations = 0;
+        let currentModel = this.modelRegistry.get(input.options?.model || 'gemini-2.5-flash-lite') || this.modelRegistry.getDefault();
+        while (!passed && iterations < MAX_ITERATIONS) {
+            iterations++;
+            const result = await this.policy.evaluate({
+                model: currentModel,
+                contextItems: pendingItems,
+                mode: this.determineMode(mode),
+                userIntent: input.rawInput
+            });
+            passed = await this.handlePolicyResult(result, pendingItems, currentModel, iterations);
+            if (passed)
+                break;
+        }
+        if (!passed) {
+            throw new Error('MaxIterationsExceeded');
+        }
+        // 3. 授权执行 (Execution Phase)
+        const resolved = await Promise.all(pendingItems.map(item => item.resolve()));
+        resolved.forEach(r => {
+            this.contextBuffer.add({
+                type: 'file',
+                path: pendingItems.find((p) => p.id.includes(r.content.substring(0, 20)))?.id || 'unknown',
+                content: r.content
+            }, true // bypassTokenLimit = true (已通过 policy 审计）
+            );
+        });
+        // 4. 正常 LLM Pipeline
+        await this.executeLLMPipeline(input, mode, currentModel, executionId);
+    }
+    /**
+     * 处理 Policy 结果
+     */
+    async handlePolicyResult(result, pendingItems, currentModel, iteration) {
+        if (result.status === 'ok') {
+            return true; // passed
+        }
+        if (result.status === 'block') {
+            await PolicyPresenter_1.PolicyPresenter.presentBlock(result);
+            return false;
+        }
+        if (result.status === 'warn') {
+            const decision = await PolicyPresenter_1.PolicyPresenter.presentWarning(result, `${currentModel.name}:${pendingItems.map(p => p.id).join(',')}`);
+            return this.applyDecision(decision, pendingItems, currentModel);
+        }
+        return false;
+    }
+    /**
+     * 应用用户决策
+     */
+    async applyDecision(decision, pendingItems, currentModel) {
+        switch (decision.type) {
+            case 'continue':
+                return true;
+            case 'abort':
+                return false;
+            case 'switch_model':
+                if (decision.targetModel) {
+                    const newModel = this.modelRegistry.get(decision.targetModel);
+                    if (newModel) {
+                        console.log(chalk_1.default.green(`\n🔄 切换至模型: ${decision.targetModel}`));
+                        currentModel = newModel;
+                        return false; // 需要重新评估
+                    }
+                }
+                console.log(chalk_1.default.yellow(`⚠️  模型 ${decision.targetModel} 未找到`));
+                return false;
+            case 'sample':
+                if (decision.strategy === 'head_tail') {
+                    console.log(chalk_1.default.cyan('\n✂ 应用 head_tail 采样...'));
+                    pendingItems = await Promise.all(pendingItems.map(item => sampler_1.ContextSampler.applySampling(item, 'head_tail')));
+                    return false; // 需要重新评估
+                }
+                return true;
+            default:
+                return true;
+        }
+    }
+    /**
+     * 执行 LLM Pipeline（原有的流程）
+     */
+    async executeLLMPipeline(input, mode, model, executionId) {
+        const intent = (0, intent_1.inferIntent)(input, mode);
+        const context = (0, context_1.buildContext)(input, this.contextBuffer);
+        const prompt = (0, prompt_1.buildPrompt)(intent, context, mode, input.rawInput);
+        let renderer;
+        let spinner;
+        if (mode === 'chat') {
+            spinner = (0, ora_1.default)(chalk_1.default.cyan('Thinking...')).start();
+            renderer = new renderer_1.StreamMarkdownRenderer(chalk_1.default.bold.blue('🤖 AI: '), spinner);
+        }
+        const result = await (0, llm_1.runLLM)({
+            prompt,
+            model: model.name,
+            stream: mode === 'chat',
+            onChunk: mode === 'chat' && renderer
+                ? (s) => renderer.onChunk(s)
+                : undefined,
+        });
+        if (mode === 'chat' && renderer) {
+            renderer.finish();
+        }
+        const isStreaming = mode === 'chat';
+        const plan = (0, interpret_1.interpretResultToPlan)(result, intent, mode, isStreaming);
+        result.plan = plan;
+        (0, record_1.saveRecord)({
+            id: executionId,
+            timestamp: Date.now(),
+            mode,
+            input,
+            prompt,
+            model: model.name,
+            llmResult: result,
+            action: plan.tasks[0]?.type === 'shell' ? {
+                type: 'execute',
+                command: plan.tasks[0].payload.command,
+                risk: plan.tasks[0].payload.risk
+            } : { type: 'print', content: result.rawText },
+        });
+        const summary = await (0, planExecutor_1.executePlan)(plan, input.options);
+        (0, skills_1.learnSkillFromRecord)({
+            id: executionId,
+            timestamp: Date.now(),
+            mode,
+            input,
+            prompt,
+            model: model.name,
+            llmResult: result,
+            action: plan.tasks[0]?.type === 'shell' ? {
+                type: 'execute',
+                command: plan.tasks[0].payload.command,
+                risk: plan.tasks[0].payload.risk
+            } : { type: 'print', content: result.rawText },
+        }, summary.success);
+        if (input.options?.verbose) {
+            console.log(`\n${'-'.repeat(50)}`);
+            console.log(`Execution ID: ${executionId}`);
+            console.log(`Model: ${model.name}`);
+            console.log(`Latency: ${result.latencyMs}ms`);
+            if (result.tokens) {
+                console.log(`Tokens: ${result.tokens.total}`);
+            }
+            console.log(`${'-'.repeat(50)}\n`);
+        }
+    }
+    /**
+     * 从输入中提取上下文 tokens (@file, #dir)
+     */
+    extractContextTokens(rawInput) {
+        return rawInput
+            .split(' ')
+            .filter(token => token.startsWith('@') || token.startsWith('#'));
+    }
+    /**
+     * 确定 mode
+     */
+    determineMode(mode) {
+        // 将 Agent mode 映射到 Policy mode
+        // chat → agent, command → command, command+exec → command
+        if (mode === 'chat') {
+            return 'agent';
+        }
+        if (mode === 'command+exec') {
+            return 'command';
+        }
+        return mode; // command 默认为 command
+    }
+}
+exports.AgentPipeline = AgentPipeline;
+//# sourceMappingURL=AgentPipelineEnhanced.js.map

package/dist/agent/AgentPipelineEnhanced.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AgentPipelineEnhanced.js","sourceRoot":"","sources":["../../src/agent/AgentPipelineEnhanced.ts"],"names":[],"mappings":";;;;;;AACA,6DAA0D;AAC1D,qCAAuC;AACvC,uCAAyC;AACzC,qCAAuC;AAEvC,+BAA+B;AAC/B,2CAAoD;AACpD,iDAA6C;AAC7C,qCAAsC;AACtC,qCAAgD;AAChD,mCAAoC;AACpC,gDAA2D;AAC3D,8CAA+B;AAC/B,kDAA0B;AAE1B,2EAAwE;AACxE,iEAA8D;AAC9D,2DAAwD;AACxD,2DAAwD;AAExD,+CAAmD;AAEnD,MAAM,uBAAuB,GAAG,CAAC,CAAC;AAElC,MAAa,aAAa;IACd,aAAa,GAAkB,IAAI,6BAAa,EAAE,CAAC;IACnD,aAAa,CAAgB;IAC7B,MAAM,CAAqB;IAEnC,YAAY,aAA6B;QACrC,IAAI,CAAC,aAAa,GAAG,aAAa,IAAI,IAAI,6BAAa,CAAC,EAAE,CAAC,CAAC;QAC5D,IAAI,CAAC,MAAM,GAAG,IAAI,uCAAkB,EAAE,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,KAAiB,EAAE,IAAe;QACxC,MAAM,EAAE,GAAG,IAAA,mBAAU,GAAE,CAAC;QAExB,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;QACnD,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YAClB,IAAI,KAAK,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,CAAC;YACtD,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,GAAG,CAAC,oBAAoB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAChE,CAAC;QACL,CAAC;QAED,iCAAe,CAAC,kBAAkB,EAAE,CAAC;IACzC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAC5B,KAAiB,EACjB,IAAe,EACf,WAAmB;QAEnB,yBAAyB;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,YAAY,GAAG,6BAAa,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAE/C,8BAA8B;QAC9B,0BAA0B;QAC1B,MAAM,cAAc,GAAG,CAAC,CAAC;QACzB,IAAI,MAAM,GAAG,KAAK,CAAC;QACnB,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CACrC,KAAK,CAAC,OAAO,EAAE,KAAK,IAAI,uBAAuB,CAClD,IAAI,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;QAErC,OAAO,CAAC,MAAM,IAAI,UAAU,GAAG,cAAc,EAAE,CAAC;YAC5C,UAAU,EAAE,CAAC;YAEb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;gBACtC,KAAK,EAAE,YAAY;gBACnB,YAAY,EAAE,YAAY;gBAC1B,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;gBAC9B,UAAU,EAAE,KAAK,CAAC,QAAQ;aAC7B,CAAC,CAAC;YAEH,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAClC,MAAM,EACN,YAAY,EACZ,YAAY,EACZ,UAAU,CACb,CAAC;YAEF,IAAI,MAAM;gBAAE,MAAM;QACtB,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC7C,CAAC;QAED,4BAA4B;QAC5B,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAE7E,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACjB,IAAI,CAAC,aAAa,CAAC,GAAG,CAClB;gBACI,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,SAAS;gBAC/F,OAAO,EAAE,CAAC,CAAC,OAAO;aACrB,EACD,IAAI,CAAC,0CAA0C;aAClD,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,qBAAqB;QACrB,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,CAAC,CAAC;IAC1E,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAC5B,MAAW,EACX,YAAmB,EACnB,YAAiB,EACjB,SAAiB;QAEjB,IAAI,MAAM,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,CAAC,SAAS;QAC1B,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;YAC5B,MAAM,iCAAe,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAC3C,OAAO,KAAK,CAAC;QACjB,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,MAAM,iCAAe,CAAC,cAAc,CACjD,MAAM,EACN,GAAG,YAAY,CAAC,IAAI,IAAI,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAClE,CAAC;YAEF,OAAO,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;QACpE,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa,CACvB,QAAsB,EACtB,YAAmB,EACnB,YAAiB;QAEjB,QAAQ,QAAQ,CAAC,IAAI,EAAE,CAAC;YACpB,KAAK,UAAU;gBACX,OAAO,IAAI,CAAC;YAEhB,KAAK,OAAO;gBACR,OAAO,KAAK,CAAC;YAEjB,KAAK,cAAc;gBACf,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;oBACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;oBAC9D,IAAI,QAAQ,EAAE,CAAC;wBACX,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,eAAe,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;wBAChE,YAAY,GAAG,QAAQ,CAAC;wBACxB,OAAO,KAAK,CAAC,CAAC,SAAS;oBAC3B,CAAC;gBACL,CAAC;gBACD,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,MAAM,CAAC,UAAU,QAAQ,CAAC,WAAW,MAAM,CAAC,CAAC,CAAC;gBAChE,OAAO,KAAK,CAAC;YAEjB,KAAK,QAAQ;gBACT,IAAI,QAAQ,CAAC,QAAQ,KAAK,WAAW,EAAE,CAAC;oBACpC,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC,CAAC;oBAClD,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CAC5B,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CACpB,wBAAc,CAAC,aAAa,CAAC,IAAI,EAAE,WAAW,CAAC,CAClD,CACJ,CAAC;oBACF,OAAO,KAAK,CAAC,CAAC,SAAS;gBAC3B,CAAC;gBACD,OAAO,IAAI,CAAC;YAEhB;gBACI,OAAO,IAAI,CAAC;QACpB,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAC5B,KAAiB,EACjB,IAAe,EACf,KAAU,EACV,WAAmB;QAEnB,MAAM,MAAM,GAAG,IAAA,oBAAW,EAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACxC,MAAM,OAAO,GAAG,IAAA,sBAAY,EAAC,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,IAAA,oBAAW,EAAC,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QAElE,IAAI,QAA4C,CAAC;QACjD,IAAI,OAAwB,CAAC;QAE7B,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;YAClB,OAAO,GAAG,IAAA,aAAG,EAAC,eAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;YACjD,QAAQ,GAAG,IAAI,iCAAsB,CAAC,eAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,CAAC;QAC/E,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAA,YAAM,EAAC;YACxB,MAAM;YACN,KAAK,EAAE,KAAK,CAAC,IAAI;YACjB,MAAM,EAAE,IAAI,KAAK,MAAM;YACvB,OAAO,EAAE,IAAI,KAAK,MAAM,IAAI,QAAQ;gBAChC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAS,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC7B,CAAC,CAAC,SAAS;SAClB,CAAC,CAAC;QAEH,IAAI,IAAI,KAAK,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,QAAQ,CAAC,MAAM,EAAE,CAAC;QACtB,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,KAAK,MAAM,CAAC;QACpC,MAAM,IAAI,GAAG,IAAA,iCAAqB,EAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QACtE,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;QAEnB,IAAA,mBAAU,EAAC;YACP,EAAE,EAAE,WAAW;YACf,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,IAAI;YACJ,KAAK;YACL,MAAM;YACN,KAAK,EAAE,KAAK,CAAC,IAAI;YACjB,SAAS,EAAE,MAAM;YACjB,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC;gBACtC,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO;gBACtC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI;aACnC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE;SACjD,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,IAAA,0BAAW,EAAC,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAEvD,IAAA,6BAAoB,EAAC;YACjB,EAAE,EAAE,WAAW;YACf,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,IAAI;YACJ,KAAK;YACL,MAAM;YACN,KAAK,EAAE,KAAK,CAAC,IAAI;YACjB,SAAS,EAAE,MAAM;YACjB,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC;gBACtC,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO;gBACtC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI;aACnC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE;SACjD,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAEpB,IAAI,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,iBAAiB,WAAW,EAAE,CAAC,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,UAAU,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YACpC,OAAO,CAAC,GAAG,CAAC,YAAY,MAAM,CAAC,SAAS,IAAI,CAAC,CAAC;YAC9C,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAChB,OAAO,CAAC,GAAG,CAAC,WAAW,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;YAClD,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,QAAgB;QACzC,OAAO,QAAQ;aACV,KAAK,CAAC,GAAG,CAAC;aACV,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;IACzE,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,IAAe;QACjC,+BAA+B;QAC/B,0DAA0D;QAC1D,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;YAClB,OAAO,OAAO,CAAC;QACnB,CAAC;QACD,IAAI,IAAI,KAAK,cAAc,EAAE,CAAC;YAC1B,OAAO,SAAS,CAAC;QACrB,CAAC;QACD,OAAO,IAAI,CAAC,CAAC,sBAAsB;IACvC,CAAC;CACJ;AA5QD,sCA4QC"}

package/dist/policy/model/ModelRegistry.d.ts

@@ -0,0 +1,38 @@
+import { ModelCapabilities } from '../../core/modelMatcher';
+import { ModelSpec } from '../token/types';
+/**
+ * ModelRegistry - 模型规格注册表
+ *
+ * 将现有的 ModelCapabilities 扩展为 ModelSpec，
+ * 提供统一的模型信息查询接口。
+ */
+export declare class ModelRegistry {
+    private models;
+    constructor(baseCapabilities: ModelCapabilities[]);
+    /**
+     * 注册模型规格
+     */
+    private register;
+    /**
+     * 根据名称获取模型规格
+     */
+    get(name: string): ModelSpec | undefined;
+    /**
+     * 获取默认模型
+     */
+    getDefault(): ModelSpec;
+    /**
+     * 查找所有支持长文本的模型
+     * 按上下文窗口大小降序排列
+     */
+    findLongContextCapable(): ModelSpec[];
+    /**
+     * 查找最佳长文本模型
+     * 返回上下文窗口最大的模型
+     */
+    findBestLongContextModel(): ModelSpec | undefined;
+    /**
+     * 列出所有已注册的模型
+     */
+    listModels(): ModelSpec[];
+}

package/dist/policy/model/ModelRegistry.js

@@ -0,0 +1,69 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModelRegistry = void 0;
+const capabilities_1 = require("../../core/capabilities");
+/**
+ * ModelRegistry - 模型规格注册表
+ *
+ * 将现有的 ModelCapabilities 扩展为 ModelSpec，
+ * 提供统一的模型信息查询接口。
+ */
+class ModelRegistry {
+    models = new Map();
+    constructor(baseCapabilities) {
+        baseCapabilities.forEach(cap => this.register(cap));
+    }
+    /**
+     * 注册模型规格
+     */
+    register(cap) {
+        const spec = {
+            name: cap.name,
+            contextWindow: cap.contextWindow ?? 32768,
+            costTier: cap.costProfile ?? 'medium',
+            longContextCapable: cap.atomicCapabilities.includes(capabilities_1.AtomicCapability.LONG_CONTEXT)
+        };
+        this.models.set(cap.name, spec);
+    }
+    /**
+     * 根据名称获取模型规格
+     */
+    get(name) {
+        return this.models.get(name);
+    }
+    /**
+     * 获取默认模型
+     */
+    getDefault() {
+        const defaultModel = this.get('gemini-2.5-flash-lite');
+        if (!defaultModel) {
+            throw new Error('Default model not found in registry');
+        }
+        return defaultModel;
+    }
+    /**
+     * 查找所有支持长文本的模型
+     * 按上下文窗口大小降序排列
+     */
+    findLongContextCapable() {
+        return Array.from(this.models.values())
+            .filter(m => m.longContextCapable)
+            .sort((a, b) => b.contextWindow - a.contextWindow);
+    }
+    /**
+     * 查找最佳长文本模型
+     * 返回上下文窗口最大的模型
+     */
+    findBestLongContextModel() {
+        const longContextModels = this.findLongContextCapable();
+        return longContextModels.length > 0 ? longContextModels[0] : undefined;
+    }
+    /**
+     * 列出所有已注册的模型
+     */
+    listModels() {
+        return Array.from(this.models.values());
+    }
+}
+exports.ModelRegistry = ModelRegistry;
+//# sourceMappingURL=ModelRegistry.js.map

package/dist/policy/model/ModelRegistry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ModelRegistry.js","sourceRoot":"","sources":["../../../src/policy/model/ModelRegistry.ts"],"names":[],"mappings":";;;AACA,0DAA2D;AAG3D;;;;;GAKG;AACH,MAAa,aAAa;IACd,MAAM,GAA2B,IAAI,GAAG,EAAE,CAAC;IAEnD,YAAY,gBAAqC;QAC7C,gBAAgB,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IACxD,CAAC;IAED;;OAEG;IACK,QAAQ,CAAC,GAAsB;QACnC,MAAM,IAAI,GAAc;YACpB,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,aAAa,EAAE,GAAG,CAAC,aAAa,IAAI,KAAK;YACzC,QAAQ,EAAE,GAAG,CAAC,WAAW,IAAI,QAAQ;YACrC,kBAAkB,EAAE,GAAG,CAAC,kBAAkB,CAAC,QAAQ,CAC/C,+BAAgB,CAAC,YAAY,CAChC;SACJ,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,IAAY;QACZ,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,UAAU;QACN,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;QACvD,IAAI,CAAC,YAAY,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QAC3D,CAAC;QACD,OAAO,YAAY,CAAC;IACxB,CAAC;IAED;;;OAGG;IACH,sBAAsB;QAClB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;aAClC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,kBAAkB,CAAC;aACjC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,aAAa,CAAC,CAAC;IAC3D,CAAC;IAED;;;OAGG;IACH,wBAAwB;QACpB,MAAM,iBAAiB,GAAG,IAAI,CAAC,sBAAsB,EAAE,CAAC;QACxD,OAAO,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3E,CAAC;IAED;;OAEG;IACH,UAAU;QACN,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAC5C,CAAC;CACJ;AAjED,sCAiEC"}

package/dist/policy/sampler.d.ts

@@ -0,0 +1,22 @@
+import { PendingContextItem, SamplingStrategy } from '../policy/token/types';
+/**
+ * ContextSampler - 上下文采样器
+ *
+ * 职责：
+ * - 实现 head_tail 采样策略
+ * - 保留文件头部和尾部，丢弃中间部分
+ */
+export declare class ContextSampler {
+    /**
+     * 对 PendingContextItem 应用采样策略
+     */
+    static applySampling(item: PendingContextItem, strategy: SamplingStrategy): Promise<PendingContextItem>;
+    /**
+     * Head-Tail 采样：保留头部和尾部各 30 行
+     */
+    private static applyHeadTail;
+    /**
+     * Random 采样：随机保留 50% 的行
+     */
+    private static applyRandom;
+}

package/dist/policy/sampler.js

@@ -0,0 +1,70 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContextSampler = void 0;
+/**
+ * ContextSampler - 上下文采样器
+ *
+ * 职责：
+ * - 实现 head_tail 采样策略
+ * - 保留文件头部和尾部，丢弃中间部分
+ */
+class ContextSampler {
+    /**
+     * 对 PendingContextItem 应用采样策略
+     */
+    static async applySampling(item, strategy) {
+        if (strategy === 'none' || item.samplingStrategy === 'none') {
+            return item;
+        }
+        if (strategy === 'head_tail' || item.samplingStrategy === 'head_tail') {
+            return this.applyHeadTail(item);
+        }
+        if (strategy === 'random') {
+            return this.applyRandom(item);
+        }
+        return item;
+    }
+    /**
+     * Head-Tail 采样：保留头部和尾部各 30 行
+     */
+    static async applyHeadTail(item) {
+        const resolved = await item.resolve();
+        const lines = resolved.content.split('\n');
+        const keepLines = 60;
+        const headLines = Math.min(30, Math.floor(lines.length / 2));
+        const tailLines = Math.min(30, Math.floor(lines.length / 2));
+        const sampledLines = [
+            ...lines.slice(0, headLines),
+            // ...lines.slice(headLines, lines.length - tailLines), // 跳过中间部分
+            ...lines.slice(-tailLines)
+        ];
+        return {
+            ...item,
+            id: `${item.id} (sampled)`,
+            resolve: async () => ({
+                content: sampledLines.join('\n'),
+                byteSize: Buffer.byteLength(sampledLines.join('\n'), 'utf-8'),
+                lineCount: sampledLines.length
+            })
+        };
+    }
+    /**
+     * Random 采样：随机保留 50% 的行
+     */
+    static async applyRandom(item) {
+        const resolved = await item.resolve();
+        const lines = resolved.content.split('\n');
+        const sampledLines = lines.filter((_, index) => Math.random() > 0.5);
+        return {
+            ...item,
+            id: `${item.id} (random)`,
+            resolve: async () => ({
+                content: sampledLines.join('\n'),
+                byteSize: Buffer.byteLength(sampledLines.join('\n'), 'utf-8'),
+                lineCount: sampledLines.length
+            })
+        };
+    }
+}
+exports.ContextSampler = ContextSampler;
+//# sourceMappingURL=sampler.js.map

package/dist/policy/sampler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sampler.js","sourceRoot":"","sources":["../../src/policy/sampler.ts"],"names":[],"mappings":";;;AAEA;;;;;;GAMG;AACH,MAAa,cAAc;IACvB;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,aAAa,CACtB,IAAwB,EACxB,QAA0B;QAE1B,IAAI,QAAQ,KAAK,MAAM,IAAI,IAAI,CAAC,gBAAgB,KAAK,MAAM,EAAE,CAAC;YAC1D,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,IAAI,QAAQ,KAAK,WAAW,IAAI,IAAI,CAAC,gBAAgB,KAAK,WAAW,EAAE,CAAC;YACpE,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;QAED,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,KAAK,CAAC,aAAa,CAC9B,IAAwB;QAExB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAE3C,MAAM,SAAS,GAAG,EAAE,CAAC;QACrB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QAC7D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QAE7D,MAAM,YAAY,GAAG;YACjB,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC;YAC5B,iEAAiE;YACjE,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC;SAC7B,CAAC;QAEF,OAAO;YACH,GAAG,IAAI;YACP,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,YAAY;YAC1B,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBAClB,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;gBAChC,QAAQ,EAAE,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC;gBAC7D,SAAS,EAAE,YAAY,CAAC,MAAM;aACjC,CAAC;SACL,CAAC;IACN,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAwB;QAExB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAE3C,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC;QAErE,OAAO;YACH,GAAG,IAAI;YACP,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,WAAW;YACzB,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBAClB,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;gBAChC,QAAQ,EAAE,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC;gBAC7D,SAAS,EAAE,YAAY,CAAC,MAAM;aACjC,CAAC;SACL,CAAC;IACN,CAAC;CACJ;AA1ED,wCA0EC"}

package/dist/policy/syntaxHandler.d.ts

@@ -0,0 +1,32 @@
+import { PendingContextItem } from '../policy/token/types';
+/**
+ * SyntaxHandler - 语法解析器和延迟加载生成器
+ *
+ * 职责：
+ * - 解析 @file 和 #dir 语法
+ * - 返回 PendingContextItem[]（不读取内容）
+ * - 提供 estimate() 和 resolve() 方法
+ */
+export declare class SyntaxHandler {
+    static parse(tokens: string[]): PendingContextItem[];
+    /**
+     * 解析文件引用 @file:path:start-end
+     */
+    private static parseFileToken;
+    /**
+     * 解析目录引用 #dir
+     */
+    private static parseDirToken;
+    /**
+     * 解析路径和行号范围
+     */
+    private static parsePathAndRange;
+    /**
+     * 应用行号范围
+     */
+    private static applyRange;
+    /**
+     * 扫描目录（递归）
+     */
+    private static scanDir;
+}

package/dist/policy/syntaxHandler.js

@@ -0,0 +1,155 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SyntaxHandler = void 0;
+const promises_1 = __importDefault(require("fs/promises"));
+const path_1 = __importDefault(require("path"));
+/**
+ * SyntaxHandler - 语法解析器和延迟加载生成器
+ *
+ * 职责：
+ * - 解析 @file 和 #dir 语法
+ * - 返回 PendingContextItem[]（不读取内容）
+ * - 提供 estimate() 和 resolve() 方法
+ */
+class SyntaxHandler {
+    static parse(tokens) {
+        const items = [];
+        for (const token of tokens) {
+            if (token.startsWith('@')) {
+                const item = this.parseFileToken(token);
+                if (item)
+                    items.push(item);
+            }
+            else if (token.startsWith('#')) {
+                const item = this.parseDirToken(token);
+                if (item)
+                    items.push(item);
+            }
+        }
+        return items;
+    }
+    /**
+     * 解析文件引用 @file:path:start-end
+     */
+    static parseFileToken(token) {
+        const raw = token.slice(1);
+        const { filePath, range } = this.parsePathAndRange(raw);
+        if (!filePath)
+            return null;
+        const absPath = path_1.default.resolve(filePath);
+        return {
+            id: absPath,
+            type: 'file',
+            originalToken: token,
+            samplingStrategy: 'head_tail',
+            estimate: async () => {
+                try {
+                    const stat = await promises_1.default.stat(absPath);
+                    return { byteSize: stat.size };
+                }
+                catch (error) {
+                    throw error;
+                }
+            },
+            resolve: async () => {
+                let content = await promises_1.default.readFile(absPath, 'utf-8');
+                if (range) {
+                    content = this.applyRange(content, range);
+                }
+                return {
+                    content,
+                    byteSize: Buffer.byteLength(content, 'utf-8'),
+                    lineCount: content.split('\n').length
+                };
+            }
+        };
+    }
+    /**
+     * 解析目录引用 #dir
+     */
+    static parseDirToken(token) {
+        const dir = token.slice(1);
+        const absPath = path_1.default.resolve(dir);
+        return {
+            id: absPath,
+            type: 'dir',
+            originalToken: token,
+            samplingStrategy: 'head_tail',
+            estimate: async () => {
+                try {
+                    const files = await this.scanDir(absPath);
+                    let total = 0;
+                    for (const f of files) {
+                        const stat = await promises_1.default.stat(f);
+                        total += stat.size;
+                    }
+                    return { byteSize: total };
+                }
+                catch (error) {
+                    throw error;
+                }
+            },
+            resolve: async () => {
+                const files = await this.scanDir(absPath);
+                let content = '';
+                for (const f of files) {
+                    content += `\n// ===== ${f} =====\n`;
+                    content += await promises_1.default.readFile(f, 'utf-8');
+                }
+                return {
+                    content,
+                    byteSize: Buffer.byteLength(content, 'utf-8')
+                };
+            }
+        };
+    }
+    /**
+     * 解析路径和行号范围
+     */
+    static parsePathAndRange(raw) {
+        const match = raw.match(/^(.*?)(?::(\d+)-(\d+))?$/);
+        if (!match)
+            return { filePath: raw };
+        const [, filePath, start, end] = match;
+        if (!start || !end)
+            return { filePath };
+        return {
+            filePath,
+            range: { start: Number(start), end: Number(end) }
+        };
+    }
+    /**
+     * 应用行号范围
+     */
+    static applyRange(content, range) {
+        if (!range)
+            return content;
+        const lines = content.split('\n');
+        return lines.slice(range.start - 1, range.end).join('\n');
+    }
+    /**
+     * 扫描目录（递归）
+     */
+    static async scanDir(dirPath) {
+        const files = [];
+        async function scan(dirPath) {
+            const entries = await promises_1.default.readdir(dirPath, { withFileTypes: true });
+            for (const entry of entries) {
+                const fullPath = path_1.default.join(dirPath, entry.name);
+                if (entry.isDirectory()) {
+                    await scan(fullPath);
+                }
+                else if (entry.isFile()) {
+                    files.push(fullPath);
+                }
+            }
+        }
+        await scan(dirPath);
+        return files;
+    }
+}
+exports.SyntaxHandler = SyntaxHandler;
+//# sourceMappingURL=syntaxHandler.js.map