autosnippet 3.4.0 → 3.4.1

package/dist/lib/external/ai/providers/OpenAiProvider.js

@@ -17,7 +17,7 @@ export class OpenAiProvider extends AiProvider {
         this.model = config.model || 'gpt-5.4-mini';
         this.apiKey = config.apiKey || process.env.ASD_OPENAI_API_KEY || '';
         this.baseUrl = config.baseUrl || OPENAI_BASE;
-        this.embedModel = config.embedModel || 'text-embedding-3-small';
+        this.embedModel = config.embedModel || process.env.ASD_EMBED_MODEL || 'text-embedding-3-small';
         this.logger = Logger.getInstance();
     }
     /**

package/dist/lib/external/mcp/handlers/bootstrap/pipeline/orchestrator.d.ts

@@ -68,6 +68,7 @@ interface BootstrapFileEntry {
 /** Task manager minimal shape */
 interface TaskManagerLike {
     isSessionValid(sessionId: string): boolean;
+    getSessionAbortSignal?(): AbortSignal | null;
     emitProgress?(event: string, data: Record<string, unknown>): void;
     [key: string]: unknown;
 }

package/dist/lib/external/mcp/handlers/bootstrap/pipeline/orchestrator.js

@@ -113,6 +113,7 @@ export async function fillDimensionsV3(view, dimensions) {
         /* not available */
     }
     const sessionId = view.bootstrapSession?.id ?? '';
+    const sessionAbortSignal = taskManager?.getSessionAbortSignal?.() ?? null;
     const isIncremental = incrementalPlan?.canIncremental && incrementalPlan?.mode === 'incremental';
     const emitter = new BootstrapEventEmitter(ctx.container);
     logger.info(`[Insight-v3] ═══ fillDimensionsV3 entered — ${isIncremental ? 'INCREMENTAL' : 'FULL'} pipeline`);
