ai.php

<?php
namespace Bitrix\Landing\Connector;
 
use Bitrix\AI\Context;
use Bitrix\AI\Engine;
use Bitrix\AI\Engine\IEngine;
use Bitrix\AI\Tuning;
use Bitrix\AI\Tuning\Type;
use Bitrix\Main\Event;
use Bitrix\Main\Entity;
use Bitrix\Main\EventResult;
use Bitrix\Main\Loader;
use Bitrix\Main\Localization\Loc;
 
class Ai
{
    private const TUNING_CODE_IMAGE = 'landing_allow_image_generate';
    private const TUNING_CODE_TEXT = 'landing_allow_text_generate';
 
    public static function isImageAvailable(): bool
    {
        if (!Loader::includeModule('ai'))
        {
            return false;
        }
 
        $engine = Engine::getByCategory('image', Context::getFake());
        if (!$engine)
        {
            return false;
        }
 
        return true;
    }
 
    public static function isImageActive(): bool
    {
        if (!self::isImageAvailable())
        {
            return false;
        }
 
        $default = false;
        $setting = (new Tuning\Manager())->getItem(self::TUNING_CODE_IMAGE);
 
        return $setting ? (bool)$setting->getValue() : $default;
    }
 
    public static function isTextAvailable(): bool
    {
        if (!Loader::includeModule('ai'))
        {
            return false;
        }
 
        $engine = Engine::getByCategory('text', Context::getFake());
        if (!$engine)
        {
            return false;
        }
 
        return true;
    }
 
    public static function isCopilotAvailable(): bool
    {
        if (!self::isTextAvailable())
        {
            return false;
        }
 
        if (\Bitrix\Main\Config\Option::get('landing', 'enable_copilot', 'N') === 'Y')
        {
            return true;
        }
 
        return false;
    }
 
    public static function isTextActive(): bool
    {
        if (!self::isTextAvailable())
        {
            return false;
        }
 
        $default = false;
        $setting = (new Tuning\Manager())->getItem(self::TUNING_CODE_TEXT);
 
        return $setting ? (bool)$setting->getValue() : $default;
    }
 
    public static function onTuningLoad(): Entity\EventResult
    {
        $result = new Entity\EventResult;
        $items = [];
        $groups = [];
 
        if (Engine::getByCategory('image', Context::getFake()))
        {
            $items[self::TUNING_CODE_IMAGE] = [
                'group' => Tuning\Defaults::GROUP_IMAGE,
                'header' => Loc::getMessage('LANDING_CONNECTOR_AI_ALLOW_IMAGE_COPILOT_DESC'),
                'title' => Loc::getMessage('LANDING_CONNECTOR_AI_ALLOW_COPILOT_TITLE'),
                'type' => Type::BOOLEAN,
                'default' => true,
                'sort' => 300,
            ];
        }
 
        if (Engine::getByCategory('text', Context::getFake()))
        {
            $items[self::TUNING_CODE_TEXT] = [
                'group' => Tuning\Defaults::GROUP_TEXT,
                'header' => Loc::getMessage('LANDING_CONNECTOR_AI_ALLOW_TEXT_COPILOT_DESC'),
                'title' => Loc::getMessage('LANDING_CONNECTOR_AI_ALLOW_COPILOT_TITLE'),
                'type' => Type::BOOLEAN,
                'default' => true,
                'sort' => 300,
            ];
        }
 
        $result->modifyFields([
            'items' => $items,
            'groups' => $groups,
        ]);
 
        return $result;
    }
 
    public static function onBeforeCompletions(Event $event): EventResult
    {
        $engine = $event->getParameter('engine');
        $category = $engine->getCategory();
        $module = $engine->getContext()->getModuleId();
 
        $config = new Tuning\Manager();
        $configItem = $config->getItem("{$module}_allow_{$category}_generate");
 
        if ($configItem && $configItem->getValue())
        {
            return new EventResult(EventResult::SUCCESS);
        }
        else
        {
            return new EventResult(EventResult::ERROR);
        }
    }
}