← All changes | _inc/lib/class-jetpack-ai-helper.php +154 -87 12.8.3 → 16.3-a.7 View file →

		@@ -11,8 +11,13 @@
11	11	use Automattic\Jetpack\Search\Plan as Search_Plan;
12	12	use Automattic\Jetpack\Status;
13	13	use Automattic\Jetpack\Status\Visitor;
14	14
	15	+// Required directly rather than relying on the plugin bootstrap: on
	16	+// WordPress.com Simple this helper is loaded outside load-jetpack.php
	17	+// (see the GUIDELINES_BANNER_DISMISSED_META_KEY note below).
	18	+require_once __DIR__ . '/class-jetpack-ai-settings.php';
	19	+
15	20	/**
16	21	* Class Jetpack_AI_Helper
17	22	*
18	23	* @since 11.8
		@@ -18,8 +23,23 @@
18	23	* @since 11.8
19	24	*/
20	25	class Jetpack_AI_Helper {
21	26	/**
	27	+ * User meta key storing whether the user dismissed the Content Guidelines
	28	+ * AI empty-state banner. Written by the guidelines-banner-dismissed REST
	29	+ * endpoint and read by the Content Guidelines admin-page preload.
	30	+ *
	31	+ * Lives here rather than on the endpoint class because on WordPress.com
	32	+ * Simple the wpcom-endpoints classes are only loaded in REST requests,
	33	+ * while the preload needs the key during admin page loads.
	34	+ *
	35	+ * @since 16.1
	36	+ *
	37	+ * @var string
	38	+ */
	39	+ const GUIDELINES_BANNER_DISMISSED_META_KEY = 'jetpack_content_guidelines_ai_banner_dismissed';
	40	+
	41	+ /**
22	42	* Allow new completion every X seconds. Will return cached result otherwise.
23	43	*
24	44	* @var int
25	45	*/
		@@ -32,15 +52,22 @@
32	52	*/
33	53	public static $image_generation_cache_timeout = MONTH_IN_SECONDS;
34	54
35	55	/**
36		- * Cache AI-assistant feature for ten seconds.
	56	+ * Cache AI-assistant feature for 60 seconds.
37	57	*
38	58	* @var int
39	59	*/
40		- public static $ai_assistant_feature_cache_timeout = 10;
	60	+ public static $ai_assistant_feature_cache_timeout = 60;
41	61
42	62	/**
	63	+ * Cache AI-assistant errors for ten seconds.
	64	+ *
	65	+ * @var int
	66	+ */
	67	+ public static $ai_assistant_feature_error_cache_timeout = 10;
	68	+
	69	+ /**
43	70	* Stores the number of JetpackAI calls in case we want to mark AI-assisted posts some way.
44	71	*
45	72	* @var int
46	73	*/
		@@ -46,8 +73,15 @@
46	73	*/
47	74	public static $post_meta_with_ai_generation_number = '_jetpack_ai_calls';
48	75
49	76	/**
	77	+ * Storing the error to prevent repeated requests to WPCOM after failure.
	78	+ *
	79	+ * @var null|WP_Error
	80	+ */
	81	+ private static $ai_assistant_failed_request = null;
	82	+
	83	+ /**
50	84	* Checks if a given request is allowed to get AI data from WordPress.com.
51	85	*
52	86	* @param WP_REST_Request $request Full details about the request.
53	87	*
		@@ -83,19 +117,55 @@
83	117	} elseif ( ( new Automattic\Jetpack\Status\Host() )->is_woa_site() ) {
84	118	$default = true;
85	119	}
86	120
87		- /**
88		- * Filter whether the AI features are enabled in the Jetpack plugin.
89		- *
90		- * @since 11.8
91		- *
92		- * @param bool $default Are AI features enabled? Defaults to false.
93		- */
94		- return apply_filters( 'jetpack_ai_enabled', $default );
	121	+ // The jetpack_ai_enabled filter runs inside the helper; the host and
	122	+ // master gates apply after the chain and cannot be filtered back on.
	123	+ return Jetpack_AI_Settings::is_ai_enabled( $default );
95	124	}
96	125
97	126	/**
	127	+ * Return true if the Content Guidelines AI surfaces should be active on the
	128	+ * current site.
	129	+ *
	130	+ * Same platforms as is_enabled() (WPCOM Simple and Atomic), plus WordPress
	131	+ * VIP sites. Kept separate from is_enabled() so widening Content Guidelines
	132	+ * to VIP does not also open the general AI proxy endpoint there.
	133	+ *
	134	+ * @since 16.2
	135	+ *
	136	+ * @return bool
	137	+ */
	138	+ public static function is_enabled_for_content_guidelines() {
	139	+ $default = false;
	140	+
	141	+ if ( defined( 'IS_WPCOM' ) && IS_WPCOM ) {
	142	+ $default = true;
	143	+ } else {
	144	+ $host = new Automattic\Jetpack\Status\Host();
	145	+ if ( $host->is_woa_site() || $host->is_vip_site() ) {
	146	+ $default = true;
	147	+ }
	148	+ }
	149	+
	150	+ // The jetpack_ai_enabled filter runs inside the helper; the host and
	151	+ // master gates apply after the chain and cannot be filtered back on.
	152	+ return Jetpack_AI_Settings::is_ai_enabled( $default );
	153	+ }
	154	+
	155	+ /**
	156	+ * Whether the current user has dismissed the Content Guidelines AI
	157	+ * empty-state banner.
	158	+ *
	159	+ * @since 16.1
	160	+ *
	161	+ * @return bool
	162	+ */
	163	+ public static function is_guidelines_banner_dismissed() {
	164	+ return (bool) get_user_meta( get_current_user_id(), self::GUIDELINES_BANNER_DISMISSED_META_KEY, true );
	165	+ }
	166	+
	167	+ /**
98	168	* Return true if the AI chat feature should be active on the current site.
99	169	*
100	170	* @todo IS_WPCOM (the endpoints need to be updated too).
101	171	*
		@@ -207,10 +277,28 @@
207	277	'content' => $content,
208	278	),
209	279	);
210	280
211		- $result = ( new OpenAI( 'openai', array( 'post_id' => $post_id ) ) )->request_chat_completion( $data );
	281	+ $openai = new OpenAI( 'openai', array( 'post_id' => $post_id ) );
	282	+ $moderation_result = $openai->moderate(
	283	+ implode(
	284	+ ' ',
	285	+ array_map(
	286	+ function ( $msg ) {
	287	+ return $msg['role'] === 'user' ? $msg['content'] : '';
	288	+ },
	289	+ $data
	290	+ )
	291	+ )
	292	+ );
212	293
	294	+ if ( is_wp_error( $moderation_result ) ) {
	295	+ return $moderation_result;
	296	+ }
	297	+
	298	+ $max_tokens = 480; // Default
	299	+ $result = $openai->request_chat_completion( $data, $max_tokens );
	300	+
213	301	if ( is_wp_error( $result ) ) {
214	302	return $result;
215	303	}
216	304
		@@ -231,9 +319,10 @@
231	319	),
232	320	wp_json_encode(
233	321	array(
234	322	'content' => $content,
235		- )
	323	+ ),
	324	+ JSON_UNESCAPED_SLASHES
236	325	),
237	326	'wpcom'
238	327	);
239	328
		@@ -305,9 +394,10 @@
305	394	),
306	395	wp_json_encode(
307	396	array(
308	397	'prompt' => $prompt,
309		- )
	398	+ ),
	399	+ JSON_UNESCAPED_SLASHES
310	400	),
311	401	'wpcom'
312	402	);
313	403
		@@ -331,104 +421,70 @@
331	421	*
332	422	* @return mixed
333	423	*/
334	424	public static function get_ai_assistance_feature() {
335		- $blog_id = Jetpack_Options::get_option( 'id' );
336		-
337		- // Try to pick the AI Assistant feature from cache.
338		- $transient_name = self::transient_name_for_ai_assistance_feature( $blog_id );
339		- $cache = get_transient( $transient_name );
340		- if ( $cache ) {
341		- return $cache;
342		- }
343		-
344	425	if ( defined( 'IS_WPCOM' ) && IS_WPCOM ) {
345		- $has_ai_assistant_feature = \wpcom_site_has_feature( 'ai-assistant' );
346		- if ( ! class_exists( 'OpenAI' ) ) {
347		- \require_lib( 'openai' );
348		- }
	426	+ // On WPCOM, we can get the ID from the site.
	427	+ $blog_id = get_current_blog_id();
	428	+ $has_ai_assistant_feature = \wpcom_site_has_feature( 'ai-assistant', $blog_id );
349	429
350		- if ( ! class_exists( 'OpenAI_Limit_Usage' ) ) {
351		- if ( is_readable( WP_PLUGIN_DIR . '/openai/openai-limit-usage.php' ) ) {
352		- require_once WP_PLUGIN_DIR . '/openai/openai-limit-usage.php';
	430	+ if ( ! class_exists( 'WPCOM\Jetpack_AI\Usage\Helper' ) ) {
	431	+ if ( is_readable( WP_CONTENT_DIR . '/lib/jetpack-ai/usage/helper.php' ) ) {
	432	+ require_once WP_CONTENT_DIR . '/lib/jetpack-ai/usage/helper.php';
353	433	} else {
354	434	return new WP_Error(
355		- 'openai_limit_usage_not_found',
356		- __( 'OpenAI_Limit_Usage class not found.', 'jetpack' )
	435	+ 'jetpack_ai_usage_helper_not_found',
	436	+ __( 'WPCOM\Jetpack_AI\Usage\Helper class not found.', 'jetpack' )
357	437	);
358	438	}
359	439	}
360	440
361		- if ( ! class_exists( 'OpenAI_Request_Count' ) ) {
362		- if ( is_readable( WP_PLUGIN_DIR . '/openai/openai-request-count.php' ) ) {
363		- require_once WP_PLUGIN_DIR . '/openai/openai-request-count.php';
	441	+ if ( ! class_exists( 'WPCOM\Jetpack_AI\Feature_Control' ) ) {
	442	+ if ( is_readable( WP_CONTENT_DIR . '/lib/jetpack-ai/feature-control.php' ) ) {
	443	+ require_once WP_CONTENT_DIR . '/lib/jetpack-ai/feature-control.php';
364	444	} else {
365	445	return new WP_Error(
366		- 'openai_request_count_not_found',
367		- __( 'OpenAI_Request_Count class not found.', 'jetpack' )
	446	+ 'jetpack_ai_feature_control_not_found',
	447	+ __( 'WPCOM\Jetpack_AI\Feature_Control class not found.', 'jetpack' )
368	448	);
369	449	}
370	450	}
371	451
372		- if ( ! class_exists( 'WPCOM\Jetpack_AI\Usage\Helper' ) ) {
373		- if ( is_readable( WP_CONTENT_DIR . '/lib/jetpack-ai/usage/helper.php' ) ) {
374		- require_once WP_CONTENT_DIR . '/lib/jetpack-ai/usage/helper.php';
375		- } else {
376		- return new WP_Error(
377		- 'jetpack_ai_usage_helper_not_found',
378		- __( 'WPCOM\Jetpack_AI\Usage\Helper class not found.', 'jetpack' )
379		- );
380		- }
381		- }
382		-
383		- $blog_id = get_current_blog_id();
384		- $is_over_limit = \OpenAI_Limit_Usage::is_blog_over_request_limit( $blog_id );
385		- $requests_limit = \OpenAI_Limit_Usage::get_free_requests_limit( $blog_id );
386		- $requests_count = \OpenAI_Request_Count::get_count( $blog_id );
387		-
388		- /*
389		- * Usage since the last plan purchase day
390		- */
391		- $usage_period_start = null;
392		- $usage_next_period_start = null;
393		- $usage_period_requests_count = 0;
394		-
395		- /*
396		- * Get current tier value, a number representing
397		- * the current tier of the site.
398		- *
399		- * - 0 represents a site with the free plan.
400		- * - 1 represents a site with the current, unlimited plan.
401		- * - 100, 200, 500 represents a site with the new plans,
402		- * with the respective number of allowed requests.
403		- */
404		- $current_tier_value = $has_ai_assistant_feature ? 1 : 0;
405		-
406		- // Check if the site requires an upgrade.
407		- $require_upgrade = $is_over_limit && ! $has_ai_assistant_feature;
408		-
409	452	// Determine the upgrade type
410	453	$upgrade_type = wpcom_is_vip( $blog_id ) ? 'vip' : 'default';
411	454
412	455	return array(
413	456	'has-feature' => $has_ai_assistant_feature,
414		- 'is-over-limit' => $is_over_limit,
415		- 'requests-count' => $requests_count,
416		- 'requests-limit' => $requests_limit,
417		- 'usage-period' => array(
418		- 'current-start' => $usage_period_start,
419		- 'next-start' => $usage_next_period_start,
420		- 'requests-count' => $usage_period_requests_count,
421		- ),
422		- 'site-require-upgrade' => $require_upgrade,
	457	+ 'is-over-limit' => WPCOM\Jetpack_AI\Usage\Helper::is_over_limit( $blog_id ),
	458	+ 'requests-count' => WPCOM\Jetpack_AI\Usage\Helper::get_all_time_requests_count( $blog_id ),
	459	+ 'requests-limit' => WPCOM\Jetpack_AI\Usage\Helper::get_free_requests_limit( $blog_id ),
	460	+ 'usage-period' => WPCOM\Jetpack_AI\Usage\Helper::get_period_data( $blog_id ),
	461	+ 'site-require-upgrade' => WPCOM\Jetpack_AI\Usage\Helper::site_requires_upgrade( $blog_id ),
423	462	'upgrade-type' => $upgrade_type,
424		- 'current-tier' => array(
425		- 'value' => $current_tier_value,
426		- ),
	463	+ 'upgrade-url' => WPCOM\Jetpack_AI\Usage\Helper::get_upgrade_url( $blog_id ),
	464	+ 'current-tier' => WPCOM\Jetpack_AI\Usage\Helper::get_current_tier( $blog_id ),
	465	+ 'next-tier' => WPCOM\Jetpack_AI\Usage\Helper::get_next_tier( $blog_id ),
427	466	'tier-plans' => WPCOM\Jetpack_AI\Usage\Helper::get_tier_plans_list(),
	467	+ 'tier-plans-enabled' => WPCOM\Jetpack_AI\Usage\Helper::ai_tier_plans_enabled(),
	468	+ 'costs' => WPCOM\Jetpack_AI\Usage\Helper::get_costs(),
	469	+ 'features-control' => WPCOM\Jetpack_AI\Feature_Control::get_features(),
428	470	);
429	471	}
430	472
	473	+ // Outside of WPCOM, we need to fetch the data from the site.
	474	+ $blog_id = Jetpack_Options::get_option( 'id' );
	475	+
	476	+ // Try to pick the AI Assistant feature from cache.
	477	+ $transient_name = self::transient_name_for_ai_assistance_feature( $blog_id );
	478	+ $cache = get_transient( $transient_name );
	479	+ if ( $cache ) {
	480	+ return $cache;
	481	+ }
	482	+
	483	+ if ( null !== static::$ai_assistant_failed_request ) {
	484	+ return static::$ai_assistant_failed_request;
	485	+ }
	486	+
431	487	$request_path = sprintf( '/sites/%d/jetpack-ai/ai-assistant-feature', $blog_id );
432	488
433	489	$wpcom_request = Client::wpcom_json_api_request_as_user(
434	490	$request_path,
		@@ -437,8 +493,9 @@
437	493	'method' => 'GET',
438	494	'headers' => array(
439	495	'X-Forwarded-For' => ( new Visitor() )->get_ip( true ),
440	496	),
	497	+ 'timeout' => 30,
441	498	),
442	499	null,
443	500	'wpcom'
444	501	);
		@@ -451,12 +508,22 @@
451	508	set_transient( $transient_name, $ai_assistant_feature_data, self::$ai_assistant_feature_cache_timeout );
452	509
453	510	return $ai_assistant_feature_data;
454	511	} else {
455		- return new WP_Error(
	512	+ $error = new WP_Error(
456	513	'failed_to_fetch_data',
457	514	esc_html__( 'Unable to fetch the requested data.', 'jetpack' ),
458		- array( 'status' => $response_code )
	515	+ array(
	516	+ 'status' => $response_code,
	517	+ 'ts' => time(),
	518	+ )
459	519	);
	520	+
	521	+ // Cache the AI Assistant feature error, for Jetpack sites, avoid API hammering.
	522	+ set_transient( $transient_name, $error, self::$ai_assistant_feature_error_cache_timeout );
	523	+
	524	+ static::$ai_assistant_failed_request = $error;
	525	+
	526	+ return $error;
460	527	}
461	528	}
462	529	}