diff --git a/src/celeste/providers/openai/images/client.py b/src/celeste/providers/openai/images/client.py index 27bc169a..eb989c26 100644 --- a/src/celeste/providers/openai/images/client.py +++ b/src/celeste/providers/openai/images/client.py @@ -97,10 +97,7 @@ async def _make_stream_request( extra_headers: dict[str, str] | None = None, **parameters: Any, ) -> AsyncGenerator[dict[str, Any], None]: - """Make streaming request to OpenAI Images API. - - Streaming is only supported for gpt-image-1. - """ + """Make streaming request to OpenAI Images API.""" if endpoint is None: endpoint = config.OpenAIImagesEndpoint.CREATE_IMAGE @@ -137,7 +134,6 @@ def _parse_usage( """Extract usage data from Images API response. Returns dict that modality clients wrap in their specific Usage type. - gpt-image-1 returns usage, DALL-E models don't. """ usage_data = response_data.get("usage") or {} return OpenAIImagesClient.map_usage_fields(usage_data) diff --git a/src/celeste/providers/openai/images/config.py b/src/celeste/providers/openai/images/config.py index 74e74ea4..1e25d384 100644 --- a/src/celeste/providers/openai/images/config.py +++ b/src/celeste/providers/openai/images/config.py @@ -8,7 +8,6 @@ class OpenAIImagesEndpoint(StrEnum): CREATE_IMAGE = "/v1/images/generations" CREATE_EDIT = "/v1/images/edits" - CREATE_VARIATION = "/v1/images/variations" BASE_URL = "https://api.openai.com"