CopilotEvent.php

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

<?php
 
namespace Bitrix\Im\V2\Analytics\Event;
 
use Bitrix\AI\Engine;
use Bitrix\Im\V2\Chat\CopilotChat;
use Bitrix\Im\V2\Integration\AI\AIHelper;
use Bitrix\Im\V2\Integration\AI\EngineManager;
use Bitrix\Im\V2\Integration\AI\RoleManager;
use Bitrix\Imbot\Bot\CopilotChatBot;
use Bitrix\Main\Result;
 
class CopilotEvent extends Event
{
    protected const ANALYTICS_STATUS = [
        'SUCCESS' => 'success',
        'ERROR_PROVIDER' => 'error_provider',
        'ERROR_B24' => 'error_b24',
        'ERROR_LIMIT_DAILY' => 'error_limit_daily',
        'ERROR_LIMIT_MONTHLY' => 'error_limit_monthly',
        'ERROR_AGREEMENT' => 'error_agreement',
        'ERROR_TURNEDOFF' => 'error_turnedoff',
        'ERROR_LIMIT_BAAS' => 'error_limit_baas',
    ];
 
    protected function getTool(): string
    {
        return 'ai';
    }
 
    protected function getCategory(string $eventName): string
    {
        return 'chat_operations';
    }
 
    protected function setDefaultParams(?Engine $engine = null, ?string $promptCode = null): self
    {
        $this
            ->setSection('copilot_tab')
            ->setCopilotP2()
            ->setCopilotP3()
            ->setCopilotP4()
            ->setCopilotP5()
        ;
 
        return $this;
    }
 
    public function setCopilotStatus(Result $result): self
    {
        if ($result->isSuccess())
        {
            $this->status = self::ANALYTICS_STATUS['SUCCESS'];
 
            return $this;
        }
 
        $error = $result->getErrors()[0];
 
        if (!isset($error))
        {
            $this->status = self::ANALYTICS_STATUS['ERROR_B24'];
 
            return $this;
        }
 
        $this->status = match ($error->getCode()) {
            CopilotChatBot::AI_ENGINE_ERROR_PROVIDER => self::ANALYTICS_STATUS['ERROR_PROVIDER'],
            CopilotChatBot::LIMIT_IS_EXCEEDED_DAILY => self::ANALYTICS_STATUS['ERROR_LIMIT_DAILY'],
            CopilotChatBot::LIMIT_IS_EXCEEDED_MONTHLY => self::ANALYTICS_STATUS['ERROR_LIMIT_MONTHLY'],
            CopilotChatBot::ERROR_AGREEMENT => self::ANALYTICS_STATUS['ERROR_AGREEMENT'],
            CopilotChatBot::LIMIT_IS_EXCEEDED_BAAS => self::ANALYTICS_STATUS['ERROR_LIMIT_BAAS'],
            default => self::ANALYTICS_STATUS['ERROR_B24'],
        };
 
        return $this;
    }
 
    public function setCopilotP1(?string $promptCode): self
    {
        $this->p1 = isset($promptCode) ? ('1st-type_' . self::convertUnderscore($promptCode)) : 'none';
 
        return $this;
    }
 
    protected function setCopilotP2(): self
    {
        $engineName = null;
 
        if ($this->chat instanceof CopilotChat)
        {
            $engineCode = $this->chat->getEngineCode() ?? EngineManager::getDefaultEngineCode();
            $engineName = (new EngineManager())->getEngineNameByCode($engineCode);
        }
 
        $engineName ??= 'none';
        $this->p2 = 'provider_' . $engineName;
 
        return $this;
    }
 
    protected function setCopilotP3(): self
    {
        $this->p3 = $this->chat->getUserCount() > 2 ? 'chatType_multiuser' : 'chatType_private';
 
        return $this;
    }
 
    protected function setCopilotP4(): self
    {
        $role = (new RoleManager())->getMainRole($this->chat->getChatId()) ?? '';
        $this->p4 = 'role_' . self::convertUnderscore($role);
 
        return $this;
    }
 
    protected function setCopilotP5(): self
    {
        $this->p5 = 'chatId_' . $this->chat->getChatId();
 
        return $this;
    }
}