feat: return the typed ImagesResponse from image_generation

Now that the regenerated core types the /v1/images/generations response (gateway
enrich_spec injects any-llm's ImagesResponse), return the typed model from both
OtariClient.image_generation and AsyncOtariClient.image_generation instead of an
untyped dict, matching embedding / rerank / moderation. Updates tests and the
README example to use attribute access.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

Author: njbrake

README.md

@@ -238,10 +238,10 @@ result = client.image_generation(
     prompt="A watercolor fox in a misty forest",
 )
 
-print(result["data"][0]["url"])
+print(result.data[0].url)
 ```
 
-The gateway returns the OpenAI-compatible image payload as a dict.
+The gateway returns a typed OpenAI-compatible `ImagesResponse`.
 
 ### Audio
 

src/otari/async_client.py

@@ -62,6 +62,7 @@
     from otari._client.models.chat_completion_chunk import ChatCompletionChunk
     from otari._client.models.count_tokens_response import CountTokensResponse
     from otari._client.models.create_embedding_response import CreateEmbeddingResponse
+    from otari._client.models.images_response import ImagesResponse
     from otari._client.models.model_object import ModelObject
     from otari._client.models.moderation_response import ModerationResponse
     from otari._client.models.rerank_response import RerankResponse
@@ -313,12 +314,11 @@ async def image_generation(
         model: str,
         prompt: str,
         **kwargs: Any,
-    ) -> dict[str, Any]:
+    ) -> ImagesResponse:
         """Generate images from a text prompt.
 
-        Returns the gateway's OpenAI-compatible image payload as a dict
-        (``{"created": ..., "data": [...]}``). The generated core models this
-        response as an opaque object, so the parsed JSON is returned unchanged.
+        Returns the gateway's OpenAI-compatible
+        :class:`~otari._client.models.images_response.ImagesResponse`.
 
         Args:
             model: Model identifier (e.g. ``"openai:dall-e-3"``).
@@ -332,7 +332,7 @@ async def image_generation(
         result = await self._call(
             lambda: self._images.create_image_v1_images_generations_post(request)
         )
-        return cast("dict[str, Any]", result)
+        return cast("ImagesResponse", result)
 
     # -- Audio --------------------------------------------------------------
 

src/otari/client.py

@@ -62,6 +62,7 @@
     from otari._client.models.chat_completion_chunk import ChatCompletionChunk
     from otari._client.models.count_tokens_response import CountTokensResponse
     from otari._client.models.create_embedding_response import CreateEmbeddingResponse
+    from otari._client.models.images_response import ImagesResponse
     from otari._client.models.model_object import ModelObject
     from otari._client.models.moderation_response import ModerationResponse
     from otari._client.models.rerank_response import RerankResponse
@@ -339,12 +340,11 @@ def image_generation(
         model: str,
         prompt: str,
         **kwargs: Any,
-    ) -> dict[str, Any]:
+    ) -> ImagesResponse:
         """Generate images from a text prompt.
 
-        Returns the gateway's OpenAI-compatible image payload as a dict
-        (``{"created": ..., "data": [...]}``). The generated core models this
-        response as an opaque object, so the parsed JSON is returned unchanged.
+        Returns the gateway's OpenAI-compatible
+        :class:`~otari._client.models.images_response.ImagesResponse`.
 
         Args:
             model: Model identifier (e.g. ``"openai:dall-e-3"``).
@@ -356,7 +356,7 @@ def image_generation(
             ImageGenerationRequest, {"model": model, "prompt": prompt, **kwargs}
         )
         result = self._call(lambda: self._images.create_image_v1_images_generations_post(request))
-        return cast("dict[str, Any]", result)
+        return cast("ImagesResponse", result)
 
     # -- Audio --------------------------------------------------------------
 

tests/unit/test_async_client.py

@@ -171,11 +171,12 @@ async def test_streaming_error_maps(self) -> None:
 
 
 class TestImages:
-    async def test_image_generation_returns_dict(self, mock_rest: Any) -> None:
+    async def test_image_generation_returns_typed(self, mock_rest: Any) -> None:
         mock = mock_rest(status=200, body=IMAGE_RESPONSE)
         client = AsyncOtariClient(api_base="http://localhost:8000", api_key="vk")
         result = await client.image_generation(model="openai:dall-e-3", prompt="a cat")
-        assert result == IMAGE_RESPONSE
+        assert result.created == 1
+        assert result.data[0].url == "https://example.com/image.png"
         assert mock.last.url.endswith("/v1/images/generations")
 
 

tests/unit/test_client.py

@@ -385,11 +385,12 @@ def test_platform_mode_streaming_sends_bearer(self) -> None:
 
 
 class TestImages:
-    def test_image_generation_returns_dict(self, mock_rest: Any) -> None:
+    def test_image_generation_returns_typed(self, mock_rest: Any) -> None:
         mock = mock_rest(status=200, body=IMAGE_RESPONSE)
         client = OtariClient(api_base="http://localhost:8000", api_key="vk")
         result = client.image_generation(model="openai:dall-e-3", prompt="a cat")
-        assert result == IMAGE_RESPONSE
+        assert result.created == 1
+        assert result.data[0].url == "https://example.com/image.png"
         assert mock.last.url.endswith("/v1/images/generations")
         body = mock.last.json_body
         assert body["model"] == "openai:dall-e-3"