@xinizai/pi-image-gen 0.1.0

package/dist/core/global-config.js

@@ -0,0 +1,172 @@
+import { randomUUID } from "node:crypto";
+import { mkdir, readFile, rename, writeFile, chmod } from "node:fs/promises";
+import { homedir } from "node:os";
+import { dirname, resolve } from "node:path";
+import { ImageGenError } from "./errors.js";
+import { normalizeBaseUrl, parseManualCapabilities } from "../utils/config.js";
+export const GLOBAL_DIR = resolve(homedir(), ".pi", "agent", "pi-image-gen");
+export const PROVIDERS_FILE = resolve(GLOBAL_DIR, "providers.json");
+export class GlobalConfigStore {
+    filePath;
+    constructor(filePath = PROVIDERS_FILE) {
+        this.filePath = filePath;
+    }
+    async read() {
+        try {
+            const parsed = JSON.parse(await readFile(this.filePath, "utf8"));
+            if (!Array.isArray(parsed.providers))
+                throw new ImageGenError("invalid_json", "全局图片 Provider 配置缺少 providers 数组。", `检查或修复 ${this.filePath}。`);
+            const providers = parsed.providers.filter(isStoredProvider).map(normalizeStoredProvider);
+            const defaultProviderId = typeof parsed.defaultProviderId === "string" && providers.some((p) => p.id === parsed.defaultProviderId) ? parsed.defaultProviderId : providers[0]?.id;
+            return { version: 1, providers, defaultProviderId };
+        }
+        catch (e) {
+            if (e instanceof Error && "code" in e && e.code === "ENOENT")
+                return { version: 1, providers: [] };
+            throw new ImageGenError("invalid_json", "全局图片 Provider 配置文件无效。", `检查或修复 ${this.filePath}。`);
+        }
+    }
+    async write(config) {
+        await mkdir(dirname(this.filePath), { recursive: true });
+        const tmp = `${this.filePath}.${process.pid}.tmp`;
+        await writeFile(tmp, JSON.stringify(config, null, 2), { encoding: "utf8", mode: 0o600 });
+        await chmod(tmp, 0o600).catch(() => undefined);
+        await rename(tmp, this.filePath);
+        await chmod(this.filePath, 0o600).catch(() => undefined);
+    }
+    async addProvider(input) {
+        const config = await this.read();
+        const now = new Date().toISOString();
+        const id = stableProviderId(input.id ?? input.name, config.providers.map((p) => p.id));
+        const provider = {
+            id,
+            name: input.name.trim(),
+            providerType: "openai-compatible",
+            baseUrl: normalizeBaseUrl(input.baseUrl),
+            apiKey: input.apiKey,
+            defaultModel: input.defaultModel || undefined,
+            defaultEditModel: undefined,
+            defaultVariationModel: undefined,
+            responseFormat: input.responseFormat ?? "auto",
+            manualCapabilities: parseManualCapabilities(input.manualCapabilities),
+            createdAt: now,
+            updatedAt: now,
+            apiKeyUpdatedAt: now,
+        };
+        config.providers.push(provider);
+        if (!config.defaultProviderId)
+            config.defaultProviderId = provider.id;
+        await this.write(config);
+        return provider;
+    }
+    async updateProvider(id, patch) {
+        const config = await this.read();
+        const idx = config.providers.findIndex((p) => p.id === id);
+        if (idx < 0)
+            throw new ImageGenError("model_unavailable", `提供商 ${id} 不存在。`, "请运行 /image-config 查看可用提供商。 ");
+        const current = config.providers[idx];
+        const apiKeyChanged = typeof patch.apiKey === "string" && patch.apiKey !== current.apiKey;
+        const next = {
+            ...current,
+            ...patch,
+            baseUrl: patch.baseUrl ? normalizeBaseUrl(patch.baseUrl) : current.baseUrl,
+            updatedAt: new Date().toISOString(),
+            apiKeyUpdatedAt: apiKeyChanged ? new Date().toISOString() : current.apiKeyUpdatedAt,
+        };
+        config.providers[idx] = next;
+        await this.write(config);
+        return next;
+    }
+    async deleteProvider(id) {
+        const config = await this.read();
+        config.providers = config.providers.filter((p) => p.id !== id);
+        if (config.defaultProviderId === id)
+            config.defaultProviderId = config.providers[0]?.id;
+        await this.write(config);
+    }
+    async setDefaultProvider(id) {
+        const config = await this.read();
+        if (!config.providers.some((p) => p.id === id))
+            throw new ImageGenError("model_unavailable", `Provider ${id} 不存在。`, "先添加 Provider。 ");
+        config.defaultProviderId = id;
+        await this.write(config);
+    }
+    async resolveProvider(idOrName) {
+        const config = await this.read();
+        if (idOrName)
+            return config.providers.find((p) => p.id === idOrName || p.name.toLowerCase() === idOrName.toLowerCase());
+        if (config.defaultProviderId)
+            return config.providers.find((p) => p.id === config.defaultProviderId);
+        if (config.providers.length === 1)
+            return config.providers[0];
+        return undefined;
+    }
+    async toRuntimeConfig(provider, env = process.env) {
+        return {
+            providerId: provider.id,
+            providerName: provider.name,
+            baseUrl: provider.baseUrl,
+            apiKey: provider.apiKey,
+            model: provider.defaultModel,
+            editModel: provider.defaultEditModel,
+            variationModel: provider.defaultVariationModel,
+            responseFormat: provider.responseFormat ?? "auto",
+            providerType: provider.providerType,
+            timeoutMs: Number(env.IMAGE_TIMEOUT_SECONDS ?? 60) * 1000,
+            cacheTtlMs: Number(env.IMAGE_CACHE_TTL_SECONDS ?? 600) * 1000,
+            outputDir: env.IMAGE_OUTPUT_DIR ? resolve(env.IMAGE_OUTPUT_DIR) : resolve(homedir(), ".pi", "agent", "generated"),
+            maxResponseBytes: Number(env.IMAGE_MAX_RESPONSE_BYTES ?? 8_000_000),
+            maxDownloadBytes: Number(env.IMAGE_MAX_DOWNLOAD_BYTES ?? 25_000_000),
+            manualCapabilities: { ...provider.manualCapabilities, ...parseManualCapabilities(env.IMAGE_MODEL_CAPABILITIES) },
+            apiKeyUpdatedAt: provider.apiKeyUpdatedAt,
+        };
+    }
+}
+export function envFallbackProvider(env = process.env) {
+    if (!env.IMAGE_BASE_URL || !env.IMAGE_API_KEY)
+        return undefined;
+    const now = new Date().toISOString();
+    return {
+        id: "env",
+        name: "环境变量配置",
+        providerType: "openai-compatible",
+        baseUrl: normalizeBaseUrl(env.IMAGE_BASE_URL),
+        apiKey: env.IMAGE_API_KEY,
+        defaultModel: env.IMAGE_MODEL || undefined,
+        defaultEditModel: env.IMAGE_EDIT_MODEL || undefined,
+        defaultVariationModel: env.IMAGE_VARIATION_MODEL || undefined,
+        responseFormat: "auto",
+        manualCapabilities: parseManualCapabilities(env.IMAGE_MODEL_CAPABILITIES),
+        createdAt: now,
+        updatedAt: now,
+        apiKeyUpdatedAt: now,
+    };
+}
+export async function resolveRuntimeConfig(providerId, env = process.env, store = new GlobalConfigStore()) {
+    const provider = await store.resolveProvider(providerId);
+    if (provider)
+        return store.toRuntimeConfig(provider, env);
+    const fallback = envFallbackProvider(env);
+    if (fallback)
+        return store.toRuntimeConfig(fallback, env);
+    throw new ImageGenError("missing_config", "还没有配置图片生成提供商。", "运行 /image-config 添加全局图片提供商。 ");
+}
+function isStoredProvider(value) {
+    if (!value || typeof value !== "object")
+        return false;
+    const p = value;
+    return typeof p.id === "string" && typeof p.name === "string" && typeof p.baseUrl === "string" && typeof p.apiKey === "string";
+}
+function normalizeStoredProvider(provider) {
+    return { ...provider, providerType: "openai-compatible", responseFormat: provider.responseFormat ?? "auto", manualCapabilities: provider.manualCapabilities ?? {}, defaultModel: provider.defaultModel || undefined, defaultEditModel: provider.defaultEditModel || undefined, defaultVariationModel: provider.defaultVariationModel || undefined };
+}
+function stableProviderId(seed, existing) {
+    const base = (seed.toLowerCase().replace(/[^a-z0-9]+/g, "-").replace(/^-|-$/g, "") || `custom-${randomUUID().slice(0, 6)}`).slice(0, 40);
+    if (!existing.includes(base))
+        return base;
+    for (let i = 1; i < 1000; i += 1)
+        if (!existing.includes(`${base}-${i}`))
+            return `${base}-${i}`;
+    return `${base}-${randomUUID().slice(0, 6)}`;
+}
+//# sourceMappingURL=global-config.js.map

