46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
from typing import Any
|
|
|
|
import requests
|
|
from django.conf import settings
|
|
|
|
from .openai_compatible import openai_image_size
|
|
from .volcano import VolcanoArkProvider
|
|
|
|
|
|
class YunqiProvider(VolcanoArkProvider):
|
|
"""YunQi AI(New API 网关)适配器:OpenAI 兼容 images/generations。
|
|
|
|
与火山 ARK 的差异:不接受 watermark/sequential_image_generation/response_format 私有参数;
|
|
size 用 OpenAI 枚举(1024x1024/1024x1536/1536x1024);gpt-image-2 固定返回 b64_json,
|
|
下游 extract_first_media_url / media_to_bytes 已兼容 base64,无需改动。
|
|
"""
|
|
|
|
def __post_init__(self) -> None:
|
|
self.api_key = self.api_key or settings.YUNQI.get("api_key")
|
|
self.base_url = self.base_url or settings.YUNQI.get("base_url")
|
|
|
|
def image_generation(
|
|
self,
|
|
*,
|
|
model: str,
|
|
prompt: str,
|
|
endpoint: str = "images/generations",
|
|
image: str | list[str] | None = None,
|
|
size: str = "1024x1536",
|
|
timeout: float = 300,
|
|
) -> dict[str, Any]:
|
|
if not self.api_key:
|
|
raise ValueError("YUNQI_API_KEY is not configured")
|
|
if image:
|
|
raise ValueError("gpt-image-2 via YunQi only supports text-to-image; reference image is not supported")
|
|
body: dict[str, Any] = {"model": model, "prompt": prompt, "size": openai_image_size(size), "n": 1}
|
|
# 实测该网关生图平均延迟 75s+,超时须显著高于火山的 180s
|
|
response = requests.post(
|
|
f"{self.base_url.rstrip('/')}/{endpoint.lstrip('/')}",
|
|
headers={"Authorization": f"Bearer {self.api_key}", "Content-Type": "application/json"},
|
|
json=body,
|
|
timeout=timeout,
|
|
)
|
|
response.raise_for_status()
|
|
return response.json()
|