chuan/WebAI2API
1
0
Fork 0
mirror of https://github.com/foxhui/WebAI2API.git synced 2026-06-16 21:03:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 添加模型图片策略管理并在服务器端进行验证

Browse Source
This commit is contained in:
foxhui
2025-11-29 23:38:20 +08:00
Unverified
parent c410f08101
commit 5c758a7288
2 changed files with 190 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/backend/models.js
+168 -57
View File
@@ -1,46 +1,151 @@
// LMArena 完整模型映射 (模型名 -> UUID)
export const LMARENA_MODEL_MAPPING = {
"gemini-3-pro-image-preview": "019aa208-5c19-7162-ae3b-0a9ddbb1e16a",
"seedream-4-high-res-fal": "32974d8d-333c-4d2e-abf3-f258c0ac1310",
"hunyuan-image-3.0": "7766a45c-1b6b-4fb8-9823-2557291e1ddd",
"gemini-2.5-flash-image-preview": "0199ef2a-583f-7088-b704-b75fd169401d",
"imagen-4.0-ultra-generate-preview-06-06": "f8aec69d-e077-4ed1-99be-d34f48559bbf",
"imagen-4.0-generate-preview-06-06": "2ec9f1a6-126f-4c65-a102-15ac401dcea4",
"wan2.5-t2i-preview": "019a5050-2875-78ed-ae3a-d9a51a438685",
"gpt-image-1": "6e855f13-55d7-4127-8656-9168a9f4dcc0",
"gpt-image-mini": "0199c238-f8ee-7f7d-afc1-7e28fcfd21cf",
"mai-image-1": "1b407d5c-1806-477c-90a5-e5c5a114f3bc",
"seedream-3": "d8771262-8248-4372-90d5-eb41910db034",
"qwen-image-prompt-extend": "9fe82ee1-c84f-417f-b0e7-cab4ae4cf3f3",
"flux-1-kontext-pro": "28a8f330-3554-448c-9f32-2c0a08ec6477",
"imagen-3.0-generate-002": "51ad1d79-61e2-414c-99e3-faeb64bb6b1b",
"ideogram-v3-quality": "73378be5-cdba-49e7-b3d0-027949871aa6",
"photon": "e7c9fa2d-6f5d-40eb-8305-0980b11c7cab",
"lucid-origin": "5a3b3520-c87d-481f-953c-1364687b6e8f",
"recraft-v3": "b88d5814-1d20-49cc-9eb6-e362f5851661",
"gemini-2.0-flash-preview-image-generation": "69bbf7d4-9f44-447e-a868-abc4f7a31810",
"dall-e-3": "bb97bc68-131c-4ea4-a59e-03a6252de0d2",
"flux-1-kontext-dev": "eb90ae46-a73a-4f27-be8b-40f090592c9a",
"imagen-4.0-fast-generate-001": "f44fd4f8-af30-480f-8ce2-80b2bdfea55e",
"hunyuan-image-2.1": "a9a26426-5377-4efa-bef9-de71e29ad943"
// 图片策略枚举
export const IMAGE_POLICY = {
OPTIONAL: 'optional', // 可带可不带（默认）
REQUIRED: 'required', // 必须有参考图
FORBIDDEN: 'forbidden' // 禁止带图
};
// GeminiBiz 支持的模型列表 (仅需验证模型 ID,不需要 UUID)
export const GEMINI_BIZ_SUPPORTED_MODELS = [
"gemini-3-pro-image-preview"
];
// LMArena 后端模型配置
export const LMARENA_MODELS = {
"gemini-3-pro-image-preview": {
codeName: "019aa208-5c19-7162-ae3b-0a9ddbb1e16a",
imagePolicy: IMAGE_POLICY.OPTIONAL
},
"seedream-4-high-res-fal": {
codeName: "32974d8d-333c-4d2e-abf3-f258c0ac1310",
imagePolicy: IMAGE_POLICY.OPTIONAL
},
"hunyuan-image-3.0": {
codeName: "7766a45c-1b6b-4fb8-9823-2557291e1ddd",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"gemini-2.5-flash-image-preview": {
codeName: "0199ef2a-583f-7088-b704-b75fd169401d",
imagePolicy: IMAGE_POLICY.OPTIONAL
},
"imagen-4.0-ultra-generate-preview-06-06": {
codeName: "f8aec69d-e077-4ed1-99be-d34f48559bbf",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"imagen-4.0-generate-preview-06-06": {
codeName: "2ec9f1a6-126f-4c65-a102-15ac401dcea4",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"wan2.5-t2i-preview": {
codeName: "019a5050-2875-78ed-ae3a-d9a51a438685",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"gpt-image-1": {
codeName: "6e855f13-55d7-4127-8656-9168a9f4dcc0",
imagePolicy: IMAGE_POLICY.OPTIONAL
},
"gpt-image-mini": {
codeName: "0199c238-f8ee-7f7d-afc1-7e28fcfd21cf",
imagePolicy: IMAGE_POLICY.OPTIONAL
},
"mai-image-1": {
codeName: "1b407d5c-1806-477c-90a5-e5c5a114f3bc",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"seedream-3": {
codeName: "d8771262-8248-4372-90d5-eb41910db034",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"qwen-image-prompt-extend": {
codeName: "9fe82ee1-c84f-417f-b0e7-cab4ae4cf3f3",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"flux-1-kontext-pro": {
codeName: "28a8f330-3554-448c-9f32-2c0a08ec6477",
imagePolicy: IMAGE_POLICY.OPTIONAL
},
"imagen-3.0-generate-002": {
codeName: "51ad1d79-61e2-414c-99e3-faeb64bb6b1b",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"ideogram-v3-quality": {
codeName: "73378be5-cdba-49e7-b3d0-027949871aa6",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"photon": {
codeName: "e7c9fa2d-6f5d-40eb-8305-0980b11c7cab",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"lucid-origin": {
codeName: "5a3b3520-c87d-481f-953c-1364687b6e8f",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"recraft-v3": {
codeName: "b88d5814-1d20-49cc-9eb6-e362f5851661",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"gemini-2.0-flash-preview-image-generation": {
codeName: "69bbf7d4-9f44-447e-a868-abc4f7a31810",
imagePolicy: IMAGE_POLICY.OPTIONAL
},
"dall-e-3": {
codeName: "bb97bc68-131c-4ea4-a59e-03a6252de0d2",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"flux-1-kontext-dev": {
codeName: "eb90ae46-a73a-4f27-be8b-40f090592c9a",
imagePolicy: IMAGE_POLICY.OPTIONAL
},
"imagen-4.0-fast-generate-001": {
codeName: "f44fd4f8-af30-480f-8ce2-80b2bdfea55e",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"flux-2-pro": {
codeName: "019abcf4-5600-7a8b-864d-9b8ab7ab7328",
imagePolicy: IMAGE_POLICY.OPTIONAL
},
"flux-2-flex": {
codeName: "019abed6-d96e-7a2b-bf69-198c28bef281",
imagePolicy: IMAGE_POLICY.OPTIONAL
},
"hunyuan-image-2.1": {
codeName: "a9a26426-5377-4efa-bef9-de71e29ad943",
imagePolicy: IMAGE_POLICY.FORBIDDEN
},
"qwen-image-edit": {
codeName: "995cf221-af30-466d-a809-8e0985f83649",
imagePolicy: IMAGE_POLICY.REQUIRED
},
"reve-v1": {
codeName: "0199e980-ba42-737b-9436-927b6e7ca73e",
imagePolicy: IMAGE_POLICY.REQUIRED
},
"reve-fast-edit": {
codeName: "019a5675-0a56-7835-abdd-1cb9e7870afa",
imagePolicy: IMAGE_POLICY.REQUIRED
}
};
// Gemini Biz 后端模型配置
export const GEMINI_BIZ_MODELS = {
"gemini-3-pro-image-preview": {
imagePolicy: IMAGE_POLICY.OPTIONAL
}
};
/**
* 获取后端对应的模型映射或列表
* 获取后端对应的模型配置表
* @param {string} backendName - 后端名称 ('lmarena' 或 'gemini_biz')
* @returns {Object|Array} LMArena 返回映射对象,GeminiBiz 返回支持的模型数组
* @returns {Object} 模型配置对象
* @private
*/
function getMapForBackend(backendName) {
if (backendName === 'gemini_biz') {
return GEMINI_BIZ_SUPPORTED_MODELS;
function getModelsConfigForBackend(backendName) {
switch (backendName) {
case 'lmarena':
return LMARENA_MODELS;
case 'gemini_biz':
return GEMINI_BIZ_MODELS;
// 将来新增其它后端：
// case 'foo_site':
// return FOO_SITE_MODELS;
default:
return {};
}
return LMARENA_MODEL_MAPPING;
}
/**
@@ -49,16 +154,8 @@ function getMapForBackend(backendName) {
* @returns {Object} OpenAI 格式的模型列表
*/
export function getModelsForBackend(backendName) {
const map = getMapForBackend(backendName);
let modelIds;
if (backendName === 'gemini_biz') {
// GeminiBiz: 直接使用支持的模型列表
modelIds = map;
} else {
// LMArena: 从映射对象中提取键
modelIds = Object.keys(map);
}
const modelsConf = getModelsConfigForBackend(backendName);
const modelIds = Object.keys(modelsConf);
return {
object: 'list',
@@ -66,7 +163,9 @@ export function getModelsForBackend(backendName) {
id,
object: 'model',
created: Math.floor(Date.now() / 1000),
owned_by: backendName === 'gemini_biz' ? 'gemini_biz' : 'lmarena'
owned_by: backendName,
// 向前端暴露图片策略
image_policy: modelsConf[id].imagePolicy || IMAGE_POLICY.OPTIONAL
}))
};
}
@@ -75,20 +174,32 @@ export function getModelsForBackend(backendName) {
* 解析模型 ID
* @param {string} backendName - 后端名称
* @param {string} modelKey - 请求的模型键
* @returns {string|null} LMArena 返回 UUID,GeminiBiz 返回模型 ID (验证通过) 或 null
* @returns {string|null} 返回内部使用的 codeName，若模型无效则返回 null
*/
export function resolveModelId(backendName, modelKey) {
if (backendName === 'gemini_biz') {
// GeminiBiz: 只验证模型是否在支持列表中
return GEMINI_BIZ_SUPPORTED_MODELS.includes(modelKey) ? modelKey : null;
const modelsConf = getModelsConfigForBackend(backendName);
const model = modelsConf[modelKey];
if (!model) return null; // 未配置的模型 -> 无效
// 无 codeName 时，退回到模型 ID 本身
return model.codeName || modelKey;
}
/**
* 获取模型的图片策略
* @param {string} backendName - 后端名称
* @param {string} modelKey - 模型键
* @returns {string} 图片策略 ('optional' | 'required' | 'forbidden')
*/
export function getImagePolicy(backendName, modelKey) {
const modelsConf = getModelsConfigForBackend(backendName);
const model = modelsConf[modelKey];
if (!model || !model.imagePolicy) {
return IMAGE_POLICY.OPTIONAL;
}
// LMArena: 返回 UUID
return LMARENA_MODEL_MAPPING[modelKey] || null;
return model.imagePolicy;
}
// 保留旧的导出以兼容 (如果有其他地方还在使用)
export const MODEL_MAPPING = LMARENA_MODEL_MAPPING;
export function getModels() {
return getModelsForBackend('lmarena');
}
server.js
+22 -1
View File
@@ -3,7 +3,7 @@ import fs from 'fs';
import path from 'path';
import sharp from 'sharp';
import { getBackend } from './lib/backend/index.js';
import { getModelsForBackend, resolveModelId } from './lib/backend/models.js';
import { getModelsForBackend, resolveModelId, getImagePolicy, IMAGE_POLICY } from './lib/backend/models.js';
import { logger } from './lib/logger.js';
import crypto from 'crypto';
@@ -314,8 +314,29 @@ async function startServer() {
logger.info('服务器', '未指定模型，使用网页默认', { id });
}
// 图片策略校验
const hasImage = imagePaths.length > 0;
const policy = data.model ? getImagePolicy(name, data.model) : IMAGE_POLICY.OPTIONAL;
if (policy === IMAGE_POLICY.REQUIRED && !hasImage) {
const errorMsg = `Model ${data.model} requires a reference image.`;
logger.warn('服务器', errorMsg, { id });
if (isQueueMode) { sseHelper.send('error', { msg: errorMsg }); sseHelper.end(); }
else { res.writeHead(400); res.end(JSON.stringify({ error: errorMsg })); }
return;
}
if (policy === IMAGE_POLICY.FORBIDDEN && hasImage) {
const errorMsg = `Model ${data.model} does not accept images.`;
logger.warn('服务器', errorMsg, { id });
if (isQueueMode) { sseHelper.send('error', { msg: errorMsg }); sseHelper.end(); }
else { res.writeHead(400); res.end(JSON.stringify({ error: errorMsg })); }
return;
}
logger.info('服务器', `[队列] 请求入队: ${prompt.slice(0, 10)}...`, { id, images: imagePaths.length });
if (isQueueMode) {
sseHelper.send('status', { status: 'queued', position: queue.length + 1 });
}