una/studio/template/scripts/BxBaseStudioAgentsAssistants.php

<?php defined('BX_DOL') or die('hack attempt');
/**
 * Copyright (c) UNA, Inc - https://una.io
 * MIT License - https://opensource.org/licenses/MIT
 *
 * @defgroup    UnaView UNA Studio Representation classes
 * @ingroup     UnaStudio
 * @{
 */

class BxBaseStudioAgentsAssistants extends BxDolStudioAgentsAssistants
{
    protected $_sUrlPage;
    protected $_sFieldName;

    public function __construct ($aOptions, $oTemplate = false)
    {
        parent::__construct ($aOptions, $oTemplate);

        $this->_sUrlPage = BX_DOL_URL_STUDIO . 'agents.php?page=assistants';

        $this->_sFieldName = 'name';
    }

    public function getPageJsObject()
    {
        return 'oBxDolStudioPageAgents';
    }

    public function performActionAdd()
    {
        $sAction = 'add';

        $oAI = BxDolAI::getInstance();

        $aForm = $this->_getForm($sAction);
        $oForm = new BxTemplFormView($aForm);
        $oForm->initChecker();

        if($oForm->isSubmittedAndValid()) {
            $aValsToAdd = ['added' => time()];

            $sName = $oForm->getCleanValue($this->_sFieldName);
            $sName = BxDolAIAssistant::getName($sName);
            BxDolForm::setSubmittedValue($this->_sFieldName, $sName, $oForm->aFormAttrs['method']);

            $iProfileId = $oForm->getCleanValue('profile_id');
            if(empty($iProfileId)) {
                $iProfileId = (int)getParam('sys_agents_profile');
                if(empty($iProfileId))
                    $iProfileId = current(bx_srv('system', 'get_options_agents_profile', [false], 'TemplServices'))['key'];

                $aValsToAdd['profile_id'] = $iProfileId;
            }

            $oAIModel = $oAI->getModelObject($oForm->getCleanValue('model_id'));
            if(($aAssistant = $oAIModel->getAssistant(['name' => $sName, 'prompt' => $oForm->getCleanValue('prompt')])) !== false)
                $aValsToAdd = array_merge($aValsToAdd, [
                    'ai_vs_id' => $aAssistant['vector_store_id'],
                    'ai_asst_id' => $aAssistant['assistant_id']
                ]);

            if(($iId = $oForm->insert($aValsToAdd)) !== false)
                $aRes = ['grid' => $this->getCode(false), 'blink' => $iId];
            else
                $aRes = ['msg' => _t('_sys_txt_error_occured')];

            return echoJson($aRes);
        }

        $sFormId = $oForm->getId();
        $sContent = BxTemplStudioFunctions::getInstance()->popupBox($sFormId . '_popup', _t('_sys_agents_assistants_popup_add'), $this->_oTemplate->parseHtmlByName('agents_automator_form.html', [
            'form_id' => $sFormId,
            'form' => $oForm->getCode(true),
            'object' => $this->_sObject,
            'action' => $sAction
        ]));

        return echoJson(['popup' => ['html' => $sContent, 'options' => ['closeOnOuterClick' => false]]]);
    }

