RequestSiteData.php

См. документацию.

<?php
declare(strict_types=1);
 
namespace Bitrix\Landing\Copilot\Generation\Step;
 
use Bitrix\Landing\Copilot\Connector\AI;
use Bitrix\Landing\Copilot\Connector\AI\Prompt;
use Bitrix\Landing\Copilot\Data\Site;
use Bitrix\Landing\Copilot\Generation\Error;
use Bitrix\Landing\Copilot\Generation\GenerationException;
use Bitrix\Landing\Copilot\Generation\Markers;
use Bitrix\Landing\Copilot\Generation\Type\Errors;
use Bitrix\Landing\Copilot\Generation\Type\GenerationErrors;
use Bitrix\Landing\Copilot\Generation\Type\RequestQuotaDto;
 
class RequestSiteData extends RequestSingle
{
    public function __construct()
    {
        parent::__construct();
        if (class_exists(self::getConnectorClass()))
        {
            $this->connector = new (self::getConnectorClass())();
        }
    }

    public static function getConnectorClass(): string
    {
        return AI\Text::class;
    }

    public static function getRequestQuota(Site $siteData): ?RequestQuotaDto
    {
        return new RequestQuotaDto(self::getConnectorClass(), 1);
    }
 
    protected function getPrompt(): Prompt
    {
        $prompt = new Prompt('landing_ai_data');
        $prompt->setMarkers(Markers::getSiteDataPromptMarkers($this->siteData));
 
        return $prompt;
    }
 
    protected function applyResponse(): bool
    {
        $result = $this->request->getResult();
 
        if ($result)
        {
            Site::initSite($this->siteData, $result);
        }
 
        return true;
    }

    public function verifyResponse(): void
    {
        $result = $this->request->getResult();
 
        if (!$result)
        {
            throw new GenerationException(GenerationErrors::notExistResponse);
        }
 
        if (
            !isset($result['siteData']['isAllowedRequest'])
            || $result['siteData']['isAllowedRequest'] !== "yes"
        )
        {
            $this->request->saveError(Error::createError(Errors::requestNotAllowed));
 
            throw new GenerationException(GenerationErrors::restrictedRequest);
        }
 
        if (
            !isset($result['siteData']['colors'])
            && !isset($result['siteData']['titles'])
        )
        {
            throw new GenerationException(
                GenerationErrors::notFullyResponse,
                "Missing required 'colors' or 'titles' in 'siteData'.",
            );
        }
    }
}