@@ -623,7 +624,7 @@ export async function fillDimensionsV3(view, dimensions) {
             // 外层超时 = 安全网 (各阶段已有独立超时: Analyst 300s + Producer 180s + 硬缓冲 60s)
             const outerTimeoutMs = 3_600_000; // 1 小时——维度分析本身耗时长
             const runResult = await Promise.race([
-                runtime.execute(message, { strategyContext }),
+                runtime.execute(message, { strategyContext, abortSignal: sessionAbortSignal }),
                 new Promise((_, reject) => setTimeout(() => reject(new Error(`Bootstrap runtime timeout for "${dimId}"`)), outerTimeoutMs)),
             ]);
             // ── 提取结果 ──

package/dist/lib/external/mcp/handlers/bootstrap/shared/bootstrap-phases.d.ts

@@ -39,6 +39,8 @@ interface BootstrapFileEntry {
     relativePath: string;
     content: string;
     targetName: string;
+    /** Whether this file belongs to a test target or matches test file naming patterns */
+    isTest: boolean;
 }
 /** Target item — either a plain string or an object with metadata */
 type TargetItem = string | {

package/dist/lib/external/mcp/handlers/bootstrap/shared/bootstrap-phases.js

@@ -67,6 +67,7 @@ export async function runPhase1_FileCollection(projectRoot, logger, options = {}
     const seenPaths = new Set();
     const allFiles = [];
     for (const t of allTargets) {
+        const isTestTarget = typeof t === 'object' && /^test/i.test(t.type || '');
         try {
             const fileList = await discoverer.getTargetFiles(t);
             for (const f of fileList) {
@@ -86,6 +87,7 @@ export async function runPhase1_FileCollection(projectRoot, logger, options = {}
                         relativePath: f.relativePath || path.basename(fp),
                         content,
                         targetName: typeof t === 'string' ? t : t.name,
+                        isTest: isTestTarget || LanguageService.isTestFile(fp),
                     });
                 }
                 catch {
@@ -382,6 +384,7 @@ export async function runPhase3_GuardAudit(allFiles, container, logger, options
         const guardFiles = allFiles.map((f) => ({
             path: f.path,
             content: f.content,
+            isTest: f.isTest,
         }));
         guardAudit = guardEngine.auditFiles(guardFiles, { scope: 'project' });
         // 写入 ViolationsStore
@@ -484,6 +487,7 @@ export async function runPhase4_DimensionResolve(params) {
             const guardFiles = allFiles.map((f) => ({
                 path: f.path,
                 content: f.content,
+                isTest: f.isTest,
             }));
             guardAudit = guardEngine.auditFiles(guardFiles, { scope: 'project' });
             logger.info(`[Bootstrap] Guard re-audit with ${guardEngine.getExternalRuleCount()} Enhancement Pack rules → ${guardAudit.summary?.totalViolations ?? 0} total violations`);

package/dist/lib/external/mcp/handlers/guard.js

@@ -10,6 +10,7 @@ import { execSync } from 'node:child_process';
 import fs from 'node:fs';
 import { readFile } from 'node:fs/promises';
 import path from 'node:path';
+import { LanguageService } from '#shared/LanguageService.js';
 import { resolveProjectRoot } from '#shared/resolveProjectRoot.js';
 import { envelope } from '../envelope.js';
 // ═══ Review 轮次追踪（模块私有） ═══════════════════
@@ -93,7 +94,7 @@ export async function guardAuditFiles(ctx, args) {
                 content = '';
             }
         }
-        return { path: absPath, content };
+        return { path: absPath, content, isTest: LanguageService.isTestFile(absPath) };
     }));
     const result = engine.auditFiles(filesToAudit, { scope });
     // 写入 ViolationsStore + GuardFeedbackLoop
@@ -224,7 +225,7 @@ export async function guardReview(ctx, args) {
     for (const fp of filePaths) {
         try {
             const code = await readFile(fp, 'utf8');
-            const auditResult = engine.auditFile(fp, code);
+            const auditResult = engine.auditFile(fp, code, { isTest: LanguageService.isTestFile(fp) });
             const violations = auditResult.violations;
             // 收集 uncertain
             if (auditResult.uncertainResults?.length) {
@@ -523,7 +524,7 @@ export async function scanProject(ctx, args) {
                     content = '';
                 }
             }
-            return { path: f.path, content };
+            return { path: f.path, content, isTest: LanguageService.isTestFile(f.path) };
         }));
         guardAudit = engine.auditFiles(filesToAudit, { scope: 'project' });
         // 写入 ViolationsStore
@@ -629,9 +630,13 @@ async function _injectEnhancementGuardRules(engine, ctx) {
     try {
         const { initEnhancementRegistry } = await import('#core/enhancement/index.js');
         const enhReg = await initEnhancementRegistry();
-        // 使用空语言+空框架列表获取所有已注册的 Pack（不过滤）
-        // 这里我们注入 ALL 规则，让 GuardCheckEngine 按 languages 字段自行过滤
-        const allPacks = enhReg.all();
+        // 仅注入无框架条件的通用 Pack 规则（如 go-web 无 frameworks 条件）
+        // 有框架条件的 Pack（如 go-grpc 需要 frameworks: ['grpc']）由 Bootstrap Phase 4
+        // 通过 resolve(lang, detectedFrameworks) 精确注入，避免非 gRPC 项目出现误报
+        const allPacks = enhReg.all().filter((pack) => {
+            const cond = pack.conditions;
+            return !cond?.frameworks?.length;
+        });
         const allGuardRules = [];
         for (const pack of allPacks) {
             try {

package/dist/lib/http/routes/ai.js

@@ -462,6 +462,10 @@ const LLM_ENV_KEYS = [
     'ASD_CLAUDE_API_KEY',
     'ASD_DEEPSEEK_API_KEY',
     'ASD_AI_PROXY',
+    'ASD_EMBED_PROVIDER',
+    'ASD_EMBED_MODEL',
+    'ASD_EMBED_BASE_URL',
+    'ASD_EMBED_API_KEY',
 ];
 /**
  * 解析 .env 内容为 key-value（仅提取 LLM 相关变量）
@@ -523,7 +527,7 @@ router.get('/env-config', async (req, res) => {
  * Body: { provider, model, apiKey, proxy? }
  */
 router.post('/env-config', validate(AiEnvConfigBody), async (req, res) => {
-    const { provider, model, apiKey, proxy } = req.body;
+    const { provider, model, apiKey, proxy, embedProvider, embedModel, embedBaseUrl, embedApiKey } = req.body;
     const envPath = _getProjectEnvPath();
     let content = existsSync(envPath) ? readFileSync(envPath, 'utf8') : '';
     // 构建 key-value 更新列表
@@ -547,6 +551,19 @@ router.post('/env-config', validate(AiEnvConfigBody), async (req, res) => {
     if (keyName && apiKey) {
         updates[keyName] = apiKey;
     }
+    // Embedding 独立配置
+    if (embedProvider) {
+        updates.ASD_EMBED_PROVIDER = embedProvider;
+        if (embedModel) {
+            updates.ASD_EMBED_MODEL = embedModel;
+        }
+        if (embedBaseUrl) {
+            updates.ASD_EMBED_BASE_URL = embedBaseUrl;
+        }
+        if (embedApiKey) {
+            updates.ASD_EMBED_API_KEY = embedApiKey;
+        }
+    }
     // 逐条合并到 .env 内容
     for (const [k, v] of Object.entries(updates)) {
         // 匹配已有行（包括被注释的行）

package/dist/lib/infrastructure/vector/IndexingPipeline.js

@@ -44,7 +44,12 @@ export class IndexingPipeline {
     constructor(options = {}) {
         this.#vectorStore = options.vectorStore || null;
         this.#aiProvider = options.aiProvider || null;
-        this.#scanDirs = options.scanDirs || ['recipes', `${KNOWLEDGE_BASE_DIR}/recipes`];
+        this.#scanDirs = options.scanDirs || [
+            'recipes',
+            'candidates',
+            `${KNOWLEDGE_BASE_DIR}/recipes`,
+            `${KNOWLEDGE_BASE_DIR}/candidates`,
+        ];
         this.#projectRoot = options.projectRoot || process.cwd();
         this.#chunkingOptions = {
             strategy: options.chunking?.strategy ?? 'auto',

package/dist/lib/injection/modules/AiModule.js

@@ -72,7 +72,28 @@ export async function initialize(c) {
     // Token 追踪 AOP（manager 自身已在构造时 wire，此处延迟注入 recorder）
     // recorder 注入放到 register() 之后（tokenUsageStore 需先注册）
     // Embedding fallback: manager 的 embedFallbackInit 回调已绑定，初始化时主动触发一次
-    const initialEmbed = createEmbedFallback(c, c.singletons.aiProvider);
+    // 优先使用独立的 embed provider（ASD_EMBED_PROVIDER），其次 fallback 机制
+    let initialEmbed = null;
+    try {
+        const aiFactory = c.singletons._aiFactory;
+        if (typeof aiFactory?.createEmbedProvider === 'function') {
+            initialEmbed = aiFactory.createEmbedProvider();
+            if (initialEmbed) {
+                logger.info('Dedicated embed provider created from ASD_EMBED_PROVIDER', {
+                    provider: initialEmbed.name,
+                });
+            }
+        }
+    }
+    catch (err) {
+        logger.warn('Failed to create dedicated embed provider', {
+            error: err.message,
+        });
+    }
+    // 若无独立 embed provider，走旧的 fallback 逻辑
+    if (!initialEmbed) {
+        initialEmbed = createEmbedFallback(c, c.singletons.aiProvider);
+    }
     if (initialEmbed) {
         manager.setEmbedProvider(initialEmbed);
         c.singletons._embedProvider = initialEmbed;

package/dist/lib/service/bootstrap/BootstrapTaskManager.d.ts

@@ -116,6 +116,13 @@ export declare class BootstrapTaskManager {
      * @param [reason='Aborted by user']
      */
     abortSession(reason?: string): void;
+    /**
+     * 获取当前 session 的 AbortSignal
+     *
+     * 用于传入 AgentRuntime.execute()，使得 abortSession() 可以立即中断正在执行的 AI 调用，
+     * 而不是等到下一个维度边界才检测到取消。
+     */
+    getSessionAbortSignal(): AbortSignal | null;
     /**
      * 验证 sessionId 是否仍然是活跃 session
      *

package/dist/lib/service/bootstrap/BootstrapTaskManager.js

@@ -119,6 +119,8 @@ export class BootstrapTaskManager {
     #currentSession = null;
     #eventBus = null;
     #signalBus = null;
+    /** 当前 session 的 AbortController，用于取消正在执行的 AI 调用 */
+    #sessionAbortController = null;
     /** 获取 RealtimeService 的 getter（延迟获取，避免循环依赖） */
     #getRealtimeService = null;
     constructor({ eventBus, signalBus, getRealtimeService } = {}) {
@@ -144,6 +146,7 @@ export class BootstrapTaskManager {
         }
         const sessionId = `bs_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
         this.#currentSession = new BootstrapSession(sessionId);
+        this.#sessionAbortController = new AbortController();
         for (const { id, meta } of taskDefs) {
             this.#currentSession.addTask(id, meta);
         }
@@ -177,6 +180,11 @@ export class BootstrapTaskManager {
                 task.error = reason;
             }
         }
+        // 触发 AbortController，中断正在执行的 AI 调用
+        if (this.#sessionAbortController) {
+            this.#sessionAbortController.abort(reason);
+            this.#sessionAbortController = null;
+        }
         session.status = 'aborted';
         session.completedAt = Date.now();
         session.summary = {
@@ -209,6 +217,15 @@ export class BootstrapTaskManager {
             },
         });
     }
+    /**
+     * 获取当前 session 的 AbortSignal
+     *
+     * 用于传入 AgentRuntime.execute()，使得 abortSession() 可以立即中断正在执行的 AI 调用，
+     * 而不是等到下一个维度边界才检测到取消。
+     */
+    getSessionAbortSignal() {
+        return this.#sessionAbortController?.signal ?? null;
+    }
     /**
      * 验证 sessionId 是否仍然是活跃 session
      *

package/dist/lib/service/guard/ComplianceReporter.js

@@ -93,7 +93,11 @@ export class ComplianceReporter {
         try {
             const { initEnhancementRegistry } = await import('#core/enhancement/index.js');
             const enhReg = await initEnhancementRegistry();
-            const allPacks = enhReg.all();
+            // 仅注入无框架条件的通用 Pack（有框架条件的由 Bootstrap resolve() 精确注入）
+            const allPacks = enhReg.all().filter((pack) => {
+                const cond = pack.conditions;
+                return !cond?.frameworks?.length;
+            });
             const allGuardRules = [];
             for (const pack of allPacks) {
                 try {

package/dist/lib/service/guard/GuardCheckEngine.d.ts

@@ -29,8 +29,10 @@ interface BuiltInRule {
     fixSuggestion?: string;
     excludePaths?: RegExp;
     excludeLinePatterns?: string[];
+    excludePrevLinePatterns?: string[];
     skipComments?: boolean;
     skipTestBlocks?: boolean;
+    skipTestFiles?: boolean;
 }
 interface GuardRule {
     id: string;
@@ -46,8 +48,11 @@ interface GuardRule {
     fixSuggestion?: string | null;
     excludePaths?: RegExp | string;
     excludeLinePatterns?: string[];
+    excludePrevLinePatterns?: string[];
     skipComments?: boolean;
     skipTestBlocks?: boolean;
+    /** When true, this rule is skipped for test files (detected by LanguageService.isTestFile) */
+    skipTestFiles?: boolean;
     astQuery?: {
         queryType: string;
         params?: Record<string, string>;
@@ -108,6 +113,8 @@ interface AuditFileResult {
 interface AuditFilesInput {
     path: string;
     content: string;
+    /** Pre-computed test file flag from LanguageService.isTestFile */
+    isTest?: boolean;
 }
 export { detectLanguage } from './GuardPatternUtils.js';
 /** GuardCheckEngine - 核心检查引擎 */
@@ -146,12 +153,13 @@ export declare class GuardCheckEngine {
      * 对代码运行静态检查
      * @param code 源代码
      * @param language 'objc'|'swift'|'javascript' 等
-     * @param options {scope, filePath}
+     * @param options {scope, filePath, isTest}
      * @returns >}
      */
     checkCode(code: string, language: string, options?: {
         scope?: string | null;
         filePath?: string;
+        isTest?: boolean;
     }): {
         reasoning: {
             whatViolated: string;
@@ -223,6 +231,7 @@ export declare class GuardCheckEngine {
      */
     auditFile(filePath: string, code: string, options?: {
         scope?: string;
+        isTest?: boolean;
     }): AuditFileResult;
     /**
      * 批量文件审计
@@ -237,6 +246,8 @@ export declare class GuardCheckEngine {
         crossFileViolations: import("./GuardCrossFileChecks.js").CrossFileViolation[];
         summary: {
             filesChecked: number;
+            testFiles: number;
+            productionFiles: number;
             totalViolations: number;
             totalErrors: number;
             totalUncertain: number;

package/dist/lib/service/guard/GuardCheckEngine.js

@@ -261,6 +261,7 @@ const BUILT_IN_RULES = {
         languages: ['go'],
         dimension: 'file',
         category: 'correctness',
+        skipTestFiles: true,
     },
     'go-no-err-ignored': {
         message: '错误值不应用 _ 忽略，应处理或明确标注',
@@ -270,6 +271,11 @@ const BUILT_IN_RULES = {
         dimension: 'file',
         category: 'correctness',
         excludePaths: /(?:^|[/\\])(?:tests?|testdata|_test)[/\\]|_test\.go$/,
+        excludeLinePatterns: [
+            '\\.\\([^)]*\\)', // type assertion: val, _ := expr.(Type) — _ 是 bool ok，不是 error
+            'RegisterFlagCompletionFunc', // cobra flag completion: flag 名由同函数字面量保证，不会失败
+            'MarkFlagRequired', // cobra flag setup: 同上
+        ],
     },
     'go-no-init-abuse': {
         message: 'init() 函数副作用难以追踪，避免在 init 中执行复杂逻辑',
@@ -287,6 +293,14 @@ const BUILT_IN_RULES = {
         dimension: 'file',
         category: 'style',
         excludePaths: /(?:^|[/\\])(?:tests?|testdata)[/\\]|_test\.go$/,
+        excludeLinePatterns: [
+            '\\bembed\\.', // //go:embed requires package-level var
+            '\\bsync\\.', // sync.Map, sync.Once, sync.Mutex etc. are designed as package-level vars
+            '\\batomic\\.', // atomic.Pointer, atomic.Value etc.
+        ],
+        excludePrevLinePatterns: [
+            '//go:embed', // //go:embed directive on previous line requires package-level var
+        ],
     },
     // ══════════════════════════════════════════════════════════
     //  Dart (Flutter)
@@ -623,7 +637,11 @@ export class GuardCheckEngine {
                     ...(rule.excludePaths ? { excludePaths: rule.excludePaths } : {}),
                     ...(rule.skipComments ? { skipComments: true } : {}),
                     ...(rule.skipTestBlocks ? { skipTestBlocks: true } : {}),
+                    ...(rule.skipTestFiles ? { skipTestFiles: true } : {}),
                     ...(rule.excludeLinePatterns ? { excludeLinePatterns: rule.excludeLinePatterns } : {}),
+                    ...(rule.excludePrevLinePatterns
+                        ? { excludePrevLinePatterns: rule.excludePrevLinePatterns }
+                        : {}),
                 });
             }
         }
@@ -662,11 +680,11 @@ export class GuardCheckEngine {
      * 对代码运行静态检查
      * @param code 源代码
      * @param language 'objc'|'swift'|'javascript' 等
-     * @param options {scope, filePath}
+     * @param options {scope, filePath, isTest}
      * @returns >}
      */
     checkCode(code, language, options = {}) {
-        const { scope = null, filePath = '' } = options;
+        const { scope = null, filePath = '', isTest = false } = options;
         const violations = [];
         // 获取匹配语言的规则
         let rules = this.getRules(language);
@@ -680,6 +698,10 @@ export class GuardCheckEngine {
                 return !re.test(filePath);
             });
         }
+        // 按 skipTestFiles 标记过滤测试文件
+        if (isTest) {
+            rules = rules.filter((r) => !r.skipTestFiles);
+        }
         // 如果有 scope，按层级过滤：project ⊇ target ⊇ file
         // project 范围包含所有维度的规则；target 包含 file+target；file 仅匹配 file
         // 'universal' 维度在所有 scope 下都生效
@@ -719,6 +741,7 @@ export class GuardCheckEngine {
             // 合并内置 + 配置级排除行模式
             const ruleId = rule.id || rule.name;
             const excludeLineRegexes = this._getExcludeLineRegexes(ruleId, rule.excludeLinePatterns);
+            const excludePrevLineRegexes = this._getExcludeLineRegexes(`${ruleId}:prev`, rule.excludePrevLinePatterns);
             for (let i = 0; i < lines.length; i++) {
                 // skipComments: 跳过注释行（doc comments / 行注释 / 块注释内）
                 if (shouldSkipComments && commentLines[i]) {
@@ -733,6 +756,12 @@ export class GuardCheckEngine {
                     if (excludeLineRegexes.length > 0 && excludeLineRegexes.some((ep) => ep.test(lines[i]))) {
                         continue;
                     }
+                    // excludePrevLinePatterns: 跳过前一行匹配排除模式的行（//go:embed 等指令注释）
+                    if (excludePrevLineRegexes.length > 0 &&
+                        i > 0 &&
+                        excludePrevLineRegexes.some((ep) => ep.test(lines[i - 1]))) {
+                        continue;
+                    }
                     violations.push({
                         ruleId: rule.id || rule.name,
                         message: rule.message,
@@ -1358,8 +1387,8 @@ export class GuardCheckEngine {
         const results = [];
         let totalViolations = 0;
         let totalErrors = 0;
-        for (const { path: filePath, content } of files) {
-            const result = this.auditFile(filePath, content, options);
+        for (const { path: filePath, content, isTest } of files) {
+            const result = this.auditFile(filePath, content, { ...options, isTest });
             results.push(result);
             totalViolations += result.summary.total;
             totalErrors += result.summary.errors;
@@ -1370,8 +1399,11 @@ export class GuardCheckEngine {
         });
         totalViolations += crossFileViolations.length;
         totalErrors += crossFileViolations.filter((v) => v.severity === 'error').length;
+        const testFileCount = files.filter((f) => f.isTest).length;
         const summary = {
             filesChecked: results.length,
+            testFiles: testFileCount,
+            productionFiles: results.length - testFileCount,
             totalViolations,
             totalErrors,
             totalUncertain: results.reduce((s, r) => s + r.summary.uncertain, 0),

package/dist/lib/service/guard/GuardCodeChecks.js

@@ -90,6 +90,7 @@ export function runCodeLevelChecks(code, language, lines, options = {}) {
     // ── Go ──
     if (language === 'go') {
         // defer 在循环内检查 — defer 在函数结束时才执行，循环内 defer 可能资源泄露
+        // 排除 go func() { defer ... } 模式（goroutine 内的 defer 是安全的）
         if (!isDisabled('go-defer-in-loop')) {
             let inLoop = false;
             for (let i = 0; i < lines.length; i++) {
@@ -98,15 +99,32 @@ export function runCodeLevelChecks(code, language, lines, options = {}) {
                     inLoop = true;
                 }
                 if (inLoop && /^\s*defer\s/.test(lines[i])) {
-                    violations.push({
-                        ruleId: 'go-defer-in-loop',
-                        message: 'defer 在循环内会延迟到函数返回时才执行，可能导致资源泄露或大量堆积',
-                        severity: 'warning',
-                        line: i + 1,
-                        snippet: lines[i].trim().slice(0, 120),
-                        dimension: 'file',
-                        fixSuggestion: '将循环体提取到独立函数中，或手动调用 Close()',
-                    });
+                    // 回溯找最近的作用域开场 { — 判断 defer 是否在匿名函数/goroutine 内
+                    let insideAnonymousFunc = false;
+                    let braceBalance = 0;
+                    for (let j = i - 1; j >= 0; j--) {
+                        const prev = lines[j];
+                        braceBalance += (prev.match(/\}/g) || []).length;
+                        braceBalance -= (prev.match(/\{/g) || []).length;
+                        if (braceBalance < 0) {
+                            // 找到包裹 defer 的最近 { — 检查该行是否包含 func 关键字
+                            if (/\bfunc\b/.test(prev)) {
+                                insideAnonymousFunc = true;
+                            }
+                            break;
+                        }
+                    }
+                    if (!insideAnonymousFunc) {
+                        violations.push({
+                            ruleId: 'go-defer-in-loop',
+                            message: 'defer 在循环内会延迟到函数返回时才执行，可能导致资源泄露或大量堆积',
+                            severity: 'warning',
+                            line: i + 1,
+                            snippet: lines[i].trim().slice(0, 120),
+                            dimension: 'file',
+                            fixSuggestion: '将循环体提取到独立函数中，或手动调用 Close()',
+                        });
+                    }
                 }
                 // 简化: 遇到 } 且缩进回到顶层，认为循环结束
                 if (inLoop && trimmed === '}' && (lines[i].match(/^\t/) || lines[i].match(/^}/))) {

package/dist/lib/service/guard/SourceFileCollector.d.ts

@@ -19,10 +19,10 @@ export declare function collectSourceFiles(dir: string, options?: {
     maxFiles?: number;
 }): Promise<string[]>;
 /**
- * 收集源文件并读取内容
+ * 收集源文件并读取内容（带测试文件标记）
  * @param dir 根目录
  * @param options collectSourceFiles 选项
- * @returns >>}
+ * @returns { path, content, isTest }[]
  */
 export declare function collectSourceFilesWithContent(dir: string, options?: {
     extensions?: Set<string>;
@@ -31,6 +31,7 @@ export declare function collectSourceFilesWithContent(dir: string, options?: {
 }): Promise<{
     path: string;
     content: string;
+    isTest: boolean;
 }[]>;
 declare const _default: {
     collectSourceFiles: typeof collectSourceFiles;

package/dist/lib/service/guard/SourceFileCollector.js

@@ -82,10 +82,10 @@ export async function collectSourceFiles(dir, options = {}) {
     return files;
 }
 /**
- * 收集源文件并读取内容
+ * 收集源文件并读取内容（带测试文件标记）
  * @param dir 根目录
  * @param options collectSourceFiles 选项
- * @returns >>}
+ * @returns { path, content, isTest }[]
  */
 export async function collectSourceFilesWithContent(dir, options = {}) {
     const paths = await collectSourceFiles(dir, options);
@@ -93,7 +93,7 @@ export async function collectSourceFilesWithContent(dir, options = {}) {
     for (const filePath of paths) {
         try {
             const content = await readFile(filePath, 'utf-8');
-            results.push({ path: filePath, content });
+            results.push({ path: filePath, content, isTest: LanguageService.isTestFile(filePath) });
         }
         catch {
             // 读取失败跳过