VOLCANO_PROVIDER = {
    "name": "volcengine",
    "display_name": "火山引擎(豆包)",
    "base_url": "https://ark.cn-beijing.volces.com/api/v3",
}

VOLCANO_MODELS = [
    {
        "display_name": "Doubao-Seed-2.0-Pro",
        "name": "doubao-seed-2-0-pro-260215",
        "capability": "text",
        "endpoint": "chat/completions",
        "metadata": {"think": True, "source": "video-flow/data/vendor/volcengine.ts"},
    },
    {
        "display_name": "Doubao-Seed-2.0-Lite",
        "name": "doubao-seed-2-0-lite-260215",
        "capability": "text",
        "endpoint": "chat/completions",
        "metadata": {"think": True, "source": "video-flow/data/vendor/volcengine.ts"},
    },
    {
        "display_name": "Seedream-5.0",
        "name": "doubao-seedream-5-0-260128",
        "capability": "image",
        "endpoint": "images/generations",
        "metadata": {
            "modes": ["text", "singleImage", "multiReference"],
            "watermark": False,
            "source": "video-flow/data/vendor/volcengine.ts",
        },
    },
    {
        "display_name": "Seedream-4.5",
        "name": "doubao-seedream-4-5-251128",
        "capability": "image",
        "endpoint": "images/generations",
        "metadata": {
            "modes": ["text", "singleImage", "multiReference"],
            "watermark": False,
            "source": "video-flow/data/vendor/volcengine.ts",
        },
    },
    {
        "display_name": "Seedance-2.0",
        "name": "doubao-seedance-2-0-260128",
        "capability": "video",
        "endpoint": "contents/generations/tasks",
        "metadata": {
            "audio": "optional",
            "modes": ["text", "startFrameOptional", "imageReference:9", "videoReference:3", "audioReference:3"],
            "durations": list(range(4, 16)),
            "resolutions": ["480p", "720p"],
            "watermark": False,
            "source": "video-flow/data/vendor/volcengine.ts",
        },
    },
    {
        "display_name": "Seedance-1.5-Pro",
        "name": "doubao-seedance-1-5-pro-251215",
        "capability": "video",
        "endpoint": "contents/generations/tasks",
        "metadata": {
            "audio": "optional",
            "modes": ["text", "startFrameOptional"],
            "durations": list(range(4, 13)),
            "resolutions": ["480p", "720p", "1080p"],
            "watermark": False,
            "source": "video-flow/data/vendor/volcengine.ts",
        },
    },
]