    public function performActionEdit()
    {
        $sAction = 'edit';

        $iId = $this->_getId();
        $aAssistant = $this->_oDb->getAssistantsBy(['sample' => 'id', 'id' => $iId]);

        $aForm = $this->_getFormEdit($sAction, $aAssistant);
        $oForm = new BxTemplFormView($aForm);
        $oForm->initChecker();

        if($oForm->isSubmittedAndValid()) {
            $aValsToAdd = [];

            $sName = $oForm->getCleanValue($this->_sFieldName);
            if($aAssistant[$this->_sFieldName] != $sName) {
                $sName = BxDolAIAssistant::getName($sName);
                BxDolForm::setSubmittedValue($this->_sFieldName, $sName, $oForm->aFormAttrs['method']);
            }

            $iProfileId = $oForm->getCleanValue('profile_id');
            if(empty($iProfileId)) {
                $iProfileId = (int)getParam('sys_agents_profile');
                if(empty($iProfileId))
                    $iProfileId = current(bx_srv('system', 'get_options_agents_profile', [false], 'TemplServices'))['key'];

                $aValsToAdd['profile_id'] = $iProfileId;
            }

            $aRes = [];
            $aResError = ['msg' => _t('_sys_txt_error_occured')];

            $oAIModel = BxDolAI::getInstance()->getModelObject($aAssistant['model_id']);
            if(($sKey = 'ai_asst_id') && !empty($aAssistant[$sKey]) && !$oAIModel->editAssistant($aAssistant[$sKey], ['name' => $sName, 'prompt' => $oForm->getCleanValue('prompt')]))
                return echoJson($aResError);

            if($oForm->update($iId, $aValsToAdd) !== false)
                $aRes = ['grid' => $this->getCode(false), 'blink' => $iId];
            else
                $aRes = $aResError;

            return echoJson($aRes);
        } 

        $sFormId = $oForm->getId();
        $sContent = BxTemplStudioFunctions::getInstance()->popupBox($sFormId . '_popup', _t('_sys_agents_assistants_popup_edit'), $this->_oTemplate->parseHtmlByName('agents_automator_form.html', [
            'form_id' => $sFormId,
            'form' => $oForm->getCode(true),
            'object' => $this->_sObject,
            'action' => $sAction
        ]));

        return echoJson(['popup' => ['html' => $sContent, 'options' => ['closeOnOuterClick' => false]]]);
    }

    public function performActionCodes()
    {
        $sAction = 'codes';

        $iId = $this->_getId();

        $sContent = BxTemplStudioFunctions::getInstance()->transBox('bx_std_agents_assistants_codes', $this->_oTemplate->parseHtmlByName('agents_assistant_codes.html', [
            'code_onclick' => '{onclick}',
            'code_button' => htmlspecialchars($this->_oTemplate->parseLink('javascript:void(0)', _t('_sys_agents_assistants_txt_add_knowledge'), [
                'class' => 'bx-btn', 
                'onclick' => '{onclick}'
            ]), ENT_NOQUOTES)
        ]));        
        $sContent = bx_replace_markers($sContent, ['onclick' => "javascript:bx_agents_action(this, 'asst', 'add_knowledge', {id: " . $iId . "})"]);
                

        return echoJson(['popup' => ['html' => $sContent, 'options' => []]]);
    }

    protected function _getActionChats($sType, $sKey, $a, $isSmall = false, $isDisabled = false, $aRow = [])
    {
    	$a['attr'] = array_merge($a['attr'], [
            "onclick" => "window.open('" . $this->_sUrlPage . '&spage=chats&aid=' . $aRow['id'] . "', '_self');"
    	]);

    	return $this->_getActionDefault ($sType, $sKey, $a, $isSmall, $isDisabled, $aRow);
    }

    protected function _getActionFiles($sType, $sKey, $a, $isSmall = false, $isDisabled = false, $aRow = [])
    {
    	$a['attr'] = array_merge($a['attr'], [
            "onclick" => "window.open('" . $this->_sUrlPage . '&spage=files&aid=' . $aRow['id'] . "', '_self');"
    	]);

    	return $this->_getActionDefault ($sType, $sKey, $a, $isSmall, $isDisabled, $aRow);
    }

    protected function _delete ($mixedId)
    {
        $iAssistantId = (int)$mixedId;
        $aAssistant = $this->_oDb->getAssistantsBy(['sample' => 'id', 'id' => $iAssistantId]);

        $mixedResult = parent::_delete($mixedId);
        if($mixedResult) {
            $oAi = BxDolAI::getInstance();
            $oAIModel = $oAi->getModelObject($aAssistant['model_id']);

            $aChats = $this->_oDb->getChatsBy(['sample' => 'assistant_id', 'assistant_id' => $iAssistantId]);
            foreach($aChats as $aChat) {
                if(($oCmts = $oAi->getAssistantChatCmtsObject($aChat['id'])) !== false)
                    $oCmts->onObjectDelete();

                if(!empty($aChat['ai_file_id'])) {
                    $oAIModel->callVectorStoresFilesDelete($aAssistant['ai_vs_id'], $aChat['ai_file_id']);
                    $oAIModel->callFilesDelete($aChat['ai_file_id']);
                }
            }

            $this->_oDb->deleteChats(['assistant_id' => $iAssistantId]);

            $this->_oDb->deleteAutomatorAssistants(['assistant_id' => $iAssistantId]);

            $oAIModel->callVectorStoresDelete($aAssistant['ai_vs_id']);
            $oAIModel->callAssistantsDelete($aAssistant['ai_asst_id']);
        }

        return $mixedResult;
    }

