Changeset 62067 for trunk/src/wp-includes/ai-client/class-wp-ai-client-prompt-builder.php

Timestamp:

03/19/2026 01:55:45 PM (7 months ago)

Author:

SergeyBiryukov

Message:

AI: Introduce wp_supports_ai() function to control LLM-related features.

This includes a WP_AI_SUPPORT constant and a wp_supports_ai filter.

When false,

• _wp_connectors_get_provider_settings() will return an empty array (short-circuiting the other functionality).
• WP_AI_Client_Prompt_Builder() will short-circuit the construction with a WP_Error(). wp_ai_client_prompt() will still return an instance, to allow for fluidity to remain intact.

Priority: WP_AI_SUPPORT > add_filter( 'wp_supports_ai', ...) > (default: true)

Follow-up to [61943], [61749], [61943].

Props justlevine, westonruter, gziolo, flixos90, romainmrhenry, ahortin, chrismcelroyseo, SergeyBiryukov.
See #64706.

File:

• trunk/src/wp-includes/ai-client/class-wp-ai-client-prompt-builder.php (modified) (6 diffs)

trunk/src/wp-includes/ai-client/class-wp-ai-client-prompt-builder.php

@@ -46 +46 @@
  *
  * @since 7.0.0
+ *
+ * @phpstan-import-type Prompt from PromptBuilder
  *
  * @method self with_text(string $text) Adds text to the current message.
@@ -171 +173 @@
          * @since 7.0.0
          *
-         * @param ProviderRegistry                                                                 $registry The provider registry for finding suitable models.
-         * @param string|MessagePart|Message|array|list<string|MessagePart|array>|list<Message>|null $prompt   Optional. Initial prompt content.
-         *                                                                                                    A string for simple text prompts,
-         *                                                                                                    a MessagePart or Message object for
-         *                                                                                                    structured content, an array for a
-         *                                                                                                    message array shape, or a list of
-         *                                                                                                    parts or messages for multi-turn
-         *                                                                                                    conversations. Default null.
+         * @param ProviderRegistry $registry The provider registry for finding suitable models.
+         * @param Prompt           $prompt   Optional. Initial prompt content.
+         *                                   A string for simple text prompts,
+         *                                   a MessagePart or Message object for
+         *                                   structured content, an array for a
+         *                                   message array shape, or a list of
+         *                                   parts or messages for multi-turn
+         *                                   conversations. Default null.
          */
         public function __construct( ProviderRegistry $registry, $prompt = null ) {
@@ -290 +292 @@
                 // Check if the prompt should be prevented for is_supported* and generate_*/convert_text_to_speech* methods.
                 if ( self::is_support_check_method( $name ) || self::is_generating_method( $name ) ) {
-                        /**
-                         * Filters whether to prevent the prompt from being executed.
-                         *
-                         * @since 7.0.0
-                         *
-                         * @param bool                        $prevent Whether to prevent the prompt. Default false.
-                         * @param WP_AI_Client_Prompt_Builder $builder A clone of the prompt builder instance (read-only).
-                         */
-                        $prevent = (bool) apply_filters( 'wp_ai_client_prevent_prompt', false, clone $this );
+                        // If AI is not supported, then there's no need to apply the filter as the prompt will be prevented anyway.
+                        $is_ai_disabled = ! wp_supports_ai();
+                        $prevent        = $is_ai_disabled;
+                        if ( ! $prevent ) {
+                                /**
+                                 * Filters whether to prevent the prompt from being executed.
+                                 *
+                                 * @since 7.0.0
+                                 *
+                                 * @param bool                        $prevent Whether to prevent the prompt. Default false.
+                                 * @param WP_AI_Client_Prompt_Builder $builder A clone of the prompt builder instance (read-only).
+                                 */
+                                $prevent = (bool) apply_filters( 'wp_ai_client_prevent_prompt', false, clone $this );
+                        }
 
                         if ( $prevent ) {
@@ -306 +313 @@
                                 }
 
+                                $error_message = $is_ai_disabled
+                                        ? __( 'AI features are not supported in this environment.' )
+                                        : __( 'Prompt execution was prevented by a filter.' );
+
                                 // For generate_* and convert_text_to_speech* methods, create a WP_Error.
                                 $this->error = new WP_Error(
                                         'prompt_prevented',
-                                        __( 'Prompt execution was prevented by a filter.' ),
+                                        $error_message,
                                         array(
                                                 'status' => 503,
@@ -424 +435 @@
                 $camel_case_name = $this->snake_to_camel_case( $name );
 
-                if ( ! is_callable( array( $this->builder, $camel_case_name ) ) ) {
+                $method = array( $this->builder, $camel_case_name );
+                if ( ! is_callable( $method ) ) {
                         throw new BadMethodCallException(
                                 sprintf(
@@ -435 +447 @@
                 }
 
-                return array( $this->builder, $camel_case_name );
+                return $method;
         }