mn-docs-mcp 0.3.0 → 0.3.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/mcp/lib.mjs +26 -3
- package/package.json +1 -1
package/mcp/lib.mjs
CHANGED
|
@@ -75,7 +75,15 @@ function formatBytes(bytes) {
|
|
|
75
75
|
function logDownloadProgress(info) {
|
|
76
76
|
if (IS_SILENT) return;
|
|
77
77
|
if (info?.status === 'download') {
|
|
78
|
-
|
|
78
|
+
// 清除当前行（如果之前有内容）
|
|
79
|
+
if (IS_STDIO) {
|
|
80
|
+
process.stderr.write('\r\x1b[K'); // 清除整行
|
|
81
|
+
process.stderr.write(color('开始下载模型...', '38;5;45') + '\n');
|
|
82
|
+
} else {
|
|
83
|
+
process.stdout.write('\r\x1b[K'); // 清除整行
|
|
84
|
+
console.log(color('开始下载模型...', '38;5;45'));
|
|
85
|
+
}
|
|
86
|
+
lastDownloadProgress = -1;
|
|
79
87
|
return;
|
|
80
88
|
}
|
|
81
89
|
if (info?.status === 'progress' && typeof info.progress === 'number') {
|
|
@@ -87,10 +95,10 @@ function logDownloadProgress(info) {
|
|
|
87
95
|
const suffix = loaded && total ? ` ${loaded}/${total}` : '';
|
|
88
96
|
const line = `${color('模型下载进度', '38;5;45')}: ${pct}%${suffix}`;
|
|
89
97
|
if (IS_STDIO) {
|
|
90
|
-
process.stderr.write(`\r${line}`);
|
|
98
|
+
process.stderr.write(`\r\x1b[K${line}`); // \x1b[K 清除从光标到行尾的内容
|
|
91
99
|
if (pct === 100) process.stderr.write('\n');
|
|
92
100
|
} else {
|
|
93
|
-
process.stdout.write(`\r${line}`);
|
|
101
|
+
process.stdout.write(`\r\x1b[K${line}`);
|
|
94
102
|
if (pct === 100) process.stdout.write('\n');
|
|
95
103
|
}
|
|
96
104
|
}
|
|
@@ -117,11 +125,23 @@ async function getExtractor() {
|
|
|
117
125
|
if (extractorPromise) return extractorPromise;
|
|
118
126
|
loadEnv();
|
|
119
127
|
setupProxy();
|
|
128
|
+
|
|
129
|
+
// 抑制 Hugging Face Transformers 的警告输出
|
|
130
|
+
env.cacheDir = path.join(MCP_DIR, 'models');
|
|
120
131
|
env.allowRemoteModels = true;
|
|
132
|
+
env.disableProgressBars = true; // 禁用库自带的进度条
|
|
133
|
+
env.disableSymlinksWarning = true; // 禁用符号链接警告
|
|
134
|
+
|
|
135
|
+
// 设置日志级别为 error，避免 info/warning 级别日志干扰
|
|
136
|
+
if (!process.env.LOG_LEVEL) {
|
|
137
|
+
process.env.LOG_LEVEL = 'error';
|
|
138
|
+
}
|
|
139
|
+
|
|
121
140
|
if (process.env.HF_ENDPOINT) {
|
|
122
141
|
env.HF_ENDPOINT = process.env.HF_ENDPOINT;
|
|
123
142
|
}
|
|
124
143
|
|
|
144
|
+
const modelDir = path.join(env.cacheDir, 'Xenova', 'bge-small-zh-v1.5');
|
|
125
145
|
const create = async () =>
|
|
126
146
|
pipeline('feature-extraction', MODEL_ID, {
|
|
127
147
|
progress_callback: logDownloadProgress,
|
|
@@ -143,7 +163,10 @@ async function getExtractor() {
|
|
|
143
163
|
throw error;
|
|
144
164
|
}
|
|
145
165
|
|
|
166
|
+
// 清除上次的进度状态，为重试做准备
|
|
167
|
+
lastDownloadProgress = -1;
|
|
146
168
|
logInfo(`模型下载失败，准备重试(${attempt}/${MAX_EXTRACTOR_RETRIES})...`);
|
|
169
|
+
await fs.rm(modelDir, { recursive: true, force: true });
|
|
147
170
|
}
|
|
148
171
|
}
|
|
149
172
|
throw new Error('模型加载失败');
|