docs(aigw): README and docker-compose updates for image generation (gpt-image-1-mini 1024x1024 low); align examples and services

Signed-off-by: Hrushikesh Patil <[email protected]>

Author: nutanix-Hrushikesh

cmd/aigw/README.md

@@ -74,17 +74,17 @@ Here are values we use for Ollama:
          -X POST http://localhost:1975/v1/images/generations \
          -H "Authorization: Bearer unused" \
          -H "Content-Type: application/json" \
-         -d '{"model":"dall-e-2","prompt":"A watercolor painting of a red fox in a birch forest","size":"512x512"}' \
+        -d '{"model":"gpt-image-1-mini","prompt":"A watercolor painting of a red fox in a birch forest","size":"1024x1024","quality":"low"}' \
          | jq -r '.data[0].b64_json' | base64 -d > image.png
        ```
 
 4. **Create embeddings**:
 
-   The `create-embeddings` service uses `curl` to send an embeddings request
+   The `embeddings` service uses `curl` to send an embeddings request
    to the AI Gateway CLI (aigw) which routes it to Ollama.
 
    ```bash
-   docker compose run --rm create-embeddings
+   docker compose run --rm embeddings
    ```
 
 5. **Shutdown the example stack**:
@@ -179,7 +179,7 @@ This configures the OTLP endpoint to otel-tui on port 4318.
 
    ```bash
    COMPOSE_PROFILES="<profile>" docker compose -f docker-compose-otel.yaml run --build --rm chat-completion
-   COMPOSE_PROFILES="<profile>" docker compose -f docker-compose-otel.yaml run --build --rm create-embeddings
+   COMPOSE_PROFILES="<profile>" docker compose -f docker-compose-otel.yaml run --build --rm embeddings
    COMPOSE_PROFILES="<profile>" docker compose -f docker-compose-otel.yaml run --build --rm mcp
    # Image generation
    COMPOSE_PROFILES="<profile>" docker compose -f docker-compose-otel.yaml run --build --rm image-generation

cmd/aigw/docker-compose-otel.yaml

@@ -105,16 +105,16 @@ services:
       - OPENAI_BASE_URL=http://aigw:1975/v1
       - OPENAI_API_KEY=unused
 
-  # create-embeddings is the standard OpenAI client (`openai` in pip), instrumented
+  # embeddings is the standard OpenAI client (`openai` in pip), instrumented
   # with the following OpenTelemetry instrumentation libraries:
   # - openinference-instrumentation-openai (embeddings spans)
   # - opentelemetry-instrumentation-httpx (HTTP client spans and trace headers)
-  create-embeddings:
+  embeddings:
     build:
       context: ../../tests/internal/testopeninference
       dockerfile: Dockerfile.openai_client
       target: create-embeddings
-    container_name: create-embeddings
+    container_name: embeddings
     profiles: ["test"]
     env_file:
       - ../../.env.ollama
@@ -144,3 +144,21 @@ services:
       - flyTo=LAX
       - --tool-arg
       - departureDate=15/12/2025
+
+  # image-generation is a simple curl-based test client for sending image
+  # generation requests to aigw.
+  image-generation:
+    image: golang:1.25
+    container_name: image-generation
+    profiles: ["test"]
+    env_file:
+      - ../../.env.ollama
+    command:
+      - sh
+      - -c
+      - |
+        curl -s -w %{http_code} \
+          -X POST http://aigw:1975/v1/images/generations \
+          -H "Authorization: Bearer unused" \
+          -H "Content-Type: application/json" \
+          -d "{\"model\":\"$$IMAGE_GENERATION_MODEL\",\"prompt\":\"A watercolor painting of a red fox in a birch forest\",\"size\":\"1024x1024\",\"quality\":\"low\"}"

cmd/aigw/docker-compose.yaml

@@ -152,6 +152,6 @@ services:
           -X POST http://aigw:1975/v1/images/generations \
           -H "Authorization: Bearer unused" \
           -H "Content-Type: application/json" \
-          -d "{\"model\":\"$$IMAGE_GENERATION_MODEL\",\"prompt\":\"A watercolor painting of a red fox in a birch forest\",\"size\":\"256x256\"}"
+          -d "{\"model\":\"$$IMAGE_GENERATION_MODEL\",\"prompt\":\"A watercolor painting of a red fox in a birch forest\",\"size\":\"1024x1024\",\"quality\":\"low\"}"
     extra_hosts:  # localhost:host-gateway trick doesn't work with aigw
       - "host.docker.internal:host-gateway"