Merge pull request #226 from dkotter/fix/increase-image-timeout

Ensure we set a higher timeout when generating images

Co-authored-by: dkotter <dkotter@git.wordpress.org>

Author: jeffpaul

includes/Abilities/Image/Generate_Image.php

@@ -15,6 +15,7 @@
 use WordPress\AI_Client\AI_Client;
 use WordPress\AiClient\Files\Enums\FileTypeEnum;
 use WordPress\AiClient\Providers\DTO\ProviderMetadata;
+use WordPress\AiClient\Providers\Http\DTO\RequestOptions;
 use WordPress\AiClient\Providers\Models\DTO\ModelMetadata;
 
 use function WordPress\AI\get_preferred_image_models;
@@ -165,8 +166,12 @@ protected function meta(): array {
 	 * @return array{data: string, provider_metadata: array<string, string>, model_metadata: array<string, string>}|\WP_Error The generated image data, provider metadata, and model metadata, or a WP_Error if there was an error.
 	 */
 	protected function generate_image( string $prompt ) { // phpcs:ignore Generic.NamingConventions.ConstructorName.OldStyle
+		$request_options = new RequestOptions();
+		$request_options->setTimeout( 90 );
+
 		// Generate the image using the AI client.
 		$result = AI_Client::prompt_with_wp_error( $prompt )
+			->using_request_options( $request_options )
 			->as_output_file_type( FileTypeEnum::inline() )
 			->using_model_preference( ...get_preferred_image_models() )
 			->generate_image_result();