feat(xai): Add XAI image generation handler and tests

src/Providers/XAI/Handlers/Images.php

@@ -0,0 +1,77 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Prism\Prism\Providers\XAI\Handlers;
+
+use Illuminate\Http\Client\PendingRequest;
+use Illuminate\Http\Client\Response as ClientResponse;
+use Prism\Prism\Images\Request;
+use Prism\Prism\Images\Response;
+use Prism\Prism\Images\ResponseBuilder;
+use Prism\Prism\Providers\XAI\Concerns\ValidatesResponses;
+use Prism\Prism\Providers\XAI\Maps\ImageRequestMap;
+use Prism\Prism\ValueObjects\GeneratedImage;
+use Prism\Prism\ValueObjects\Meta;
+use Prism\Prism\ValueObjects\Usage;
+
+class Images
+{
+    use ValidatesResponses;
+
+    public function __construct(protected PendingRequest $client) {}
+
+    public function handle(Request $request): Response
+    {
+        $response = $this->sendRequest($request);
+
+        $this->validateResponse($response);
+
+        $data = $response->json();
+
+        $images = $this->extractImages($data);
+
+        $responseBuilder = new ResponseBuilder(
+            usage: new Usage(
+                promptTokens: data_get($data, 'usage.input_tokens', data_get($data, 'usage.prompt_tokens', 0)),
+                completionTokens: data_get($data, 'usage.output_tokens', data_get($data, 'usage.completion_tokens', 0)),
+            ),
+            meta: new Meta(
+                id: data_get($data, 'id', 'img_'.bin2hex(random_bytes(8))),
+                model: data_get($data, 'model', $request->model()),
+                rateLimits: [],
+            ),
+            images: $images,
+            raw: $data,
+        );
+
+        return $responseBuilder->toResponse();
+    }
+
+    protected function sendRequest(Request $request): ClientResponse
+    {
+        /** @var ClientResponse $response */
+        $response = $this->client->post('images/generations', ImageRequestMap::map($request));
+
+        return $response;
+    }
+
+    /**
+     * @param  array<string, mixed>  $data
+     * @return GeneratedImage[]
+     */
+    protected function extractImages(array $data): array
+    {
+        $images = [];
+
+        foreach (data_get($data, 'data', []) as $imageData) {
+            $images[] = new GeneratedImage(
+                url: data_get($imageData, 'url'),
+                base64: data_get($imageData, 'b64_json'),
+                revisedPrompt: data_get($imageData, 'revised_prompt'),
+            );
+        }
+
+        return $images;
+    }
+}

src/Providers/XAI/Maps/ImageRequestMap.php

@@ -0,0 +1,40 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Prism\Prism\Providers\XAI\Maps;
+
+use Illuminate\Support\Arr;
+use Prism\Prism\Images\Request;
+
+class ImageRequestMap
+{
+    /**
+     * @return array<string, mixed>
+     */
+    public static function map(Request $request): array
+    {
+        $baseData = [
+            'model' => $request->model(),
+            'prompt' => $request->prompt(),
+        ];
+
+        $providerOptions = $request->providerOptions();
+
+        $supportedOptions = [
+            'n' => $providerOptions['n'] ?? null,
+            'response_format' => $providerOptions['response_format'] ?? null,
+            'aspect_ratio' => $providerOptions['aspect_ratio'] ?? null,
+            'resolution' => $providerOptions['resolution'] ?? null,
+        ];
+
+        // Include any additional options not explicitly handled above
+        $additionalOptions = array_diff_key($providerOptions, $supportedOptions);
+
+        return array_merge(
+            $baseData,
+            Arr::whereNotNull($supportedOptions),
+            $additionalOptions
+        );
+    }
+}

src/Providers/XAI/XAI.php

@@ -10,7 +10,10 @@
 use Prism\Prism\Concerns\InitializesClient;
 use Prism\Prism\Exceptions\PrismException;
 use Prism\Prism\Exceptions\PrismRateLimitedException;
+use Prism\Prism\Images\Request as ImagesRequest;
+use Prism\Prism\Images\Response as ImagesResponse;
 use Prism\Prism\Providers\Provider;
+use Prism\Prism\Providers\XAI\Handlers\Images;
 use Prism\Prism\Providers\XAI\Handlers\Stream;
 use Prism\Prism\Providers\XAI\Handlers\Structured;
 use Prism\Prism\Providers\XAI\Handlers\Text;
@@ -47,6 +50,17 @@ public function stream(TextRequest $request): Generator
         return $handler->handle($request);
     }
 
+    #[\Override]
+    public function images(ImagesRequest $request): ImagesResponse
+    {
+        $handler = new Images($this->client(
+            $request->clientOptions(),
+            $request->clientRetry()
+        ));
+
+        return $handler->handle($request);
+    }
+
     #[\Override]
     public function structured(StructuredRequest $request): StructuredResponse
     {