61b5ef30a9
* refactor(memhub): trust embedded SDK native stores Co-authored-by: Cursor <cursoragent@cursor.com> * refactor(memhub): remove model content permission UX Co-authored-by: Cursor <cursoragent@cursor.com> --------- Co-authored-by: alvinfei <alvinfei@tencent.com> Co-authored-by: Cursor <cursoragent@cursor.com>
462 lines
14 KiB
TypeScript
462 lines
14 KiB
TypeScript
import type { ModelProfile } from "@one-memory/contracts";
|
||
import type { CoreReader } from "./core-client.js";
|
||
import type {
|
||
CoreDeleteOperations,
|
||
CoreOperations,
|
||
} from "./core-operations.js";
|
||
import {
|
||
executeModelProfile,
|
||
type FlowDocument,
|
||
planModelProfile,
|
||
} from "./headless-flows.js";
|
||
import { approvalFingerprint, listHosts, readHost } from "./host-flow.js";
|
||
import { parseSelection, type WizardPrompter } from "./init-wizard.js";
|
||
import { executeModelProbe, planModelProbe } from "./model-probe-flow.js";
|
||
import {
|
||
executeModelProfileActivate,
|
||
planModelProfileActivate,
|
||
} from "./model-write-flows.js";
|
||
|
||
type ProviderPreset = {
|
||
id: string;
|
||
label: string;
|
||
provider_kind: "openai_compatible" | "ollama";
|
||
base_url: string;
|
||
model: string;
|
||
key_required: boolean;
|
||
};
|
||
|
||
export const CLI_PROVIDER_PRESETS: readonly ProviderPreset[] = [
|
||
{
|
||
id: "openai",
|
||
label: "OpenAI",
|
||
provider_kind: "openai_compatible",
|
||
base_url: "https://api.openai.com/v1",
|
||
model: "gpt-4o-mini",
|
||
key_required: true,
|
||
},
|
||
{
|
||
id: "deepseek",
|
||
label: "DeepSeek",
|
||
provider_kind: "openai_compatible",
|
||
base_url: "https://api.deepseek.com",
|
||
model: "deepseek-flash",
|
||
key_required: true,
|
||
},
|
||
{
|
||
id: "dashscope",
|
||
label: "阿里云百炼（通义千问）",
|
||
provider_kind: "openai_compatible",
|
||
base_url: "https://dashscope.aliyuncs.com/compatible-mode/v1",
|
||
model: "qwen-flash",
|
||
key_required: true,
|
||
},
|
||
{
|
||
id: "moonshot",
|
||
label: "Kimi（月之暗面）",
|
||
provider_kind: "openai_compatible",
|
||
base_url: "https://api.moonshot.cn/v1",
|
||
model: "kimi-k3",
|
||
key_required: true,
|
||
},
|
||
{
|
||
id: "zhipu",
|
||
label: "智谱 GLM",
|
||
provider_kind: "openai_compatible",
|
||
base_url: "https://open.bigmodel.cn/api/paas/v4",
|
||
model: "glm-4-flash",
|
||
key_required: true,
|
||
},
|
||
{
|
||
id: "volcengine",
|
||
label: "火山方舟（豆包）",
|
||
provider_kind: "openai_compatible",
|
||
base_url: "https://ark.cn-beijing.volces.com/api/v3",
|
||
model: "doubao-seed-2-1-pro-260628",
|
||
key_required: true,
|
||
},
|
||
{
|
||
id: "hunyuan",
|
||
label: "腾讯混元",
|
||
provider_kind: "openai_compatible",
|
||
base_url: "https://api.hunyuan.cloud.tencent.com/v1",
|
||
model: "hunyuan-turbos-latest",
|
||
key_required: true,
|
||
},
|
||
{
|
||
id: "siliconflow",
|
||
label: "硅基流动 SiliconFlow",
|
||
provider_kind: "openai_compatible",
|
||
base_url: "https://api.siliconflow.cn/v1",
|
||
model: "deepseek-ai/DeepSeek-V3",
|
||
key_required: true,
|
||
},
|
||
{
|
||
id: "openrouter",
|
||
label: "OpenRouter",
|
||
provider_kind: "openai_compatible",
|
||
base_url: "https://openrouter.ai/api/v1",
|
||
model: "",
|
||
key_required: true,
|
||
},
|
||
{
|
||
id: "ollama",
|
||
label: "Ollama（本机）",
|
||
provider_kind: "ollama",
|
||
base_url: "http://127.0.0.1:11434",
|
||
model: "",
|
||
key_required: false,
|
||
},
|
||
{
|
||
id: "custom",
|
||
label: "自定义 OpenAI 兼容服务",
|
||
provider_kind: "openai_compatible",
|
||
base_url: "",
|
||
model: "",
|
||
key_required: true,
|
||
},
|
||
];
|
||
|
||
export type InteractiveModelSetupState =
|
||
| {
|
||
state: "configured";
|
||
profile: ModelProfile;
|
||
installed_hosts: string[];
|
||
}
|
||
| { state: "skipped" }
|
||
| { state: "cancelled" }
|
||
| { state: "failed"; code: string };
|
||
|
||
function receiptProfile(document: FlowDocument): ModelProfile | null {
|
||
if (!(document.state === "committed" || document.state === "recovered"))
|
||
return null;
|
||
const value = document.receipt as Partial<ModelProfile>;
|
||
return typeof value.profile_id === "string" &&
|
||
typeof value.revision === "number"
|
||
? (value as ModelProfile)
|
||
: null;
|
||
}
|
||
|
||
function probePassed(value: unknown): boolean {
|
||
if (value === null || typeof value !== "object") return false;
|
||
const receipt = value as {
|
||
state?: unknown;
|
||
receipt?: { result?: { capabilities?: Record<string, unknown> } };
|
||
};
|
||
const capabilities = receipt.receipt?.result?.capabilities;
|
||
return (
|
||
receipt.state === "completed" &&
|
||
capabilities?.model_available === true &&
|
||
capabilities?.embedding_available === true
|
||
);
|
||
}
|
||
|
||
async function answerWithDefault(
|
||
prompter: WizardPrompter,
|
||
question: string,
|
||
fallback: string,
|
||
): Promise<string> {
|
||
const value = (
|
||
await prompter.ask(`${question}（默认 ${fallback}）: `)
|
||
).trim();
|
||
return value || fallback;
|
||
}
|
||
|
||
async function chooseProvider(
|
||
prompter: WizardPrompter,
|
||
): Promise<ProviderPreset | null> {
|
||
prompter.write("选择模型服务：");
|
||
CLI_PROVIDER_PRESETS.forEach((preset, index) => {
|
||
prompter.write(` ${index + 1}) ${preset.label}`);
|
||
});
|
||
prompter.write(" 0) 跳过，暂时只导入原文");
|
||
while (true) {
|
||
const answer = (
|
||
await prompter.ask("请选择（默认 DeepSeek，可输入 back 返回）: ")
|
||
).trim();
|
||
if (answer.toLowerCase() === "back") return null;
|
||
if (answer === "0" || answer.toLowerCase() === "skip") return null;
|
||
const index = answer === "" ? 1 : Number(answer) - 1;
|
||
if (
|
||
Number.isSafeInteger(index) &&
|
||
index >= 0 &&
|
||
index < CLI_PROVIDER_PRESETS.length
|
||
) {
|
||
return CLI_PROVIDER_PRESETS[index]!;
|
||
}
|
||
prompter.write("输入无法识别，请输入服务编号、0 或 back。");
|
||
}
|
||
}
|
||
|
||
async function retryOrSkip(prompter: WizardPrompter): Promise<boolean> {
|
||
while (true) {
|
||
const answer = (await prompter.ask("输入 r 重新填写，或 0 跳过模型配置: "))
|
||
.trim()
|
||
.toLowerCase();
|
||
if (answer === "" || answer === "r" || answer === "retry") return true;
|
||
if (answer === "0" || answer === "skip") return false;
|
||
prompter.write("输入无法识别，请输入 r 或 0。");
|
||
}
|
||
}
|
||
|
||
async function chooseDetectedHosts(
|
||
prompter: WizardPrompter,
|
||
hostHome: string,
|
||
hostApi: {
|
||
list: typeof listHosts;
|
||
read: typeof readHost;
|
||
},
|
||
): Promise<string[]> {
|
||
const listed = await hostApi.list(hostHome);
|
||
if (!("hosts" in listed) || listed.hosts.length === 0) return [];
|
||
const answer = (await prompter.ask("接入检测到的应用？[Y/n]: "))
|
||
.trim()
|
||
.toLowerCase();
|
||
if (["n", "no", "否", "0"].includes(answer)) return [];
|
||
const detected = listed.hosts.filter((host) => host.present);
|
||
if (detected.length === 0) {
|
||
prompter.write(
|
||
"没有检测到可接入的应用，已跳过。之后可运行 memhub host install 接入。",
|
||
);
|
||
return [];
|
||
}
|
||
prompter.write("检测到以下应用：");
|
||
detected.forEach((host, index) => {
|
||
prompter.write(` ${index + 1}) ${host.label}`);
|
||
});
|
||
let selected: number[] | null = null;
|
||
while (selected === null) {
|
||
selected = parseSelection(
|
||
await prompter.ask("选择要接入的应用（回车=全选，如 1,3）: "),
|
||
detected.length,
|
||
);
|
||
if (selected === null)
|
||
prompter.write("输入无法识别，请重新输入编号或 all。");
|
||
}
|
||
return selected.flatMap((index) => {
|
||
const host = detected[index];
|
||
return host ? [host.host_id] : [];
|
||
});
|
||
}
|
||
|
||
/** OpenClaw has a layout but the connection client refuses it, so installing
|
||
* it from the wizard always throws. Keep the list aligned with what the
|
||
* client can actually install instead of failing the whole run.
|
||
*/
|
||
const CLI_INSTALLABLE_HOSTS = new Set(["openclaw"]);
|
||
|
||
/** Hosts the CLI cannot install but the App can. The summary must never call
|
||
* them "failed": the failure is the CLI's, not the host's. */
|
||
const APP_ONLY_HOSTS = new Set(["openclaw"]);
|
||
|
||
async function installHosts(
|
||
dataDir: string,
|
||
hostIds: readonly string[],
|
||
hostApi: { read: typeof readHost },
|
||
): Promise<{ installed: string[]; failed: string[] }> {
|
||
const installed: string[] = [];
|
||
const failed: string[] = [];
|
||
for (const hostId of hostIds) {
|
||
try {
|
||
const preview = await hostApi.read(dataDir, hostId, "previewInstall");
|
||
const fingerprint =
|
||
"result" in preview ? approvalFingerprint(preview.result) : null;
|
||
if (fingerprint === null) {
|
||
failed.push(hostId);
|
||
continue;
|
||
}
|
||
const applied = await hostApi.read(dataDir, hostId, "install", [
|
||
fingerprint,
|
||
]);
|
||
if ("state" in applied) failed.push(hostId);
|
||
else installed.push(hostId);
|
||
} catch (error) {
|
||
failed.push(hostId);
|
||
if (CLI_INSTALLABLE_HOSTS.has(hostId)) continue;
|
||
if (
|
||
!(error instanceof Error) ||
|
||
!error.message.startsWith("connection_client_")
|
||
) {
|
||
throw error;
|
||
}
|
||
}
|
||
}
|
||
return { installed, failed };
|
||
}
|
||
|
||
function writeSummary(
|
||
prompter: WizardPrompter,
|
||
profile: ModelProfile,
|
||
installedHosts: readonly string[],
|
||
failedHosts: readonly { hostId: string; label: string }[],
|
||
): void {
|
||
prompter.write("配置摘要：");
|
||
prompter.write(` 模型：${profile.name} / ${profile.model}`);
|
||
prompter.write(
|
||
` 应用接入：${installedHosts.length > 0 ? installedHosts.join("、") : "已跳过"}`,
|
||
);
|
||
for (const failure of failedHosts) {
|
||
if (APP_ONLY_HOSTS.has(failure.hostId)) {
|
||
prompter.write(` 请在 App 里接入：${failure.label}`);
|
||
continue;
|
||
}
|
||
prompter.write(` 接入失败：${failure.label}`);
|
||
}
|
||
prompter.write(
|
||
"下一步：运行 memhub status 查看状态，或 memhub recall <问题> 查记忆。",
|
||
);
|
||
}
|
||
|
||
export async function runInteractiveModelSetup(
|
||
prompter: WizardPrompter,
|
||
clients: {
|
||
reader: CoreReader;
|
||
operations: CoreOperations & CoreDeleteOperations;
|
||
},
|
||
options: {
|
||
dataDir: string;
|
||
hostHome: string;
|
||
installHosts: boolean;
|
||
hostApi?: { list: typeof listHosts; read: typeof readHost };
|
||
},
|
||
): Promise<InteractiveModelSetupState> {
|
||
if (!prompter.askSecret) return { state: "skipped" };
|
||
while (true) {
|
||
const preset = await chooseProvider(prompter);
|
||
if (preset === null) {
|
||
prompter.write("已跳过模型配置，只保留原文导入。");
|
||
return { state: "skipped" };
|
||
}
|
||
const baseUrl = await answerWithDefault(
|
||
prompter,
|
||
"模型服务地址",
|
||
preset.base_url,
|
||
);
|
||
if (baseUrl.toLowerCase() === "back") continue;
|
||
const model = await answerWithDefault(prompter, "模型名称", preset.model);
|
||
if (model.toLowerCase() === "back") continue;
|
||
if (!model) {
|
||
prompter.write("模型名称不能为空，请重新选择。");
|
||
continue;
|
||
}
|
||
const apiKey = preset.key_required
|
||
? await prompter.askSecret("API Key（输入不会显示）: ")
|
||
: undefined;
|
||
if (apiKey?.toLowerCase() === "back") continue;
|
||
if (preset.key_required && !apiKey) {
|
||
prompter.write("未填写 API Key。输入 back 返回，或重新选择模型服务。");
|
||
continue;
|
||
}
|
||
const hostApi = options.hostApi ?? { list: listHosts, read: readHost };
|
||
const selectedHosts = options.installHosts
|
||
? await chooseDetectedHosts(prompter, options.hostHome, hostApi)
|
||
: [];
|
||
const hostLabels = options.installHosts
|
||
? await hostApi.list(options.hostHome).then((listed) => {
|
||
if (!("hosts" in listed)) return new Map<string, string>();
|
||
return new Map(
|
||
listed.hosts.map((host) => [host.host_id, host.label]),
|
||
);
|
||
})
|
||
: new Map<string, string>();
|
||
const planned = planModelProfile({
|
||
name: preset.label,
|
||
provider_kind: preset.provider_kind,
|
||
base_url: baseUrl,
|
||
model,
|
||
embedding_provider_kind: "local_onnx",
|
||
embedding_model: "bge-small-zh-v1.5",
|
||
embedding_dimensions: 512,
|
||
timeout_seconds: 90,
|
||
max_concurrency: 4,
|
||
disable_thinking: true,
|
||
});
|
||
if (
|
||
planned.state !== "preview" ||
|
||
planned.plan.kind !== "model_profile_create"
|
||
) {
|
||
return { state: "failed", code: "model_config_invalid" };
|
||
}
|
||
const created = await executeModelProfile(
|
||
planned.plan,
|
||
apiKey,
|
||
clients.reader,
|
||
clients.operations,
|
||
);
|
||
const profile = receiptProfile(created);
|
||
if (profile === null) {
|
||
prompter.write("模型配置未保存。");
|
||
if (await retryOrSkip(prompter)) continue;
|
||
return { state: "skipped" };
|
||
}
|
||
|
||
try {
|
||
const probePlan = await planModelProbe(
|
||
clients.reader,
|
||
profile.profile_id,
|
||
);
|
||
const probed =
|
||
probePlan.state === "preview"
|
||
? await executeModelProbe(
|
||
probePlan.plan,
|
||
clients.reader,
|
||
clients.operations,
|
||
)
|
||
: probePlan;
|
||
if (!probePassed(probed)) {
|
||
await clients.operations.delete(
|
||
`/api/v1/model-profiles/${encodeURIComponent(profile.profile_id)}`,
|
||
);
|
||
prompter.write("模型连接未通过。可以重新填写，或输入 0 跳过后续配置。");
|
||
if (await retryOrSkip(prompter)) continue;
|
||
prompter.write("已跳过模型配置，只保留原文导入。");
|
||
return { state: "skipped" };
|
||
}
|
||
} catch (error) {
|
||
await clients.operations
|
||
.delete(
|
||
`/api/v1/model-profiles/${encodeURIComponent(profile.profile_id)}`,
|
||
)
|
||
.catch(() => undefined);
|
||
throw error;
|
||
}
|
||
|
||
const activation = planModelProfileActivate({
|
||
profileId: profile.profile_id,
|
||
});
|
||
if (
|
||
activation.state !== "preview" ||
|
||
activation.plan.kind !== "model_profile_activate"
|
||
) {
|
||
return { state: "failed", code: "model_activate_plan_invalid" };
|
||
}
|
||
const activated = await executeModelProfileActivate(
|
||
activation.plan,
|
||
clients.operations,
|
||
);
|
||
if (activated.state !== "committed") {
|
||
return { state: "failed", code: "model_activate_failed" };
|
||
}
|
||
|
||
const { installed, failed } = await installHosts(
|
||
options.dataDir,
|
||
selectedHosts,
|
||
hostApi,
|
||
);
|
||
writeSummary(
|
||
prompter,
|
||
profile,
|
||
installed,
|
||
failed.map((hostId) => ({
|
||
hostId,
|
||
label: hostLabels.get(hostId) ?? hostId,
|
||
})),
|
||
);
|
||
return {
|
||
state: "configured",
|
||
profile,
|
||
installed_hosts: installed,
|
||
};
|
||
}
|
||
}
|