package/dist/core/global-config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"global-config.js","sourceRoot":"","sources":["../../src/core/global-config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAC7E,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAE/E,MAAM,CAAC,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,CAAC,CAAC;AAC7E,MAAM,CAAC,MAAM,cAAc,GAAG,OAAO,CAAC,UAAU,EAAE,gBAAgB,CAAC,CAAC;AAEpE,MAAM,OAAO,iBAAiB;IACC;IAA7B,YAA6B,WAAW,cAAc;QAAzB,aAAQ,GAAR,QAAQ,CAAiB;IAAG,CAAC;IAE1D,KAAK,CAAC,IAAI;QACR,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAkC,CAAC;YAClG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC;gBAAE,MAAM,IAAI,aAAa,CAAC,cAAc,EAAE,kCAAkC,EAAE,SAAS,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC;YAC7I,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YACzF,MAAM,iBAAiB,GAAG,OAAO,MAAM,CAAC,iBAAiB,KAAK,QAAQ,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;YACjL,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,iBAAiB,EAAE,CAAC;QACtD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,YAAY,KAAK,IAAI,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ;gBAAE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;YACnG,MAAM,IAAI,aAAa,CAAC,cAAc,EAAE,uBAAuB,EAAE,SAAS,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC;QAC9F,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAA4B;QACtC,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACzD,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,QAAQ,IAAI,OAAO,CAAC,GAAG,MAAM,CAAC;QAClD,MAAM,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACzF,MAAM,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC/C,MAAM,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QACjC,MAAM,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAuK;QACvL,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACjC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,EAAE,GAAG,gBAAgB,CAAC,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACvF,MAAM,QAAQ,GAAyB;YACrC,EAAE;YACF,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE;YACvB,YAAY,EAAE,mBAAmB;YACjC,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC;YACxC,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,SAAS;YAC7C,gBAAgB,EAAE,SAAS;YAC3B,qBAAqB,EAAE,SAAS;YAChC,cAAc,EAAE,KAAK,CAAC,cAAc,IAAI,MAAM;YAC9C,kBAAkB,EAAE,uBAAuB,CAAC,KAAK,CAAC,kBAAkB,CAAC;YACrE,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;YACd,eAAe,EAAE,GAAG;SACrB,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChC,IAAI,CAAC,MAAM,CAAC,iBAAiB;YAAE,MAAM,CAAC,iBAAiB,GAAG,QAAQ,CAAC,EAAE,CAAC;QACtE,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACzB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAU,EAAE,KAA8D;QAC7F,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACjC,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3D,IAAI,GAAG,GAAG,CAAC;YAAE,MAAM,IAAI,aAAa,CAAC,mBAAmB,EAAE,OAAO,EAAE,OAAO,EAAE,6BAA6B,CAAC,CAAC;QAC3G,MAAM,OAAO,GAAG,MAAM,CAAC,SAAS,CAAC,GAAG,CAAE,CAAC;QACvC,MAAM,aAAa,GAAG,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC;QAC1F,MAAM,IAAI,GAAyB;YACjC,GAAG,OAAO;YACV,GAAG,KAAK;YACR,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO;YAC1E,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,eAAe,EAAE,aAAa,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,eAAe;SACpF,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;QAC7B,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAU;QAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACjC,MAAM,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/D,IAAI,MAAM,CAAC,iBAAiB,KAAK,EAAE;YAAE,MAAM,CAAC,iBAAiB,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QACxF,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,EAAU;QACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACjC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC;YAAE,MAAM,IAAI,aAAa,CAAC,mBAAmB,EAAE,YAAY,EAAE,OAAO,EAAE,gBAAgB,CAAC,CAAC;QACtI,MAAM,CAAC,iBAAiB,GAAG,EAAE,CAAC;QAC9B,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,QAAiB;QACrC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACjC,IAAI,QAAQ;YAAE,OAAO,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;QACxH,IAAI,MAAM,CAAC,iBAAiB;YAAE,OAAO,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,iBAAiB,CAAC,CAAC;QACrG,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAC9D,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,QAA8B,EAAE,MAAyB,OAAO,CAAC,GAAG;QACxF,OAAO;YACL,UAAU,EAAE,QAAQ,CAAC,EAAE;YACvB,YAAY,EAAE,QAAQ,CAAC,IAAI;YAC3B,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,KAAK,EAAE,QAAQ,CAAC,YAAY;YAC5B,SAAS,EAAE,QAAQ,CAAC,gBAAgB;YACpC,cAAc,EAAE,QAAQ,CAAC,qBAAqB;YAC9C,cAAc,EAAE,QAAQ,CAAC,cAAc,IAAI,MAAM;YACjD,YAAY,EAAE,QAAQ,CAAC,YAAY;YACnC,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,qBAAqB,IAAI,EAAE,CAAC,GAAG,IAAI;YACzD,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC,GAAG,IAAI;YAC7D,SAAS,EAAE,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC;YACjH,gBAAgB,EAAE,MAAM,CAAC,GAAG,CAAC,wBAAwB,IAAI,SAAS,CAAC;YACnE,gBAAgB,EAAE,MAAM,CAAC,GAAG,CAAC,wBAAwB,IAAI,UAAU,CAAC;YACpE,kBAAkB,EAAE,EAAE,GAAG,QAAQ,CAAC,kBAAkB,EAAE,GAAG,uBAAuB,CAAC,GAAG,CAAC,wBAAwB,CAAC,EAAE;YAChH,eAAe,EAAE,QAAQ,CAAC,eAAe;SAC1C,CAAC;IACJ,CAAC;CACF;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAyB,OAAO,CAAC,GAAG;IACtE,IAAI,CAAC,GAAG,CAAC,cAAc,IAAI,CAAC,GAAG,CAAC,aAAa;QAAE,OAAO,SAAS,CAAC;IAChE,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACrC,OAAO;QACL,EAAE,EAAE,KAAK;QACT,IAAI,EAAE,QAAQ;QACd,YAAY,EAAE,mBAAmB;QACjC,OAAO,EAAE,gBAAgB,CAAC,GAAG,CAAC,cAAc,CAAC;QAC7C,MAAM,EAAE,GAAG,CAAC,aAAa;QACzB,YAAY,EAAE,GAAG,CAAC,WAAW,IAAI,SAAS;QAC1C,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,IAAI,SAAS;QACnD,qBAAqB,EAAE,GAAG,CAAC,qBAAqB,IAAI,SAAS;QAC7D,cAAc,EAAE,MAAM;QACtB,kBAAkB,EAAE,uBAAuB,CAAC,GAAG,CAAC,wBAAwB,CAAC;QACzE,SAAS,EAAE,GAAG;QACd,SAAS,EAAE,GAAG;QACd,eAAe,EAAE,GAAG;KACrB,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,UAAmB,EAAE,MAAyB,OAAO,CAAC,GAAG,EAAE,KAAK,GAAG,IAAI,iBAAiB,EAAE;IACnI,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IACzD,IAAI,QAAQ;QAAE,OAAO,KAAK,CAAC,eAAe,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC1D,MAAM,QAAQ,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;IAC1C,IAAI,QAAQ;QAAE,OAAO,KAAK,CAAC,eAAe,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC1D,MAAM,IAAI,aAAa,CAAC,gBAAgB,EAAE,eAAe,EAAE,8BAA8B,CAAC,CAAC;AAC7F,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,MAAM,CAAC,GAAG,KAAsC,CAAC;IACjD,OAAO,OAAO,CAAC,CAAC,EAAE,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,MAAM,KAAK,QAAQ,CAAC;AACjI,CAAC;AAED,SAAS,uBAAuB,CAAC,QAA8B;IAC7D,OAAO,EAAE,GAAG,QAAQ,EAAE,YAAY,EAAE,mBAAmB,EAAE,cAAc,EAAE,QAAQ,CAAC,cAAc,IAAI,MAAM,EAAE,kBAAkB,EAAE,QAAQ,CAAC,kBAAkB,IAAI,EAAE,EAAE,YAAY,EAAE,QAAQ,CAAC,YAAY,IAAI,SAAS,EAAE,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB,IAAI,SAAS,EAAE,qBAAqB,EAAE,QAAQ,CAAC,qBAAqB,IAAI,SAAS,EAAE,CAAC;AACtV,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAY,EAAE,QAAkB;IACxD,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,UAAU,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACzI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IAC1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC;QAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,IAAI,IAAI,CAAC,EAAE,CAAC;YAAE,OAAO,GAAG,IAAI,IAAI,CAAC,EAAE,CAAC;IAChG,OAAO,GAAG,IAAI,IAAI,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AAC/C,CAAC"}

