Merge pull request #811 from 10up/feature/openai-moderation-model

Add the new OpenAI Moderation model

includes/Classifai/Providers/OpenAI/Moderation.php

@@ -22,6 +22,13 @@ class Moderation extends Provider {
 	 */
 	protected $moderation_url = 'https://api.openai.com/v1/moderations';
 
+	/**
+	 * OpenAI Moderation model
+	 *
+	 * @var string
+	 */
+	protected $model = 'omni-moderation-latest';
+
 	/**
 	 * OpenAI Moderation constructor.
 	 *
@@ -31,6 +38,28 @@ public function __construct( $feature_instance = null ) {
 		$this->feature_instance = $feature_instance;
 	}
 
+	/**
+	 * Get the model name.
+	 *
+	 * @return string
+	 */
+	public function get_model(): string {
+		/**
+		 * Filter the model name.
+		 *
+		 * Useful if you want to use a different model, like
+		 * text-moderation-latest.
+		 *
+		 * @since x.x.x
+		 * @hook classifai_openai_moderation_model
+		 *
+		 * @param {string} $model The default model to use.
+		 *
+		 * @return {string} The model to use.
+		 */
+		return apply_filters( 'classifai_openai_moderation_model', $this->model );
+	}
+
 	/**
 	 * Register what we need for the provider.
 	 */
@@ -191,6 +220,7 @@ public function moderate_comment( int $comment_id = 0 ) {
 			'classifai_openai_moderation_request_body',
 			[
 				'input' => $comment->comment_content,
+				'model' => $this->get_model(),
 			],
 			$comment_id
 		);