Implement image generation support

cli.php

@@ -156,7 +156,11 @@ function logError(string $message, int $exit_code = 1): void
 }
 
 try {
-    $result = $promptBuilder->generateTextResult();
+    if ($outputFormat === 'image-json' || $outputFormat === 'image-base64') {
+        $result = $promptBuilder->generateImageResult();
+    } else {
+        $result = $promptBuilder->generateTextResult();
+    }
 } catch (InvalidArgumentException $e) {
     logError('Invalid arguments while trying to generate text result: ' . $e->getMessage());
 } catch (ResponseException $e) {
@@ -173,6 +177,12 @@ function logError(string $message, int $exit_code = 1): void
     case 'candidates-json':
         $output = json_encode($result->getCandidates(), JSON_PRETTY_PRINT);
         break;
+    case 'image-json':
+        $output = json_encode($result->toFile(), JSON_PRETTY_PRINT);
+        break;
+    case 'image-base64':
+        $output = $result->toFile()->getBase64Data();
+        break;
     case 'message-text':
     default:
         $output = $result->toText();

src/ProviderImplementations/Google/GoogleImageGenerationModel.php

@@ -0,0 +1,30 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WordPress\AiClient\ProviderImplementations\Google;
+
+use WordPress\AiClient\Providers\Http\DTO\Request;
+use WordPress\AiClient\Providers\Http\Enums\HttpMethodEnum;
+use WordPress\AiClient\Providers\OpenAiCompatibleImplementation\AbstractOpenAiCompatibleImageGenerationModel;
+
+/**
+ * Class for a Google image generation model.
+ *
+ * @since n.e.x.t
+ */
+class GoogleImageGenerationModel extends AbstractOpenAiCompatibleImageGenerationModel
+{
+    /**
+     * @inheritDoc
+     */
+    protected function createRequest(HttpMethodEnum $method, string $path, array $headers = [], $data = null): Request
+    {
+        return new Request(
+            $method,
+            GoogleProvider::BASE_URI . '/openai/' . ltrim($path, '/'),
+            $headers,
+            $data
+        );
+    }
+}

src/ProviderImplementations/Google/GoogleModelMetadataDirectory.php

@@ -158,10 +158,13 @@ protected function parseResponseToModelMetadataList(Response $response): array
             new SupportedOption(OptionEnum::outputFileType(), [FileTypeEnum::inline()]),
             new SupportedOption(OptionEnum::outputMediaOrientation(), [
                 MediaOrientationEnum::square(),
-                MediaOrientationEnum::landscape(),
-                MediaOrientationEnum::portrait(),
+                // The following orientations are normally supported, but not when using the OpenAI compatible endpoint.
+                // MediaOrientationEnum::landscape(),
+                // MediaOrientationEnum::portrait(),
             ]),
-            new SupportedOption(OptionEnum::outputMediaAspectRatio(), ['1:1', '16:9', '4:3', '9:16', '3:4']),
+            // Aspect ratio is normally supported, but not when using the OpenAI compatible endpoint.
+            // new SupportedOption(OptionEnum::outputMediaAspectRatio(), ['1:1', '16:9', '4:3', '9:16', '3:4']),
+            new SupportedOption(OptionEnum::customOptions()),
         ];
 
         $modelsData = (array) $responseData['models'];

src/ProviderImplementations/Google/GoogleProvider.php

@@ -38,10 +38,7 @@ protected static function createModel(
                 return new GoogleTextGenerationModel($modelMetadata, $providerMetadata);
             }
             if ($capability->isImageGeneration()) {
-                // TODO: Implement GoogleImageGenerationModel.
-                throw new RuntimeException(
-                    'Google image generation model class is not yet implemented.'
-                );
+                return new GoogleImageGenerationModel($modelMetadata, $providerMetadata);
             }
         }
 

src/ProviderImplementations/OpenAi/OpenAiImageGenerationModel.php

@@ -0,0 +1,50 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WordPress\AiClient\ProviderImplementations\OpenAi;
+
+use WordPress\AiClient\Providers\Http\DTO\Request;
+use WordPress\AiClient\Providers\Http\Enums\HttpMethodEnum;
+use WordPress\AiClient\Providers\OpenAiCompatibleImplementation\AbstractOpenAiCompatibleImageGenerationModel;
+
+/**
+ * Class for an OpenAI image generation model.
+ *
+ * @since n.e.x.t
+ */
+class OpenAiImageGenerationModel extends AbstractOpenAiCompatibleImageGenerationModel
+{
+    /**
+     * @inheritDoc
+     */
+    protected function createRequest(HttpMethodEnum $method, string $path, array $headers = [], $data = null): Request
+    {
+        return new Request(
+            $method,
+            OpenAiProvider::BASE_URI . '/' . ltrim($path, '/'),
+            $headers,
+            $data
+        );
+    }
+
+    /**
+     * @inheritDoc
+     */
+    protected function prepareGenerateImageParams(array $prompt): array
+    {
+        $params = parent::prepareGenerateImageParams($prompt);
+
+        /*
+         * Only the newer 'gpt-image-' models support passing a MIME type ('output_format').
+         * Conversely, they do not support 'response_format', but always return a base64 encoded image.
+         */
+        if (isset($params['model']) && is_string($params['model']) && 0 === strpos($params['model'], 'gpt-image-')) {
+            unset($params['response_format']);
+        } else {
+            unset($params['output_format']);
+        }
+
+        return $params;
+    }
+}

src/ProviderImplementations/OpenAi/OpenAiModelMetadataDirectory.php

@@ -126,6 +126,7 @@ protected function parseResponseToModelMetadataList(Response $response): array
                 MediaOrientationEnum::portrait(),
             ]),
             new SupportedOption(OptionEnum::outputMediaAspectRatio(), ['1:1', '7:4', '4:7']),
+            new SupportedOption(OptionEnum::customOptions()),
         ];
         $gptImageOptions = [
             new SupportedOption(OptionEnum::inputModalities(), [[ModalityEnum::text()]]),
@@ -139,6 +140,7 @@ protected function parseResponseToModelMetadataList(Response $response): array
                 MediaOrientationEnum::portrait(),
             ]),
             new SupportedOption(OptionEnum::outputMediaAspectRatio(), ['1:1', '3:2', '2:3']),
+            new SupportedOption(OptionEnum::customOptions()),
         ];
         $ttsCapabilities = [
             CapabilityEnum::textToSpeechConversion(),
@@ -154,6 +156,7 @@ protected function parseResponseToModelMetadataList(Response $response): array
                 'audio/aac',
             ]),
             new SupportedOption(OptionEnum::outputSpeechVoice()),
+            new SupportedOption(OptionEnum::customOptions()),
         ];
 
         $modelsData = (array) $responseData['data'];

src/ProviderImplementations/OpenAi/OpenAiProvider.php

@@ -38,10 +38,7 @@ protected static function createModel(
                 return new OpenAiTextGenerationModel($modelMetadata, $providerMetadata);
             }
             if ($capability->isImageGeneration()) {
-                // TODO: Implement OpenAiImageGenerationModel.
-                throw new RuntimeException(
-                    'OpenAI image generation model class is not yet implemented.'
-                );
+                return new OpenAiImageGenerationModel($modelMetadata, $providerMetadata);
             }
             if ($capability->isTextToSpeechConversion()) {
                 // TODO: Implement OpenAiTextToSpeechConversionModel.