autosnippet 3.0.11 → 3.0.13

package/lib/service/chat/tools/infrastructure.js

@@ -0,0 +1,222 @@
+/**
+ * infrastructure.js — 基础设施类工具 (7)
+ *
+ * 29. graph_impact_analysis  知识图谱影响分析
+ * 30. rebuild_index          向量索引重建
+ * 31. query_audit_log        审计日志查询
+ * 32. load_skill             加载 Skill 文档
+ * 33. create_skill           创建项目级 Skill
+ * 34. suggest_skills         推荐创建 Skill
+ * 34. bootstrap_knowledge    冷启动知识库
+ */
+
+import fs from 'node:fs';
+import path from 'node:path';
+import Logger from '../../../infrastructure/logging/Logger.js';
+import { SKILLS_DIR, PROJECT_SKILLS_DIR } from './_shared.js';
+
+// ────────────────────────────────────────────────────────────
+// 29. graph_impact_analysis
+// ────────────────────────────────────────────────────────────
+export const graphImpactAnalysis = {
+  name: 'graph_impact_analysis',
+  description: '知识图谱影响范围分析 — 查找修改某个 Recipe 后可能受影响的所有下游依赖。',
+  parameters: {
+    type: 'object',
+    properties: {
+      recipeId: { type: 'string', description: 'Recipe ID' },
+      maxDepth: { type: 'number', description: '最大深度，默认 3' },
+    },
+    required: ['recipeId'],
+  },
+  handler: async (params, ctx) => {
+    const kgService = ctx.container.get('knowledgeGraphService');
+    const impacted = kgService.getImpactAnalysis(params.recipeId, 'recipe', params.maxDepth || 3);
+    return { recipeId: params.recipeId, impactedCount: impacted.length, impacted };
+  },
+};
+
+// ────────────────────────────────────────────────────────────
+// 30. rebuild_index
+// ────────────────────────────────────────────────────────────
+export const rebuildIndex = {
+  name: 'rebuild_index',
+  description:
+    '向量索引重建 — 重新扫描 Recipe 文件并更新向量索引（用于索引过期或新增大量 Recipe 后）。',
+  parameters: {
+    type: 'object',
+    properties: {
+      force: { type: 'boolean', description: '强制重建（跳过增量检测），默认 false' },
+      dryRun: { type: 'boolean', description: '仅预览不实际写入，默认 false' },
+    },
+  },
+  handler: async (params, ctx) => {
+    const pipeline = ctx.container.get('indexingPipeline');
+    return pipeline.run({ force: params.force || false, dryRun: params.dryRun || false });
+  },
+};
+
+// ────────────────────────────────────────────────────────────
+// 31. query_audit_log
+// ────────────────────────────────────────────────────────────
+export const queryAuditLog = {
+  name: 'query_audit_log',
+  description: '审计日志查询 — 查看系统操作历史（谁在什么时间做了什么操作）。',
+  parameters: {
+    type: 'object',
+    properties: {
+      action: {
+        type: 'string',
+        description: '按操作类型过滤 (create_candidate/approve_candidate/create_guard_rule 等)',
+      },
+      actor: { type: 'string', description: '按操作者过滤' },
+      limit: { type: 'number', description: '返回数量，默认 20' },
+    },
+  },
+  handler: async (params, ctx) => {
+    const auditLogger = ctx.container.get('auditLogger');
+    const { action, actor, limit = 20 } = params;
+
+    if (actor) {
+      return auditLogger.getByActor(actor, limit);
+    }
+    if (action) {
+      return auditLogger.getByAction(action, limit);
+    }
+    return auditLogger.getStats();
+  },
+};
+
+// ────────────────────────────────────────────────────────────
+// 32. load_skill — 按需加载 Agent Skill 文档
+// ────────────────────────────────────────────────────────────
+export const loadSkill = {
+  name: 'load_skill',
+  description:
+    '加载指定的 Agent Skill 文档，获取领域操作指南和最佳实践参考。可用于冷启动指南 (autosnippet-coldstart)、语言参考 (autosnippet-reference-swift/objc/jsts/python/java/kotlin/go/dart/rust) 等。',
+  parameters: {
+    type: 'object',
+    properties: {
+      skillName: {
+        type: 'string',
+        description: 'Skill 目录名（如 autosnippet-coldstart, autosnippet-reference-swift, autosnippet-reference-go 等）',
+      },
+    },
+    required: ['skillName'],
+  },
+  handler: async (params) => {
+    // 项目级 Skills 优先（覆盖同名内置 Skill）
+    const projectSkillPath = path.join(PROJECT_SKILLS_DIR, params.skillName, 'SKILL.md');
+    const builtinSkillPath = path.join(SKILLS_DIR, params.skillName, 'SKILL.md');
+    const skillPath = fs.existsSync(projectSkillPath) ? projectSkillPath : builtinSkillPath;
+    try {
+      const content = fs.readFileSync(skillPath, 'utf8');
+      const source = skillPath === projectSkillPath ? 'project' : 'builtin';
+      return { skillName: params.skillName, source, content };
+    } catch {
+      const available = new Set();
+      try {
+        fs.readdirSync(SKILLS_DIR, { withFileTypes: true })
+          .filter((d) => d.isDirectory())
+          .forEach((d) => available.add(d.name));
+      } catch {}
+      try {
+        fs.readdirSync(PROJECT_SKILLS_DIR, { withFileTypes: true })
+          .filter((d) => d.isDirectory())
+          .forEach((d) => available.add(d.name));
+      } catch {}
+      return { error: `Skill "${params.skillName}" not found`, availableSkills: [...available] };
+    }
+  },
+};
+
+// ────────────────────────────────────────────────────────────
+// 33. create_skill — 创建项目级 Skill
+// ────────────────────────────────────────────────────────────
+export const createSkillTool = {
+  name: 'create_skill',
+  description:
+    '创建项目级 Skill 文档，写入 AutoSnippet/skills/<name>/SKILL.md。Skill 是 Agent 的领域知识增强文档。创建后自动更新编辑器索引。',
+  parameters: {
+    type: 'object',
+    properties: {
+      name: {
+        type: 'string',
+        description: 'Skill 名称（kebab-case，如 my-auth-guide），3-64 字符',
+      },
+      description: { type: 'string', description: 'Skill 一句话描述（写入 frontmatter）' },
+      content: { type: 'string', description: 'Skill 正文内容（Markdown 格式，不含 frontmatter）' },
+      overwrite: { type: 'boolean', description: '如果同名 Skill 已存在，是否覆盖（默认 false）' },
+    },
+    required: ['name', 'description', 'content'],
+  },
+  handler: async (params, ctx) => {
+    const { createSkill } = await import('../../../external/mcp/handlers/skill.js');
+    // 根据 ChatAgent 的 source 推断 createdBy
+    const createdBy = ctx?.source === 'system' ? 'system-ai' : 'user-ai';
+    const raw = createSkill(null, { ...params, createdBy });
+    try {
+      return JSON.parse(raw);
+    } catch {
+      return { success: false, error: raw };
+    }
+  },
+};
+
+// ────────────────────────────────────────────────────────────
+// 34. suggest_skills — 基于使用模式推荐 Skill 创建
+// ────────────────────────────────────────────────────────────
+export const suggestSkills = {
+  name: 'suggest_skills',
+  description:
+    '基于项目使用模式分析，推荐创建 Skill。分析 Guard 违规频率、Memory 偏好积累、Recipe 分布缺口、候选积压率。返回推荐列表（含 name/description/rationale/priority），可据此直接调用 create_skill 创建。',
+  parameters: {
+    type: 'object',
+    properties: {},
+    required: [],
+  },
+  handler: async (_params, ctx) => {
+    const { SkillAdvisor } = await import('../../skills/SkillAdvisor.js');
+    const database = ctx?.container?.get?.('database') || null;
+    const projectRoot = ctx?.projectRoot || process.cwd();
+    const advisor = new SkillAdvisor(projectRoot, { database });
+    return advisor.suggest();
+  },
+};
+
+// ────────────────────────────────────────────────────────────
+// 34. bootstrap_knowledge — 冷启动知识库初始化
+// ────────────────────────────────────────────────────────────
+export const bootstrapKnowledgeTool = {
+  name: 'bootstrap_knowledge',
+  description:
+    '冷启动知识库初始化（纯启发式，不使用 AI）: SPM Target 扫描 → 依赖图谱 → Guard 审计 → 9 维度 Candidate 自动创建。支持 Skill 增强维度定义。产出为初稿候选，后续由 DAG pipeline 自动编排 AI 增强（enrich → refine）。',
+  parameters: {
+    type: 'object',
+    properties: {
+      maxFiles: { type: 'number', description: '最大扫描文件数，默认 500' },
+      skipGuard: { type: 'boolean', description: '是否跳过 Guard 审计，默认 false' },
+      contentMaxLines: { type: 'number', description: '每文件读取最大行数，默认 120' },
+      loadSkills: {
+        type: 'boolean',
+        description: '是否加载 Skills 增强维度定义（推荐开启），默认 true',
+      },
+    },
+  },
+  handler: async (params, ctx) => {
+    const { bootstrapKnowledge } = await import('../../../external/mcp/handlers/bootstrap.js');
+    const logger = Logger.getInstance();
+    const result = await bootstrapKnowledge(
+      { container: ctx.container, logger },
+      {
+        maxFiles: params.maxFiles || 500,
+        skipGuard: params.skipGuard || false,
+        contentMaxLines: params.contentMaxLines || 120,
+        loadSkills: params.loadSkills ?? true,
+      }
+    );
+    // bootstrapKnowledge 返回 envelope JSON string，解析提取 data
+    const parsed = typeof result === 'string' ? JSON.parse(result) : result;
+    return parsed?.data || parsed;
+  },
+};

