<?php
/**
 * Trait ATFPP\AI_Translate\Services\Traits\With_API_Client_Trait
 *
 * @since 0.7.0
 * @package ai-services
 */

namespace ATFPP\AI_Translate\Services\Traits;

use ATFPP\AI_Translate\Services\Contracts\Generative_AI_API_Client;
use RuntimeException;

/**
 * Trait for a service or model which implements the With_API_Client interface.
 *
 * @since 0.7.0
 */
trait With_API_Client_Trait {

	/**
	 * The AI API client instance.
	 *
	 * @since 0.7.0
	 * @var Generative_AI_API_Client
	 */
	private $api_client;

	/**
	 * Gets the API client instance.
	 *
	 * @since 0.7.0
	 *
	 * @return Generative_AI_API_Client The API client instance.
	 *
	 * @throws RuntimeException Thrown if the API client is not set.
	 */
	final public function get_api_client(): Generative_AI_API_Client {
		if ( ! $this->api_client instanceof Generative_AI_API_Client ) {
			throw new RuntimeException( 'API client must be set in the constructor.' );
		}

		return $this->api_client;
	}

	/**
	 * Sets the API client instance.
	 *
	 * @since 0.7.0
	 *
	 * @param Generative_AI_API_Client $api_client The API client instance.
	 */
	final protected function set_api_client( Generative_AI_API_Client $api_client ): void {
		$this->api_client = $api_client;
	}
}
