pi-model-manager 0.1.0

package/rescue.ts

@@ -0,0 +1,69 @@
+// rescue.ts
+//
+// 当前会话模型在以下场景可能失效：
+//   1) 模型被删除（unregister）
+//   2) 模型被改名（同接入内 id 变更）
+//   3) 接入被改名 / 删除
+//   4) 接入的 baseUrl/apiKey 变更后 in-memory 模型 stale
+//
+// withModelRescue 统一处理：
+//   - 优先切换到调用方指定的 preferred（rename 场景）
+//   - 否则切到 modelRegistry.getAvailable()[0]
+//   - 都不行则只 notify warning
+// 调用方负责先同步 registry/runtime，避免救援阶段再次触发全目录刷新。
+
+import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+
+interface AffectedRef {
+	providerId: string;
+	modelId?: string;
+}
+
+export interface RescueOptions {
+	reason: string;
+	preferred?: { providerId: string; modelId: string };
+}
+
+export async function withModelRescue(
+	ctx: ExtensionCommandContext,
+	pi: ExtensionAPI,
+	affected: AffectedRef,
+	options: RescueOptions,
+): Promise<void> {
+	const current = ctx.model;
+	if (!current) return;
+
+	const isAffected = current.provider === affected.providerId
+		&& (affected.modelId === undefined || current.id === affected.modelId);
+	const isUsable = !!ctx.modelRegistry.find(current.provider, current.id)
+		&& ctx.modelRegistry.hasConfiguredAuth(current);
+
+	if (!isAffected && isUsable) return;
+
+	if (options.preferred) {
+		const replacement = ctx.modelRegistry.find(options.preferred.providerId, options.preferred.modelId);
+		if (replacement && ctx.modelRegistry.hasConfiguredAuth(replacement)) {
+			if (await pi.setModel(replacement)) {
+				ctx.ui.notify(
+					`${options.reason}，已切换到 ${options.preferred.providerId}/${options.preferred.modelId}`,
+					"info",
+				);
+				return;
+			}
+		}
+	}
+
+	const fallback = ctx.modelRegistry.getAvailable()[0];
+	if (fallback && await pi.setModel(fallback)) {
+		ctx.ui.notify(
+			`${options.reason}，已自动切换到 ${fallback.provider}/${fallback.id}`,
+			"info",
+		);
+		return;
+	}
+
+	ctx.ui.notify(
+		`${options.reason}，但当前没有其它可用模型。请使用 /model-manager 添加模型或 /login 配置认证。`,
+		"warning",
+	);
+}

package/runtime-base-url.ts

@@ -0,0 +1,54 @@
+// runtime-base-url.ts
+//
+// 将用户在 TUI 中填写的 Base URL 转成 pi 各内置 SDK 实际需要的 baseUrl。
+// 约定：state.json 保存用户输入；注册到 pi 和 bootstrap 到 models.json 时使用运行时 URL。
+
+import type { ApiKind } from "./types.ts";
+
+function trimTrailingSlashes(value: string): string {
+	return value.trim().replace(/\/+$/, "");
+}
+
+function stripTrailingV1(baseUrl: string): string {
+	const trimmed = trimTrailingSlashes(baseUrl);
+	return trimmed.toLowerCase().endsWith("/v1") ? trimmed.slice(0, -3) : trimmed;
+}
+
+function hasRootPathWithoutQueryOrHash(url: URL): boolean {
+	return (url.pathname === "" || url.pathname === "/") && !url.search && !url.hash;
+}
+
+function appendV1ForRootUrl(baseUrl: string): string {
+	const trimmed = trimTrailingSlashes(baseUrl);
+	try {
+		const parsed = new URL(trimmed);
+		if (hasRootPathWithoutQueryOrHash(parsed)) {
+			return `${trimmed}/v1`;
+		}
+	} catch {
+		return trimmed;
+	}
+	return trimmed;
+}
+
+function appendGoogleGenerativeApiVersionForRootUrl(baseUrl: string): string {
+	const trimmed = trimTrailingSlashes(baseUrl);
+	try {
+		const parsed = new URL(trimmed);
+		// Google Generative Language 的根域名不是可直接请求的模型 API 根路径；
+		// pi 的 google-generative-ai 适配器需要带版本段的 baseUrl。
+		if (parsed.hostname === "generativelanguage.googleapis.com" && hasRootPathWithoutQueryOrHash(parsed)) {
+			return `${trimmed}/v1beta`;
+		}
+	} catch {
+		return trimmed;
+	}
+	return trimmed;
+}
+
+export function resolveRuntimeBaseUrl(api: ApiKind, baseUrl: string): string {
+	if (api === "anthropic-messages") return stripTrailingV1(baseUrl);
+	if (api === "openai-completions" || api === "openai-responses") return appendV1ForRootUrl(baseUrl);
+	if (api === "google-generative-ai") return appendGoogleGenerativeApiVersionForRootUrl(baseUrl);
+	return trimTrailingSlashes(baseUrl);
+}

package/state-cache.ts

@@ -0,0 +1,55 @@
+// state-cache.ts
+//
+// 请求路径上的轻量 StateDocument 缓存。外部改动由文件签名检测，插件自身 mutation 还会主动失效。
+
+import { getModelsJsonPath, readFileSignature, sameFileSignature, type FileSignature } from "./models-json-manager.ts";
+import { readState, getStatePath } from "./state-store.ts";
+import type { StateDocument } from "./types.ts";
+
+
+interface StateSignature {
+	modelsJson: FileSignature;
+	metadataState: FileSignature;
+}
+
+let cachedState: { signature: StateSignature; state: StateDocument } | undefined;
+
+
+async function readStateSignature(): Promise<StateSignature> {
+	const [modelsJson, metadataState] = await Promise.all([
+		readFileSignature(getModelsJsonPath()),
+		readFileSignature(getStatePath()),
+	]);
+	return { modelsJson, metadataState };
+}
+
+
+function sameStateSignature(a: StateSignature, b: StateSignature): boolean {
+	return sameFileSignature(a.modelsJson, b.modelsJson)
+		&& sameFileSignature(a.metadataState, b.metadataState);
+}
+
+export function invalidateStateCache(): void {
+	cachedState = undefined;
+}
+
+export async function readCachedState(): Promise<StateDocument> {
+	const signature = await readStateSignature();
+	if (cachedState && sameStateSignature(cachedState.signature, signature)) {
+		return cachedState.state;
+	}
+
+	// StateDocument 横跨两个文件；只缓存同一稳定读取窗口内得到的组合。
+	for (let attempt = 0; attempt < 3; attempt += 1) {
+		const before = await readStateSignature();
+		const state = await readState();
+		const after = await readStateSignature();
+		if (sameStateSignature(before, after)) {
+			cachedState = { signature: after, state };
+			return state;
+		}
+	}
+
+	// 文件持续变化时返回最后一次读取结果但不缓存，下一请求会重新尝试。
+	return readState();
+}