package/lib/service/chat/tools/knowledge-graph.js

@@ -0,0 +1,234 @@
+/**
+ * knowledge-graph.js — 知识图谱类工具 (3)
+ *
+ * 10. check_duplicate     候选查重
+ * 11. discover_relations  AI 关系发现
+ * 12. add_graph_edge      手动添加图谱边
+ */
+
+import { findSimilarRecipes } from '../../candidate/SimilarityService.js';
+
+// ────────────────────────────────────────────────────────────
+// 10. check_duplicate
+// ────────────────────────────────────────────────────────────
+export const checkDuplicate = {
+  name: 'check_duplicate',
+  description:
+    '候选查重 — 检测候选代码是否与已有 Recipe 重复（基于标题/摘要/代码的 Jaccard 相似度）。',
+  parameters: {
+    type: 'object',
+    properties: {
+      candidate: { type: 'object', description: '候选对象 { title, summary, code, usageGuide }' },
+      candidateId: { type: 'string', description: '或提供候选 ID，从数据库读取' },
+      projectRoot: { type: 'string', description: '项目根目录（可选，默认当前项目）' },
+      threshold: { type: 'number', description: '相似度阈值，默认 0.5' },
+    },
+  },
+  handler: async (params, ctx) => {
+    let cand = params.candidate;
+    const projectRoot = params.projectRoot || ctx.projectRoot;
+    const threshold = params.threshold ?? 0.5;
+
+    // 如果提供 candidateId，从数据库读取条目信息
+    if (!cand && params.candidateId) {
+      try {
+        const knowledgeService = ctx.container.get('knowledgeService');
+        const found = await knowledgeService.get(params.candidateId);
+        if (found) {
+          const json = typeof found.toJSON === 'function' ? found.toJSON() : found;
+          cand = {
+            title: json.title || '',
+            summary: json.description || '',
+            code: json.content?.pattern || '',
+            usageGuide: '',
+          };
+        }
+      } catch {
+        /* ignore */
+      }
+    }
+
+    if (!cand) {
+      return { similar: [], message: 'No candidate provided' };
+    }
+
+    const similar = findSimilarRecipes(projectRoot, cand, {
+      threshold,
+      topK: 10,
+    });
+
+    return {
+      similar,
+      hasDuplicate: similar.some((s) => s.similarity >= 0.7),
+      highestSimilarity: similar.length > 0 ? similar[0].similarity : 0,
+      _meta: {
+        confidence: similar.length === 0 ? 'none' : similar[0].similarity >= 0.7 ? 'high' : 'low',
+        hint:
+          similar.length === 0
+            ? '未发现相似 Recipe，可放心提交。'
+            : similar[0].similarity >= 0.7
+              ? '发现高度相似 Recipe，建议人工审核是否重复。'
+              : '有低相似度匹配，大概率不是重复。',
+      },
+    };
+  },
+};
+
+// ────────────────────────────────────────────────────────────
+// 11. discover_relations
+// ────────────────────────────────────────────────────────────
+export const discoverRelations = {
+  name: 'discover_relations',
+  description:
+    'AI 知识图谱关系发现 — 分析 Recipe 对之间的潜在关系（requires/extends/enforces/calls 等），并自动写入知识图谱。',
+  parameters: {
+    type: 'object',
+    properties: {
+      recipePairs: {
+        type: 'array',
+        items: {
+          type: 'object',
+          properties: {
+            a: {
+              type: 'object',
+              properties: {
+                id: { type: 'string' },
+                title: { type: 'string' },
+                category: { type: 'string' },
+                code: { type: 'string' },
+              },
+            },
+            b: {
+              type: 'object',
+              properties: {
+                id: { type: 'string' },
+                title: { type: 'string' },
+                category: { type: 'string' },
+                code: { type: 'string' },
+              },
+            },
+          },
+        },
+        description:
+          'Recipe 对数组 [{ a: {id, title, category, code}, b: {id, title, category, code} }]',
+      },
+      dryRun: { type: 'boolean', description: '仅分析不写入，默认 false' },
+    },
+    required: ['recipePairs'],
+  },
+  handler: async (params, ctx) => {
+    if (!ctx.aiProvider) {
+      return { error: 'AI provider not available' };
+    }
+
+    const { recipePairs, dryRun = false } = params;
+    if (!recipePairs || recipePairs.length === 0) {
+      return { relations: [] };
+    }
+
+    // 构建 LLM prompt
+    const pairsText = recipePairs
+      .map(
+        (p, i) => `
+--- Pair #${i + 1} ---
+Recipe A [${p.a.id}]: ${p.a.title} (${p.a.category}/${p.a.language || ''})
+${p.a.code ? `Code: ${p.a.code.substring(0, 300)}` : ''}
+
+Recipe B [${p.b.id}]: ${p.b.title} (${p.b.category}/${p.b.language || ''})
+${p.b.code ? `Code: ${p.b.code.substring(0, 300)}` : ''}`
+      )
+      .join('\n');
+
+    const prompt = `# Role
+You are a Software Architect analyzing relationships between code recipes (knowledge units).
+
+# Goal
+For each Recipe pair below, determine if there is a meaningful relationship.
+
+# Relationship Types
+- requires: A needs B to function
+- extends: A builds upon / enriches B
+- enforces: A enforces rules defined in B
+- depends_on: A depends on B
+- inherits: A inherits from B (class/protocol)
+- implements: A implements interface/protocol defined in B
+- calls: A calls API defined in B
+- prerequisite: B must be learned/applied before A
+- none: No meaningful relationship
+
+# Output
+Return a JSON array. For each pair with a relationship (skip "none"):
+{ "index": 0, "from_id": "...", "to_id": "...", "relation": "requires", "confidence": 0.85, "reason": "A uses the network client defined in B" }
+
+Return ONLY a JSON array. No markdown, no extra text. Return [] if no relationships found.
+
+# Recipe Pairs
+${pairsText}`;
+
+    const parsed = await ctx.aiProvider.chatWithStructuredOutput(prompt, {
+      openChar: '[',
+      closeChar: ']',
+      temperature: 0.2,
+    });
+    const relations = Array.isArray(parsed) ? parsed : [];
+
+    // 写入知识图谱（除非 dryRun）
+    if (!dryRun && relations.length > 0) {
+      try {
+        const kgService = ctx.container.get('knowledgeGraphService');
+        for (const rel of relations) {
+          if (rel.from_id && rel.to_id && rel.relation && rel.relation !== 'none') {
+            kgService.addEdge(rel.from_id, 'recipe', rel.to_id, 'recipe', rel.relation, {
+              confidence: rel.confidence || 0.5,
+              reason: rel.reason || '',
+              source: 'ai-discovery',
+            });
+          }
+        }
+      } catch {
+        /* KG not available */
+      }
+    }
+
+    return {
+      analyzed: recipePairs.length,
+      relations: relations.filter((r) => r.relation !== 'none'),
+      written: dryRun ? 0 : relations.filter((r) => r.relation !== 'none').length,
+    };
+  },
+};
+
+// ────────────────────────────────────────────────────────────
+// 12. add_graph_edge
+// ────────────────────────────────────────────────────────────
+export const addGraphEdge = {
+  name: 'add_graph_edge',
+  description: '手动添加知识图谱关系边（从 A 到 B 的关系）。',
+  parameters: {
+    type: 'object',
+    properties: {
+      fromId: { type: 'string', description: '源节点 ID' },
+      fromType: { type: 'string', description: '源节点类型 (recipe/candidate)' },
+      toId: { type: 'string', description: '目标节点 ID' },
+      toType: { type: 'string', description: '目标节点类型 (recipe/candidate)' },
+      relation: {
+        type: 'string',
+        description:
+          '关系类型 (requires/extends/enforces/depends_on/inherits/implements/calls/prerequisite)',
+      },
+      weight: { type: 'number', description: '权重 0-1，默认 1.0' },
+    },
+    required: ['fromId', 'fromType', 'toId', 'toType', 'relation'],
+  },
+  handler: async (params, ctx) => {
+    const kgService = ctx.container.get('knowledgeGraphService');
+    return kgService.addEdge(
+      params.fromId,
+      params.fromType,
+      params.toId,
+      params.toType,
+      params.relation,
+      { weight: params.weight || 1.0, source: 'manual' }
+    );
+  },
+};