Merge pull request #545 from 10up/feature/modify-title-prompt

Change the prompt we send for generating titles to use a `system` message
10up · Jul 31, 2023 · 9af6fea · 9af6fea
2 parents e667db4 + 70c8737
commit 9af6fea
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/includes/Classifai/Providers/OpenAI/ChatGPT.php b/includes/Classifai/Providers/OpenAI/ChatGPT.php
@@ -654,7 +654,7 @@ public function generate_titles( int $post_id = 0, array $args = [] ) {
 		 *
 		 * @return {string} Prompt.
 		 */
-		$prompt = apply_filters( 'classifai_chatgpt_title_prompt', 'Write an SEO-friendly title for the following content that will encourage readers to clickthrough, staying within a range of 40 to 60 characters', $post_id, $args );
+		$prompt = apply_filters( 'classifai_chatgpt_title_prompt', 'Write an SEO-friendly title for the following content that will encourage readers to clickthrough, staying within a range of 40 to 60 characters.', $post_id, $args );
 
 		/**
 		 * Filter the request body before sending to ChatGPT.
@@ -672,9 +672,13 @@ public function generate_titles( int $post_id = 0, array $args = [] ) {
 			[
 				'model'       => $this->chatgpt_model,
 				'messages'    => [
+					[
+						'role'    => 'system',
+						'content' => $prompt,
+					],
 					[
 						'role'    => 'user',
-						'content' => esc_html( $prompt ) . ': ' . $this->get_content( $post_id, absint( $args['num'] ) * 15, false, $args['content'] ) . '',
+						'content' => $this->get_content( $post_id, absint( $args['num'] ) * 15, false, $args['content'] ) . '',
 					],
 				],
 				'temperature' => 0.9,