mcp-probe-kit 3.0.19 → 3.0.22

package/build/lib/memory-orchestration.js

@@ -1,5 +1,35 @@
 import { createMemoryClient } from './memory-client.js';
-export async function loadMemoryInjectionContext(query) {
+import { getMemoryConfig } from './memory-config.js';
+function kindSearchPreferences(kind) {
+    switch (kind) {
+        case 'bugfix':
+            return { preferTypes: ['bugfix'], preferTags: ['bugfix', 'root-cause'] };
+        case 'ui':
+            return { preferTypes: ['component', 'pattern'], preferTags: ['ui', 'pattern'] };
+        case 'feature':
+            return { preferTypes: ['pattern', 'code'], preferTags: ['feature', 'pattern'] };
+        default:
+            return { preferTypes: [], preferTags: [] };
+    }
+}
+export function truncateInjectionText(value, maxChars) {
+    if (value.length <= maxChars) {
+        return value;
+    }
+    return `${value.slice(0, Math.max(0, maxChars - 3))}...`;
+}
+async function loadFullAssets(results) {
+    const client = createMemoryClient();
+    if (!client.isReadEnabled() || results.length === 0) {
+        return {};
+    }
+    const entries = await Promise.all(results.map(async (item) => {
+        const asset = await client.getAsset(item.id);
+        return asset ? [item.id, asset] : null;
+    }));
+    return Object.fromEntries(entries.filter((entry) => entry !== null));
+}
+export async function loadMemoryInjectionContext(query, kind = 'default') {
     const client = createMemoryClient();
     if (!client.isEnabled()) {
         return {
@@ -8,16 +38,23 @@ export async function loadMemoryInjectionContext(query) {
             degraded: false,
             query,
             results: [],
+            assetsById: {},
         };
     }
     try {
-        const results = await client.search(query);
+        const prefs = kindSearchPreferences(kind);
+        const results = await client.search(query, {
+            preferTypes: prefs.preferTypes,
+            preferTags: prefs.preferTags,
+        });
+        const assetsById = await loadFullAssets(results);
         return {
             enabled: true,
             available: true,
             degraded: false,
             query,
             results,
+            assetsById,
         };
     }
     catch (error) {
@@ -27,6 +64,7 @@ export async function loadMemoryInjectionContext(query) {
             degraded: true,
             query,
             results: [],
+            assetsById: {},
             error: error instanceof Error ? error.message : String(error),
         };
     }
@@ -39,7 +77,44 @@ function formatMemoryResultLabel(item) {
             : '历史资产';
     return `${item.name} [${item.type}] (${kind})`;
 }
+export function shouldShowSourceInSearch(item, config = getMemoryConfig()) {
+    if (config.searchShowSource) {
+        return Boolean(item.sourcePath);
+    }
+    if (!config.repoId || !item.sourceProject || !item.sourcePath) {
+        return false;
+    }
+    return item.sourceProject === config.repoId;
+}
+function formatSourceHint(item, config) {
+    if (!shouldShowSourceInSearch(item, config)) {
+        return '';
+    }
+    return `\n   - 来源: ${item.sourcePath}`;
+}
+function formatAssetBody(asset, config) {
+    const lines = [
+        `### ${asset.name}`,
+        `- asset_id: ${asset.id}`,
+        asset.description ? `- 描述: ${asset.description}` : '',
+        asset.usage ? `- 适用: ${asset.usage}` : '',
+        asset.tags.length > 0 ? `- 标签: ${asset.tags.join(', ')}` : '',
+        '',
+        truncateInjectionText(asset.content, config.injectionContentMaxChars),
+    ].filter(Boolean);
+    return lines.join('\n');
+}
+function formatResultBlock(item, index, context, config) {
+    const label = formatMemoryResultLabel(item);
+    const asset = context.assetsById[item.id];
+    const header = `${index + 1}. ${label} score=${item.score.toFixed(3)}\n   - 摘要: ${item.summary}${formatSourceHint(item, config)}`;
+    if (asset?.content) {
+        return `${header}\n\n${formatAssetBody(asset, config)}\n`;
+    }
+    return `${header}\n   - 全文加载失败，可手动: read_memory_asset {"asset_id": "${item.id}"}\n`;
+}
 export function renderMemoryGuideSection(context) {
+    const config = getMemoryConfig();
     if (!context.enabled) {
         return '';
     }
@@ -49,13 +124,11 @@ export function renderMemoryGuideSection(context) {
     if (context.results.length === 0) {
         return `\n\n## 🧠 记忆系统\n- 状态: 已启用\n- 检索结果: 未找到高相关记录（含历史 Bug 修复与可复用模式）\n- 处理: 继续主流程；Bug 修复验证通过后必须 \`memorize_asset\` 沉淀；功能/UI 有可复用产出再沉淀\n`;
     }
+    const loadedCount = context.results.filter((item) => context.assetsById[item.id]?.content).length;
     const items = context.results
-        .map((item, index) => {
-        const label = formatMemoryResultLabel(item);
-        return `${index + 1}. ${label} score=${item.score.toFixed(3)}\n   - 摘要: ${item.summary}\n   - 读取: read_memory_asset {\"asset_id\": \"${item.id}\"}${item.sourcePath ? `\n   - 来源: ${item.sourcePath}` : ''}`;
-    })
+        .map((item, index) => formatResultBlock(item, index, context, config))
         .join('\n');
-    return `\n\n## 🧠 记忆系统\n- 状态: 已启用\n- 指令: 开干前先复用下列记录（含历史 Bug 现象/根因/改法）；相关条目用 \`read_memory_asset\` 读全文，避免重复踩坑\n- 检索结果:\n${items}\n`;
+    return `\n\n## 🧠 记忆系统\n- 状态: 已启用\n- 指令: 下列为已自动加载的历史经验全文（${loadedCount}/${context.results.length} 条）；开干前直接复用，无需再调 \`read_memory_asset\`\n- 检索结果:\n${items}`;
 }
 export function buildMemoryPlanStep(kind = 'default') {
     if (kind === 'bugfix') {
@@ -68,7 +141,7 @@ export function buildMemoryPlanStep(kind = 'default') {
                 type: 'bugfix',
                 description: '[现象、报错信息、复现条件]',
                 summary: '[检索用：关键词 + 根因 + 修复要点，一句话]',
-                content: '【现象】...\n【根因】...\n【修复】具体改动文件与关键代码/配置\n【验证】如何确认已修好',
+                content: '【现象】...\n【根因】...\n【修复】具体改动与关键代码/配置\n【验证】如何确认已修好',
                 usage: '[再次遇到何种症状时可参考]',
                 tags: ['bugfix', 'root-cause'],
                 confidence: 0.85,

package/build/lib/memory-payload.d.ts

@@ -0,0 +1,21 @@
+/**
+ * Unify legacy Qdrant payloads (kind/title/source) with the current asset schema.
+ */
+export declare function normalizeMemoryPayload(payload: Record<string, unknown>): Record<string, unknown>;
+export declare function payloadToMemoryFields(payload: Record<string, unknown>): {
+    id: string;
+    name: string;
+    type: string;
+    description: string;
+    summary: string;
+    content: string;
+    tags: string[];
+    confidence: number;
+    sourceProject?: string;
+    sourcePath?: string;
+    usage?: string;
+    contentHash?: string;
+    normalizedContentHash?: string;
+    createdAt: string;
+    updatedAt: string;
+};

package/build/lib/memory-payload.js

@@ -0,0 +1,65 @@
+function truncate(value, maxChars) {
+    if (value.length <= maxChars) {
+        return value;
+    }
+    return `${value.slice(0, Math.max(0, maxChars - 3))}...`;
+}
+/**
+ * Unify legacy Qdrant payloads (kind/title/source) with the current asset schema.
+ */
+export function normalizeMemoryPayload(payload) {
+    const normalized = { ...payload };
+    if (!normalized.name && typeof normalized.title === 'string') {
+        normalized.name = normalized.title;
+    }
+    if (!normalized.type && typeof normalized.kind === 'string') {
+        const kind = normalized.kind;
+        normalized.type = kind === 'extracted_pattern' ? 'pattern' : kind;
+    }
+    if (!normalized.description) {
+        if (typeof normalized.title === 'string') {
+            normalized.description = normalized.title;
+        }
+        else if (typeof normalized.source === 'string') {
+            normalized.description = `Legacy memory from ${normalized.source}`;
+        }
+    }
+    if (!normalized.summary && typeof normalized.content === 'string') {
+        normalized.summary = truncate(normalized.content, 280);
+    }
+    if (!normalized.createdAt && typeof normalized.created_at === 'string') {
+        normalized.createdAt = normalized.created_at;
+    }
+    if (!normalized.updatedAt) {
+        if (typeof normalized.updated_at === 'string') {
+            normalized.updatedAt = normalized.updated_at;
+        }
+        else if (typeof normalized.createdAt === 'string') {
+            normalized.updatedAt = normalized.createdAt;
+        }
+    }
+    return normalized;
+}
+export function payloadToMemoryFields(payload) {
+    const p = normalizeMemoryPayload(payload);
+    const tags = Array.isArray(p.tags)
+        ? p.tags.filter((item) => typeof item === 'string' && item.trim().length > 0)
+        : [];
+    return {
+        id: String(p.id || ''),
+        name: String(p.name || ''),
+        type: String(p.type || ''),
+        description: String(p.description || ''),
+        summary: String(p.summary || ''),
+        content: String(p.content || ''),
+        tags,
+        confidence: typeof p.confidence === 'number' && Number.isFinite(p.confidence) ? p.confidence : 0.5,
+        sourceProject: typeof p.sourceProject === 'string' ? p.sourceProject : undefined,
+        sourcePath: typeof p.sourcePath === 'string' ? p.sourcePath : undefined,
+        usage: typeof p.usage === 'string' ? p.usage : undefined,
+        contentHash: typeof p.contentHash === 'string' ? p.contentHash : undefined,
+        normalizedContentHash: typeof p.normalizedContentHash === 'string' ? p.normalizedContentHash : undefined,
+        createdAt: String(p.createdAt || ''),
+        updatedAt: String(p.updatedAt || ''),
+    };
+}

package/build/lib/shadcn-ui.d.ts

@@ -0,0 +1,11 @@
+/**
+ * 判断是否适合使用 shadcn/ui 实现路径
+ */
+export declare function isShadcnStack(stack?: string): boolean;
+export declare function isShadcnCategory(category?: string): boolean;
+export declare function isThemeCategory(category?: string): boolean;
+export declare function isGuidelineCategory(category?: string): boolean;
+export declare function formatShadcnResult(data: Record<string, any>): string;
+export declare function formatThemeResult(data: Record<string, any>): string;
+export declare function formatGuidelineResult(data: Record<string, any>): string;
+export declare function pickThemeForProductType(themes: Record<string, any>[], productType: string): Record<string, any> | undefined;

package/build/lib/shadcn-ui.js

@@ -0,0 +1,78 @@
+/**
+ * 判断是否适合使用 shadcn/ui 实现路径
+ */
+export function isShadcnStack(stack) {
+    if (!stack)
+        return false;
+    const normalized = stack.toLowerCase();
+    return (normalized.includes('react') ||
+        normalized.includes('next') ||
+        normalized.includes('shadcn') ||
+        normalized.includes('tailwind'));
+}
+export function isShadcnCategory(category) {
+    return Boolean(category?.startsWith('shadcn-'));
+}
+export function isThemeCategory(category) {
+    return category === 'ui-themes';
+}
+export function isGuidelineCategory(category) {
+    return category === 'ui-guidelines-vercel';
+}
+export function formatShadcnResult(data) {
+    const lines = [
+        `- **名称**: ${data.name || data.title}`,
+        `- **类型**: ${data.type || 'unknown'}`,
+        `- **描述**: ${data.description || '—'}`,
+        `- **安装**: \`${data.installCommand || `npx shadcn@latest add ${data.name}`}\``,
+    ];
+    if (Array.isArray(data.registryDependencies) && data.registryDependencies.length > 0) {
+        lines.push(`- **依赖组件**: ${data.registryDependencies.join(', ')}`);
+    }
+    if (Array.isArray(data.files) && data.files.length > 0) {
+        lines.push(`- **文件**: ${data.files.slice(0, 5).join(', ')}${data.files.length > 5 ? '…' : ''}`);
+    }
+    return lines.join('\n');
+}
+export function formatThemeResult(data) {
+    const lines = [
+        `- **主题**: ${data.title || data.name}`,
+        `- **描述**: ${data.description || '—'}`,
+        `- **基色**: ${data.baseColor || '—'}`,
+        `- **适合**: ${Array.isArray(data.bestFor) ? data.bestFor.join(', ') : '—'}`,
+        `- **用法**: 将 \`globalsCssSnippet\` 粘贴到 \`app/globals.css\`（shadcn new-york）`,
+    ];
+    return lines.join('\n');
+}
+export function formatGuidelineResult(data) {
+    return [
+        `- **级别**: ${data.level || '—'}`,
+        `- **章节**: ${data.section || '—'} / ${data.subsection || '—'}`,
+        `- **规则**: ${data.rule || data.description || '—'}`,
+        `- **来源**: Vercel Web Interface Guidelines`,
+    ].join('\n');
+}
+export function pickThemeForProductType(themes, productType) {
+    if (themes.length === 0)
+        return undefined;
+    const normalized = productType.toLowerCase();
+    const scored = themes.map((theme) => {
+        const bestFor = Array.isArray(theme.bestFor) ? theme.bestFor : [];
+        const score = bestFor.reduce((acc, item) => {
+            const token = item.toLowerCase();
+            if (normalized.includes(token) || token.includes(normalized)) {
+                return acc + 2;
+            }
+            if (normalized.split(/\s+/).some((part) => token.includes(part) && part.length > 2)) {
+                return acc + 1;
+            }
+            return acc;
+        }, 0);
+        return { theme, score };
+    });
+    scored.sort((a, b) => b.score - a.score);
+    if (scored[0]?.score > 0) {
+        return scored[0].theme;
+    }
+    return themes.find((theme) => theme.name === 'zinc-neutral') || themes[0];
+}