RequestSingle.php

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

<?php
declare(strict_types=1);
 
namespace Bitrix\Landing\Copilot\Generation\Step;
 
use Bitrix\Landing\Copilot\Connector\AI\Prompt;
use Bitrix\Landing\Copilot\Data\Site;
use Bitrix\Landing\Copilot\Generation\GenerationException;
use Bitrix\Landing\Copilot\Generation\Request;
use Bitrix\Landing\Copilot\Generation\Type\GenerationErrors;
use Bitrix\Landing\Copilot\Generation\Type\RequestQuotaDto;
 
abstract class RequestSingle extends AIStep
{
    protected ?Request $request = null;

    public function initialize(): void
    {
        $existsRequests = Request::getByGeneration($this->generation->getId(), $this->stepId);
        if (!empty($existsRequests))
        {
            $this->request = array_shift($existsRequests);
        }
    }

    public function execute(): bool
    {
        if (!parent::execute())
        {
            return false;
        }
 
        if (!isset($this->siteData, $this->stepId))
        {
            return false;
        }
 
        if (!isset($this->request))
        {
            $this->request = new Request($this->generation->getId(), $this->stepId);
            $this->request->send($this->getPrompt(), $this->connector);
        }
        elseif ($this->request->getError())
        {
            throw new GenerationException(
                GenerationErrors::notCorrectResponse,
                $this->request->getError()->getMessage(),
            );
        }
 
        if ($this->request->isReceived())
        {
            $this->verifyResponse();
            if ($this->applyResponse())
            {
                $this->request->setApplied();
            }
            $this->changed = true;
        }
 
        return true;
    }

    public function isStarted(): bool
    {
        return isset($this->request);
    }

    public function isFinished(): bool
    {
        return
            isset($this->request)
            && $this->request->isApplied();
    }

    public function clearErrors(): void
    {
        if (
            $this->request
            && $this->request->getError()
        )
        {
            $this->request->setDeleted();
            $this->request = null;
        }
    }

    abstract protected function getPrompt(): Prompt;

    abstract protected function applyResponse(): bool;

    abstract public function verifyResponse(): void;

    abstract public static function getRequestQuota(Site $siteData): ?RequestQuotaDto;
}