autosnippet 2.9.0 → 2.10.0

package/lib/service/chat/tools.js

@@ -33,7 +33,7 @@
  * │ 17. generate_guard_rule  AI 生成 Guard 规则       │
  * └─────────────────────────────────────────────────────┘
  * ┌─── 生命周期操作类 (7) ─────────────────────────────┐
- * │ 18. submit_candidate     提交候选                │
+ * │ 18. submit_knowledge     提交候选                │
  * │ 19. approve_candidate    批准候选                │
  * │ 20. reject_candidate     驳回候选                │
  * │ 21. publish_recipe       发布 Recipe              │
@@ -853,19 +853,19 @@ const searchRecipes = {
     },
   },
   handler: async (params, ctx) => {
-    const recipeService = ctx.container.get('recipeService');
+    const knowledgeService = ctx.container.get('knowledgeService');
     const { keyword, category, language, knowledgeType, limit = 10 } = params;
 
     if (keyword) {
-      return recipeService.searchRecipes(keyword, { page: 1, pageSize: limit });
+      return knowledgeService.search(keyword, { page: 1, pageSize: limit });
     }
 
-    const filters = {};
+    const filters = { lifecycle: 'active' };
     if (category)      filters.category = category;
     if (language)      filters.language = language;
     if (knowledgeType) filters.knowledgeType = knowledgeType;
 
-    return recipeService.listRecipes(filters, { page: 1, pageSize: limit });
+    return knowledgeService.list(filters, { page: 1, pageSize: limit });
   },
 };
 
@@ -886,19 +886,20 @@ const searchCandidates = {
     },
   },
   handler: async (params, ctx) => {
-    const candidateService = ctx.container.get('candidateService');
+    const knowledgeService = ctx.container.get('knowledgeService');
     const { keyword, status, language, category, limit = 10 } = params;
 
     if (keyword) {
-      return candidateService.searchCandidates(keyword, { page: 1, pageSize: limit });
+      return knowledgeService.search(keyword, { page: 1, pageSize: limit });
     }
 
+    // V3: status 映射为 lifecycle
     const filters = {};
-    if (status)   filters.status = status;
+    if (status)   filters.lifecycle = status;
     if (language) filters.language = language;
     if (category) filters.category = category;
 
-    return candidateService.listCandidates(filters, { page: 1, pageSize: limit });
+    return knowledgeService.list(filters, { page: 1, pageSize: limit });
   },
 };
 
@@ -916,10 +917,13 @@ const getRecipeDetail = {
     required: ['recipeId'],
   },
   handler: async (params, ctx) => {
-    const recipeRepo = ctx.container.get('recipeRepository');
-    const recipe = await recipeRepo.findById(params.recipeId);
-    if (!recipe) return { error: `Recipe '${params.recipeId}' not found` };
-    return recipe;
+    const knowledgeService = ctx.container.get('knowledgeService');
+    try {
+      const entry = await knowledgeService.get(params.recipeId);
+      return typeof entry.toJSON === 'function' ? entry.toJSON() : entry;
+    } catch {
+      return { error: `Knowledge entry '${params.recipeId}' not found` };
+    }
   },
 };
 
