WuYongJie
00cef71be7
## Summary 生图线路确认欠费后立即停止新请求，跳过同一上游账户的其他模型，保留已生成结果和已提交任务。欠费状态须经管理员恢复，不再定时自动解封，也不会被旧任务的成功回执清除。旧模型选择仅使用明确配置且尺寸、比例和参考图数量兼容的备用模型；固定模型模式仍要求用户自行选择。 98K 文生图优先使用其公开声明的任务接口，原生传递分辨率和比例，清除文生图请求中的示例参考图片。 ## Scope - [x] API / backend logic - [x] Settings / admin configuration ## Required Verification 验证普通 Pro、Banana 2、两个 Plus 和 GPT Image 2 共 10 个真实文生图及参考图编辑用例，均获得可下载的 2048×2048 图片。625 项 helper 回归通过，1 项跳过；TypeScript、日志/品牌/CI/环境/凭据/元数据/测试/产物检查通过。全仓 Biome 仍有历史格式和 lint 诊断（1005 项）。构建机执行生产构建及路由烟测；生产切换前核对任务与请求，保留旧构建和模型配置回滚记录。 ## Release Readiness 仅修改本次生图运行逻辑与测试，凭据和验证产物保存在仓库外或忽略目录中。内部生成接口的权限与访问方式保持兼容。
29 lines
3.0 KiB
TypeScript
29 lines
3.0 KiB
TypeScript
import { serverLoader } from "./canvas-download-server-loader";
|
|
import * as generationErrors from "@/lib/ai/runtime/image-generation-error";
|
|
|
|
export function agentImageProviderHarness(options: { generate?: (id: string, count: number, call: number) => string; models?: any[] } = {}) {
|
|
const calls: any[] = [], paused = new Set<string>();
|
|
const models = options.models || ["primary", "backup"].map((id, i) => ({ model: { id, modelId: id, displayName: id, isEnabled: true, supportsImageGen: true, supportsVision: true, isDefault: !i, imageGenConfig: { maxImages: 12 }, capabilities: { kind: "image" } }, provider: { id: "same-provider", type: "openai-image", apiKey: "synthetic", isEnabled: true } }));
|
|
const mocks: Record<string, any> = {
|
|
"@/lib/ai/runtime/image-generation-error": generationErrors,
|
|
"@/lib/db/queries": { getEnabledAiModelsWithProvider: async () => models.filter(entry => entry.model.isEnabled), getAiModelById: async ({ id }: { id: string }) => models.find(entry => entry.model.id === id)?.model },
|
|
"@/lib/ai/rmbg/service": {},
|
|
"@/lib/net/url-access": { filterReachableRemoteUrls: async (urls: string[]) => ({ kept: urls, dropped: [] }) },
|
|
"@/lib/canvas/image-concurrency": { acquireCanvasImageModelSlot: async () => ({ waitMs: 0, maxConcurrent: 999, release: async () => {} }), normalizeCanvasImageModelKey: (v: string) => v, recordCanvasImageModelExecutionDuration() {} },
|
|
"@/lib/ai/runtime/model-health": { recordAiModelHealthSuccess: async () => {}, recordAiModelHealthFailure: async () => {}, classifyAiModelHealthFailure: () => ({ reason: "provider_error" }), shouldFailoverAiModelError: () => false },
|
|
"@/lib/ai/runtime/agent-model-circuit-store": { agentImageCircuitStore: { isPaused: async (e: any) => paused.has(e.model.id), fail: async (e: any) => { paused.add(e.model.id); } } },
|
|
"@/lib/ai/image-model-pricing": { getModelGenerationCreditCost: ({ imageGenConfig }: any) => imageGenConfig?.creditCost || 0 },
|
|
"@/lib/storage/service": {},
|
|
};
|
|
for (const name of ["gemini-image", "alibaba-image", "bfl-image", "ruxa-image", "fal-image", "snapgen-image", "openai-image", "bananaimage", "visionary", "atlascloud", "vip98k", "stepfun-image", "midjourney", "runway", "krea", "volcano-vision"]) mocks[`@/lib/ai/providers/${name}`] = {};
|
|
mocks["@/lib/ai/providers/openai-image"] = { createOpenAIImage: () => ({ chat: (id: string) => ({ doGenerate: async (params: any) => {
|
|
calls.push({ id, ...params });
|
|
const text = options.generate?.(id, params.imageParams.imageCount, calls.length) || ``;
|
|
return { text };
|
|
} }) }) };
|
|
const generate = serverLoader(mocks)("lib/canvas/generate-image.ts").generateCanvasImage;
|
|
const request = { prompt: "a cat", model: "primary", imageCount: 1, imageSize: "2K", aspectRatio: "3:4", agentImageMode: "auto", allowFailover: true, referenceImages: ["https://fixture.invalid/source.png"], negativePrompt: "no text", providerImageParams: { seed: 0, watermark: false } };
|
|
return { generate, request, calls, paused };
|
|
}
|
|
|