package/dist/core/image-service.d.ts

@@ -0,0 +1,14 @@
+import { ModelDiscoveryCache } from "./cache.js";
+import { ModelRegistry } from "./model-registry.js";
+import type { DiscoveryResult, EditParams, GenerateParams, ImageGenConfig, ImageOperationResult, ImageProvider, VariationParams } from "./types.js";
+export declare class ImageService {
+    private readonly config;
+    private readonly provider;
+    private readonly cache;
+    constructor(config: ImageGenConfig, provider: ImageProvider, cache?: ModelDiscoveryCache);
+    discover(refresh?: boolean, signal?: AbortSignal): Promise<DiscoveryResult>;
+    registry(refresh?: boolean, signal?: AbortSignal): Promise<ModelRegistry>;
+    generate(params: GenerateParams, signal?: AbortSignal): Promise<ImageOperationResult>;
+    edit(params: EditParams, signal?: AbortSignal): Promise<ImageOperationResult>;
+    variation(params: VariationParams, signal?: AbortSignal): Promise<ImageOperationResult>;
+}

package/dist/core/image-service.js

@@ -0,0 +1,42 @@
+import { ModelDiscoveryCache } from "./cache.js";
+import { ModelRegistry } from "./model-registry.js";
+export class ImageService {
+    config;
+    provider;
+    cache;
+    constructor(config, provider, cache = new ModelDiscoveryCache()) {
+        this.config = config;
+        this.provider = provider;
+        this.cache = cache;
+    }
+    async discover(refresh = false, signal) {
+        if (!refresh) {
+            const cached = await this.cache.get(this.config);
+            if (cached)
+                return cached;
+        }
+        const models = await this.provider.discoverModels(signal);
+        const result = { providerId: this.config.providerId, providerName: this.config.providerName, providerType: this.provider.type, baseUrl: this.config.baseUrl, models, discoveredAt: new Date().toISOString(), apiKeyFingerprint: this.cache.apiKeyFingerprint(this.config) };
+        await this.cache.set(this.config, result);
+        return result;
+    }
+    async registry(refresh = false, signal) {
+        return new ModelRegistry((await this.discover(refresh, signal)).models);
+    }
+    async generate(params, signal) {
+        const registry = await this.registry(false, signal);
+        const model = registry.select("image_generation", params.model, this.config.model);
+        return this.provider.generate({ ...params, model: model.id, modelCapabilities: model }, signal);
+    }
+    async edit(params, signal) {
+        const registry = await this.registry(false, signal);
+        const model = registry.select("image_edit", params.model, this.config.editModel ?? this.config.model);
+        return this.provider.edit({ ...params, model: model.id, modelCapabilities: model }, signal);
+    }
+    async variation(params, signal) {
+        const registry = await this.registry(false, signal);
+        const model = registry.select("image_variation", params.model, this.config.variationModel ?? this.config.model);
+        return this.provider.variation({ ...params, model: model.id, modelCapabilities: model }, signal);
+    }
+}
+//# sourceMappingURL=image-service.js.map

