<?php

/**
 * Class AI_WRITEX_GIMINI_AI_Tool
 *
 * Handles the interaction with the GIMINI AI API.
 */
class AI_WRITEX_GIMINI_AI_Tool
{
    private $api_key;
    private $model;

    public function __construct(string $api_key, string $model)
    {
        $this->api_key = $api_key;
        $this->model = $model;
    }

    public function get_api_url(): string
    {
        return "https://generativelanguage.googleapis.com/v1beta/models/{$this->model}:generateContent?key={$this->api_key}";
    }

    public function get_headers(): array
    {
        return [
            'Content-Type' => 'application/json',
        ];
    }

    public function prepare_payload(string $prompt): array
    {
        return [
            'contents' => [
                [
                    'parts' => [
                        ['text' => $prompt]
                    ]
                ]
            ]
        ];
    }
}