pi-model-manager 0.1.0

package/tui/model-picker.ts

@@ -0,0 +1,266 @@
+// tui/model-picker.ts
+//
+// 已拉取模型列表的本地选择器：8 行分页 + 直接输入搜索 + 匹配排序。
+
+import type { ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+import { CURSOR_MARKER, Key, matchesKey, truncateToWidth } from "@earendil-works/pi-tui";
+
+const MODEL_PAGE_SIZE = 8;
+const EXACT_MATCH_SCORE = 50_000;
+const PREFIX_MATCH_SCORE = 40_000;
+const SEGMENT_PREFIX_MATCH_SCORE = 30_000;
+const SUBSTRING_MATCH_SCORE = 20_000;
+const FUZZY_MATCH_SCORE = 10_000;
+
+interface RankedModelId {
+	modelId: string;
+	score: number;
+}
+
+function clampSelectedIndex(index: number, modelCount: number): number {
+	if (modelCount <= 0) return 0;
+	return Math.min(Math.max(0, index), modelCount - 1);
+}
+
+function getPageCount(modelCount: number): number {
+	if (modelCount <= 0) return 0;
+	return Math.ceil(modelCount / MODEL_PAGE_SIZE);
+}
+
+function getPageStart(selectedIndex: number): number {
+	return Math.floor(selectedIndex / MODEL_PAGE_SIZE) * MODEL_PAGE_SIZE;
+}
+
+function normalizeSearchText(text: string): string {
+	return text.trim().toLowerCase();
+}
+
+function splitSearchTokens(searchText: string): string[] {
+	const normalized = normalizeSearchText(searchText);
+	if (!normalized) return [];
+	return normalized.split(/\s+/).filter(Boolean);
+}
+
+function splitModelSegments(normalizedModelId: string): string[] {
+	return normalizedModelId.split(/[\s/:._-]+/).filter(Boolean);
+}
+
+function getFuzzySpan(normalizedModelId: string, token: string): number | undefined {
+	let searchStart = 0;
+	let firstMatchIndex = -1;
+	let lastMatchIndex = -1;
+	for (const character of token) {
+		const matchIndex = normalizedModelId.indexOf(character, searchStart);
+		if (matchIndex < 0) return undefined;
+		if (firstMatchIndex < 0) firstMatchIndex = matchIndex;
+		lastMatchIndex = matchIndex;
+		searchStart = matchIndex + character.length;
+	}
+	return lastMatchIndex - firstMatchIndex + 1;
+}
+
+function rankTokenWithinModel(normalizedModelId: string, token: string): number | undefined {
+	if (normalizedModelId === token) return EXACT_MATCH_SCORE;
+	if (normalizedModelId.startsWith(token)) return PREFIX_MATCH_SCORE - normalizedModelId.length;
+
+	const segmentIndex = splitModelSegments(normalizedModelId).findIndex((segment) => segment.startsWith(token));
+	if (segmentIndex >= 0) {
+		return SEGMENT_PREFIX_MATCH_SCORE - segmentIndex * 100 - normalizedModelId.length;
+	}
+
+	const substringIndex = normalizedModelId.indexOf(token);
+	if (substringIndex >= 0) {
+		return SUBSTRING_MATCH_SCORE - substringIndex * 100 - normalizedModelId.length;
+	}
+
+	const fuzzySpan = getFuzzySpan(normalizedModelId, token);
+	if (fuzzySpan === undefined) return undefined;
+	return FUZZY_MATCH_SCORE - fuzzySpan * 10 - normalizedModelId.length;
+}
+
+function rankModelId(modelId: string, searchText: string): number | undefined {
+	const tokens = splitSearchTokens(searchText);
+	if (tokens.length === 0) return 0;
+
+	const normalizedModelId = modelId.toLowerCase();
+	let combinedScore = 0;
+	for (const token of tokens) {
+		const tokenScore = rankTokenWithinModel(normalizedModelId, token);
+		if (tokenScore === undefined) return undefined;
+		combinedScore += tokenScore;
+	}
+	return combinedScore - normalizedModelId.length;
+}
+
+function sortModelIdsForSearch(modelIds: string[], searchText: string): string[] {
+	const tokens = splitSearchTokens(searchText);
+	if (tokens.length === 0) return [...modelIds].sort((a, b) => a.localeCompare(b));
+
+	const rankedModelIds: RankedModelId[] = [];
+	for (const modelId of modelIds) {
+		const score = rankModelId(modelId, searchText);
+		if (score !== undefined) rankedModelIds.push({ modelId, score });
+	}
+
+	rankedModelIds.sort((a, b) => {
+		const scoreOrder = b.score - a.score;
+		if (scoreOrder !== 0) return scoreOrder;
+		const lengthOrder = a.modelId.length - b.modelId.length;
+		if (lengthOrder !== 0) return lengthOrder;
+		return a.modelId.localeCompare(b.modelId);
+	});
+	return rankedModelIds.map((model) => model.modelId);
+}
+
+function isPlainTextInput(data: string): boolean {
+	if (!data) return false;
+	for (const character of data) {
+		const codePoint = character.codePointAt(0);
+		if (codePoint === undefined || codePoint < 32 || codePoint === 127) return false;
+	}
+	return !data.includes("\x1b");
+}
+
+function formatPageText(modelCount: number, selectedIndex: number): string {
+	const pageCount = getPageCount(modelCount);
+	if (pageCount === 0) return "页码 0 / 0";
+	return `页码 ${Math.floor(selectedIndex / MODEL_PAGE_SIZE) + 1} / ${pageCount}`;
+}
+
+export async function pickModelIdFromList(
+	ctx: ExtensionCommandContext,
+	title: string,
+	modelIds: string[],
+	currentModelId: string,
+): Promise<string | undefined> {
+	const sortedModelIds = [...new Set(modelIds)].sort((a, b) => a.localeCompare(b));
+	const normalizedCurrentModelId = currentModelId.trim();
+
+	return ctx.ui.custom<string | undefined>((tui, theme, _keybindings, done) => {
+		let searchText = "";
+		let matchedModelIds = sortModelIdsForSearch(sortedModelIds, searchText);
+		let selectedIndex = clampSelectedIndex(matchedModelIds.indexOf(normalizedCurrentModelId), matchedModelIds.length);
+		let focused = false;
+
+		const requestRender = (): void => {
+			selectedIndex = clampSelectedIndex(selectedIndex, matchedModelIds.length);
+			tui.requestRender();
+		};
+
+		const replaceSearchText = (nextSearchText: string): void => {
+			searchText = nextSearchText;
+			matchedModelIds = sortModelIdsForSearch(sortedModelIds, searchText);
+			const selectedModelId = normalizeSearchText(searchText) ? undefined : normalizedCurrentModelId;
+			const selectedModelIndex = selectedModelId ? matchedModelIds.indexOf(selectedModelId) : -1;
+			selectedIndex = clampSelectedIndex(selectedModelIndex >= 0 ? selectedModelIndex : 0, matchedModelIds.length);
+			requestRender();
+		};
+
+		const moveSelection = (delta: number): void => {
+			selectedIndex = clampSelectedIndex(selectedIndex + delta, matchedModelIds.length);
+			requestRender();
+		};
+
+		const movePage = (delta: number): void => {
+			if (matchedModelIds.length === 0) return;
+			const pageCount = getPageCount(matchedModelIds.length);
+			const currentPageIndex = Math.floor(selectedIndex / MODEL_PAGE_SIZE);
+			const rowOffset = selectedIndex % MODEL_PAGE_SIZE;
+			const nextPageIndex = Math.min(Math.max(0, currentPageIndex + delta), pageCount - 1);
+			selectedIndex = clampSelectedIndex(nextPageIndex * MODEL_PAGE_SIZE + rowOffset, matchedModelIds.length);
+			requestRender();
+		};
+
+		return {
+			get focused(): boolean {
+				return focused;
+			},
+			set focused(value: boolean) {
+				focused = value;
+			},
+			invalidate(): void {},
+			handleInput(data: string): void {
+				if (matchesKey(data, Key.escape) || matchesKey(data, Key.ctrl("c"))) {
+					done(undefined);
+					return;
+				}
+				if (matchesKey(data, Key.enter)) {
+					const selectedModelId = matchedModelIds[selectedIndex];
+					if (selectedModelId) done(selectedModelId);
+					return;
+				}
+				if (matchesKey(data, Key.backspace)) {
+					if (searchText.length > 0) replaceSearchText(Array.from(searchText).slice(0, -1).join(""));
+					return;
+				}
+				if (matchesKey(data, Key.up)) {
+					moveSelection(-1);
+					return;
+				}
+				if (matchesKey(data, Key.down)) {
+					moveSelection(1);
+					return;
+				}
+				if (matchesKey(data, Key.pageUp) || matchesKey(data, Key.left)) {
+					movePage(-1);
+					return;
+				}
+				if (matchesKey(data, Key.pageDown) || matchesKey(data, Key.right)) {
+					movePage(1);
+					return;
+				}
+				if (matchesKey(data, Key.home)) {
+					selectedIndex = 0;
+					requestRender();
+					return;
+				}
+				if (matchesKey(data, Key.end)) {
+					selectedIndex = Math.max(0, matchedModelIds.length - 1);
+					requestRender();
+					return;
+				}
+				if (isPlainTextInput(data)) {
+					replaceSearchText(searchText + data);
+				}
+			},
+			render(width: number): string[] {
+				selectedIndex = clampSelectedIndex(selectedIndex, matchedModelIds.length);
+				const pageStart = getPageStart(selectedIndex);
+				const visibleModelIds = matchedModelIds.slice(pageStart, pageStart + MODEL_PAGE_SIZE);
+				const border = theme.fg("borderMuted", "─".repeat(Math.max(0, Math.min(width, 100))));
+				const queryCursor = focused ? `${CURSOR_MARKER}${theme.fg("accent", "▌")}` : "";
+				const searchLine = searchText
+					? `${theme.fg("muted", "搜索: ")}${searchText}${queryCursor}`
+					: `${theme.fg("muted", "搜索: ")}${queryCursor}${theme.fg("dim", "<直接输入搜索>")}`;
+				const lines: string[] = [
+					border,
+					theme.fg("accent", theme.bold(title)),
+					searchLine,
+					theme.fg("dim", `匹配 ${matchedModelIds.length} / ${sortedModelIds.length} · ${formatPageText(matchedModelIds.length, selectedIndex)}`),
+					"",
+				];
+
+				if (visibleModelIds.length === 0) {
+					lines.push(theme.fg("warning", "没有匹配模型；Backspace 删除搜索词，Esc 返回后可手动输入。"));
+				} else {
+					for (let rowIndex = 0; rowIndex < visibleModelIds.length; rowIndex += 1) {
+						const absoluteIndex = pageStart + rowIndex;
+						const modelId = visibleModelIds[rowIndex]!;
+						const selected = absoluteIndex === selectedIndex;
+						const currentSuffix = modelId === normalizedCurrentModelId ? theme.fg("dim", "  ← 当前") : "";
+						const prefix = selected ? "❯ " : "  ";
+						const rowText = `${prefix}${modelId}${currentSuffix}`;
+						lines.push(selected ? theme.fg("accent", rowText) : rowText);
+					}
+				}
+
+				lines.push(
+					"",
+					theme.fg("dim", "↑↓ 选择 · ←→/PgUp/PgDn 翻页 · 输入搜索 · Backspace 删除 · Enter 确认 · Esc 取消"),
+					border,
+				);
+				return lines.map((line) => truncateToWidth(line, width));
+			},
+		};
+	});
+}

package/tui/models-json-panel.ts

@@ -0,0 +1,269 @@
+// tui/models-json-panel.ts
+//
+// 二级面板：查看 / 管理 pi 的 ~/.pi/agent/models.json。
+//
+// 说明：models.json 是 provider/model 定义权威源；本面板是高级入口。
+// 删除扩展管理中的同名条目时，会同步清理插件元数据、enabledModels 与当前运行时注册。
+
+import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+import { formatUnknownError } from "../common.ts";
+import {
+	MODELS_JSON_PATH,
+	readModelsJson,
+	type ModelsJsonDocument,
+	type ModelsJsonModelEntry,
+	type ModelsJsonProviderEntry,
+} from "../models-json-manager.ts";
+import { deleteModelsJsonModelConfiguration, deleteModelsJsonProviderConfiguration } from "../models-json-mutations.ts";
+import { getModelFullId } from "../state-document.ts";
+import { readState } from "../state-store.ts";
+import { showPersistentShortcutMenu, type MenuCursor } from "./persistent-menu.ts";
+import { formatTableHeader, joinFixedColumns } from "./ui-helpers.ts";
+
+interface ProviderRow {
+	providerId: string;
+	label: string;
+}
+
+interface ProviderMenuRow {
+	modelId: string;
+	model: ModelsJsonModelEntry;
+	label: string;
+}
+
+function modelCount(entry: ModelsJsonProviderEntry): number {
+	return Array.isArray(entry.models) ? entry.models.length : 0;
+}
+
+function formatApiLabel(api: string | undefined): string {
+	if (api === "openai-responses") return "Responses";
+	if (api === "openai-completions") return "Chat";
+	if (api === "anthropic-messages") return "Claude";
+	if (api === "google-generative-ai") return "Gemini";
+	return api ?? "未知";
+}
+
+function formatTokenLimit(value: unknown): string {
+	if (typeof value !== "number" || !Number.isFinite(value)) return "-";
+	if (value >= 1_000_000) return `${(value / 1_000_000).toFixed(value % 1_000_000 === 0 ? 0 : 1)}M`;
+	if (value >= 1_000) return `${Math.round(value / 1_000)}K`;
+	return String(value);
+}
+
+function sanitizeEndpoint(value: string | undefined): string {
+	return (value ?? "<未配置>").replace(/([?&](?:key|api_key|api-key)=)[^&]+/gi, "$1REDACTED");
+}
+
+function formatModelInput(input: unknown): string {
+	return Array.isArray(input) && input.includes("image") ? "文本,视觉" : "文本";
+}
+
+function formatNativeModelNameCell(model: ModelsJsonModelEntry): string {
+	return model.name && model.name !== model.id ? model.name : "默认";
+}
+
+function formatNativeModelThinkingCell(model: ModelsJsonModelEntry): string {
+	return model.reasoning === true ? "开" : "关";
+}
+
+const NATIVE_PROVIDER_HEADER = formatTableHeader(joinFixedColumns([
+	{ text: "Provider", width: 24 },
+	{ text: "API", width: 9 },
+	{ text: "Models", width: 6, align: "right" },
+	{ text: "Source", width: 8 },
+]));
+
+const NATIVE_MODEL_HEADER = formatTableHeader(joinFixedColumns([
+	{ text: "模型 ID", width: 30 },
+	{ text: "显示名", width: 16 },
+	{ text: "输入", width: 10 },
+	{ text: "Thinking", width: 8 },
+	{ text: "上下文", width: 7, align: "right" },
+	{ text: "输出", width: 7, align: "right" },
+	{ text: "Headers", width: 8, align: "right" },
+]));
+
+function formatNativeModelRow(model: ModelsJsonModelEntry): string {
+	return joinFixedColumns([
+		{ text: model.id, width: 30 },
+		{ text: formatNativeModelNameCell(model), width: 16 },
+		{ text: formatModelInput(model.input), width: 10 },
+		{ text: formatNativeModelThinkingCell(model), width: 8 },
+		{ text: formatTokenLimit(model.contextWindow), width: 7, align: "right" },
+		{ text: formatTokenLimit(model.maxTokens), width: 7, align: "right" },
+		{ text: String(Object.keys(model.headers ?? {}).length), width: 8, align: "right" },
+	]);
+}
+
+function formatProviderRow(providerId: string, entry: ModelsJsonProviderEntry, managedInState: boolean): string {
+	return joinFixedColumns([
+		{ text: providerId, width: 24 },
+		{ text: formatApiLabel(entry.api), width: 9 },
+		{ text: String(modelCount(entry)), width: 6, align: "right" },
+		{ text: managedInState ? "managed" : "native", width: 8 },
+	]);
+}
+
+function formatProviderDetailLines(providerId: string, entry: ModelsJsonProviderEntry, managedInState: boolean): string[] {
+	const title = entry.name && entry.name !== providerId ? `${entry.name} (${providerId})` : providerId;
+	return [
+		title,
+		`  endpoint  ${sanitizeEndpoint(entry.baseUrl)}`,
+		`  api       ${formatApiLabel(entry.api)} · ${modelCount(entry)} models · ${managedInState ? "extension metadata" : "native only"}`,
+		`  source    ${MODELS_JSON_PATH}`,
+	];
+}
+
+function buildProviderRows(document: ModelsJsonDocument, managedIds: ReadonlySet<string>): ProviderRow[] {
+	return Object.keys(document.providers)
+		.sort((a, b) => a.localeCompare(b))
+		.map((providerId) => ({
+			providerId,
+			label: formatProviderRow(providerId, document.providers[providerId]!, managedIds.has(providerId)),
+		}));
+}
+
+function buildProviderMenuRows(entry: ModelsJsonProviderEntry): ProviderMenuRow[] {
+	return [...(entry.models ?? [])]
+		.sort((a, b) => a.id.localeCompare(b.id))
+		.map((model) => ({
+			modelId: model.id,
+			model,
+			label: formatNativeModelRow(model),
+		}));
+}
+
+
+async function deleteProvider(pi: ExtensionAPI, ctx: ExtensionCommandContext, providerId: string): Promise<void> {
+	const ok = await ctx.ui.confirm(`删除 ${providerId}`, "将从 models.json 删除整个接入，并同步清理插件元数据与 enabledModels。");
+	if (!ok) return;
+	try {
+		await deleteModelsJsonProviderConfiguration(pi, ctx, providerId);
+	} catch (error) {
+		ctx.ui.notify(`删除失败：${formatUnknownError(error)}`, "error");
+	}
+}
+
+async function deleteModel(pi: ExtensionAPI, ctx: ExtensionCommandContext, providerId: string, modelId: string): Promise<void> {
+	const fullId = getModelFullId(providerId, modelId);
+	const docForPrompt = await readModelsJson();
+	const modelCount = docForPrompt.providers[providerId]?.models?.length ?? 0;
+	const ok = await ctx.ui.confirm(
+		`删除 ${fullId}`,
+		modelCount <= 1 ? "这是该接入下最后一个模型；删除后该接入也会从 models.json 中移除。" : "只从 models.json 的模型列表移除该模型。",
+	);
+	if (!ok) return;
+	try {
+		await deleteModelsJsonModelConfiguration(pi, ctx, providerId, modelId);
+	} catch (error) {
+		ctx.ui.notify(`删除失败：${formatUnknownError(error)}`, "error");
+	}
+}
+
+type ModelsJsonProviderShortcut = "delete-model";
+
+async function showProviderMenu(
+	pi: ExtensionAPI,
+	ctx: ExtensionCommandContext,
+	providerId: string,
+	managedInState: boolean,
+): Promise<void> {
+	const cursor: MenuCursor = { index: 0 };
+	while (true) {
+		const doc = await readModelsJson();
+		const entry = doc.providers[providerId];
+		if (!entry) {
+			ctx.ui.notify(`接入不存在：${providerId}`, "warning");
+			return;
+		}
+		const rows = buildProviderMenuRows(entry);
+		const action = await showPersistentShortcutMenu<ModelsJsonProviderShortcut>(
+			ctx,
+			`/model-manager / 原生配置 / ${providerId}`,
+			"",
+			rows.map((row, index) => ({ id: `${index}`, label: row.label })),
+			cursor,
+			[{ input: "d", shortcut: "delete-model" }],
+			{
+				summaryLines: formatProviderDetailLines(providerId, entry, managedInState).slice(1, 3),
+				tableHeader: NATIVE_MODEL_HEADER,
+				visibleRows: Math.min(12, Math.max(1, rows.length)),
+				footer: "↑↓ 选择   d 删除模型   Esc 返回",
+				emptyLabel: "暂无模型；返回上一级可删除接入",
+			},
+		);
+		if (action.type === "cancel") return;
+		if (action.type === "shortcut") {
+			const selectedModelId = rows[cursor.index]?.modelId;
+			if (!selectedModelId) {
+				ctx.ui.notify("没有可删除的模型；删除接入请返回上一级按 d。", "info");
+				continue;
+			}
+			await deleteModel(pi, ctx, providerId, selectedModelId);
+			continue;
+		}
+		ctx.ui.notify("models.json 模型只能删除；请按 d 删除选中模型。", "info");
+	}
+}
+
+type ModelsJsonShortcut = "delete-provider";
+
+export async function runModelsJsonPanel(pi: ExtensionAPI, ctx: ExtensionCommandContext): Promise<void> {
+	const cursor: MenuCursor = { index: 0 };
+	while (true) {
+		let document: ModelsJsonDocument;
+		let managedIds: Set<string>;
+		try {
+			document = await readModelsJson();
+			managedIds = await readState()
+				.then((state) => new Set(Object.keys(state.providers)))
+				.catch(() => new Set<string>());
+		} catch (error) {
+			ctx.ui.notify(`读取 models.json 失败：${formatUnknownError(error)}`, "error");
+			return;
+		}
+
+		const rows = buildProviderRows(document, managedIds);
+		const menuRows = rows.map((row, index) => ({ id: `${index}`, label: row.label }));
+		const action = await showPersistentShortcutMenu<ModelsJsonShortcut>(
+			ctx,
+			"/model-manager / 原生配置",
+			"",
+			menuRows,
+			cursor,
+			[{ input: "d", shortcut: "delete-provider" }],
+			{
+				summaryLines: [
+					"Pi / workflow 直接读取的最终模型配置",
+					MODELS_JSON_PATH,
+				],
+				tableHeader: NATIVE_PROVIDER_HEADER,
+				getDetailLines: (selectedRow) => {
+					const row = rows[Number.parseInt(selectedRow?.id ?? "", 10)];
+					const entry = row ? document.providers[row.providerId] : undefined;
+					return row && entry ? formatProviderDetailLines(row.providerId, entry, managedIds.has(row.providerId)) : [];
+				},
+				footer: "↑↓ 选择   Enter 查看接入   d 删除接入   Esc 返回",
+				emptyLabel: "models.json 中暂无接入",
+			},
+		);
+		if (action.type === "cancel") return;
+		if (action.type === "shortcut") {
+			const selectedProviderId = rows[cursor.index]?.providerId;
+			if (!selectedProviderId) {
+				ctx.ui.notify("没有可删除的接入。", "info");
+				continue;
+			}
+			await deleteProvider(pi, ctx, selectedProviderId);
+			continue;
+		}
+		const index = Number.parseInt(action.id, 10);
+		if (!Number.isFinite(index) || index < 0 || index >= rows.length) continue;
+		const row = rows[index]!;
+		await showProviderMenu(pi, ctx, row.providerId, managedIds.has(row.providerId));
+	}
+}
+
+export function getModelsJsonPanelPath(): string {
+	return MODELS_JSON_PATH;
+}