package/dist/core/image-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image-service.js","sourceRoot":"","sources":["../../src/core/image-service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAGpD,MAAM,OAAO,YAAY;IACM;IAAyC;IAA0C;IAAhH,YAA6B,MAAsB,EAAmB,QAAuB,EAAmB,QAAQ,IAAI,mBAAmB,EAAE;QAApH,WAAM,GAAN,MAAM,CAAgB;QAAmB,aAAQ,GAAR,QAAQ,CAAe;QAAmB,UAAK,GAAL,KAAK,CAA4B;IAAG,CAAC;IAErJ,KAAK,CAAC,QAAQ,CAAC,OAAO,GAAG,KAAK,EAAE,MAAoB;QAClD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACjD,IAAI,MAAM;gBAAE,OAAO,MAAM,CAAC;QAC5B,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAoB,EAAE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,iBAAiB,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7R,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAO,GAAG,KAAK,EAAE,MAAoB;QAClD,OAAO,IAAI,aAAa,CAAC,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAsB,EAAE,MAAoB;QACzD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACpD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,kBAAkB,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACnF,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,GAAG,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE,EAAE,MAAM,CAAC,CAAC;IAClG,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAkB,EAAE,MAAoB;QACjD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACpD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACtG,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,GAAG,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE,EAAE,MAAM,CAAC,CAAC;IAC9F,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAuB,EAAE,MAAoB;QAC3D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACpD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,iBAAiB,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAChH,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,GAAG,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE,EAAE,MAAM,CAAC,CAAC;IACnG,CAAC;CACF"}

