diff --git a/includes/Abilities/Image/Generate_Image.php b/includes/Abilities/Image/Generate_Image.php index b257df1a..f7c55257 100644 --- a/includes/Abilities/Image/Generate_Image.php +++ b/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, model_metadata: array}|\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();