Merge pull request #29 from WordPress/enhance/default-http-timeout

Author: JasonTheAdams

includes/Builders/Prompt_Builder.php

@@ -112,6 +112,23 @@ class Prompt_Builder {
 	 */
 	public function __construct( ProviderRegistry $registry, $prompt = null ) {
 		$this->builder = new PromptBuilder( $registry, $prompt );
+
+		/**
+		 * Filters the default request timeout in seconds for AI Client HTTP requests.
+		 *
+		 * @since n.e.x.t
+		 *
+		 * @param int $default_timeout The default timeout in seconds.
+		 */
+		$default_timeout = (int) apply_filters( 'wp_ai_client_default_request_timeout', 30 );
+
+		$this->builder->usingRequestOptions(
+			RequestOptions::fromArray(
+				array(
+					RequestOptions::KEY_TIMEOUT => $default_timeout,
+				)
+			)
+		);
 	}
 
 	/**

tests/phpunit/tests/Builders/Prompt_Builder_Tests.php

@@ -29,6 +29,7 @@
 use WordPress\AiClient\Providers\DTO\ProviderMetadata;
 use WordPress\AiClient\Providers\DTO\ProviderModelsMetadata;
 use WordPress\AiClient\Providers\Enums\ProviderTypeEnum;
+use WordPress\AiClient\Providers\Http\DTO\RequestOptions;
 use WordPress\AiClient\Providers\Models\Contracts\ModelInterface;
 use WordPress\AiClient\Providers\Models\DTO\ModelConfig;
 use WordPress\AiClient\Providers\Models\DTO\ModelMetadata;
@@ -139,6 +140,39 @@ public function test_instantiation_with_prompt(): void {
 		$this->assertInstanceOf( Prompt_Builder::class, $prompt_builder );
 	}
 
+	/**
+	 * Test that the constructor sets the default request timeout.
+	 */
+	public function test_constructor_sets_default_request_timeout(): void {
+		$builder = new Prompt_Builder( AiClient::defaultRegistry() );
+
+		/** @var RequestOptions $request_options */
+		$request_options = $this->get_wrapped_prompt_builder_property_value( $builder, 'requestOptions' );
+
+		$this->assertInstanceOf( RequestOptions::class, $request_options );
+		$this->assertEquals( 30, $request_options->getTimeout() );
+	}
+
+	/**
+	 * Test that the constructor allows overriding the default request timeout.
+	 */
+	public function test_constructor_allows_overriding_request_timeout(): void {
+		add_filter(
+			'wp_ai_client_default_request_timeout',
+			static function () {
+				return 45;
+			}
+		);
+
+		$builder = new Prompt_Builder( AiClient::defaultRegistry() );
+
+		/** @var RequestOptions $request_options */
+		$request_options = $this->get_wrapped_prompt_builder_property_value( $builder, 'requestOptions' );
+
+		$this->assertInstanceOf( RequestOptions::class, $request_options );
+		$this->assertEquals( 45, $request_options->getTimeout() );
+	}
+
 	/**
 	 * Test method chaining with fluent methods.
 	 *