package/dist/core/model-browser.d.ts

@@ -0,0 +1,5 @@
+import type { DiscoveryResult, ModelBrowserOptions } from "./types.js";
+export declare const DEFAULT_MODEL_LIMIT = 20;
+export declare const MAX_MODEL_LIMIT = 50;
+export declare function renderModelBrowser(result: DiscoveryResult, options: ModelBrowserOptions): string;
+export declare function renderRefreshSummary(result: DiscoveryResult): string;

package/dist/core/model-browser.js

@@ -0,0 +1,76 @@
+import { ModelRegistry } from "./model-registry.js";
+export const DEFAULT_MODEL_LIMIT = 20;
+export const MAX_MODEL_LIMIT = 50;
+export function renderModelBrowser(result, options) {
+    const limit = Math.min(Math.max(1, options.limit), MAX_MODEL_LIMIT);
+    const page = Math.max(1, options.page);
+    const registry = new ModelRegistry(result.models);
+    const all = registry.all();
+    const keyword = options.search?.trim().toLowerCase();
+    const filtered = keyword ? all.filter((m) => `${m.id} ${JSON.stringify(m.rawMetadata)}`.toLowerCase().includes(keyword)) : all;
+    const groups = [["image_generation", "图片生成"], ["image_edit", "图片编辑"], ["image_variation", "图片变体"], ["vision", "视觉理解"], ["unknown", "未知能力"]];
+    if (options.includeText || keyword)
+        groups.push(["text", "文本模型"]);
+    const lines = [
+        `提供商：${result.providerName} (${result.providerId})`,
+        `模型总数：${all.length}`,
+        keyword ? `搜索：${keyword}` : "默认视图：只显示图片相关模型",
+        `显示限制：每组 ${limit} 个，第 ${page} 页`,
+        "",
+    ];
+    for (const [cap, label] of groups) {
+        const models = filtered.filter((m) => m.capabilities.includes(cap));
+        const start = (page - 1) * limit;
+        const shown = models.slice(start, start + limit);
+        const totalPages = Math.max(1, Math.ceil(models.length / limit));
+        lines.push(`${label}（${models.length}）${models.length > limit ? ` - 第 ${Math.min(page, totalPages)} / ${totalPages} 页` : ""}`);
+        if (shown.length === 0)
+            lines.push("- 无");
+        else
+            shown.forEach((m, i) => lines.push(`${start + i + 1}. ${formatModel(m)}`));
+        if (models.length > start + shown.length)
+            lines.push(`... 还有 ${models.length - start - shown.length} 个。请在 /image-config 中使用“搜索模型”或调整分页查看。`);
+        lines.push("");
+    }
+    try {
+        lines.push(`当前默认生图模型：${registry.select("image_generation").id}`);
+    }
+    catch {
+        lines.push("当前默认生图模型：无");
+    }
+    if (all.length > filtered.length)
+        lines.push(`已过滤：${all.length - filtered.length}`);
+    if (!keyword)
+        lines.push("请在 /image-config 中使用“搜索模型”搜索完整模型列表。");
+    return truncateLines(lines, 180, 12_000);
+}
+export function renderRefreshSummary(result) {
+    const summary = new ModelRegistry(result.models).summary();
+    return [
+        `提供商已刷新：${result.providerName} (${result.providerId})`,
+        `模型总数：${result.models.length}`,
+        `图片生成：${summary.image_generation.length}`,
+        `图片编辑：${summary.image_edit.length}`,
+        `图片变体：${summary.image_variation.length}`,
+        `视觉理解：${summary.vision.length}`,
+        `未知能力：${summary.unknown.length}`,
+    ].join("\n");
+}
+function formatModel(model) {
+    const source = model.capabilitySource === "heuristic" ? " [heuristic]" : model.capabilitySource === "manual" ? " [manual]" : "";
+    return `${model.id}${source}`;
+}
+function truncateLines(lines, maxLines, maxChars) {
+    const out = [];
+    let chars = 0;
+    for (const line of lines) {
+        if (out.length >= maxLines || chars + line.length + 1 > maxChars) {
+            out.push("... 输出已截断。请在 /image-config 中使用搜索、分页或选择提供商缩小结果。");
+            break;
+        }
+        out.push(line);
+        chars += line.length + 1;
+    }
+    return out.join("\n");
+}
+//# sourceMappingURL=model-browser.js.map

