@dsh-plugin/dsh-synapse 0.0.0-stage → 0.1.0-dev.41044228975
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/LICENSE +661 -0
- package/NOTICE.md +18 -0
- package/README.md +136 -2
- package/cordis.patch.yml +3 -0
- package/lib/bot.cjs +8314 -0
- package/lib/bot.cjs.map +7 -0
- package/lib/build.json +1 -0
- package/lib/client.js +46 -0
- package/lib/host/agent.d.ts +21 -0
- package/lib/host/agent.js +151 -0
- package/lib/host/agent.js.map +1 -0
- package/lib/host/bot.d.ts +17 -0
- package/lib/host/bot.js +158 -0
- package/lib/host/bot.js.map +1 -0
- package/lib/host/client.d.ts +2 -0
- package/lib/host/client.js +19 -0
- package/lib/host/client.js.map +1 -0
- package/lib/host/config.d.ts +24 -0
- package/lib/host/config.js +192 -0
- package/lib/host/config.js.map +1 -0
- package/lib/host/files.d.ts +29 -0
- package/lib/host/files.js +188 -0
- package/lib/host/files.js.map +1 -0
- package/lib/host/gateway.d.ts +33 -0
- package/lib/host/gateway.js +237 -0
- package/lib/host/gateway.js.map +1 -0
- package/lib/host/github.d.ts +47 -0
- package/lib/host/github.js +219 -0
- package/lib/host/github.js.map +1 -0
- package/lib/host/http.d.ts +4 -0
- package/lib/host/http.js +41 -0
- package/lib/host/http.js.map +1 -0
- package/lib/host/index.d.ts +32 -0
- package/lib/host/index.js +192 -0
- package/lib/host/index.js.map +1 -0
- package/lib/host/inference-worker.d.ts +1 -0
- package/lib/host/inference-worker.js +72 -0
- package/lib/host/inference-worker.js.map +1 -0
- package/lib/host/mcp.d.ts +138 -0
- package/lib/host/mcp.js +104 -0
- package/lib/host/mcp.js.map +1 -0
- package/lib/host/memory.d.ts +26 -0
- package/lib/host/memory.js +155 -0
- package/lib/host/memory.js.map +1 -0
- package/lib/host/panel.d.ts +1 -0
- package/lib/host/panel.html +174 -0
- package/lib/host/panel.js +388 -0
- package/lib/host/panel.js.map +1 -0
- package/lib/host/runtime.d.ts +21 -0
- package/lib/host/runtime.js +145 -0
- package/lib/host/runtime.js.map +1 -0
- package/lib/host/types.d.ts +112 -0
- package/lib/host/types.js +2 -0
- package/lib/host/types.js.map +1 -0
- package/package.json +94 -4
- package/scripts/build.mjs +90 -0
- package/scripts/check.mjs +13 -0
- package/scripts/dump-npm-latest.mjs +30 -0
- package/scripts/model-smoke.mjs +25 -0
- package/src/agent.ts +203 -0
- package/src/bot.ts +173 -0
- package/src/client.ts +20 -0
- package/src/config.ts +217 -0
- package/src/files.ts +206 -0
- package/src/gateway.ts +276 -0
- package/src/github.ts +263 -0
- package/src/http.ts +41 -0
- package/src/index.ts +227 -0
- package/src/inference-worker.ts +89 -0
- package/src/mcp.ts +112 -0
- package/src/memory.ts +178 -0
- package/src/panel.html +174 -0
- package/src/panel.ts +440 -0
- package/src/runtime.ts +164 -0
- package/src/types.ts +106 -0
- package/test/core.test.mjs +279 -0
- package/test/fixtures/mcp-server.mjs +14 -0
- package/test/integration.test.mjs +253 -0
- package/test/panel.test.mjs +73 -0
- package/tsconfig.json +18 -0
- package/vendor/gugle-chat-bot/LICENSE +661 -0
- package/vendor/gugle-chat-bot/package.json +57 -0
- package/vendor/gugle-chat-bot/src/allow-rule.yml +9345 -0
- package/vendor/gugle-chat-bot/src/api.json +4285 -0
- package/vendor/gugle-chat-bot/src/command/index.ts +141 -0
- package/vendor/gugle-chat-bot/src/config/features.ts +548 -0
- package/vendor/gugle-chat-bot/src/config/index.ts +184 -0
- package/vendor/gugle-chat-bot/src/config/manager.ts +357 -0
- package/vendor/gugle-chat-bot/src/constants/index.ts +13 -0
- package/vendor/gugle-chat-bot/src/custom.ts +548 -0
- package/vendor/gugle-chat-bot/src/event/index.ts +118 -0
- package/vendor/gugle-chat-bot/src/features/admin/index.ts +414 -0
- package/vendor/gugle-chat-bot/src/features/bili/image/impl/video.ts +39 -0
- package/vendor/gugle-chat-bot/src/features/bili/image/index.ts +37 -0
- package/vendor/gugle-chat-bot/src/features/bili/index.ts +65 -0
- package/vendor/gugle-chat-bot/src/features/bili/template/style/video.css +97 -0
- package/vendor/gugle-chat-bot/src/features/bili/template/video.html +57 -0
- package/vendor/gugle-chat-bot/src/features/discord-bridge/index.ts +1128 -0
- package/vendor/gugle-chat-bot/src/features/discord-bridge/markdown.ts +105 -0
- package/vendor/gugle-chat-bot/src/features/discord-bridge/slash.ts +355 -0
- package/vendor/gugle-chat-bot/src/features/discord-bridge/source.ts +96 -0
- package/vendor/gugle-chat-bot/src/features/discord-bridge/template/message.html +14 -0
- package/vendor/gugle-chat-bot/src/features/discord-bridge/template/style/message.css +144 -0
- package/vendor/gugle-chat-bot/src/features/github/api.ts +93 -0
- package/vendor/gugle-chat-bot/src/features/github/binding.ts +161 -0
- package/vendor/gugle-chat-bot/src/features/github/image/impl/issue.ts +132 -0
- package/vendor/gugle-chat-bot/src/features/github/image/impl/pr.ts +144 -0
- package/vendor/gugle-chat-bot/src/features/github/image/impl/release.ts +160 -0
- package/vendor/gugle-chat-bot/src/features/github/image/index.ts +310 -0
- package/vendor/gugle-chat-bot/src/features/github/index.ts +302 -0
- package/vendor/gugle-chat-bot/src/features/github/template/issue.html +118 -0
- package/vendor/gugle-chat-bot/src/features/github/template/pull_request.html +125 -0
- package/vendor/gugle-chat-bot/src/features/github/template/release.html +53 -0
- package/vendor/gugle-chat-bot/src/features/github/template/style/issue.css +468 -0
- package/vendor/gugle-chat-bot/src/features/github/template/style/pull_request.css +469 -0
- package/vendor/gugle-chat-bot/src/features/github/template/style/release.css +412 -0
- package/vendor/gugle-chat-bot/src/features/hermes/client.ts +184 -0
- package/vendor/gugle-chat-bot/src/features/hermes/config.ts +210 -0
- package/vendor/gugle-chat-bot/src/features/hermes/context.ts +67 -0
- package/vendor/gugle-chat-bot/src/features/hermes/history.ts +86 -0
- package/vendor/gugle-chat-bot/src/features/hermes/index.ts +1533 -0
- package/vendor/gugle-chat-bot/src/features/hermes/message.ts +319 -0
- package/vendor/gugle-chat-bot/src/features/hermes/reaction.ts +181 -0
- package/vendor/gugle-chat-bot/src/features/hermes/render.ts +118 -0
- package/vendor/gugle-chat-bot/src/features/hermes/reply-units.ts +331 -0
- package/vendor/gugle-chat-bot/src/features/hermes/skills.ts +406 -0
- package/vendor/gugle-chat-bot/src/features/hermes/template/approval.html +30 -0
- package/vendor/gugle-chat-bot/src/features/hermes/template/progress.html +21 -0
- package/vendor/gugle-chat-bot/src/features/hermes/template/style/approval.css +81 -0
- package/vendor/gugle-chat-bot/src/features/hermes/template/style/progress.css +120 -0
- package/vendor/gugle-chat-bot/src/features/hermes/trigger.ts +86 -0
- package/vendor/gugle-chat-bot/src/features/hermes/types.ts +327 -0
- package/vendor/gugle-chat-bot/src/features/management/index.ts +156 -0
- package/vendor/gugle-chat-bot/src/features/minecraft/index.ts +127 -0
- package/vendor/gugle-chat-bot/src/features/modrinth/index.ts +71 -0
- package/vendor/gugle-chat-bot/src/features/parentheses/index.ts +117 -0
- package/vendor/gugle-chat-bot/src/features/peak-valley-timer/index.ts +452 -0
- package/vendor/gugle-chat-bot/src/features/poke/index.ts +24 -0
- package/vendor/gugle-chat-bot/src/features/version-tracker/index.ts +78 -0
- package/vendor/gugle-chat-bot/src/features/welcome/index.ts +58 -0
- package/vendor/gugle-chat-bot/src/github-ip.json +5814 -0
- package/vendor/gugle-chat-bot/src/image/index.ts +278 -0
- package/vendor/gugle-chat-bot/src/index.ts +695 -0
- package/vendor/gugle-chat-bot/src/logger/index.ts +54 -0
- package/vendor/gugle-chat-bot/src/result.json +359 -0
- package/vendor/gugle-chat-bot/src/type/bili.d.ts +825 -0
- package/vendor/gugle-chat-bot/src/type/event.d.ts +146 -0
- package/vendor/gugle-chat-bot/src/type/github.d.ts +446 -0
- package/vendor/gugle-chat-bot/src/type/index.d.ts +254 -0
- package/vendor/gugle-chat-bot/tsconfig.json +46 -0
|
@@ -0,0 +1,388 @@
|
|
|
1
|
+
const $ = (id) => document.getElementById(id);
|
|
2
|
+
function node(tag, text, attrs = {}) {
|
|
3
|
+
const n = document.createElement(tag);
|
|
4
|
+
if (text !== undefined)
|
|
5
|
+
n.textContent = text;
|
|
6
|
+
Object.assign(n, attrs);
|
|
7
|
+
return n;
|
|
8
|
+
}
|
|
9
|
+
function notify(text, error = false) {
|
|
10
|
+
$("notice").textContent = text;
|
|
11
|
+
$("notice").className = error ? "error" : "";
|
|
12
|
+
}
|
|
13
|
+
async function request(path, data) {
|
|
14
|
+
const r = await fetch(path, {
|
|
15
|
+
cache: "no-store",
|
|
16
|
+
...(data === undefined
|
|
17
|
+
? {}
|
|
18
|
+
: {
|
|
19
|
+
method: "POST",
|
|
20
|
+
headers: { "Content-Type": "application/json" },
|
|
21
|
+
body: JSON.stringify(data),
|
|
22
|
+
}),
|
|
23
|
+
});
|
|
24
|
+
const result = await r.json();
|
|
25
|
+
if (!r.ok || result.ok === false)
|
|
26
|
+
throw new Error(result.error || result.message || `HTTP ${r.status}`);
|
|
27
|
+
return result;
|
|
28
|
+
}
|
|
29
|
+
const api = (action, data) => request("/api/dsh-synapse/" + action, data);
|
|
30
|
+
function button(label, action) {
|
|
31
|
+
const b = node("button", label);
|
|
32
|
+
b.onclick = async () => {
|
|
33
|
+
b.disabled = true;
|
|
34
|
+
try {
|
|
35
|
+
await action();
|
|
36
|
+
}
|
|
37
|
+
catch (e) {
|
|
38
|
+
notify(e.message, true);
|
|
39
|
+
}
|
|
40
|
+
finally {
|
|
41
|
+
b.disabled = false;
|
|
42
|
+
}
|
|
43
|
+
};
|
|
44
|
+
return b;
|
|
45
|
+
}
|
|
46
|
+
function labeled(parent, label, input) {
|
|
47
|
+
const id = "field-" + ++sequence;
|
|
48
|
+
input.id = id;
|
|
49
|
+
const l = node("label", label, { htmlFor: id });
|
|
50
|
+
parent.append(l, input);
|
|
51
|
+
return input;
|
|
52
|
+
}
|
|
53
|
+
let sequence = 0, dirty = false, view, draft, active = "配置";
|
|
54
|
+
const tabs = ["配置", "Skill", "MCP", "Hermes 记忆", "GitHub 审查"];
|
|
55
|
+
for (const label of tabs) {
|
|
56
|
+
const b = button(label, async () => {
|
|
57
|
+
if (dirty && !confirm("当前修改尚未保存,放弃修改并切换?"))
|
|
58
|
+
return;
|
|
59
|
+
dirty = false;
|
|
60
|
+
active = label;
|
|
61
|
+
await render();
|
|
62
|
+
});
|
|
63
|
+
b.setAttribute("role", "tab");
|
|
64
|
+
$("tabs").append(b);
|
|
65
|
+
}
|
|
66
|
+
async function loadConfig() {
|
|
67
|
+
view = (await request("/api/dshloader/config/dsh-synapse")).value;
|
|
68
|
+
draft = structuredClone(view.value);
|
|
69
|
+
}
|
|
70
|
+
async function saveConfig() {
|
|
71
|
+
const saved = await request("/api/dshloader/config/dsh-synapse", {
|
|
72
|
+
value: draft,
|
|
73
|
+
expectedRevision: view.revision,
|
|
74
|
+
});
|
|
75
|
+
view = saved.value;
|
|
76
|
+
draft = structuredClone(view.value);
|
|
77
|
+
dirty = false;
|
|
78
|
+
notify("已保存。通道正在重新连接;可刷新状态检查结果。");
|
|
79
|
+
}
|
|
80
|
+
function field(parent, f, obj, key = f.key) {
|
|
81
|
+
if (f.type === "group" || f.type === "object") {
|
|
82
|
+
const box = node("fieldset");
|
|
83
|
+
box.append(node("legend", f.label));
|
|
84
|
+
parent.append(box);
|
|
85
|
+
const target = f.type === "object" ? (obj[key] ??= {}) : obj;
|
|
86
|
+
for (const child of f.fields)
|
|
87
|
+
field(box, child, target);
|
|
88
|
+
return;
|
|
89
|
+
}
|
|
90
|
+
if (f.type === "array") {
|
|
91
|
+
const box = node("fieldset");
|
|
92
|
+
box.append(node("legend", f.label));
|
|
93
|
+
parent.append(box);
|
|
94
|
+
const list = node("div");
|
|
95
|
+
box.append(list);
|
|
96
|
+
const redraw = () => {
|
|
97
|
+
list.replaceChildren();
|
|
98
|
+
(obj[key] ||= []).forEach((_, i) => {
|
|
99
|
+
const row = node("div", undefined, { className: "row" });
|
|
100
|
+
field(row, { ...f.item, label: `第 ${i + 1} 项` }, obj[key], i);
|
|
101
|
+
row.append(button("删除", () => {
|
|
102
|
+
obj[key].splice(i, 1);
|
|
103
|
+
dirty = true;
|
|
104
|
+
redraw();
|
|
105
|
+
}));
|
|
106
|
+
list.append(row);
|
|
107
|
+
});
|
|
108
|
+
};
|
|
109
|
+
redraw();
|
|
110
|
+
box.append(button("添加", () => {
|
|
111
|
+
obj[key].push(defaultOf(f.item));
|
|
112
|
+
dirty = true;
|
|
113
|
+
redraw();
|
|
114
|
+
}));
|
|
115
|
+
return;
|
|
116
|
+
}
|
|
117
|
+
let control;
|
|
118
|
+
if (f.type === "boolean") {
|
|
119
|
+
control = node("input", undefined, {
|
|
120
|
+
type: "checkbox",
|
|
121
|
+
checked: obj[key] === true,
|
|
122
|
+
});
|
|
123
|
+
control.onchange = () => {
|
|
124
|
+
obj[key] = control.checked;
|
|
125
|
+
dirty = true;
|
|
126
|
+
};
|
|
127
|
+
}
|
|
128
|
+
else if (f.type === "select") {
|
|
129
|
+
control = node("select");
|
|
130
|
+
for (const opt of f.options)
|
|
131
|
+
control.append(node("option", opt.label, { value: opt.value }));
|
|
132
|
+
control.value = obj[key];
|
|
133
|
+
control.onchange = () => {
|
|
134
|
+
obj[key] = control.value;
|
|
135
|
+
dirty = true;
|
|
136
|
+
};
|
|
137
|
+
}
|
|
138
|
+
else {
|
|
139
|
+
const isJSON = String(key).toLowerCase().includes("json");
|
|
140
|
+
control = node(f.type === "textarea" || isJSON ? "textarea" : "input");
|
|
141
|
+
if (control.tagName === "INPUT")
|
|
142
|
+
control.type =
|
|
143
|
+
f.type === "password"
|
|
144
|
+
? "password"
|
|
145
|
+
: f.type === "number"
|
|
146
|
+
? "number"
|
|
147
|
+
: "text";
|
|
148
|
+
const secret = obj[key] && typeof obj[key] === "object";
|
|
149
|
+
control.value = secret ? "" : (obj[key] ?? "");
|
|
150
|
+
if (secret)
|
|
151
|
+
control.placeholder = "已保存(不回显)。留空保持不变;输入替换。";
|
|
152
|
+
control.oninput = () => {
|
|
153
|
+
obj[key] = f.type === "number" ? Number(control.value) : control.value;
|
|
154
|
+
dirty = true;
|
|
155
|
+
};
|
|
156
|
+
if (f.min !== undefined)
|
|
157
|
+
control.min = f.min;
|
|
158
|
+
if (f.max !== undefined)
|
|
159
|
+
control.max = f.max;
|
|
160
|
+
}
|
|
161
|
+
labeled(parent, f.label, control);
|
|
162
|
+
if (f.description)
|
|
163
|
+
parent.append(node("div", f.description, { className: "muted" }));
|
|
164
|
+
if (f.type === "password")
|
|
165
|
+
parent.append(button("清空此值", () => {
|
|
166
|
+
obj[key] = "";
|
|
167
|
+
control.value = "";
|
|
168
|
+
control.placeholder = "已清空,保存后生效";
|
|
169
|
+
dirty = true;
|
|
170
|
+
}));
|
|
171
|
+
}
|
|
172
|
+
function defaultOf(f) {
|
|
173
|
+
if (f.default !== undefined)
|
|
174
|
+
return structuredClone(f.default);
|
|
175
|
+
if (f.type === "object")
|
|
176
|
+
return Object.fromEntries(f.fields.map((x) => [x.key, defaultOf(x)]));
|
|
177
|
+
if (f.type === "array")
|
|
178
|
+
return [];
|
|
179
|
+
if (f.type === "boolean")
|
|
180
|
+
return false;
|
|
181
|
+
if (f.type === "select")
|
|
182
|
+
return f.options[0].value;
|
|
183
|
+
if (f.type === "number")
|
|
184
|
+
return f.min || 0;
|
|
185
|
+
return "";
|
|
186
|
+
}
|
|
187
|
+
async function configPanel(root, onlyMcp = false) {
|
|
188
|
+
await loadConfig();
|
|
189
|
+
root.append(node("p", onlyMcp
|
|
190
|
+
? "添加或修改 MCP 服务,保存后可预览工具。仅允许列表中的工具可被聊天自动调用。"
|
|
191
|
+
: "所有通道和上游功能均保存在当前 profile;保存使用修订号检查并发修改。"));
|
|
192
|
+
for (const f of view.template.fields)
|
|
193
|
+
if (onlyMcp ? f.key === "mcp" : f.key !== "mcp")
|
|
194
|
+
field(root, f, draft);
|
|
195
|
+
root.append(button("保存配置", saveConfig));
|
|
196
|
+
if (!onlyMcp) {
|
|
197
|
+
root.append(button("载入上游功能 JSON 以预览 / 修改", async () => {
|
|
198
|
+
const rows = await api("features");
|
|
199
|
+
for (const row of rows) {
|
|
200
|
+
const f = draft.features.find((f) => f.id === row.id);
|
|
201
|
+
if (f)
|
|
202
|
+
f.json = JSON.stringify(row.value, null, 2);
|
|
203
|
+
}
|
|
204
|
+
dirty = true;
|
|
205
|
+
root.replaceChildren();
|
|
206
|
+
root.append(node("p", "已载入当前功能配置,可能包含密钥。修改后保存。"));
|
|
207
|
+
for (const f of view.template.fields)
|
|
208
|
+
if (f.key !== "mcp")
|
|
209
|
+
field(root, f, draft);
|
|
210
|
+
root.append(button("保存配置", saveConfig));
|
|
211
|
+
}));
|
|
212
|
+
}
|
|
213
|
+
else {
|
|
214
|
+
const box = node("section");
|
|
215
|
+
root.append(box);
|
|
216
|
+
for (const s of draft.mcp) {
|
|
217
|
+
box.append(button(`预览 ${s.name} 工具`, async () => {
|
|
218
|
+
preview.textContent = JSON.stringify(await api("mcp/tools", { name: s.name }), null, 2);
|
|
219
|
+
}));
|
|
220
|
+
}
|
|
221
|
+
const preview = node("pre", "工具列表显示在这里");
|
|
222
|
+
box.append(preview);
|
|
223
|
+
}
|
|
224
|
+
}
|
|
225
|
+
async function skillPanel(root) {
|
|
226
|
+
const layout = node("div", undefined, { className: "two" }), list = node("div"), editor = node("div");
|
|
227
|
+
layout.append(list, editor);
|
|
228
|
+
root.append(layout);
|
|
229
|
+
const name = labeled(editor, "Skill 名称", node("input")), text = labeled(editor, "Markdown 文档", node("textarea"));
|
|
230
|
+
text.style.minHeight = "400px";
|
|
231
|
+
let revision = null, inherited = false;
|
|
232
|
+
name.oninput = text.oninput = () => {
|
|
233
|
+
dirty = true;
|
|
234
|
+
};
|
|
235
|
+
const file = labeled(editor, "导入 SKILL.md", node("input", undefined, {
|
|
236
|
+
type: "file",
|
|
237
|
+
accept: ".md,text/markdown,text/plain",
|
|
238
|
+
}));
|
|
239
|
+
file.onchange = async () => {
|
|
240
|
+
const f = file.files?.[0];
|
|
241
|
+
if (f) {
|
|
242
|
+
text.value = await f.text();
|
|
243
|
+
dirty = true;
|
|
244
|
+
}
|
|
245
|
+
};
|
|
246
|
+
const preview = node("pre", "选择 Skill 预览或创建新 Skill。");
|
|
247
|
+
editor.append(preview);
|
|
248
|
+
const refresh = async () => {
|
|
249
|
+
list.replaceChildren();
|
|
250
|
+
for (const s of await api("skills"))
|
|
251
|
+
list.append(button(s.name + (s.inherited ? " · Home" : ""), () => {
|
|
252
|
+
name.value = s.name;
|
|
253
|
+
text.value = s.content;
|
|
254
|
+
revision = s.revision;
|
|
255
|
+
inherited = s.inherited;
|
|
256
|
+
preview.textContent = s.content;
|
|
257
|
+
dirty = false;
|
|
258
|
+
}));
|
|
259
|
+
};
|
|
260
|
+
editor.append(button("预览文档", () => {
|
|
261
|
+
preview.textContent = text.value;
|
|
262
|
+
}), button("新建", () => {
|
|
263
|
+
name.value = "";
|
|
264
|
+
text.value = "";
|
|
265
|
+
revision = null;
|
|
266
|
+
inherited = false;
|
|
267
|
+
dirty = false;
|
|
268
|
+
}), button("保存 Skill", async () => {
|
|
269
|
+
const saved = await api("skills/save", {
|
|
270
|
+
name: name.value,
|
|
271
|
+
content: text.value,
|
|
272
|
+
revision,
|
|
273
|
+
});
|
|
274
|
+
revision = saved.revision;
|
|
275
|
+
dirty = false;
|
|
276
|
+
notify("Skill 已保存,可被运行中的聊天检索。");
|
|
277
|
+
await refresh();
|
|
278
|
+
}), button("删除 Skill", async () => {
|
|
279
|
+
if (inherited)
|
|
280
|
+
throw new Error("Home Skill 为继承项,请在 Home 中管理");
|
|
281
|
+
if (!confirm(`删除 ${name.value}?`))
|
|
282
|
+
return;
|
|
283
|
+
await api("skills/delete", { name: name.value, revision });
|
|
284
|
+
name.value = "";
|
|
285
|
+
text.value = "";
|
|
286
|
+
revision = null;
|
|
287
|
+
dirty = false;
|
|
288
|
+
await refresh();
|
|
289
|
+
notify("Skill 已删除");
|
|
290
|
+
}));
|
|
291
|
+
await refresh();
|
|
292
|
+
}
|
|
293
|
+
async function memoryPanel(root) {
|
|
294
|
+
const channel = labeled(root, "记忆分类", node("select"));
|
|
295
|
+
channel.append(node("option", "聊天(QQ / Discord 共享)", { value: "chat" }), node("option", "GitHub 审查", { value: "review" }));
|
|
296
|
+
const markdown = labeled(root, "粘贴 Hermes 记忆 Markdown", node("textarea"));
|
|
297
|
+
markdown.oninput = () => {
|
|
298
|
+
dirty = true;
|
|
299
|
+
};
|
|
300
|
+
const preview = node("pre", "导入前可预览分块。");
|
|
301
|
+
root.append(button("预览分块", async () => {
|
|
302
|
+
preview.textContent = JSON.stringify(await api("memory/import", {
|
|
303
|
+
markdown: markdown.value,
|
|
304
|
+
channel: channel.value,
|
|
305
|
+
preview: true,
|
|
306
|
+
}), null, 2);
|
|
307
|
+
}), button("批量导入", async () => {
|
|
308
|
+
notify("正在生成本地嵌入;首次下载模型可能需要几分钟…");
|
|
309
|
+
preview.textContent = JSON.stringify(await api("memory/import", {
|
|
310
|
+
markdown: markdown.value,
|
|
311
|
+
channel: channel.value,
|
|
312
|
+
}), null, 2);
|
|
313
|
+
dirty = false;
|
|
314
|
+
notify("记忆导入完成");
|
|
315
|
+
}));
|
|
316
|
+
const query = labeled(root, "检索记忆", node("input"));
|
|
317
|
+
root.append(button("混合检索", async () => {
|
|
318
|
+
preview.textContent = JSON.stringify(await api("memory/search", {
|
|
319
|
+
query: query.value,
|
|
320
|
+
channel: channel.value,
|
|
321
|
+
}), null, 2);
|
|
322
|
+
}), button("最近记忆", async () => {
|
|
323
|
+
preview.textContent = JSON.stringify(await api(`memory?channel=${channel.value}`), null, 2);
|
|
324
|
+
}), preview);
|
|
325
|
+
}
|
|
326
|
+
async function reviewPanel(root) {
|
|
327
|
+
root.append(node("p", "Webhook:" + location.origin + "/hooks/dsh-synapse/github"), node("p", "配置 GitHub Secret 和仓库允许列表后启用;未开启自动发布时,结果仅保存在此处。"));
|
|
328
|
+
const list = node("div");
|
|
329
|
+
root.append(button("刷新任务", refresh), list);
|
|
330
|
+
async function refresh() {
|
|
331
|
+
list.replaceChildren();
|
|
332
|
+
for (const job of await api("reviews")) {
|
|
333
|
+
const a = node("article");
|
|
334
|
+
a.append(node("strong", `${job.event} · ${job.state} · ${job.id}`), node("pre", job.error || job.result || "等待处理"));
|
|
335
|
+
if (job.state === "failed")
|
|
336
|
+
a.append(button("重试", async () => {
|
|
337
|
+
await api("reviews/retry", { id: job.id });
|
|
338
|
+
await refresh();
|
|
339
|
+
}));
|
|
340
|
+
list.append(a);
|
|
341
|
+
}
|
|
342
|
+
}
|
|
343
|
+
await refresh();
|
|
344
|
+
}
|
|
345
|
+
async function render() {
|
|
346
|
+
for (const b of $("tabs").children)
|
|
347
|
+
b.setAttribute("aria-selected", String(b.textContent === active));
|
|
348
|
+
const root = $("content");
|
|
349
|
+
root.replaceChildren();
|
|
350
|
+
notify("");
|
|
351
|
+
try {
|
|
352
|
+
if (active === "配置")
|
|
353
|
+
await configPanel(root);
|
|
354
|
+
if (active === "MCP")
|
|
355
|
+
await configPanel(root, true);
|
|
356
|
+
if (active === "Skill")
|
|
357
|
+
await skillPanel(root);
|
|
358
|
+
if (active === "Hermes 记忆")
|
|
359
|
+
await memoryPanel(root);
|
|
360
|
+
if (active === "GitHub 审查")
|
|
361
|
+
await reviewPanel(root);
|
|
362
|
+
}
|
|
363
|
+
catch (e) {
|
|
364
|
+
notify(e.message, true);
|
|
365
|
+
}
|
|
366
|
+
}
|
|
367
|
+
async function status() {
|
|
368
|
+
try {
|
|
369
|
+
const s = await api("status");
|
|
370
|
+
$("status").textContent = s.lastError
|
|
371
|
+
? "需要处理:" + s.lastError
|
|
372
|
+
: "机器人:" + s.bot;
|
|
373
|
+
}
|
|
374
|
+
catch (e) {
|
|
375
|
+
$("status").textContent = e.message;
|
|
376
|
+
}
|
|
377
|
+
}
|
|
378
|
+
window.addEventListener("beforeunload", (e) => {
|
|
379
|
+
if (dirty) {
|
|
380
|
+
e.preventDefault();
|
|
381
|
+
e.returnValue = "";
|
|
382
|
+
}
|
|
383
|
+
});
|
|
384
|
+
await render();
|
|
385
|
+
await status();
|
|
386
|
+
setInterval(status, 10000);
|
|
387
|
+
export {};
|
|
388
|
+
//# sourceMappingURL=panel.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"panel.js","sourceRoot":"","sources":["../../src/panel.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;AAC9C,SAAS,IAAI,CACX,GAAM,EACN,IAAa,EACb,QAAiC,EAAE;IAEnC,MAAM,CAAC,GAAG,QAAQ,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IACtC,IAAI,IAAI,KAAK,SAAS;QAAE,CAAC,CAAC,WAAW,GAAG,IAAI,CAAC;IAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACxB,OAAO,CAAC,CAAC;AACX,CAAC;AACD,SAAS,MAAM,CAAC,IAAI,EAAE,KAAK,GAAG,KAAK;IACjC,CAAC,CAAC,QAAQ,CAAC,CAAC,WAAW,GAAG,IAAI,CAAC;IAC/B,CAAC,CAAC,QAAQ,CAAC,CAAC,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;AAC/C,CAAC;AACD,KAAK,UAAU,OAAO,CAAC,IAAI,EAAE,IAAK;IAChC,MAAM,CAAC,GAAG,MAAM,KAAK,CAAC,IAAI,EAAE;QAC1B,KAAK,EAAE,UAAU;QACjB,GAAG,CAAC,IAAI,KAAK,SAAS;YACpB,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC;KACP,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;IAC9B,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,MAAM,CAAC,EAAE,KAAK,KAAK;QAC9B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,OAAO,IAAI,QAAQ,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IACxE,OAAO,MAAM,CAAC;AAChB,CAAC;AACD,MAAM,GAAG,GAAG,CAAC,MAAM,EAAE,IAAK,EAAE,EAAE,CAAC,OAAO,CAAC,mBAAmB,GAAG,MAAM,EAAE,IAAI,CAAC,CAAC;AAC3E,SAAS,MAAM,CAAC,KAAK,EAAE,MAAM;IAC3B,MAAM,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAChC,CAAC,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QACrB,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC;YACH,MAAM,MAAM,EAAE,CAAC;QACjB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC1B,CAAC;gBAAS,CAAC;YACT,CAAC,CAAC,QAAQ,GAAG,KAAK,CAAC;QACrB,CAAC;IACH,CAAC,CAAC;IACF,OAAO,CAAC,CAAC;AACX,CAAC;AACD,SAAS,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK;IACnC,MAAM,EAAE,GAAG,QAAQ,GAAG,EAAE,QAAQ,CAAC;IACjC,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC;IACd,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;IAChD,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACxB,OAAO,KAAK,CAAC;AACf,CAAC;AACD,IAAI,QAAQ,GAAG,CAAC,EACd,KAAK,GAAG,KAAK,EACb,IAAI,EACJ,KAAK,EACL,MAAM,GAAG,IAAI,CAAC;AAChB,MAAM,IAAI,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;AAC9D,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;IACzB,MAAM,CAAC,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE;QACjC,IAAI,KAAK,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC;YAAE,OAAO;QACnD,KAAK,GAAG,KAAK,CAAC;QACd,MAAM,GAAG,KAAK,CAAC;QACf,MAAM,MAAM,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IACH,CAAC,CAAC,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC9B,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;AACtB,CAAC;AACD,KAAK,UAAU,UAAU;IACvB,IAAI,GAAG,CAAC,MAAM,OAAO,CAAC,mCAAmC,CAAC,CAAC,CAAC,KAAK,CAAC;IAClE,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACtC,CAAC;AACD,KAAK,UAAU,UAAU;IACvB,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,mCAAmC,EAAE;QAC/D,KAAK,EAAE,KAAK;QACZ,gBAAgB,EAAE,IAAI,CAAC,QAAQ;KAChC,CAAC,CAAC;IACH,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC;IACnB,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpC,KAAK,GAAG,KAAK,CAAC;IACd,MAAM,CAAC,yBAAyB,CAAC,CAAC;AACpC,CAAC;AACD,SAAS,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC,CAAC,GAAG;IACxC,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7B,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACnB,MAAM,MAAM,GAAG,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;QAC7D,KAAK,MAAM,KAAK,IAAI,CAAC,CAAC,MAAM;YAAE,KAAK,CAAC,GAAG,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QACxD,OAAO;IACT,CAAC;IACD,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7B,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACnB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACjB,MAAM,MAAM,GAAG,GAAG,EAAE;YAClB,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;gBACjC,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;gBACzD,KAAK,CAAC,GAAG,EAAE,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC9D,GAAG,CAAC,MAAM,CACR,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;oBAChB,GAAG,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;oBACtB,KAAK,GAAG,IAAI,CAAC;oBACb,MAAM,EAAE,CAAC;gBACX,CAAC,CAAC,CACH,CAAC;gBACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC;QACF,MAAM,EAAE,CAAC;QACT,GAAG,CAAC,MAAM,CACR,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;YAChB,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YACjC,KAAK,GAAG,IAAI,CAAC;YACb,MAAM,EAAE,CAAC;QACX,CAAC,CAAC,CACH,CAAC;QACF,OAAO;IACT,CAAC;IACD,IAAI,OAAO,CAAC;IACZ,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QACzB,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE;YACjC,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI;SAC3B,CAAC,CAAC;QACH,OAAO,CAAC,QAAQ,GAAG,GAAG,EAAE;YACtB,GAAG,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;YAC3B,KAAK,GAAG,IAAI,CAAC;QACf,CAAC,CAAC;IACJ,CAAC;SAAM,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzB,KAAK,MAAM,GAAG,IAAI,CAAC,CAAC,OAAO;YACzB,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAClE,OAAO,CAAC,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;QACzB,OAAO,CAAC,QAAQ,GAAG,GAAG,EAAE;YACtB,GAAG,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;YACzB,KAAK,GAAG,IAAI,CAAC;QACf,CAAC,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC1D,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,IAAI,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACvE,IAAI,OAAO,CAAC,OAAO,KAAK,OAAO;YAC7B,OAAO,CAAC,IAAI;gBACV,CAAC,CAAC,IAAI,KAAK,UAAU;oBACnB,CAAC,CAAC,UAAU;oBACZ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ;wBACnB,CAAC,CAAC,QAAQ;wBACV,CAAC,CAAC,MAAM,CAAC;QACjB,MAAM,MAAM,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,OAAO,GAAG,CAAC,GAAG,CAAC,KAAK,QAAQ,CAAC;QACxD,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;QAC/C,IAAI,MAAM;YACR,OAAO,CAAC,WAAW,GAAG,uBAAuB,CAAC;QAChD,OAAO,CAAC,OAAO,GAAG,GAAG,EAAE;YACrB,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;YACvE,KAAK,GAAG,IAAI,CAAC;QACf,CAAC,CAAC;QACF,IAAI,CAAC,CAAC,GAAG,KAAK,SAAS;YAAE,OAAO,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC;QAC7C,IAAI,CAAC,CAAC,GAAG,KAAK,SAAS;YAAE,OAAO,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC;IAC/C,CAAC;IACD,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAClC,IAAI,CAAC,CAAC,WAAW;QACf,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;IACpE,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU;QACvB,MAAM,CAAC,MAAM,CACX,MAAM,CAAC,MAAM,EAAE,GAAG,EAAE;YAClB,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;YACd,OAAO,CAAC,KAAK,GAAG,EAAE,CAAC;YACnB,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC;YAClC,KAAK,GAAG,IAAI,CAAC;QACf,CAAC,CAAC,CACH,CAAC;AACN,CAAC;AACD,SAAS,SAAS,CAAC,CAAC;IAClB,IAAI,CAAC,CAAC,OAAO,KAAK,SAAS;QAAE,OAAO,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAC/D,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ;QACrB,OAAO,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACxE,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,EAAE,CAAC;IAClC,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS;QAAE,OAAO,KAAK,CAAC;IACvC,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IACnD,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IAC3C,OAAO,EAAE,CAAC;AACZ,CAAC;AACD,KAAK,UAAU,WAAW,CAAC,IAAI,EAAE,OAAO,GAAG,KAAK;IAC9C,MAAM,UAAU,EAAE,CAAC;IACnB,IAAI,CAAC,MAAM,CACT,IAAI,CACF,GAAG,EACH,OAAO;QACL,CAAC,CAAC,0CAA0C;QAC5C,CAAC,CAAC,wCAAwC,CAC7C,CACF,CAAC;IACF,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM;QAClC,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,KAAK;YAAE,KAAK,CAAC,IAAI,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;IACzE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;IACxC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,CACT,MAAM,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,UAAU,CAAC,CAAC;YACnC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,MAAM,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;gBACtD,IAAI,CAAC;oBAAE,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACrD,CAAC;YACD,KAAK,GAAG,IAAI,CAAC;YACb,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,IAAI,CAAC,MAAM,CACT,IAAI,CAAC,GAAG,EAAE,yBAAyB,CAAC,CACrC,CAAC;YACF,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM;gBAClC,IAAI,CAAC,CAAC,GAAG,KAAK,KAAK;oBAAE,KAAK,CAAC,IAAI,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACjB,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;YAC1B,GAAG,CAAC,MAAM,CACR,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,KAAK,EAAE,KAAK,IAAI,EAAE;gBACnC,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,SAAS,CAClC,MAAM,GAAG,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,EACxC,IAAI,EACJ,CAAC,CACF,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QACzC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACtB,CAAC;AACH,CAAC;AACD,KAAK,UAAU,UAAU,CAAC,IAAI;IAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EACzD,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAClB,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IACvB,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC5B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACpB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,EAAE,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,EACrD,IAAI,GAAG,OAAO,CAAC,MAAM,EAAE,aAAa,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;IAC1D,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,OAAO,CAAC;IAC/B,IAAI,QAAQ,GAAG,IAAI,EACjB,SAAS,GAAG,KAAK,CAAC;IACpB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,GAAG,GAAG,EAAE;QACjC,KAAK,GAAG,IAAI,CAAC;IACf,CAAC,CAAC;IACF,MAAM,IAAI,GAAG,OAAO,CAClB,MAAM,EACN,aAAa,EACb,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE;QACvB,IAAI,EAAE,MAAM;QACZ,MAAM,EAAE,8BAA8B;KACvC,CAAC,CACH,CAAC;IACF,IAAI,CAAC,QAAQ,GAAG,KAAK,IAAI,EAAE;QACzB,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1B,IAAI,CAAC,EAAE,CAAC;YACN,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;YAC5B,KAAK,GAAG,IAAI,CAAC;QACf,CAAC;IACH,CAAC,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,EAAE,wBAAwB,CAAC,CAAC;IACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACvB,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;QACzB,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,KAAK,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,QAAQ,CAAC;YACjC,IAAI,CAAC,MAAM,CACT,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE;gBACnD,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC;gBACpB,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC;gBACvB,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC;gBACtB,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC;gBACxB,OAAO,CAAC,WAAW,GAAG,CAAC,CAAC,OAAO,CAAC;gBAChC,KAAK,GAAG,KAAK,CAAC;YAChB,CAAC,CAAC,CACH,CAAC;IACN,CAAC,CAAC;IACF,MAAM,CAAC,MAAM,CACX,MAAM,CAAC,MAAM,EAAE,GAAG,EAAE;QAClB,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC;IACnC,CAAC,CAAC,EACF,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;QAChB,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAChB,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAChB,QAAQ,GAAG,IAAI,CAAC;QAChB,SAAS,GAAG,KAAK,CAAC;QAClB,KAAK,GAAG,KAAK,CAAC;IAChB,CAAC,CAAC,EACF,MAAM,CAAC,UAAU,EAAE,KAAK,IAAI,EAAE;QAC5B,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,aAAa,EAAE;YACrC,IAAI,EAAE,IAAI,CAAC,KAAK;YAChB,OAAO,EAAE,IAAI,CAAC,KAAK;YACnB,QAAQ;SACT,CAAC,CAAC;QACH,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC1B,KAAK,GAAG,KAAK,CAAC;QACd,MAAM,CAAC,uBAAuB,CAAC,CAAC;QAChC,MAAM,OAAO,EAAE,CAAC;IAClB,CAAC,CAAC,EACF,MAAM,CAAC,UAAU,EAAE,KAAK,IAAI,EAAE;QAC5B,IAAI,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QAC9D,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,KAAK,GAAG,CAAC;YAAE,OAAO;QAC1C,MAAM,GAAG,CAAC,eAAe,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC3D,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAChB,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAChB,QAAQ,GAAG,IAAI,CAAC;QAChB,KAAK,GAAG,KAAK,CAAC;QACd,MAAM,OAAO,EAAE,CAAC;QAChB,MAAM,CAAC,WAAW,CAAC,CAAC;IACtB,CAAC,CAAC,CACH,CAAC;IACF,MAAM,OAAO,EAAE,CAAC;AAClB,CAAC;AACD,KAAK,UAAU,WAAW,CAAC,IAAI;IAC7B,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;IACtD,OAAO,CAAC,MAAM,CACZ,IAAI,CAAC,QAAQ,EAAE,qBAAqB,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,EACxD,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CACjD,CAAC;IACF,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,EAAE,uBAAuB,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;IAC1E,QAAQ,CAAC,OAAO,GAAG,GAAG,EAAE;QACtB,KAAK,GAAG,IAAI,CAAC;IACf,CAAC,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IACzC,IAAI,CAAC,MAAM,CACT,MAAM,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE;QACxB,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,SAAS,CAClC,MAAM,GAAG,CAAC,eAAe,EAAE;YACzB,QAAQ,EAAE,QAAQ,CAAC,KAAK;YACxB,OAAO,EAAE,OAAO,CAAC,KAAK;YACtB,OAAO,EAAE,IAAI;SACd,CAAC,EACF,IAAI,EACJ,CAAC,CACF,CAAC;IACJ,CAAC,CAAC,EACF,MAAM,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE;QACxB,MAAM,CAAC,yBAAyB,CAAC,CAAC;QAClC,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,SAAS,CAClC,MAAM,GAAG,CAAC,eAAe,EAAE;YACzB,QAAQ,EAAE,QAAQ,CAAC,KAAK;YACxB,OAAO,EAAE,OAAO,CAAC,KAAK;SACvB,CAAC,EACF,IAAI,EACJ,CAAC,CACF,CAAC;QACF,KAAK,GAAG,KAAK,CAAC;QACd,MAAM,CAAC,QAAQ,CAAC,CAAC;IACnB,CAAC,CAAC,CACH,CAAC;IACF,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IACnD,IAAI,CAAC,MAAM,CACT,MAAM,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE;QACxB,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,SAAS,CAClC,MAAM,GAAG,CAAC,eAAe,EAAE;YACzB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO,EAAE,OAAO,CAAC,KAAK;SACvB,CAAC,EACF,IAAI,EACJ,CAAC,CACF,CAAC;IACJ,CAAC,CAAC,EACF,MAAM,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE;QACxB,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,SAAS,CAClC,MAAM,GAAG,CAAC,kBAAkB,OAAO,CAAC,KAAK,EAAE,CAAC,EAC5C,IAAI,EACJ,CAAC,CACF,CAAC;IACJ,CAAC,CAAC,EACF,OAAO,CACR,CAAC;AACJ,CAAC;AACD,KAAK,UAAU,WAAW,CAAC,IAAI;IAC7B,IAAI,CAAC,MAAM,CACT,IAAI,CAAC,GAAG,EAAE,UAAU,GAAG,QAAQ,CAAC,MAAM,GAAG,2BAA2B,CAAC,EACrE,IAAI,CACF,GAAG,EACH,gDAAgD,CACjD,CACF,CAAC;IACF,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IACzB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,CAAC;IAC3C,KAAK,UAAU,OAAO;QACpB,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,KAAK,MAAM,GAAG,IAAI,MAAM,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACvC,MAAM,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;YAC1B,CAAC,CAAC,MAAM,CACN,IAAI,CAAC,QAAQ,EAAE,GAAG,GAAG,CAAC,KAAK,MAAM,GAAG,CAAC,KAAK,MAAM,GAAG,CAAC,EAAE,EAAE,CAAC,EACzD,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,MAAM,IAAI,MAAM,CAAC,CAC/C,CAAC;YACF,IAAI,GAAG,CAAC,KAAK,KAAK,QAAQ;gBACxB,CAAC,CAAC,MAAM,CACN,MAAM,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE;oBACtB,MAAM,GAAG,CAAC,eAAe,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC3C,MAAM,OAAO,EAAE,CAAC;gBAClB,CAAC,CAAC,CACH,CAAC;YACJ,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IACD,MAAM,OAAO,EAAE,CAAC;AAClB,CAAC;AACD,KAAK,UAAU,MAAM;IACnB,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,QAAQ;QAChC,CAAC,CAAC,YAAY,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC,CAAC,WAAW,KAAK,MAAM,CAAC,CAAC,CAAC;IACpE,MAAM,IAAI,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;IAC1B,IAAI,CAAC,eAAe,EAAE,CAAC;IACvB,MAAM,CAAC,EAAE,CAAC,CAAC;IACX,IAAI,CAAC;QACH,IAAI,MAAM,KAAK,IAAI;YAAE,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC;QAC7C,IAAI,MAAM,KAAK,KAAK;YAAE,MAAM,WAAW,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACpD,IAAI,MAAM,KAAK,OAAO;YAAE,MAAM,UAAU,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,MAAM,KAAK,WAAW;YAAE,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC;QACpD,IAAI,MAAM,KAAK,WAAW;YAAE,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC1B,CAAC;AACH,CAAC;AACD,KAAK,UAAU,MAAM;IACnB,IAAI,CAAC;QACH,MAAM,CAAC,GAAG,MAAM,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC,CAAC,QAAQ,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,SAAS;YACnC,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,SAAS;YACvB,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC;IACrB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,CAAC,CAAC,QAAQ,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,OAAO,CAAC;IACtC,CAAC;AACH,CAAC;AACD,MAAM,CAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,EAAE;IAC5C,IAAI,KAAK,EAAE,CAAC;QACV,CAAC,CAAC,cAAc,EAAE,CAAC;QACnB,CAAC,CAAC,WAAW,GAAG,EAAE,CAAC;IACrB,CAAC;AACH,CAAC,CAAC,CAAC;AACH,MAAM,MAAM,EAAE,CAAC;AACf,MAAM,MAAM,EAAE,CAAC;AACf,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import { Worker } from "node:worker_threads";
|
|
2
|
+
export declare function installRuntime(root: string, signal?: AbortSignal): Promise<string>;
|
|
3
|
+
export declare class Inference {
|
|
4
|
+
data: string;
|
|
5
|
+
pending: Map<number, {
|
|
6
|
+
resolve: (value: number[]) => void;
|
|
7
|
+
reject: (error: Error) => void;
|
|
8
|
+
timer: NodeJS.Timeout;
|
|
9
|
+
}>;
|
|
10
|
+
counter: number;
|
|
11
|
+
closed: boolean;
|
|
12
|
+
starting: Promise<void>;
|
|
13
|
+
worker: Worker;
|
|
14
|
+
installer: AbortController;
|
|
15
|
+
constructor(data: string);
|
|
16
|
+
start(): Promise<void>;
|
|
17
|
+
call(method: any, args: any): Promise<number[]>;
|
|
18
|
+
embed(text: any, query?: boolean): Promise<number[]>;
|
|
19
|
+
rerank(query: any, documents: any): Promise<number[]>;
|
|
20
|
+
close(): Promise<void>;
|
|
21
|
+
}
|
|
@@ -0,0 +1,145 @@
|
|
|
1
|
+
import { Worker } from "node:worker_threads";
|
|
2
|
+
import { mkdir, readFile, writeFile } from "node:fs/promises";
|
|
3
|
+
import { createRequire } from "node:module";
|
|
4
|
+
import { spawn } from "node:child_process";
|
|
5
|
+
import path from "node:path";
|
|
6
|
+
const VERSION = "3.8.1";
|
|
7
|
+
export async function installRuntime(root, signal) {
|
|
8
|
+
await mkdir(root, { recursive: true });
|
|
9
|
+
const file = path.join(root, "package.json");
|
|
10
|
+
const manifest = {
|
|
11
|
+
name: "dsh-synapse-inference-runtime",
|
|
12
|
+
private: true,
|
|
13
|
+
version: "1.0.0",
|
|
14
|
+
dependencies: { "@huggingface/transformers": VERSION },
|
|
15
|
+
};
|
|
16
|
+
try {
|
|
17
|
+
const req = createRequire(file);
|
|
18
|
+
req.resolve("@huggingface/transformers");
|
|
19
|
+
const pkg = JSON.parse(await readFile(path.join(root, "node_modules/@huggingface/transformers/package.json"), "utf8"));
|
|
20
|
+
if (pkg.version === VERSION)
|
|
21
|
+
return file;
|
|
22
|
+
}
|
|
23
|
+
catch { }
|
|
24
|
+
await writeFile(file, JSON.stringify(manifest, null, 2));
|
|
25
|
+
// npm is launched without a command shell, including on Windows.
|
|
26
|
+
const candidates = [
|
|
27
|
+
process.env.npm_execpath,
|
|
28
|
+
path.join(path.dirname(process.execPath), "node_modules/npm/bin/npm-cli.js"),
|
|
29
|
+
path.resolve(path.dirname(process.execPath), "../lib/node_modules/npm/bin/npm-cli.js"),
|
|
30
|
+
...(process.env.PATH || "")
|
|
31
|
+
.split(path.delimiter)
|
|
32
|
+
.flatMap((dir) => [
|
|
33
|
+
path.join(dir, "node_modules/npm/bin/npm-cli.js"),
|
|
34
|
+
path.resolve(dir, "../lib/node_modules/npm/bin/npm-cli.js"),
|
|
35
|
+
]),
|
|
36
|
+
...(process.platform === "win32"
|
|
37
|
+
? []
|
|
38
|
+
: ["/usr/share/nodejs/npm/bin/npm-cli.js"]),
|
|
39
|
+
].filter(Boolean);
|
|
40
|
+
let npm;
|
|
41
|
+
for (const c of candidates) {
|
|
42
|
+
try {
|
|
43
|
+
await readFile(c);
|
|
44
|
+
npm = c;
|
|
45
|
+
break;
|
|
46
|
+
}
|
|
47
|
+
catch { }
|
|
48
|
+
}
|
|
49
|
+
if (!npm)
|
|
50
|
+
throw new Error("找不到 npm-cli.js;请安装 Node.js 的 npm 组件");
|
|
51
|
+
await new Promise((resolve, reject) => {
|
|
52
|
+
const child = spawn(process.execPath, [npm, "install", "--omit=dev", "--no-audit", "--no-fund"], { cwd: root, windowsHide: true, stdio: "ignore", signal });
|
|
53
|
+
const timer = setTimeout(() => {
|
|
54
|
+
child.kill();
|
|
55
|
+
reject(new Error("推理运行时安装超时"));
|
|
56
|
+
}, 600000);
|
|
57
|
+
child.on("error", (e) => {
|
|
58
|
+
clearTimeout(timer);
|
|
59
|
+
reject(e);
|
|
60
|
+
});
|
|
61
|
+
child.on("exit", (code) => {
|
|
62
|
+
clearTimeout(timer);
|
|
63
|
+
code === 0
|
|
64
|
+
? resolve()
|
|
65
|
+
: reject(new Error(`推理运行时安装失败 (${code})`));
|
|
66
|
+
});
|
|
67
|
+
});
|
|
68
|
+
return file;
|
|
69
|
+
}
|
|
70
|
+
export class Inference {
|
|
71
|
+
data;
|
|
72
|
+
pending;
|
|
73
|
+
counter;
|
|
74
|
+
closed;
|
|
75
|
+
starting;
|
|
76
|
+
worker;
|
|
77
|
+
installer = new AbortController();
|
|
78
|
+
constructor(data) {
|
|
79
|
+
this.data = data;
|
|
80
|
+
this.pending = new Map();
|
|
81
|
+
this.counter = 0;
|
|
82
|
+
this.closed = false;
|
|
83
|
+
}
|
|
84
|
+
async start() {
|
|
85
|
+
if (this.closed)
|
|
86
|
+
throw new Error("推理 Worker 已关闭");
|
|
87
|
+
if (!this.starting)
|
|
88
|
+
this.starting = (async () => {
|
|
89
|
+
const manifest = await installRuntime(path.join(this.data, "runtime"), this.installer.signal);
|
|
90
|
+
if (this.closed)
|
|
91
|
+
throw new Error("推理 Worker 已关闭");
|
|
92
|
+
this.worker = new Worker(new URL("./inference-worker.js", import.meta.url), { workerData: { manifest, cache: path.join(this.data, "models") } });
|
|
93
|
+
this.worker.on("message", ({ id, value, error }) => {
|
|
94
|
+
const p = this.pending.get(id);
|
|
95
|
+
if (!p)
|
|
96
|
+
return;
|
|
97
|
+
clearTimeout(p.timer);
|
|
98
|
+
this.pending.delete(id);
|
|
99
|
+
error ? p.reject(new Error(error)) : p.resolve(value);
|
|
100
|
+
});
|
|
101
|
+
const fail = (e) => {
|
|
102
|
+
for (const p of this.pending.values()) {
|
|
103
|
+
clearTimeout(p.timer);
|
|
104
|
+
p.reject(e);
|
|
105
|
+
}
|
|
106
|
+
this.pending.clear();
|
|
107
|
+
};
|
|
108
|
+
this.worker.on("error", fail);
|
|
109
|
+
this.worker.on("exit", (code) => {
|
|
110
|
+
this.closed = true;
|
|
111
|
+
fail(new Error(`推理 Worker 退出 (${code})`));
|
|
112
|
+
});
|
|
113
|
+
})();
|
|
114
|
+
return this.starting;
|
|
115
|
+
}
|
|
116
|
+
async call(method, args) {
|
|
117
|
+
await this.start();
|
|
118
|
+
if (this.closed)
|
|
119
|
+
throw new Error("推理 Worker 已关闭");
|
|
120
|
+
if (this.pending.size >= 100)
|
|
121
|
+
throw new Error("推理队列已满");
|
|
122
|
+
return new Promise((resolve, reject) => {
|
|
123
|
+
const id = ++this.counter;
|
|
124
|
+
const timer = setTimeout(() => {
|
|
125
|
+
this.pending.delete(id);
|
|
126
|
+
reject(new Error("模型下载或推理超时"));
|
|
127
|
+
}, 600000);
|
|
128
|
+
this.pending.set(id, { resolve, reject, timer });
|
|
129
|
+
this.worker.postMessage({ id, method, args });
|
|
130
|
+
});
|
|
131
|
+
}
|
|
132
|
+
embed(text, query = false) {
|
|
133
|
+
return this.call("embed", { text, query });
|
|
134
|
+
}
|
|
135
|
+
rerank(query, documents) {
|
|
136
|
+
return this.call("rerank", { query, documents });
|
|
137
|
+
}
|
|
138
|
+
async close() {
|
|
139
|
+
this.closed = true;
|
|
140
|
+
this.installer.abort();
|
|
141
|
+
await this.starting?.catch(() => { });
|
|
142
|
+
await this.worker?.terminate();
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
//# sourceMappingURL=runtime.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runtime.js","sourceRoot":"","sources":["../../src/runtime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,MAAM,OAAO,GAAG,OAAO,CAAC;AACxB,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,IAAY,EAAE,MAAoB;IACrE,MAAM,KAAK,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACvC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IAC7C,MAAM,QAAQ,GAAG;QACf,IAAI,EAAE,+BAA+B;QACrC,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,OAAO;QAChB,YAAY,EAAE,EAAE,2BAA2B,EAAE,OAAO,EAAE;KACvD,CAAC;IACF,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;QAChC,GAAG,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CACpB,MAAM,QAAQ,CACZ,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,qDAAqD,CAAC,EACtE,MAAM,CACP,CACF,CAAC;QACF,IAAI,GAAG,CAAC,OAAO,KAAK,OAAO;YAAE,OAAO,IAAI,CAAC;IAC3C,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;IACV,MAAM,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACzD,iEAAiE;IACjE,MAAM,UAAU,GAAG;QACjB,OAAO,CAAC,GAAG,CAAC,YAAY;QACxB,IAAI,CAAC,IAAI,CACP,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAC9B,iCAAiC,CAClC;QACD,IAAI,CAAC,OAAO,CACV,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAC9B,wCAAwC,CACzC;QACD,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC;aACxB,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;aACrB,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;YAChB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,iCAAiC,CAAC;YACjD,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,wCAAwC,CAAC;SAC5D,CAAC;QACJ,GAAG,CAAC,OAAO,CAAC,QAAQ,KAAK,OAAO;YAC9B,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,CAAC,sCAAsC,CAAC,CAAC;KAC9C,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClB,IAAI,GAAG,CAAC;IACR,KAAK,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,QAAQ,CAAC,CAAC,CAAC,CAAC;YAClB,GAAG,GAAG,CAAC,CAAC;YACR,MAAM;QACR,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;IACD,IAAI,CAAC,GAAG;QAAE,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACjE,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,KAAK,GAAG,KAAK,CACjB,OAAO,CAAC,QAAQ,EAChB,CAAC,GAAG,EAAE,SAAS,EAAE,YAAY,EAAE,YAAY,EAAE,WAAW,CAAC,EACzD,EAAE,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,CAC1D,CAAC;QACF,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,KAAK,CAAC,IAAI,EAAE,CAAC;YACb,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;QACjC,CAAC,EAAE,MAAM,CAAC,CAAC;QACX,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;YACtB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,CAAC,CAAC,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YACxB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,IAAI,KAAK,CAAC;gBACR,CAAC,CAAC,OAAO,EAAE;gBACX,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,cAAc,IAAI,GAAG,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IACH,OAAO,IAAI,CAAC;AACd,CAAC;AACD,MAAM,OAAO,SAAS;IACpB,IAAI,CAAS;IACb,OAAO,CAOL;IACF,OAAO,CAAS;IAChB,MAAM,CAAU;IAChB,QAAQ,CAAgB;IACxB,MAAM,CAAS;IACf,SAAS,GAAG,IAAI,eAAe,EAAE,CAAC;IAElC,YAAY,IAAY;QACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,OAAO,GAAG,IAAI,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IACD,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;QAClD,IAAI,CAAC,IAAI,CAAC,QAAQ;YAChB,IAAI,CAAC,QAAQ,GAAG,CAAC,KAAK,IAAI,EAAE;gBAC1B,MAAM,QAAQ,GAAG,MAAM,cAAc,CACnC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,EAC/B,IAAI,CAAC,SAAS,CAAC,MAAM,CACtB,CAAC;gBACF,IAAI,IAAI,CAAC,MAAM;oBAAE,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;gBAClD,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CACtB,IAAI,GAAG,CAAC,uBAAuB,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EACjD,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,EAAE,CACpE,CAAC;gBACF,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE;oBACjD,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBAC/B,IAAI,CAAC,CAAC;wBAAE,OAAO;oBACf,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;oBACtB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBACxB,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACxD,CAAC,CAAC,CAAC;gBACH,MAAM,IAAI,GAAG,CAAC,CAAC,EAAE,EAAE;oBACjB,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;wBACtC,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;wBACtB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;oBACd,CAAC;oBACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBACvB,CAAC,CAAC;gBACF,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBAC9B,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;oBAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;oBACnB,IAAI,CAAC,IAAI,KAAK,CAAC,iBAAiB,IAAI,GAAG,CAAC,CAAC,CAAC;gBAC5C,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,EAAE,CAAC;QACP,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI;QACrB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;QAClD,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,GAAG;YAAE,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC;QACxD,OAAO,IAAI,OAAO,CAAW,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC/C,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC;YAC1B,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACxB,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YACjC,CAAC,EAAE,MAAM,CAAC,CAAC;YACX,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YACjD,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC;IACD,KAAK,CAAC,IAAI,EAAE,KAAK,GAAG,KAAK;QACvB,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC7C,CAAC;IACD,MAAM,CAAC,KAAK,EAAE,SAAS;QACrB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;IACnD,CAAC;IACD,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACvB,MAAM,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACrC,MAAM,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC;IACjC,CAAC;CACF"}
|