class-jetpack-ai-helper.php — Jetpack – WP Security, Backup, Speed, & Growth 12.1.3

jetpack / _inc / lib / class-jetpack-ai-helper.php

jetpack / _inc / lib Last commit date

class-jetpack-ai-helper.php

279 lines

1	<?php
2	/**
3	* API helper for the AI blocks.
4	*
5	* @package automattic/jetpack
6	* @since 11.8
7	*/
8
9	use Automattic\Jetpack\Connection\Client;
10	use Automattic\Jetpack\Connection\Manager;
11	use Automattic\Jetpack\Status;
12
13	/**
14	* Class Jetpack_AI_Helper
15	*
16	* @since 11.8
17	*/
18	class Jetpack_AI_Helper {
19	/**
20	* Allow new completion every X seconds. Will return cached result otherwise.
21	*
22	* @var int
23	*/
24	public static $text_completion_cooldown_seconds = 15;
25
26	/**
27	* Cache images for a prompt for a month.
28	*
29	* @var int
30	*/
31	public static $image_generation_cache_timeout = MONTH_IN_SECONDS;
32
33	/**
34	* Stores the number of JetpackAI calls in case we want to mark AI-assisted posts some way.
35	*
36	* @var int
37	*/
38	public static $post_meta_with_ai_generation_number = '_jetpack_ai_calls';
39
40	/**
41	* Checks if a given request is allowed to get AI data from WordPress.com.
42	*
43	* @param WP_REST_Request $request Full details about the request.
44	*
45	* @return true\|WP_Error True if the request has access, WP_Error object otherwise.
46	*/
47	public static function get_status_permission_check( $request ) { // phpcs:ignore Generic.CodeAnalysis.UnusedFunctionParameter, VariableAnalysis.CodeAnalysis.VariableAnalysis.UnusedVariable
48
49	/*
50	* This may need to be updated
51	* to take into account the different ways we can make requests
52	* (from a WordPress.com site, from a Jetpack site).
53	*/
54	if ( ! current_user_can( 'edit_posts' ) ) {
55	return new WP_Error(
56	'rest_forbidden',
57	__( 'Sorry, you are not allowed to access Jetpack AI help on this site.', 'jetpack' ),
58	array( 'status' => rest_authorization_required_code() )
59	);
60	}
61
62	return true;
63	}
64
65	/**
66	* Return true if these features should be active on the current site.
67	* Currently, it's limited to WPCOM Simple and Atomic.
68	*/
69	public static function is_enabled() {
70	$default = false;
71
72	if ( defined( 'IS_WPCOM' ) && IS_WPCOM ) {
73	$default = true;
74	} elseif ( ( new Automattic\Jetpack\Status\Host() )->is_woa_site() ) {
75	$default = true;
76	}
77
78	/**
79	* Filter whether the AI features are enabled in the Jetpack plugin.
80	*
81	* @since 11.8
82	*
83	* @param bool $default Are AI features enabled? Defaults to false.
84	*/
85	return apply_filters( 'jetpack_ai_enabled', $default );
86	}
87
88	/**
89	* Get the name of the transient for image generation. Unique per prompt and allows for reuse of results for the same prompt across entire WPCOM.
90	* I expext "puppy" to always be from cache.
91	*
92	* @param string $prompt - Supplied prompt.
93	*/
94	public static function transient_name_for_image_generation( $prompt ) {
95	return 'jetpack_openai_image_' . md5( $prompt );
96	}
97
98	/**
99	* Get the name of the transient for text completion. Unique per user, but not per text. Serves more as a cooldown.
100	*/
101	public static function transient_name_for_completion() {
102	return 'jetpack_openai_completion_' . get_current_user_id(); // Cache for each user, so that other users dont get weird cached version from somebody else.
103	}
104
105	/**
106	* Mark the edited post as "touched" by AI stuff.
107	*
108	* @param int $post_id Post ID for which the content is being generated.
109	* @return void
110	*/
111	private static function mark_post_as_ai_assisted( $post_id ) {
112	if ( ! $post_id ) {
113	return;
114	}
115	$previous = get_post_meta( $post_id, self::$post_meta_with_ai_generation_number, true );
116	if ( ! $previous ) {
117	$previous = 0;
118	} elseif ( ! is_numeric( $previous ) ) {
119	// Data corrupted, nothing to do.
120	return;
121	}
122	$new_value = intval( $previous ) + 1;
123	update_post_meta( $post_id, self::$post_meta_with_ai_generation_number, $new_value );
124	}
125
126	/**
127	* Get text back from WordPress.com based off a starting text.
128	*
129	* @param string $content The content that's already been typed in the block.
130	* @param int $post_id Post ID for which the content is being generated.
131	* @return mixed
132	*/
133	public static function get_gpt_completion( $content, $post_id ) {
134	$content = wp_strip_all_tags( $content );
135	$cache = get_transient( self::transient_name_for_completion() );
136	if ( $cache ) {
137	return $cache;
138	}
139
140	if ( ( new Status() )->is_offline_mode() ) {
141	return new WP_Error(
142	'dev_mode',
143	__( 'Jetpack AI is not available in offline mode.', 'jetpack' )
144	);
145	}
146
147	$site_id = Manager::get_site_id();
148	if ( is_wp_error( $site_id ) ) {
149	return $site_id;
150	}
151
152	if ( defined( 'IS_WPCOM' ) && IS_WPCOM ) {
153	if ( ! class_exists( 'OpenAI' ) ) {
154	\require_lib( 'openai' );
155	}
156
157	// Set the content for chatGPT endpoint
158	$data = array(
159	array(
160	'role' => 'user',
161	'content' => $content,
162	),
163	);
164
165	$result = ( new OpenAI( 'openai', array( 'post_id' => $post_id ) ) )->request_chat_completion( $data );
166
167	if ( is_wp_error( $result ) ) {
168	return $result;
169	}
170
171	$response = $result->choices[0]->message->content;
172
173	// In case of Jetpack we are setting a transient on the WPCOM and not the remote site. I think the 'get_current_user_id' may default for the connection owner at this point but we'll deal with this later.
174	set_transient( self::transient_name_for_completion(), $response, self::$text_completion_cooldown_seconds );
175	self::mark_post_as_ai_assisted( $post_id );
176	return $response;
177	}
178
179	$response = Client::wpcom_json_api_request_as_user(
180	sprintf( '/sites/%d/jetpack-ai/completions', $site_id ),
181	2,
182	array(
183	'method' => 'post',
184	'headers' => array( 'content-type' => 'application/json' ),
185	),
186	wp_json_encode(
187	array(
188	'content' => $content,
189	)
190	),
191	'wpcom'
192	);
193
194	if ( is_wp_error( $response ) ) {
195	return $response;
196	}
197
198	$data = json_decode( wp_remote_retrieve_body( $response ) );
199
200	if ( wp_remote_retrieve_response_code( $response ) >= 400 ) {
201	return new WP_Error( $data->code, $data->message, $data->data );
202	}
203	set_transient( self::transient_name_for_completion(), $data, self::$text_completion_cooldown_seconds );
204	self::mark_post_as_ai_assisted( $post_id );
205
206	return $data;
207	}
208
209	/**
210	* Get an array of image objects back from WordPress.com based off a prompt.
211	*
212	* @param string $prompt The prompt to generate images for.
213	* @param int $post_id Post ID for which the content is being generated.
214	* @return mixed
215	*/
216	public static function get_dalle_generation( $prompt, $post_id ) {
217	$cache = get_transient( self::transient_name_for_image_generation( $prompt ) );
218	if ( $cache ) {
219	self::mark_post_as_ai_assisted( $post_id );
220	return $cache;
221	}
222
223	if ( ( new Status() )->is_offline_mode() ) {
224	return new WP_Error(
225	'dev_mode',
226	__( 'Jetpack AI is not available in offline mode.', 'jetpack' )
227	);
228	}
229
230	$site_id = Manager::get_site_id();
231	if ( is_wp_error( $site_id ) ) {
232	return $site_id;
233	}
234
235	if ( defined( 'IS_WPCOM' ) && IS_WPCOM ) {
236	if ( ! class_exists( 'OpenAI' ) ) {
237	\require_lib( 'openai' );
238	}
239
240	$result = ( new OpenAI( 'openai', array( 'post_id' => $post_id ) ) )->request_dalle_generation( $prompt );
241	if ( is_wp_error( $result ) ) {
242	return $result;
243	}
244	set_transient( self::transient_name_for_image_generation( $prompt ), $result, self::$image_generation_cache_timeout );
245	self::mark_post_as_ai_assisted( $post_id );
246	return $result;
247	}
248
249	$response = Client::wpcom_json_api_request_as_user(
250	sprintf( '/sites/%d/jetpack-ai/images/generations', $site_id ),
251	2,
252	array(
253	'method' => 'post',
254	'headers' => array( 'content-type' => 'application/json' ),
255	),
256	wp_json_encode(
257	array(
258	'prompt' => $prompt,
259	)
260	),
261	'wpcom'
262	);
263
264	if ( is_wp_error( $response ) ) {
265	return $response;
266	}
267
268	$data = json_decode( wp_remote_retrieve_body( $response ) );
269
270	if ( wp_remote_retrieve_response_code( $response ) >= 400 ) {
271	return new WP_Error( $data->code, $data->message, $data->data );
272	}
273	set_transient( self::transient_name_for_image_generation( $prompt ), $data, self::$image_generation_cache_timeout );
274	self::mark_post_as_ai_assisted( $post_id );
275
276	return $data;
277	}
278	}
279