package/dist/core/model-browser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-browser.js","sourceRoot":"","sources":["../../src/core/model-browser.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,MAAM,CAAC,MAAM,mBAAmB,GAAG,EAAE,CAAC;AACtC,MAAM,CAAC,MAAM,eAAe,GAAG,EAAE,CAAC;AAElC,MAAM,UAAU,kBAAkB,CAAC,MAAuB,EAAE,OAA4B;IACtF,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,CAAC,EAAE,eAAe,CAAC,CAAC;IACpE,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IACvC,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAClD,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC;IAC3B,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACrD,MAAM,QAAQ,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;IAC/H,MAAM,MAAM,GAAqC,CAAC,CAAC,kBAAkB,EAAE,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,MAAM,CAAC,EAAE,CAAC,iBAAiB,EAAE,MAAM,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC;IAC9K,IAAI,OAAO,CAAC,WAAW,IAAI,OAAO;QAAE,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IAElE,MAAM,KAAK,GAAG;QACZ,OAAO,MAAM,CAAC,YAAY,KAAK,MAAM,CAAC,UAAU,GAAG;QACnD,QAAQ,GAAG,CAAC,MAAM,EAAE;QACpB,OAAO,CAAC,CAAC,CAAC,MAAM,OAAO,EAAE,CAAC,CAAC,CAAC,gBAAgB;QAC5C,WAAW,KAAK,QAAQ,IAAI,IAAI;QAChC,EAAE;KACH,CAAC;IAEF,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;QACpE,MAAM,KAAK,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;QACjC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,KAAK,CAAC,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC;QACjE,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,MAAM,UAAU,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC/H,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;;YACrC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,GAAG,CAAC,KAAK,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAChF,IAAI,MAAM,CAAC,MAAM,GAAG,KAAK,GAAG,KAAK,CAAC,MAAM;YAAE,KAAK,CAAC,IAAI,CAAC,UAAU,MAAM,CAAC,MAAM,GAAG,KAAK,GAAG,KAAK,CAAC,MAAM,uCAAuC,CAAC,CAAC;QAC5I,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC;IAED,IAAI,CAAC;QAAC,KAAK,CAAC,IAAI,CAAC,YAAY,QAAQ,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAAC,CAAC;IAAC,MAAM,CAAC;QAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAAC,CAAC;IAC7G,IAAI,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC,MAAM;QAAE,KAAK,CAAC,IAAI,CAAC,OAAO,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IACpF,IAAI,CAAC,OAAO;QAAE,KAAK,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;IAChE,OAAO,aAAa,CAAC,KAAK,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAuB;IAC1D,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,CAAC;IAC3D,OAAO;QACL,UAAU,MAAM,CAAC,YAAY,KAAK,MAAM,CAAC,UAAU,GAAG;QACtD,QAAQ,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE;QAC9B,QAAQ,OAAO,CAAC,gBAAgB,CAAC,MAAM,EAAE;QACzC,QAAQ,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE;QACnC,QAAQ,OAAO,CAAC,eAAe,CAAC,MAAM,EAAE;QACxC,QAAQ,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE;QAC/B,QAAQ,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE;KACjC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,WAAW,CAAC,KAAgB;IACnC,MAAM,MAAM,GAAG,KAAK,CAAC,gBAAgB,KAAK,WAAW,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,gBAAgB,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;IAChI,OAAO,GAAG,KAAK,CAAC,EAAE,GAAG,MAAM,EAAE,CAAC;AAChC,CAAC;AAED,SAAS,aAAa,CAAC,KAAe,EAAE,QAAgB,EAAE,QAAgB;IACxE,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,GAAG,CAAC,MAAM,IAAI,QAAQ,IAAI,KAAK,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,GAAG,QAAQ,EAAE,CAAC;YACjE,GAAG,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;YAC3D,MAAM;QACR,CAAC;QACD,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,KAAK,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IAC3B,CAAC;IACD,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACxB,CAAC"}

package/dist/core/model-registry.d.ts

@@ -0,0 +1,10 @@
+import type { ModelCapability, ModelInfo } from "./types.js";
+export declare class ModelRegistry {
+    private readonly models;
+    constructor(models: ModelInfo[]);
+    all(): ModelInfo[];
+    byCapability(cap: ModelCapability): ModelInfo[];
+    get(id: string): ModelInfo | undefined;
+    select(capability: ModelCapability, explicit?: string, configured?: string): ModelInfo;
+    summary(): Record<ModelCapability, string[]>;
+}

package/dist/core/model-registry.js

