Add --service arg

Author: swissspidy

src/AI/AiClient.php

@@ -35,6 +35,7 @@ class AiClient {
 	public function __construct(
 		private readonly array $tools,
 		private readonly bool $approval_mode,
+		private readonly ?string $service,
 		private readonly ?string $model
 	) {}
 
@@ -106,6 +107,7 @@ static function () {
 		try {
 			$service = ai_services()->get_available_service(
 				[
+					'slugs'        => $this->service ? [ $this->service ] : null,
 					'capabilities' => [
 						AI_Capability::MULTIMODAL_INPUT,
 						AI_Capability::TEXT_GENERATION,

src/AiCommand.php

@@ -39,6 +39,11 @@ class AiCommand extends WP_CLI_Command {
 	 * [--approval-mode]
 	 * : Approve tool usage before running.
 	 *
+	 * [--service]
+	 * : Manually specify the AI service to use.
+	 * Depends on the available AI services.
+	 * Examples: 'google', 'anthropic', 'openai'.
+	 *
 	 * [--model]
 	 * : Manually specify the LLM model that should be used.
 	 * Depends on the available AI services.
@@ -79,9 +84,10 @@ public function __invoke( array $args, array $assoc_args ): void {
 		$tools    = $this->get_tools( $sessions );
 
 		$approval_mode = (bool) Utils\get_flag_value( $assoc_args, 'approval-mode', false );
+		$service       = Utils\get_flag_value( $assoc_args, 'service' );
 		$model         = Utils\get_flag_value( $assoc_args, 'model' );
 
-		$ai_client = new AiClient( $tools, $approval_mode, $model );
+		$ai_client = new AiClient( $tools, $approval_mode, $service, $model );
 
 		$ai_client->call_ai_service_with_prompt( $args[0] );
 	}