@ppdocs/mcp 2.6.26 → 2.6.28

package/dist/vector/index.d.ts

@@ -1,6 +1,6 @@
 /**
  * 向量检索服务 - 基于 transformers.js 的语义搜索
- * @version 0.1
+ * @version 0.2
  */
 interface VectorSearchResult {
     id: string;

package/dist/vector/index.js

@@ -1,13 +1,32 @@
 /**
  * 向量检索服务 - 基于 transformers.js 的语义搜索
- * @version 0.1
+ * @version 0.2
  */
 import { pipeline } from '@xenova/transformers';
 import * as fs from 'fs';
 import * as path from 'path';
-// 模型配置
-const MODEL_NAME = 'Xenova/all-MiniLM-L6-v2'; // 23MB, 384维, 中英双语
+// 默认模型配置
+const DEFAULT_MODEL = 'Xenova/multilingual-e5-small';
 const VECTOR_DIM = 384;
+// 当前加载的模型名
+let currentModel = null;
+/**
+ * 获取向量配置 (从 ~/.ppdocs/config/vector.json 读取)
+ */
+function getVectorConfig() {
+    const baseDir = process.env.PPDOCS_DATA_DIR || path.join(process.env.HOME || process.env.USERPROFILE || '', '.ppdocs');
+    const configPath = path.join(baseDir, 'config', 'vector.json');
+    try {
+        if (fs.existsSync(configPath)) {
+            const content = fs.readFileSync(configPath, 'utf-8');
+            return JSON.parse(content);
+        }
+    }
+    catch (e) {
+        console.warn('[Vector] Failed to load config:', e);
+    }
+    return { model: DEFAULT_MODEL, quantized: true };
+}
 // 单例模式
 let extractor = null;
 let isLoading = false;
@@ -18,15 +37,19 @@ const vectorIndex = new Map();
  * 加载嵌入模型 (懒加载, 首次调用时下载)
  */
 async function getExtractor() {
-    if (extractor)
+    const config = getVectorConfig();
+    // 如果模型变更，重新加载
+    if (extractor && currentModel === config.model) {
         return extractor;
-    if (isLoading && loadPromise) {
+    }
+    if (isLoading && loadPromise && currentModel === config.model) {
         return loadPromise;
     }
     isLoading = true;
-    console.log('[Vector] Loading embedding model...');
-    loadPromise = pipeline('feature-extraction', MODEL_NAME, {
-        quantized: true, // 使用量化版本，更小更快
+    currentModel = config.model;
+    console.log(`[Vector] Loading embedding model: ${config.model}...`);
+    loadPromise = pipeline('feature-extraction', config.model, {
+        quantized: config.quantized,
     });
     extractor = await loadPromise;
     isLoading = false;
@@ -100,9 +123,10 @@ async function saveIndex(projectId) {
         if (!fs.existsSync(dir)) {
             fs.mkdirSync(dir, { recursive: true });
         }
+        const config = getVectorConfig();
         fs.writeFileSync(indexPath, JSON.stringify({
             version: 1,
-            model: MODEL_NAME,
+            model: config.model,
             dimension: VECTOR_DIM,
             updatedAt: new Date().toISOString(),
             entries
@@ -188,9 +212,10 @@ export async function buildIndex(projectId, nodes) {
  */
 export async function getIndexStats(projectId) {
     const entries = await loadIndex(projectId);
+    const config = getVectorConfig();
     return {
         count: entries.length,
-        model: MODEL_NAME,
+        model: config.model,
         dimension: VECTOR_DIM,
         loaded: extractor !== null
     };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ppdocs/mcp",
-  "version": "2.6.26",
+  "version": "2.6.28",
   "description": "ppdocs MCP Server - Knowledge Graph for Claude",
   "type": "module",
   "main": "dist/index.js",