@@ -0,0 +1,44 @@
+import { ImageGenError } from "./errors.js";
+export class ModelRegistry {
+    models;
+    constructor(models) {
+        this.models = models;
+    }
+    all() { return [...this.models].sort((a, b) => a.id.localeCompare(b.id)); }
+    byCapability(cap) { return this.all().filter((m) => m.capabilities.includes(cap)); }
+    get(id) { return this.models.find((m) => m.id === id); }
+    select(capability, explicit, configured) {
+        const preferred = explicit || configured;
+        if (preferred) {
+            const model = this.get(preferred);
+            if (!model)
+                throw new ImageGenError("model_unavailable", `模型 ${preferred} 不在已发现模型列表中。`, "请在 /image-config 中刷新模型，或确认模型名称。 ");
+            if (!model.capabilities.includes(capability) && !model.capabilities.includes("unknown")) {
+                throw new ImageGenError("model_unavailable", `模型 ${preferred} 未声明支持 ${capability}。`, "请在 /image-config 中查看可用模型，或用 IMAGE_MODEL_CAPABILITIES 手动声明能力。 ");
+            }
+            if (model.capabilities.includes("unknown")) {
+                throw new ImageGenError("capability_unknown", `无法自动判断模型 ${preferred} 是否支持 ${capability}。`, "请设置 IMAGE_MODEL_CAPABILITIES='model:image_generation' 之类的手动能力声明。 ");
+            }
+            return model;
+        }
+        const capable = this.byCapability(capability);
+        if (capable.length > 0)
+            return capable[0];
+        const unknown = this.byCapability("unknown");
+        if (unknown.length > 0) {
+            throw new ImageGenError("capability_unknown", `没有可确认支持 ${capability} 的模型，但发现 unknown 模型。`, `请在 /image-config 中查看模型，并用 IMAGE_MODEL 或 IMAGE_MODEL_CAPABILITIES 手动指定。`);
+        }
+        throw new ImageGenError("model_unavailable", `没有发现支持 ${capability} 的模型。`, "检查 API 是否支持图片接口，或手动配置对应模型。 ");
+    }
+    summary() {
+        return {
+            image_generation: this.byCapability("image_generation").map((m) => m.id),
+            image_edit: this.byCapability("image_edit").map((m) => m.id),
+            image_variation: this.byCapability("image_variation").map((m) => m.id),
+            vision: this.byCapability("vision").map((m) => m.id),
+            text: this.byCapability("text").map((m) => m.id),
+            unknown: this.byCapability("unknown").map((m) => m.id),
+        };
+    }
+}
+//# sourceMappingURL=model-registry.js.map

package/dist/core/model-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry.js","sourceRoot":"","sources":["../../src/core/model-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAG5C,MAAM,OAAO,aAAa;IACK;IAA7B,YAA6B,MAAmB;QAAnB,WAAM,GAAN,MAAM,CAAa;IAAG,CAAC;IAEpD,GAAG,KAAkB,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACxF,YAAY,CAAC,GAAoB,IAAiB,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAClH,GAAG,CAAC,EAAU,IAA2B,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;IAEvF,MAAM,CAAC,UAA2B,EAAE,QAAiB,EAAE,UAAmB;QACxE,MAAM,SAAS,GAAG,QAAQ,IAAI,UAAU,CAAC;QACzC,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAClC,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,aAAa,CAAC,mBAAmB,EAAE,MAAM,SAAS,cAAc,EAAE,kCAAkC,CAAC,CAAC;YAC5H,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACxF,MAAM,IAAI,aAAa,CAAC,mBAAmB,EAAE,MAAM,SAAS,UAAU,UAAU,GAAG,EAAE,+DAA+D,CAAC,CAAC;YACxJ,CAAC;YACD,IAAI,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAI,aAAa,CAAC,oBAAoB,EAAE,YAAY,SAAS,SAAS,UAAU,GAAG,EAAE,mEAAmE,CAAC,CAAC;YAClK,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QAC9C,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,OAAO,CAAC,CAAC,CAAE,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,aAAa,CAAC,oBAAoB,EAAE,WAAW,UAAU,sBAAsB,EAAE,wEAAwE,CAAC,CAAC;QACvK,CAAC;QACD,MAAM,IAAI,aAAa,CAAC,mBAAmB,EAAE,UAAU,UAAU,OAAO,EAAE,6BAA6B,CAAC,CAAC;IAC3G,CAAC;IAED,OAAO;QACL,OAAO;YACL,gBAAgB,EAAE,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxE,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5D,eAAe,EAAE,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtE,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACpD,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAChD,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACvD,CAAC;IACJ,CAAC;CACF"}

package/dist/core/provider.d.ts

@@ -0,0 +1,2 @@
+import type { ImageGenConfig, ImageProvider } from "./types.js";
+export declare function resolveProvider(config: ImageGenConfig): ImageProvider;

package/dist/core/provider.js

@@ -0,0 +1,8 @@
+import { ImageGenError } from "./errors.js";
+import { OpenAICompatibleProvider } from "../providers/openai-compatible.js";
+export function resolveProvider(config) {
+    if (config.providerType === "openai-compatible")
+        return new OpenAICompatibleProvider(config);
+    throw new ImageGenError("unsupported_provider", `不支持的 Provider：${config.providerType}`, "当前版本只实现 OpenAI-compatible。 ");
+}
+//# sourceMappingURL=provider.js.map

