Class OpenAICompatibleGenerator

backend/generators/openai_compatible.py:11–134

OpenAI 兼容接口图片生成器

class OpenAICompatibleGenerator(ImageGeneratorBase):
    """OpenAI 兼容接口图片生成器"""

    def __init__(self, config: Dict[str, Any]):
        super().__init__(config)
        logger.debug("初始化 OpenAICompatibleGenerator...")

        if not self.api_key:
            logger.error("OpenAI 兼容 API Key 未配置")
            raise ValueError(
                "OpenAI 兼容 API Key 未配置。\n"
                "解决方案：在系统设置页面编辑该服务商，填写 API Key"
            )

        if not self.base_url:
            logger.error("OpenAI 兼容 API Base URL 未配置")
            raise ValueError(
                "OpenAI 兼容 API Base URL 未配置。\n"
                "解决方案：在系统设置页面编辑该服务商，填写 Base URL"
            )

        self.policy = ImageProviderPolicy.from_config(
            config,
            default_model='gpt-image-2',
            default_endpoint='/v1/images/generations',
        )
        self.client = ImageApiClient(self.policy)
        self.base_url = self.policy.base_url
        self.default_model = self.policy.model
        self.endpoint_type = self.policy.endpoint_type

        logger.info(f"OpenAICompatibleGenerator 初始化完成: base_url={self.base_url}, model={self.default_model}, endpoint={self.endpoint_type}")

    def validate_config(self) -> bool:
        """验证配置"""
        return bool(self.api_key and self.base_url)

    def generate_image(
        self,
        prompt: str,
        size: str = "1024x1024",
        model: str = None,
        quality: str = "standard",
        **kwargs
    ) -> bytes:
        """
        生成图片

        Args:
            prompt: 提示词
            size: 图片尺寸 (如 "1024x1024", "2048x2048", "4096x4096")
            model: 模型名称
            quality: 质量 ("standard" 或 "hd")
            **kwargs: 其他参数

        Returns:
            图片二进制数据
        """