Consistently handle AI temperature parameter between services, expecting a value between 0.0 and 1.0.

felixarntz · felixarntz · commit e7ae611b34ed · 2024-11-16T09:55:32.000-08:00
diff --git a/includes/Google/Google_AI_Model.php b/includes/Google/Google_AI_Model.php
@@ -505,7 +505,8 @@ private static function get_generation_config_transformers(): array {
 				return $config->get_max_output_tokens();
 			},
 			'temperature'      => static function ( Generation_Config $config ) {
-				return $config->get_temperature();
+				// In the Google AI API temperature ranges from 0.0 to 2.0.
+				return $config->get_temperature() * 2.0;
 			},
 			'topP'             => static function ( Generation_Config $config ) {
 				return $config->get_top_p();
diff --git a/includes/Services/API/Types/Generation_Config.php b/includes/Services/API/Types/Generation_Config.php
@@ -133,7 +133,7 @@ public function get_max_output_tokens(): int {
 	 *
 	 * @since 0.2.0
 	 *
-	 * @return float The temperature (between 0.0 and 2.0), or 0.0 if not set.
+	 * @return float The temperature (between 0.0 and 1.0), or 0.0 if not set.
 	 */
 	public function get_temperature(): float {
 		return $this->sanitized_args['temperature'] ?? 0.0;
@@ -282,7 +282,7 @@ public static function get_json_schema(): array {
 					'description' => __( 'Floating point value to control the randomness of the output.', 'ai-services' ),
 					'type'        => 'number',
 					'minimum'     => 0.0,
-					'maximum'     => 2.0,
+					'maximum'     => 1.0,
 				),
 				'topP'             => array(
 					'description' => __( 'The maximum cumulative probability of tokens to consider when sampling.', 'ai-services' ),
@@ -366,8 +366,8 @@ private function sanitize_args( array $args ): array {
 	 * @throws InvalidArgumentException Thrown if the type is not supported.
 	 */
 	private function sanitize_arg( $value, string $type, string $arg_name ) {
-		if ( 'temperature' === $arg_name && ( (float) $value < 0.0 || (float) $value > 2.0 ) ) {
-			throw new InvalidArgumentException( 'Temperature must be between 0.0 and 2.0.' );
+		if ( 'temperature' === $arg_name && ( (float) $value < 0.0 || (float) $value > 1.0 ) ) {
+			throw new InvalidArgumentException( 'Temperature must be between 0.0 and 1.0.' );
 		}
 
 		switch ( $type ) {
