wehub-resource-sync
e4dcfc49aa
Tests / Import Check (Python 3.13) (push) Has been cancelled
Tests / Import Check (Python 3.14) (push) Has been cancelled
Tests / Python Tests (Python 3.11) (push) Has been cancelled
Tests / Python Tests (Python 3.12) (push) Has been cancelled
Tests / Python Tests (Python 3.14) (push) Has been cancelled
Tests / Test Summary (push) Has been cancelled
Tests / Lint and Format (push) Has been cancelled
Tests / Web Node Tests (push) Has been cancelled
Tests / Import Check (Python 3.11) (push) Has been cancelled
Tests / Import Check (Python 3.12) (push) Has been cancelled
Tests / Python Tests (Python 3.13) (push) Has been cancelled
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""Resolved runtime configuration for image-generation providers.
|
|
|
|
This dataclass is the read-side adapter between the model catalog
|
|
(``services.imagegen``) and the HTTP adapter. It mirrors the shape of
|
|
:class:`TTSConfig` so a single OpenAI-compatible adapter can cover OpenAI
|
|
DALL·E / gpt-image, Volcengine Ark Seedream and compatible gateways by swapping
|
|
``base_url`` + ``api_key`` + ``model``.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass, field
|
|
|
|
from deeptutor.services.generation_http import AUTH_BEARER
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class ImagegenConfig:
|
|
"""Resolved text-to-image configuration for one generation call."""
|
|
|
|
model: str
|
|
provider_name: str = "openai"
|
|
adapter: str = "openai_compat"
|
|
auth_style: str = AUTH_BEARER
|
|
api_key: str = ""
|
|
base_url: str = ""
|
|
api_version: str | None = None
|
|
extra_headers: dict[str, str] = field(default_factory=dict)
|
|
# Provider/model-specific generation knobs. Empty → omit from the request
|
|
# and let the provider use its default.
|
|
size: str = "" # e.g. "1024x1024"
|
|
quality: str = "" # e.g. "standard" | "hd"
|
|
style: str = "" # e.g. "natural" | "vivid"
|
|
response_format: str = "" # "" | "url" | "b64_json"
|
|
# Image generation is slow; allow generous wall-clock per request.
|
|
request_timeout: int = 120
|
|
|
|
|
|
__all__ = ["ImagegenConfig"]
|