Add MediaOrientationEnum and include outputMediaOrientation param for model config.

Author: felixarntz

src/Files/Enums/MediaOrientationEnum.php

@@ -0,0 +1,43 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WordPress\AiClient\Files\Enums;
+
+use WordPress\AiClient\Common\AbstractEnum;
+
+/**
+ * Represents the type of file storage.
+ *
+ * @method static self square() Returns the square orientation
+ * @method static self landscape() Returns the landscape orientation.
+ * @method static self portrait() Returns the portrait orientation.
+ * @method bool isSquare() Checks if this is an square orientation
+ * @method bool isLandscape() Checks if this is a landscape orientation.
+ * @method bool isPortrait() Checks if this is a portrait orientation.
+ *
+ * @since n.e.x.t
+ */
+class MediaOrientationEnum extends AbstractEnum
+{
+    /**
+     * Square orientation.
+     *
+     * @var string
+     */
+    public const SQUARE = 'square';
+
+    /**
+     * Landscape orientation.
+     *
+     * @var string
+     */
+    public const LANDSCAPE = 'landscape';
+
+    /**
+     * Portrait orientation.
+     *
+     * @var string
+     */
+    public const PORTRAIT = 'portrait';
+}

src/Providers/Models/DTO/ModelConfig.php

@@ -7,6 +7,7 @@
 use InvalidArgumentException;
 use WordPress\AiClient\Common\AbstractDataTransferObject;
 use WordPress\AiClient\Files\Enums\FileTypeEnum;
+use WordPress\AiClient\Files\Enums\MediaOrientationEnum;
 use WordPress\AiClient\Messages\Enums\ModalityEnum;
 use WordPress\AiClient\Tools\DTO\Tool;
 
@@ -38,6 +39,7 @@
  *     outputFileType?: string,
  *     outputMimeType?: string,
  *     outputSchema?: array<string, mixed>,
+ *     outputMediaOrientation?: string,
  *     customOptions?: array<string, mixed>
  * }
  *
@@ -61,6 +63,7 @@ class ModelConfig extends AbstractDataTransferObject
     public const KEY_OUTPUT_FILE_TYPE = 'outputFileType';
     public const KEY_OUTPUT_MIME_TYPE = 'outputMimeType';
     public const KEY_OUTPUT_SCHEMA = 'outputSchema';
+    public const KEY_OUTPUT_MEDIA_ORIENTATION = 'outputMediaOrientation';
     public const KEY_CUSTOM_OPTIONS = 'customOptions';
 
     /**
@@ -143,6 +146,11 @@ class ModelConfig extends AbstractDataTransferObject
      */
     protected ?array $outputSchema = null;
 
+    /**
+     * @var MediaOrientationEnum|null Output media orientation.
+     */
+    protected ?MediaOrientationEnum $outputMediaOrientation = null;
+
     /**
      * @var array<string, mixed> Custom provider-specific options.
      */
@@ -558,6 +566,30 @@ public function getOutputSchema(): ?array
         return $this->outputSchema;
     }
 
+    /**
+     * Sets the output media orientation.
+     *
+     * @since n.e.x.t
+     *
+     * @param MediaOrientationEnum $outputMediaOrientation The output media orientation.
+     */
+    public function setOutputMediaOrientation(MediaOrientationEnum $outputMediaOrientation): void
+    {
+        $this->outputMediaOrientation = $outputMediaOrientation;
+    }
+
+    /**
+     * Gets the output media orientation.
+     *
+     * @since n.e.x.t
+     *
+     * @return MediaOrientationEnum|null The output media orientation.
+     */
+    public function getOutputMediaOrientation(): ?MediaOrientationEnum
+    {
+        return $this->outputMediaOrientation;
+    }
+
     /**
      * Sets a single custom option.
      *
@@ -687,6 +719,11 @@ public static function getJsonSchema(): array
                     'additionalProperties' => true,
                     'description' => 'Output schema (JSON schema).',
                 ],
+                self::KEY_OUTPUT_MEDIA_ORIENTATION => [
+                    'type' => 'string',
+                    'enum' => MediaOrientationEnum::getValues(),
+                    'description' => 'Output media orientation.',
+                ],
                 self::KEY_CUSTOM_OPTIONS => [
                     'type' => 'object',
                     'additionalProperties' => true,
@@ -779,6 +816,10 @@ static function (ModalityEnum $modality): string {
             $data[self::KEY_OUTPUT_SCHEMA] = $this->outputSchema;
         }
 
+        if ($this->outputMediaOrientation !== null) {
+            $data[self::KEY_OUTPUT_MEDIA_ORIENTATION] = $this->outputMediaOrientation->value;
+        }
+
         $data[self::KEY_CUSTOM_OPTIONS] = $this->customOptions;
 
         return $data;
@@ -862,6 +903,10 @@ public static function fromArray(array $array): self
             $config->setOutputSchema($array[self::KEY_OUTPUT_SCHEMA]);
         }
 
+        if (isset($array[self::KEY_OUTPUT_MEDIA_ORIENTATION])) {
+            $config->setOutputMediaOrientation(MediaOrientationEnum::from($array[self::KEY_OUTPUT_MEDIA_ORIENTATION]));
+        }
+
         if (isset($array[self::KEY_CUSTOM_OPTIONS])) {
             $config->setCustomOptions($array[self::KEY_CUSTOM_OPTIONS]);
         }