BxBaseStudioAgents.php 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378
  1. <?php defined('BX_DOL') or die('hack attempt');
  2. /**
  3. * Copyright (c) UNA, Inc - https://una.io
  4. * MIT License - https://opensource.org/licenses/MIT
  5. *
  6. * @defgroup UnaView UNA Studio Representation classes
  7. * @ingroup UnaStudio
  8. * @{
  9. */
  10. class BxBaseStudioAgents extends BxDolStudioAgents
  11. {
  12. protected $sSubpageUrl;
  13. protected $aPageJsOptions;
  14. protected $aMenuItems;
  15. protected $aGridObjects;
  16. public function __construct($sPage = '')
  17. {
  18. parent::__construct($sPage);
  19. $this->sSubpageUrl = BX_DOL_URL_STUDIO . 'agents.php?page=';
  20. $this->aPageJs = array_merge($this->aPageJs, ['agents.js']);
  21. $this->aPageCss = array_merge($this->aPageCss, ['cmts.css', 'agents.css']);
  22. $this->sPageJsClass = 'BxDolStudioPageAgents';
  23. $this->sPageJsObject = 'oBxDolStudioPageAgents';
  24. $this->aPageJsOptions = [
  25. 'sActionUrl' => BX_DOL_URL_STUDIO . 'agents.php',
  26. 'sPageUrl' => $this->sSubpageUrl
  27. ];
  28. $this->aMenuItems = [
  29. BX_DOL_STUDIO_AGENTS_TYPE_SETTINGS => ['icon' => 'mi-agt-settings.svg'],
  30. BX_DOL_STUDIO_AGENTS_TYPE_AI_PROVIDERS => ['icon' => 'mi-agt-providers.svg'],
  31. BX_DOL_STUDIO_AGENTS_TYPE_TOOLS => ['icon' => 'mi-agt-helpers.svg'],
  32. BX_DOL_STUDIO_AGENTS_TYPE_VECTOR_STORE => ['icon' => 'mi-agt-helpers.svg'],
  33. BX_DOL_STUDIO_AGENTS_TYPE_AGENTS => ['icon' => 'mi-agt-assistants.svg'],
  34. /*
  35. * Hidden for now. Most probably they will be removed.
  36. *
  37. BX_DOL_STUDIO_AGENTS_TYPE_PROVIDERS => ['icon' => 'mi-agt-providers.svg'],
  38. BX_DOL_STUDIO_AGENTS_TYPE_HELPERS => ['icon' => 'mi-agt-helpers.svg'],
  39. BX_DOL_STUDIO_AGENTS_TYPE_AUTOMATORS => ['icon' => 'mi-agt-automators.svg'],
  40. */
  41. ];
  42. $this->aGridObjects = [
  43. BX_DOL_STUDIO_AGENTS_TYPE_AI_PROVIDERS => 'sys_studio_agents_models',
  44. BX_DOL_STUDIO_AGENTS_TYPE_VECTOR_STORE => 'sys_studio_agents_vector_store',
  45. BX_DOL_STUDIO_AGENTS_TYPE_TOOLS => 'sys_studio_agents_tools',
  46. BX_DOL_STUDIO_AGENTS_TYPE_ASSISTANTS => 'sys_studio_agents_assistants',
  47. BX_DOL_STUDIO_AGENTS_TYPE_ASSISTANTS . '_chats' => 'sys_studio_agents_assistants_chats',
  48. BX_DOL_STUDIO_AGENTS_TYPE_ASSISTANTS . '_files' => 'sys_studio_agents_assistants_files',
  49. BX_DOL_STUDIO_AGENTS_TYPE_AGENTS => 'sys_studio_agents_agents',
  50. /*
  51. * Hidden for now. Most probably they will be removed.
  52. *
  53. BX_DOL_STUDIO_AGENTS_TYPE_AUTOMATORS => 'sys_studio_agents_automators',
  54. BX_DOL_STUDIO_AGENTS_TYPE_PROVIDERS => 'sys_studio_agents_providers',
  55. BX_DOL_STUDIO_AGENTS_TYPE_HELPERS => 'sys_studio_agents_helpers',
  56. *
  57. */
  58. ];
  59. }
  60. public function getPageJsCode($aOptions = [], $bWrap = true)
  61. {
  62. return parent::getPageJsCode(array_merge($aOptions, $this->aPageJsOptions), $bWrap);
  63. }
  64. public function getPageCaption()
  65. {
  66. return parent::getPageCaption() . $this->getPageJsCode();
  67. }
  68. public function getPageMenu($aMenu = [], $aMarkers = [])
  69. {
  70. $sJsObject = $this->getPageJsObject();
  71. $aMenu = [];
  72. foreach($this->aMenuItems as $sMenuItem => $aItem)
  73. $aMenu[] = [
  74. 'name' => $sMenuItem,
  75. 'icon' => $aItem['icon'],
  76. 'icon_bg' => true,
  77. 'link' => $this->sSubpageUrl . $sMenuItem,
  78. 'title' => _t('_adm_lmi_cpt_' . $sMenuItem),
  79. 'selected' => $sMenuItem == $this->sPage
  80. ];
  81. return parent::getPageMenu($aMenu);
  82. }
  83. protected function getSettings()
  84. {
  85. $oOptions = new BxTemplStudioOptions(BX_DOL_STUDIO_STG_TYPE_DEFAULT, [
  86. 'agents_general',
  87. 'agents_usage',
  88. ]);
  89. $this->aPageCss = array_merge($this->aPageCss, $oOptions->getCss());
  90. $this->aPageJs = array_merge($this->aPageJs, $oOptions->getJs());
  91. return $oOptions->getCode();
  92. }
  93. protected function getAiProviders()
  94. {
  95. $this->aPageJsOptions = array_merge($this->aPageJsOptions, [
  96. 'sPageUrl' => $this->sSubpageUrl . 'providers',
  97. 'sActionUrlGrid' => bx_append_url_params(BX_DOL_URL_ROOT . 'grid.php', [
  98. 'o' => 'sys_studio_agents_providers'
  99. ])
  100. ]);
  101. return $this->getGrid($this->aGridObjects[BX_DOL_STUDIO_AGENTS_TYPE_AI_PROVIDERS]);
  102. }
  103. protected function getTools()
  104. {
  105. $this->aPageJsOptions['sPageUrl'] .= 'tools';
  106. return $this->getGrid($this->aGridObjects[BX_DOL_STUDIO_AGENTS_TYPE_TOOLS]);
  107. }
  108. protected function getVectorstore()
  109. {
  110. $this->aPageJsOptions['sPageUrl'] .= 'vector_store';
  111. return $this->getGrid($this->aGridObjects[BX_DOL_STUDIO_AGENTS_TYPE_VECTOR_STORE]);
  112. }
  113. protected function getAssistants()
  114. {
  115. $oAi = BxDolAI::getInstance();
  116. $oTemplate = BxDolStudioTemplate::getInstance();
  117. $this->aPageJsOptions['sPageUrl'] .= 'assistants';
  118. $sSubPage = '';
  119. if(($sSubPage = bx_get('spage')) !== false)
  120. $sSubPage = bx_process_input($sSubPage, BX_DATA_TEXT);
  121. $iAssistantId = 0;
  122. if(($iAssistantId = bx_get('aid')) !== false)
  123. $iAssistantId = bx_process_input($iAssistantId, BX_DATA_INT);
  124. $iChatId = 0;
  125. if(($iChatId = bx_get('cid')) !== false)
  126. $iChatId = bx_process_input($iChatId, BX_DATA_INT);
  127. if($iAssistantId && $iChatId) {
  128. $aResult = [];
  129. $sAssistantUrl = $this->sSubpageUrl . 'assistants&spage=chats&aid=' . $iAssistantId;
  130. $aAssistant = $oAi->getAssistantById($iAssistantId);
  131. if(!empty($aAssistant) && is_array($aAssistant)) {
  132. $aChat = $oAi->getAssistantChatById($iChatId);
  133. if(!empty($aChat) && is_array($aChat))
  134. $aResult[] = $oTemplate->parseHtmlByName('agents_assistant_info.html', [
  135. 'assistant_name' => $aAssistant['name'],
  136. 'assistant_info' => $aAssistant['description'],
  137. 'bx_if:show_chat' => [
  138. 'condition' => true,
  139. 'content' => [
  140. 'chat_name' => $aChat['name'],
  141. 'chat_info' => $aChat['description'],
  142. ]
  143. ],
  144. 'url_back' => $sAssistantUrl
  145. ]);
  146. }
  147. if(($oCmts = $oAi->getAssistantChatCmtsObject($iChatId, $oTemplate)) !== false) {
  148. $this->aPageJsOptions = array_merge($this->aPageJsOptions, [
  149. 'sPageUrl' => $sAssistantUrl . '&cid=' . $iChatId,
  150. 'sActionUrlCmts' => bx_append_url_params(BX_DOL_URL_ROOT . 'cmts.php', [
  151. 'sys' => $oCmts->getSystemName(),
  152. 'id' => $iChatId
  153. ])
  154. ]);
  155. $aResult[] = $oCmts->getCommentsBlock();
  156. }
  157. else
  158. $aResult[] = MsgBox(_t('_error occured'));
  159. return $aResult;
  160. }
  161. else if($iAssistantId) {
  162. $aResult = [];
  163. $aAssistant = $oAi->getAssistantById($iAssistantId);
  164. if(!empty($aAssistant) && is_array($aAssistant))
  165. $aResult[] = $oTemplate->parseHtmlByName('agents_assistant_info.html', [
  166. 'assistant_name' => $aAssistant['name'],
  167. 'assistant_info' => $aAssistant['description'],
  168. 'bx_if:show_chat' => [
  169. 'condition' => false,
  170. 'content' => [
  171. 'chat_name' => '',
  172. 'chat_info' => '',
  173. ]
  174. ],
  175. 'url_back' => $this->aPageJsOptions['sPageUrl']
  176. ]);
  177. switch($sSubPage) {
  178. case 'chats':
  179. $aResult[] = $this->getGrid($this->aGridObjects[BX_DOL_STUDIO_AGENTS_TYPE_ASSISTANTS . '_chats']);
  180. break;
  181. case 'files':
  182. $aResult[] = $this->getGrid($this->aGridObjects[BX_DOL_STUDIO_AGENTS_TYPE_ASSISTANTS . '_files']);
  183. break;
  184. }
  185. return $aResult;
  186. }
  187. return $this->getGrid($this->aGridObjects[BX_DOL_STUDIO_AGENTS_TYPE_ASSISTANTS]);
  188. }
  189. protected function getAgents()
  190. {
  191. return $this->getGrid($this->aGridObjects[BX_DOL_STUDIO_AGENTS_TYPE_AGENTS]);
  192. }
  193. /*
  194. * Isn't used for now. Most probably they will be removed.
  195. *
  196. protected function getAutomators()
  197. {
  198. $oTemplate = BxDolStudioTemplate::getInstance();
  199. $this->aPageJsOptions['sPageUrl'] .= 'automators';
  200. if(($iId = bx_get('id')) !== false) {
  201. if(($oCmts = BxDolAI::getInstance()->getAutomatorCmtsObject($iId, $oTemplate)) !== false) {
  202. $this->aPageJsOptions = array_merge($this->aPageJsOptions, [
  203. 'sPageUrl' => $this->sSubpageUrl . 'automators&id=' . $iId,
  204. 'sActionUrlCmts' => bx_append_url_params(BX_DOL_URL_ROOT . 'cmts.php', [
  205. 'sys' => $oCmts->getSystemName(),
  206. 'id' => $iId
  207. ])
  208. ]);
  209. return $oCmts->getCommentsBlock();
  210. }
  211. else
  212. return MsgBox(_t('_error occured'));
  213. }
  214. return $this->getGrid($this->aGridObjects[BX_DOL_STUDIO_AGENTS_TYPE_AUTOMATORS]);
  215. }
  216. protected function getHelpers()
  217. {
  218. $oTemplate = BxDolStudioTemplate::getInstance();
  219. $this->aPageJsOptions['sPageUrl'] .= 'helpers';
  220. if(($iId = bx_get('id')) !== false) {
  221. $this->aPageJsOptions = array_merge($this->aPageJsOptions, [
  222. 'sPageUrl' => $this->sSubpageUrl . 'helpers&id=' . $iId,
  223. ]);
  224. $aHelper = BxDolAI::getInstance()->getHelperById($iId);
  225. $aForm = $this->_getHelpersForm('tune', $aHelper);
  226. $oForm = new BxTemplFormView($aForm);
  227. $oForm->initChecker();
  228. if($oForm->isSubmittedAndValid()) {
  229. if($oForm->update($iId) !== false) {
  230. $sMessage = $oForm->getCleanValue('message');
  231. $oForm->aInputs['result']['value'] = BxDolAI::callHelper($iId, $sMessage);
  232. }
  233. }
  234. return $oForm->getCode();
  235. }
  236. return $this->getGrid($this->aGridObjects[BX_DOL_STUDIO_AGENTS_TYPE_HELPERS]);
  237. }
  238. protected function getProviders()
  239. {
  240. $this->aPageJsOptions = array_merge($this->aPageJsOptions, [
  241. 'sPageUrl' => $this->sSubpageUrl . 'providers',
  242. 'sActionUrlGrid' => bx_append_url_params(BX_DOL_URL_ROOT . 'grid.php', [
  243. 'o' => 'sys_studio_agents_providers'
  244. ])
  245. ]);
  246. return $this->getGrid($this->aGridObjects[BX_DOL_STUDIO_AGENTS_TYPE_PROVIDERS]);
  247. }
  248. *
  249. */
  250. protected function getGrid($sObjectName, $bObject = false)
  251. {
  252. $oGrid = BxDolGrid::getObjectInstance($sObjectName);
  253. if(!$oGrid)
  254. return '';
  255. return $bObject ? $oGrid : $oGrid->getCode();
  256. }
  257. protected function _getHelpersForm($sAction, $aHelper = [])
  258. {
  259. $aForm = array(
  260. 'form_attrs' => array(
  261. 'id' => 'bx_std_agents_helpers_' . $sAction,
  262. 'action' => $this->aPageJsOptions['sPageUrl'],
  263. 'method' => 'post',
  264. ),
  265. 'params' => array (
  266. 'db' => array(
  267. 'table' => 'sys_agents_helpers',
  268. 'key' => 'id',
  269. 'submit_name' => 'do_submit',
  270. ),
  271. ),
  272. 'inputs' => [
  273. 'prompt' => [
  274. 'type' => 'textarea',
  275. 'name' => 'prompt',
  276. 'caption' => _t('_sys_agents_helpers_field_prompt'),
  277. 'value' => isset($aHelper['prompt']) ? $aHelper['prompt'] : '',
  278. 'required' => '1',
  279. 'checker' => [
  280. 'func' => 'Avail',
  281. 'params' => [],
  282. 'error' => _t('_sys_agents_helpers_field_prompt_err'),
  283. ],
  284. 'db' => [
  285. 'pass' => 'Xss',
  286. ]
  287. ],
  288. 'message' => [
  289. 'type' => 'textarea',
  290. 'name' => 'message',
  291. 'caption' => _t('_sys_agents_helpers_field_message'),
  292. 'value' => '',
  293. 'required' => '1',
  294. 'checker' => [
  295. 'func' => 'Avail',
  296. 'params' => [],
  297. 'error' => _t('_sys_agents_helpers_field_message_err'),
  298. ],
  299. ],
  300. 'result' => [
  301. 'type' => 'textarea',
  302. 'name' => 'result',
  303. 'caption' => _t('_sys_agents_helpers_field_result'),
  304. 'value' => '',
  305. 'attrs' => [
  306. 'disabled' => 'disabled'
  307. ]
  308. ],
  309. 'submit' => [
  310. 'type' => 'submit',
  311. 'name' => 'do_submit',
  312. 'value' => _t('_sys_submit'),
  313. ],
  314. ],
  315. );
  316. return $aForm;
  317. }
  318. }
  319. /** @} */