@@ -931,13 +935,8 @@ const getProjectStats = {
   description: '获取项目知识库的整体统计：Recipe 数量/分类分布、候选项数量/状态分布、知识图谱节点/边数。',
   parameters: { type: 'object', properties: {} },
   handler: async (_params, ctx) => {
-    const recipeService = ctx.container.get('recipeService');
-    const candidateService = ctx.container.get('candidateService');
-
-    const [recipeStats, candidateStats] = await Promise.all([
-      recipeService.getRecipeStats(),
-      candidateService.getCandidateStats(),
-    ]);
+    const knowledgeService = ctx.container.get('knowledgeService');
+    const stats = await knowledgeService.getStats();
 
     // 尝试获取知识图谱统计
     let graphStats = null;
@@ -947,8 +946,7 @@ const getProjectStats = {
     } catch { /* KG not available */ }
 
     return {
-      recipes: recipeStats,
-      candidates: candidateStats,
+      knowledge: stats,
       knowledgeGraph: graphStats,
     };
   },
@@ -1000,8 +998,9 @@ const searchKnowledge = {
     // 降级: RetrievalFunnel + 全量候选
     try {
       const funnel = ctx.container.get('retrievalFunnel');
-      const recipeRepo = ctx.container.get('recipeRepository');
-      const allRecipes = await recipeRepo.findAll?.() || [];
+      const knowledgeRepo = ctx.container.get('knowledgeRepository');
+      const allResult = await knowledgeRepo.findWithPagination({}, { page: 1, pageSize: 500 });
+      const allRecipes = allResult?.items || [];
 
       // 规范化为 funnel 输入格式
       const candidates = allRecipes.map(r => ({
@@ -1088,7 +1087,7 @@ const extractRecipes = {
   },
   handler: async (params, ctx) => {
     if (!ctx.aiProvider) return { error: 'AI provider not available' };
-    const { targetName, files } = params;
+    const { targetName, files, comprehensive } = params;
 
     // 加载语言参考 Skill（如有），注入到 AI 提取 prompt
     let skillReference = null;
@@ -1124,6 +1123,7 @@ const extractRecipes = {
     const extractOpts = {};
     if (skillReference) extractOpts.skillReference = skillReference;
     if (astContext) extractOpts.astContext = astContext;
+    if (comprehensive) extractOpts.comprehensive = true;
 
     // 首选：使用当前 aiProvider
     try {
@@ -1169,12 +1169,10 @@ const enrichCandidate = {
   },
   handler: async (params, ctx) => {
     if (!ctx.aiProvider) return { error: 'AI provider not available' };
-    const candidateService = ctx.container.get('candidateService');
-    return candidateService.enrichCandidates(
-      params.candidateIds,
-      ctx.aiProvider,
-      { userId: 'agent' },
-    );
+    // V3: 使用 MCP handler enrichCandidates 的逻辑
+    const { enrichCandidates: enrichFn } = await import('../../external/mcp/handlers/candidate.js');
+    const result = await enrichFn(ctx, { candidateIds: params.candidateIds });
+    return result?.data || result;
   },
 };
 
@@ -1194,20 +1192,14 @@ const refineBootstrapCandidates = {
   },
   handler: async (params, ctx) => {
     if (!ctx.aiProvider) return { error: 'AI provider not available' };
-    const candidateService = ctx.container.get('candidateService');
-
-    // 接入 BootstrapTaskManager 双通道推送 refine:* 事件到前端
-    let onProgress = null;
-    try {
-      const taskManager = ctx.container.get('bootstrapTaskManager');
-      onProgress = (eventName, data) => taskManager.emitProgress(eventName, data);
-    } catch { /* optional: no realtime push */ }
-
-    return candidateService.refineBootstrapCandidates(
-      ctx.aiProvider,
-      { candidateIds: params.candidateIds, userPrompt: params.userPrompt, dryRun: params.dryRun, onProgress },
-      { userId: 'agent' },
-    );
+    // V3: 委托给 bootstrap handler 的 refine 逻辑
+    const { bootstrapRefine } = await import('../../external/mcp/handlers/bootstrap.js');
+    const result = await bootstrapRefine(ctx, {
+      candidateIds: params.candidateIds,
+      userPrompt: params.userPrompt,
+      dryRun: params.dryRun,
+    });
+    return result?.data || result;
   },
 };
 
@@ -1231,18 +1223,18 @@ const checkDuplicate = {
     const projectRoot = params.projectRoot || ctx.projectRoot;
     const threshold = params.threshold ?? 0.5;
 
-    // 如果提供 candidateId，从数据库读取候选信息
+    // 如果提供 candidateId，从数据库读取条目信息
     if (!cand && params.candidateId) {
       try {
-        const candidateRepo = ctx.container.get('candidateRepository');
-        const found = await candidateRepo.findById(params.candidateId);
+        const knowledgeService = ctx.container.get('knowledgeService');
+        const found = await knowledgeService.get(params.candidateId);
         if (found) {
-          const meta = found.metadata || {};
+          const json = typeof found.toJSON === 'function' ? found.toJSON() : found;
           cand = {
-            title: meta.title || '',
-            summary: meta.summary_cn || meta.summary || '',
-            code: found.code || '',
-            usageGuide: meta.usageGuide_cn || meta.usageGuide || '',
+            title: json.title || '',
+            summary: json.summary_cn || json.description || '',
+            code: json.content?.pattern || '',
+            usageGuide: json.usage_guide_cn || '',
           };
         }
       } catch { /* ignore */ }
@@ -1442,8 +1434,9 @@ const getRecommendations = {
     },
   },
   handler: async (params, ctx) => {
-    const recipeService = ctx.container.get('recipeService');
-    return recipeService.getRecommendations(params.limit || 10);
+    const knowledgeService = ctx.container.get('knowledgeService');
+    // V3: 推荐 = 活跃条目按使用量排序
+    return knowledgeService.list({ lifecycle: 'active' }, { page: 1, pageSize: params.limit || 10 });
   },
 };
 
@@ -1632,24 +1625,24 @@ const DIMENSION_DISPLAY_GROUP = {
 };
 
 // ────────────────────────────────────────────────────────────
-// Bootstrap 维度类型校验 — submit_candidate / submit_with_check 共用
+// Bootstrap 维度类型校验 — submit_knowledge / submit_with_check 共用
 // 基于 dimensionMeta 类型标注系统，而非关键词模糊匹配
 // ────────────────────────────────────────────────────────────
 
 /**
  * 基于维度元数据 (dimensionMeta) 检查提交是否合法
  * @param {{ id: string, outputType: 'candidate'|'skill'|'dual', allowedKnowledgeTypes: string[] }} dimensionMeta
- * @param {object} params - submit_candidate 的参数
+ * @param {object} params - submit_knowledge 的参数
  * @param {object} [logger]
  * @returns {{ status: string, reason: string } | null} 不合法返回 rejected，合法返回 null
  */
 function _checkDimensionType(dimensionMeta, params, logger) {
   // 1. Skill-only 维度不允许提交 Candidate
   if (dimensionMeta.outputType === 'skill') {
-    logger?.info(`[submit_candidate] ✗ rejected — dimension "${dimensionMeta.id}" is skill-only, cannot submit candidates`);
+    logger?.info(`[submit_knowledge] ✗ rejected — dimension "${dimensionMeta.id}" is skill-only, cannot submit candidates`);
     return {
       status: 'rejected',
-      reason: `当前维度 "${dimensionMeta.id}" 的输出类型为 skill-only，不允许调用 submit_candidate。请只在最终回复中提供 dimensionDigest JSON。`,
+      reason: `当前维度 "${dimensionMeta.id}" 的输出类型为 skill-only，不允许调用 submit_knowledge。请只在最终回复中提供 dimensionDigest JSON。`,
     };
   }
 
@@ -1658,7 +1651,7 @@ function _checkDimensionType(dimensionMeta, params, logger) {
   if (allowed.length > 0 && params.knowledgeType) {
     if (!allowed.includes(params.knowledgeType)) {
       const corrected = allowed[0];
-      logger?.warn(`[submit_candidate] knowledgeType "${params.knowledgeType}" → "${corrected}" (auto-corrected for dimension "${dimensionMeta.id}")`);
+      logger?.warn(`[submit_knowledge] knowledgeType "${params.knowledgeType}" → "${corrected}" (auto-corrected for dimension "${dimensionMeta.id}")`);
       params.knowledgeType = corrected;
     }
   }
@@ -1666,10 +1659,10 @@ function _checkDimensionType(dimensionMeta, params, logger) {
   return null;
 }
 
-// 16. submit_candidate
+// 16. submit_knowledge
 // ────────────────────────────────────────────────────────────
 const submitCandidate = {
-  name: 'submit_candidate',
+  name: 'submit_knowledge',
   description: '提交新的代码候选项到知识库审核队列。',
   parameters: {
     type: 'object',
@@ -1688,7 +1681,7 @@ const submitCandidate = {
     required: ['code', 'language', 'category', 'title'],
   },
   handler: async (params, ctx) => {
-    const candidateService = ctx.container.get('candidateService');
+    const knowledgeService = ctx.container.get('knowledgeService');
 
     // ── Bootstrap 维度类型校验 (基于 dimensionMeta 类型标注) ──
     const dimMeta = ctx._dimensionMeta;
@@ -1706,8 +1699,7 @@ const submitCandidate = {
         params.knowledgeType = dimMeta.allowedKnowledgeTypes[0];
       }
 
-      // Bootstrap 模式: 将 category 覆盖为展示分组 ID，确保前端按合并后的分组展示
-      // AI 可能填 "UI"/"Core" 等功能分类，但前端通过 BOOTSTRAP_DIM_LABELS 按展示分组分组
+      // Bootstrap 模式: 将 category 覆盖为展示分组 ID
       params.category = DIMENSION_DISPLAY_GROUP[dimMeta.id] || dimMeta.id;
 
       // ── CandidateGuardrail 质量验证 (Bootstrap 模式) ──
@@ -1717,7 +1709,7 @@ const submitCandidate = {
       );
       const guardResult = guardrail.validate(params);
       if (!guardResult.valid) {
-        ctx.logger?.info(`[submit_candidate] ✗ guardrail rejected: ${guardResult.error}`);
+        ctx.logger?.info(`[submit_knowledge] ✗ guardrail rejected: ${guardResult.error}`);
         return {
           status: 'rejected',
           error: guardResult.error,
@@ -1726,11 +1718,10 @@ const submitCandidate = {
       }
     }
 
-    // 将所有顶层字段展开到 item — LLM 可能把 title/summary/tags 等
-    // 放在顶层而非 metadata 中（production prompt 指引）
+    // V3: 将顶层字段映射为 wire format
     const { code, language, category, source, reasoning, metadata, ...rest } = params;
 
-    // 防御性修复: AI 可能提交 reasoning.sources = [] 空数组，自动从 filePaths 补充
+    // 防御性修复: AI 可能提交 reasoning.sources = [] 空数组
     const finalReasoning = reasoning || { whyStandard: 'Submitted via ChatAgent', sources: ['agent'], confidence: 0.7 };
     if (Array.isArray(finalReasoning.sources) && finalReasoning.sources.length === 0) {
       const fallbackSources = params.filePaths || rest.filePaths;
@@ -1741,18 +1732,31 @@ const submitCandidate = {
       }
     }
 
-    const item = {
-      code,
-      language,
-      category,
-      ...rest,                     // 顶层扩展字段 (title, summary, knowledgeType, tags 等)
-      ...metadata,                 // metadata 对象 (如有)
-      reasoning: finalReasoning,
-    };
-    // Bootstrap 模式额外注入 targetName — 前端优先按 meta.targetName 分组
     const displayGroup = dimMeta ? (DIMENSION_DISPLAY_GROUP[dimMeta.id] || dimMeta.id) : null;
-    const extraMeta = (dimMeta && ctx.source === 'system') ? { targetName: displayGroup } : {};
-    return candidateService.createFromToolParams(item, source || 'agent', extraMeta, { userId: 'agent' });
+    const data = {
+      title:          rest.title || '',
+      description:    rest.summary || rest.description || '',
+      language:       language || '',
+      category:       category || 'general',
+      knowledge_type: rest.knowledgeType || 'code-pattern',
+      tags:           rest.tags || [],
+      summary_cn:     rest.summary || '',
+      source:         source || 'agent',
+      content: {
+        pattern:  code || '',
+        rationale: rest.rationale || '',
+      },
+      reasoning: {
+        why_standard: finalReasoning.whyStandard || finalReasoning.why_standard || 'Submitted via ChatAgent',
+        sources: finalReasoning.sources || ['agent'],
+        confidence: finalReasoning.confidence || 0.7,
+      },
+      ...(metadata || {}),
+    };
+    if (dimMeta && ctx.source === 'system' && displayGroup) {
+      data.tags = [...new Set([...(data.tags || []), displayGroup])];
+    }
+    return knowledgeService.create(data, { userId: 'agent' });
   },
 };
 
@@ -1770,8 +1774,8 @@ const approveCandidate = {
     required: ['candidateId'],
   },
   handler: async (params, ctx) => {
-    const candidateService = ctx.container.get('candidateService');
-    return candidateService.approveCandidate(params.candidateId, { userId: 'agent' });
+    const knowledgeService = ctx.container.get('knowledgeService');
+    return knowledgeService.approve(params.candidateId, { userId: 'agent' });
   },
 };
 
@@ -1790,8 +1794,8 @@ const rejectCandidate = {
     required: ['candidateId', 'reason'],
   },
   handler: async (params, ctx) => {
-    const candidateService = ctx.container.get('candidateService');
-    return candidateService.rejectCandidate(params.candidateId, params.reason, { userId: 'agent' });
+    const knowledgeService = ctx.container.get('knowledgeService');
+    return knowledgeService.reject(params.candidateId, params.reason, { userId: 'agent' });
   },
 };
 
@@ -1809,8 +1813,8 @@ const publishRecipe = {
     required: ['recipeId'],
   },
   handler: async (params, ctx) => {
-    const recipeService = ctx.container.get('recipeService');
-    return recipeService.publishRecipe(params.recipeId, { userId: 'agent' });
+    const knowledgeService = ctx.container.get('knowledgeService');
+    return knowledgeService.publish(params.recipeId, { userId: 'agent' });
   },
 };
 
@@ -1829,8 +1833,8 @@ const deprecateRecipe = {
     required: ['recipeId', 'reason'],
   },
   handler: async (params, ctx) => {
-    const recipeService = ctx.container.get('recipeService');
-    return recipeService.deprecateRecipe(params.recipeId, params.reason, { userId: 'agent' });
+    const knowledgeService = ctx.container.get('knowledgeService');
+    return knowledgeService.deprecate(params.recipeId, params.reason, { userId: 'agent' });
   },
 };
 
@@ -1849,8 +1853,8 @@ const updateRecipe = {
     required: ['recipeId', 'updates'],
   },
   handler: async (params, ctx) => {
-    const recipeService = ctx.container.get('recipeService');
-    return recipeService.updateRecipe(params.recipeId, params.updates, { userId: 'agent' });
+    const knowledgeService = ctx.container.get('knowledgeService');
+    return knowledgeService.update(params.recipeId, params.updates, { userId: 'agent' });
   },
 };
 
@@ -1869,9 +1873,9 @@ const recordUsage = {
     required: ['recipeId'],
   },
   handler: async (params, ctx) => {
-    const recipeService = ctx.container.get('recipeService');
+    const knowledgeService = ctx.container.get('knowledgeService');
     const type = params.type || 'adoption';
-    await recipeService.incrementUsage(params.recipeId, type);
+    await knowledgeService.incrementUsage(params.recipeId, type);
     return { success: true, recipeId: params.recipeId, type };
   },
 };
@@ -1894,9 +1898,13 @@ const qualityScore = {
     let recipe = params.recipe;
 
     if (!recipe && params.recipeId) {
-      const recipeRepo = ctx.container.get('recipeRepository');
-      recipe = await recipeRepo.findById(params.recipeId);
-      if (!recipe) return { error: `Recipe '${params.recipeId}' not found` };
+      const knowledgeService = ctx.container.get('knowledgeService');
+      try {
+        const entry = await knowledgeService.get(params.recipeId);
+        recipe = typeof entry.toJSON === 'function' ? entry.toJSON() : entry;
+      } catch {
+        return { error: `Knowledge entry '${params.recipeId}' not found` };
+      }
     }
     if (!recipe) return { error: 'Provide recipeId or recipe object' };
 
@@ -2210,13 +2218,8 @@ const knowledgeOverview = {
     const [statsResult, feedbackResult] = await Promise.all([
       (async () => {
         try {
-          const recipeService = ctx.container.get('recipeService');
-          const candidateService = ctx.container.get('candidateService');
-          const [rs, cs] = await Promise.all([
-            recipeService.getRecipeStats(),
-            candidateService.getCandidateStats(),
-          ]);
-          return { recipes: rs, candidates: cs };
+          const knowledgeService = ctx.container.get('knowledgeService');
+          return knowledgeService.getStats();
         } catch { return null; }
       })(),
       (async () => {
@@ -2229,8 +2232,7 @@ const knowledgeOverview = {
     ]);
 
     if (statsResult) {
-      result.recipes = statsResult.recipes;
-      result.candidates = statsResult.candidates;
+      result.knowledge = statsResult;
     }
 
     // 知识图谱统计
@@ -2256,7 +2258,7 @@ const knowledgeOverview = {
 // ────────────────────────────────────────────────────────────
 const submitWithCheck = {
   name: 'submit_with_check',
-  description: '安全提交候选：先执行查重检测，无重复则自动提交。如果发现高度相似 Recipe 则阻止并返回相似列表。一次调用完成 check_duplicate + submit_candidate。',
+  description: '安全提交候选：先执行查重检测，无重复则自动提交。如果发现高度相似 Recipe 则阻止并返回相似列表。一次调用完成 check_duplicate + submit_knowledge。',
   parameters: {
     type: 'object',
     properties: {
@@ -2273,7 +2275,7 @@ const submitWithCheck = {
     const { code, language, category, title, summary, threshold = 0.7 } = params;
     const projectRoot = ctx.projectRoot;
 
-    // ── Bootstrap 维度类型校验 (与 submit_candidate 共用逻辑) ──
+    // ── Bootstrap 维度类型校验 (与 submit_knowledge 共用逻辑) ──
     const dimMeta = ctx._dimensionMeta;
     if (dimMeta && ctx.source === 'system') {
       const rejected = _checkDimensionType(dimMeta, params, ctx.logger);
@@ -2311,28 +2313,33 @@ const submitWithCheck = {
       };
     }
 
-    // Step 2: 提交
+    // Step 2: 提交 (V3: 使用 knowledgeService.create)
     try {
-      const candidateService = ctx.container.get('candidateService');
+      const knowledgeService = ctx.container.get('knowledgeService');
       const { code: _c, language: _l, category: _cat, title: _t, summary: _s, threshold: _th, source: paramSource, reasoning: userReasoning, metadata, ...rest } = params;
-      const item = {
-        code,
-        language,
-        category,
-        ...rest,                     // 顶层扩展字段 (title, summary, knowledgeType, tags 等)
-        ...metadata,                 // metadata 对象 (如有)
-        title: title || '',
-        summary: summary || '',
-        reasoning: userReasoning || { whyStandard: 'Submitted via submit_with_check', sources: ['agent'], confidence: 0.7 },
+      const swcReasoning = userReasoning || { whyStandard: 'Submitted via submit_with_check', sources: ['agent'], confidence: 0.7 };
+      const data = {
+        title:          title || '',
+        description:    summary || '',
+        language:       language || '',
+        category:       category || 'general',
+        knowledge_type: rest.knowledgeType || 'code-pattern',
+        tags:           rest.tags || [],
+        summary_cn:     summary || '',
+        source:         paramSource || 'agent',
+        content: { pattern: code || '' },
+        reasoning: {
+          why_standard: swcReasoning.whyStandard || swcReasoning.why_standard || 'Submitted via submit_with_check',
+          sources: swcReasoning.sources || ['agent'],
+          confidence: swcReasoning.confidence || 0.7,
+        },
+        ...(metadata || {}),
       };
-      // Bootstrap 模式注入展示分组 targetName
-      const swcDisplayGroup = dimMeta ? (DIMENSION_DISPLAY_GROUP[dimMeta.id] || dimMeta.id) : null;
-      const swcExtraMeta = (dimMeta && ctx.source === 'system') ? { targetName: swcDisplayGroup } : {};
-      const created = await candidateService.createFromToolParams(item, paramSource || 'agent', swcExtraMeta, { userId: 'agent' });
+      const created = await knowledgeService.create(data, { userId: 'agent' });
 
       return {
         submitted: true,
-        candidate: created,
+        entry: typeof created.toJSON === 'function' ? created.toJSON() : created,
         similar: similar.length > 0 ? similar : [],
         _meta: {
           confidence: 'high',
@@ -2457,7 +2464,7 @@ const reviewMyOutput = {
   },
   handler: async (params, context) => {
     const submitted = (context._sessionToolCalls || []).filter(
-      tc => tc.tool === 'submit_candidate' || tc.tool === 'submit_with_check',
+      tc => tc.tool === 'submit_knowledge' || tc.tool === 'submit_with_check',
     );
 
     if (submitted.length === 0) {