    protected function _getFormEdit($sAction, $aAssistant = [])
    {
        $aForm = $this->_getForm($sAction, $aAssistant);
        $aForm['form_attrs']['action'] .= '&id=' . $aAssistant['id'];

        if(!isset($aForm['inputs']['model_id']['attrs']))
            $aForm['inputs']['model_id']['attrs'] = [];

        $aForm['inputs']['model_id']['attrs'] = array_merge($aForm['inputs']['model_id']['attrs'], [
            'disabled' => 'disabled'
        ]);
        unset($aForm['inputs']['model_id']['db']);

        return $aForm;
    }

    protected function _getForm($sAction, $aAssistant = [])
    {
        $sJsObject = $this->getPageJsObject();

        $aForm = [
            'form_attrs' => [
                'id' => 'bx_std_agents_assistants_' . $sAction,
                'action' => BX_DOL_URL_ROOT . bx_append_url_params('grid.php', ['o' => 'sys_studio_agents_assistants', 'a' => $sAction]),
                'method' => 'post',
            ],
            'params' => [
                'db' => [
                    'table' => 'sys_agents_assistants',
                    'key' => 'id',
                    'submit_name' => 'do_submit',
                ],
            ],
            'inputs' => [
                'name' => [
                    'type' => 'text',
                    'name' => 'name',
                    'required' => '1',
                    'caption' => _t('_sys_agents_assistants_field_name'),
                    'value' => isset($aAssistant['name']) ? $aAssistant['name'] : '',
                    'required' => '1',
                    'checker' => [
                        'func' => 'Avail',
                        'params' => [],
                        'error' => _t('_sys_agents_form_field_err_enter'),
                    ],
                    'db' => [
                        'pass' => 'Xss',
                    ]
                ],
                'model_id' => [
                    'type' => 'select',
                    'name' => 'model_id',
                    'caption' => _t('_sys_agents_assistants_field_model_id'),
                    'info' => '',
                    'value' => isset($aAssistant['model_id']) ? $aAssistant['model_id'] : BxDolAI::getDefaultModel(),
                    'values' => $this->_oDb->getModelsBy(['sample' => 'all_pairs', 'for_asst' => 1, 'active' => 1, 'hidden' => 0]),
                    'required' => '1',
                    'db' => [
                        'pass' => 'Int',
                    ]
                ],
                'profile_id' => [
                    'type' => 'select',
                    'name' => 'profile_id',
                    'caption' => _t('_sys_agents_assistants_field_profile_id'),
                    'info' => _t('_sys_agents_assistants_field_profile_id_inf'),
                    'value' => isset($aAssistant['profile_id']) ? $aAssistant['profile_id'] : 0,
                    'values' => bx_srv('system', 'get_options_agents_profile', [], 'TemplServices'),
                    'required' => '0',
                    'db' => [
                        'pass' => 'Int',
                    ]
                ],
                'description' => [
                    'type' => 'textarea',
                    'name' => 'description',
                    'caption' => _t('_sys_agents_assistants_field_description'),
                    'value' => isset($aAssistant['description']) ? $aAssistant['description'] : '',
                    'required' => '1',
                    'checker' => [
                        'func' => 'Avail',
                        'params' => [],
                        'error' => _t('_sys_agents_form_field_err_enter'),
                    ],
                    'db' => [
                        'pass' => 'Xss',
                    ]
                ],
                'prompt' => [
                    'type' => 'textarea',
                    'name' => 'prompt',
                    'caption' => _t('_sys_agents_assistants_field_prompt'),
                    'value' => isset($aAssistant['prompt']) ? $aAssistant['prompt'] : '',
                    'required' => '1',
                    'checker' => [
                        'func' => 'Avail',
                        'params' => [],
                        'error' => _t('_sys_agents_assistants_field_prompt_err'),
                    ],
                    'db' => [
                        'pass' => 'Xss',
                    ]
                ],
                'submit' => $this->_getFormControls(),
            ],
        ];

        return $aForm;
    }
}

/** @} */