fix: 图片生成支持豆包5.0 pro

This commit is contained in:
hp0912 2026-08-19 23:30:46 +08:00
parent 89a2ad2368
commit c2869a07eb
4 changed files with 6 additions and 2 deletions

View File

@ -36,7 +36,7 @@ description: "图片修改、图生图工具。基于输入的一张或多张图
}, },
"model": { "model": {
"type": "string", "type": "string",
"description": "画图模型选择(可选):即梦4.5(jimeng-4.5) / 即梦4.6(jimeng-4.6) / 即梦4.7(jimeng-4.7) / 即梦5.0(jimeng-5.0) / 即梦5.0 Pro(jimeng-5.0-pro) / 豆包5.0 Pro(doubao-seedream-5-0-pro) / 豆包图生图(doubao-seededit-3.0-i2i) / 造相基础版(Z-Image) / 造相蒸馏版(Z-Image-Turbo) / 造相图片编辑(Qwen-Image-Edit-2511) / OpenAI GPT Image(gpt-image-2),默认: 空(none)。", "description": "画图模型选择(可选):即梦4.5(jimeng-4.5) / 即梦4.6(jimeng-4.6) / 即梦4.7(jimeng-4.7) / 即梦5.0(jimeng-5.0) / 即梦5.0 Pro(jimeng-5.0-pro) / 豆包5.0 Pro(doubao-seedream-5-0-pro) / 豆包5.0 Lite(doubao-seedream-5-0) / 豆包图生图(doubao-seededit-3.0-i2i) / 造相基础版(Z-Image) / 造相蒸馏版(Z-Image-Turbo) / 造相图片编辑(Qwen-Image-Edit-2511) / OpenAI GPT Image(gpt-image-2),默认: 空(none)。",
"enum": [ "enum": [
"none", "none",
"jimeng-4.5", "jimeng-4.5",
@ -45,6 +45,7 @@ description: "图片修改、图生图工具。基于输入的一张或多张图
"jimeng-5.0", "jimeng-5.0",
"jimeng-5.0-pro", "jimeng-5.0-pro",
"doubao-seedream-5-0-pro", "doubao-seedream-5-0-pro",
"doubao-seedream-5-0",
"doubao-seededit-3.0-i2i", "doubao-seededit-3.0-i2i",
"Z-Image", "Z-Image",
"Z-Image-Turbo", "Z-Image-Turbo",

View File

@ -464,6 +464,7 @@ def call_doubao(config: dict, prompt: str, model: str, image: str) -> list[str]:
model_map = { model_map = {
"doubao-seedream-5-0-pro": "doubao-seedream-5-0-pro-260628", "doubao-seedream-5-0-pro": "doubao-seedream-5-0-pro-260628",
"doubao-seedream-5-0": "doubao-seedream-5-0-260128",
"doubao-seededit-3.0-i2i": "doubao-seededit-3-0-i2i-250628", "doubao-seededit-3.0-i2i": "doubao-seededit-3-0-i2i-250628",
} }
actual_model = model_map.get(model, model) actual_model = model_map.get(model, model)

View File

@ -34,7 +34,7 @@ description: "AI绘图工具,当用户想通过文本生成图像时,可以
}, },
"model": { "model": {
"type": "string", "type": "string",
"description": "画图模型选择(可选):即梦4.5(jimeng-4.5) / 即梦4.6(jimeng-4.6) / 即梦4.7(jimeng-4.7) / 即梦5.0(jimeng-5.0) / 即梦5.0 Pro(jimeng-5.0-pro) / 豆包5.0 Pro(doubao-seedream-5-0-pro) / 豆包4.5(doubao-seedream-4.5) / 豆包4.0(doubao-seedream-4.0) / 豆包文生图(doubao-seedream-3.0-t2i) / 豆包图生图(doubao-seededit-3.0-i2i) / 造相基础版(Z-Image) / 造相蒸馏版(Z-Image-Turbo) / 造相图片编辑(Qwen-Image-Edit-2511) / OpenAI GPT Image(gpt-image-2),默认: 空(none)。", "description": "画图模型选择(可选):即梦4.5(jimeng-4.5) / 即梦4.6(jimeng-4.6) / 即梦4.7(jimeng-4.7) / 即梦5.0(jimeng-5.0) / 即梦5.0 Pro(jimeng-5.0-pro) / 豆包5.0 Pro(doubao-seedream-5-0-pro) / 豆包5.0 Lite(doubao-seedream-5-0) / 豆包4.5(doubao-seedream-4.5) / 豆包4.0(doubao-seedream-4.0) / 豆包文生图(doubao-seedream-3.0-t2i) / 豆包图生图(doubao-seededit-3.0-i2i) / 造相基础版(Z-Image) / 造相蒸馏版(Z-Image-Turbo) / 造相图片编辑(Qwen-Image-Edit-2511) / OpenAI GPT Image(gpt-image-2),默认: 空(none)。",
"enum": [ "enum": [
"none", "none",
"jimeng-4.5", "jimeng-4.5",
@ -43,6 +43,7 @@ description: "AI绘图工具,当用户想通过文本生成图像时,可以
"jimeng-5.0", "jimeng-5.0",
"jimeng-5.0-pro", "jimeng-5.0-pro",
"doubao-seedream-5-0-pro", "doubao-seedream-5-0-pro",
"doubao-seedream-5-0",
"doubao-seedream-4.5", "doubao-seedream-4.5",
"doubao-seedream-4.0", "doubao-seedream-4.0",
"doubao-seedream-3.0-t2i", "doubao-seedream-3.0-t2i",

View File

@ -438,6 +438,7 @@ def call_doubao(config: dict, prompt: str, model: str) -> list[str]:
# Map friendly model names to actual endpoint model IDs # Map friendly model names to actual endpoint model IDs
model_map = { model_map = {
"doubao-seedream-5-0-pro": "doubao-seedream-5-0-pro-260628", "doubao-seedream-5-0-pro": "doubao-seedream-5-0-pro-260628",
"doubao-seedream-5-0": "doubao-seedream-5-0-260128",
"doubao-seedream-4.5": "doubao-seedream-4-5-251128", "doubao-seedream-4.5": "doubao-seedream-4-5-251128",
"doubao-seedream-4.0": "doubao-seedream-4-0-251128", "doubao-seedream-4.0": "doubao-seedream-4-0-251128",
"doubao-seedream-3.0-t2i": "doubao-seedream-3-0-t2i-250415", "doubao-seedream-3.0-t2i": "doubao-seedream-3-0-t2i-250415",