package/dist/core/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../src/core/provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,OAAO,EAAE,wBAAwB,EAAE,MAAM,mCAAmC,CAAC;AAE7E,MAAM,UAAU,eAAe,CAAC,MAAsB;IACpD,IAAI,MAAM,CAAC,YAAY,KAAK,mBAAmB;QAAE,OAAO,IAAI,wBAAwB,CAAC,MAAM,CAAC,CAAC;IAC7F,MAAM,IAAI,aAAa,CAAC,sBAAsB,EAAE,iBAAiB,MAAM,CAAC,YAAY,EAAE,EAAE,6BAA6B,CAAC,CAAC;AACzH,CAAC"}

package/dist/core/types.d.ts

@@ -0,0 +1,122 @@
+export type ModelCapability = "text" | "vision" | "image_generation" | "image_edit" | "image_variation" | "unknown";
+export interface ModelInfo {
+    id: string;
+    provider: string;
+    capabilities: ModelCapability[];
+    capabilitySource: "metadata" | "heuristic" | "manual" | "unknown";
+    inputModalities: string[];
+    outputModalities: string[];
+    supportsSize: boolean;
+    supportsQuality: boolean;
+    supportsAspectRatio: boolean;
+    supportsImageInput: boolean;
+    supportsMultipleImages: boolean;
+    rawMetadata: unknown;
+}
+export type ImageResponseFormat = "auto" | "url" | "b64_json";
+export interface StoredProviderConfig {
+    id: string;
+    name: string;
+    providerType: "openai-compatible";
+    baseUrl: string;
+    apiKey: string;
+    defaultModel?: string | undefined;
+    defaultEditModel?: string | undefined;
+    defaultVariationModel?: string | undefined;
+    /** auto: 优先 url，服务端明确拒绝时仅重试一次 b64_json。 */
+    responseFormat?: ImageResponseFormat | undefined;
+    manualCapabilities: Record<string, ModelCapability[]>;
+    createdAt: string;
+    updatedAt: string;
+    apiKeyUpdatedAt: string;
+}
+export interface GlobalImageGenConfig {
+    version: 1;
+    defaultProviderId?: string | undefined;
+    providers: StoredProviderConfig[];
+}
+export interface ImageGenConfig {
+    providerId: string;
+    providerName: string;
+    baseUrl: string;
+    apiKey: string;
+    model?: string | undefined;
+    editModel?: string | undefined;
+    variationModel?: string | undefined;
+    responseFormat: ImageResponseFormat;
+    providerType: "openai-compatible";
+    timeoutMs: number;
+    cacheTtlMs: number;
+    outputDir: string;
+    maxResponseBytes: number;
+    maxDownloadBytes: number;
+    manualCapabilities: Record<string, ModelCapability[]>;
+    apiKeyUpdatedAt: string;
+}
+export interface DiscoveryResult {
+    providerId: string;
+    providerName: string;
+    providerType: string;
+    baseUrl: string;
+    models: ModelInfo[];
+    discoveredAt: string;
+    apiKeyFingerprint: string;
+}
+export interface GeneratedImage {
+    path: string;
+    url?: string;
+    mimeType: string;
+    index: number;
+}
+export interface ImageOperationResult {
+    provider: string;
+    providerId: string;
+    model: string;
+    images: GeneratedImage[];
+    rawInfo?: Record<string, unknown>;
+}
+export interface GenerateParams {
+    prompt: string;
+    provider?: string;
+    model?: string;
+    size?: string;
+    aspect_ratio?: string;
+    quality?: string;
+    output_format?: "png" | "jpeg" | "jpg" | "webp";
+    n?: number;
+    modelCapabilities?: Pick<ModelInfo, "supportsSize" | "supportsQuality" | "supportsAspectRatio" | "supportsMultipleImages">;
+}
+export interface EditParams {
+    image: string;
+    prompt: string;
+    provider?: string;
+    model?: string;
+    size?: string;
+    output_format?: "png" | "jpeg" | "jpg" | "webp";
+    modelCapabilities?: Pick<ModelInfo, "supportsSize">;
+}
+export interface VariationParams {
+    image: string;
+    provider?: string;
+    model?: string;
+    size?: string;
+    n?: number;
+    modelCapabilities?: Pick<ModelInfo, "supportsSize" | "supportsMultipleImages">;
+}
+export interface ModelBrowserOptions {
+    provider?: string;
+    search?: string;
+    page: number;
+    limit: number;
+    refresh: boolean;
+    refreshAll: boolean;
+    includeText: boolean;
+}
+export interface ImageProvider {
+    readonly type: string;
+    readonly name: string;
+    discoverModels(signal?: AbortSignal): Promise<ModelInfo[]>;
+    generate(params: GenerateParams, signal?: AbortSignal): Promise<ImageOperationResult>;
+    edit(params: EditParams, signal?: AbortSignal): Promise<ImageOperationResult>;
+    variation(params: VariationParams, signal?: AbortSignal): Promise<ImageOperationResult>;
+}

package/dist/core/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/core/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":""}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+export default function imageGenExtension(pi: ExtensionAPI): void;