Files
EasySoft-ZenTaoPMS/module/ai/control.php
T
2024-02-19 14:35:39 +08:00

1053 lines
40 KiB
PHP

<?php
/**
* The control file of ai module of ZenTaoPMS.
*
* @copyright Copyright 2009-2023 禅道软件(青岛)有限公司(ZenTao Software (Qingdao) Co., Ltd. www.zentao.net)
* @license ZPL(https://zpl.pub/page/zplv12.html) or AGPL(https://www.gnu.org/licenses/agpl-3.0.en.html)
* @author Wenrui LI <liwenrui@easycorp.ltd>
* @package ai
* @link https://www.zentao.net
*/
class ai extends control
{
/**
* AI model.
*
* @var aiModel
* @access public
*/
public $ai;
/**
* User model.
*
* @var userModel
* @access public
*/
public $user;
public function __construct($module = '', $method = '')
{
parent::__construct($module, $method);
$this->loadModel('user');
}
/**
* Check for user's privs and model configs, redirect to most relevant page.
*
* @access public
* @return void
*/
public function adminIndex()
{
$modelConfig = new stdclass();
$modelConfigData = $this->loadModel('setting')->getItems('owner=system&module=ai');
foreach($modelConfigData as $item) $modelConfig->{$item->key} = $item->value;
$modelConfigured = !empty($modelConfig->key) && !empty($modelConfig->type) && (empty($modelConfig->proxyType) || !empty($modelConfig->proxyAddr));
/* Redirect to model edit ifuser has priv and model is unconfigured. */
if(commonModel::hasPriv('ai', 'editModel') && !$modelConfigured) return $this->locate($this->createLink('ai', 'editModel'));
/* Redirect to prompts ifuser has priv. */
if(commonModel::hasPriv('ai', 'prompts')) return $this->locate($this->createLink('ai', 'prompts'));
/* Redirect to models ifuser has priv. */
if(commonModel::hasPriv('ai', 'models')) return $this->locate($this->createLink('ai', 'models'));
/* User has no priv, deny access. */
return $this->send(array('result' => 'fail', 'message' => $this->lang->error->accessDenied, 'locate' => $this->createLink('admin', 'index')));
}
/**
* List models.
*
* TODO: not fully implemented yet, currently shows the only model config.
*
* @access public
* @return void
*/
public function models()
{
$modelConfig = new stdclass();
$modelConfig->type = '';
$modelConfig->key = '';
$modelConfig->proxyType = '';
$modelConfig->proxyAddr = '';
$modelConfig->description = '';
$modelConfig->status = '';
$storedModelConfig = $this->loadModel('setting')->getItems('owner=system&module=ai');
foreach($storedModelConfig as $item) $modelConfig->{$item->key} = $item->value;
$this->view->modelConfig = $modelConfig;
$this->view->title = $this->lang->ai->models->title;
$this->display();
}
/**
* Edit model configuration, store in system.ai settings.
*
* @access public
* @return void
*/
public function editModel()
{
if(strtolower($this->server->request_method) == 'post')
{
$modelConfig = fixer::input('post')->get();
$currentVendor = empty($modelConfig->vendor) ? key($this->lang->ai->models->vendorList->{empty($modelConfig->type) ? key($this->lang->ai->models->typeList) : $modelConfig->type}) : $modelConfig->vendor;
$vendorRequiredFields = $this->config->ai->vendorList[$currentVendor]['requiredFields'];
$errors = array();
if(empty($modelConfig->type)) $errors[] = sprintf($this->lang->ai->validate->noEmpty, $this->lang->ai->models->type);
foreach($vendorRequiredFields as $field)
{
if(empty($modelConfig->$field)) $errors[] = sprintf($this->lang->ai->validate->noEmpty, $this->lang->ai->models->$field);
}
if(!empty($modelConfig->proxyType) && empty($modelConfig->proxyAddr))
{
$errors[] = sprintf($this->lang->ai->validate->noEmpty, $this->lang->ai->models->proxyAddr);
}
if(!empty($errors)) return $this->send(array('result' => 'fail', 'message' => implode(' ', $errors)));
$this->loadModel('setting')->setItems('system.ai', $modelConfig);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('models') . '#app=admin'));
}
$modelConfig = new stdclass();
$modelConfig->type = '';
$modelConfig->key = '';
$modelConfig->proxyType = '';
$modelConfig->proxyAddr = '';
$modelConfig->description = '';
$modelConfig->status = '';
$storedModelConfig = $this->loadModel('setting')->getItems('owner=system&module=ai');
foreach($storedModelConfig as $item) $modelConfig->{$item->key} = $item->value;
$this->view->modelConfig = $modelConfig;
$this->view->title = $this->lang->ai->models->title;
$this->display();
}
public function view($id)
{
$this->locate($this->createLink('ai', 'miniProgramChat', "id=$id"));
}
/**
* Mini program chat.
*
* @param string $id
* @access public
* @return void
*/
public function miniProgramChat($id)
{
$miniProgram = $this->ai->getMiniProgramByID($id);
$this->view->miniProgram = $miniProgram;
$this->view->fields = $this->ai->getMiniProgramFields($id);
$this->view->collectedIDs = $this->ai->getCollectedMiniProgramIDs($this->app->user->id);
$this->view->title = "{$this->lang->ai->miniPrograms->common}#{$miniProgram->id} $miniProgram->name";
$this->display();
}
/**
* Mini program square.
*
* @param string $category
* @param int $recTotal
* @param int $recPerPage
* @param int $pageID
* @access public
* @return void
*/
public function square($category = '', $recTotal = 0, $recPerPage = 15, $pageID = 1)
{
$this->app->loadClass('pager', true);
$pager = new pager($recTotal, $recPerPage, $pageID);
$collectedIDs = null;
if($category === '')
{
$collectedIDs = $this->ai->getCollectedMiniProgramIDs($this->app->user->id);
$category = empty($collectedIDs)
? 'discovery'
: 'collection';
}
if($category === 'collection')
{
$collectedIDs = $this->ai->getCollectedMiniProgramIDs($this->app->user->id, $pager);
$miniPrograms = $this->ai->getMiniProgramsByID($collectedIDs);
}
else if($category === 'discovery')
{
$popularIDs = $this->ai->getPopularMiniProgramIDs($pager);
$miniPrograms = $this->ai->getMiniProgramsByID($popularIDs);
}
else if($category === 'latest')
{
$miniPrograms = $this->ai->getMiniPrograms('', 'active', 'createdDate_desc', $pager);
}
else
{
$miniPrograms = $this->ai->getMiniPrograms($category, 'active', 'createdDate_desc', $pager);
}
$usedCustomCategories = $this->ai->getUsedCustomCategories();
$categoryList = array_merge($this->lang->ai->miniPrograms->categoryList, $this->ai->getCustomCategories());
$usedCustomCategoryList = array();
foreach($categoryList as $key => $value)
{
if(in_array($key, $usedCustomCategories)) $usedCustomCategoryList[$key] = $value;
}
$this->view->collectedIDs = empty($collectedIDs) ? $this->ai->getCollectedMiniProgramIDs($this->app->user->id) : $collectedIDs;
$this->view->category = $category;
$this->view->categoryList = array_merge($this->lang->ai->miniPrograms->squareCategories, $usedCustomCategoryList);
$this->view->pager = $pager;
$this->view->miniPrograms = $miniPrograms ?: array();
$this->view->title = $this->lang->ai->miniPrograms->common;
$this->display();
}
/**
* Collect mini program by appid.
*
* @param string $appID
* @param string $delete
* @access public
* @return void
*/
public function collectMiniProgram($appID, $delete = 'false')
{
$this->ai->collectMiniProgram($this->app->user->id, $appID, $delete);
return $this->send(array('status' => ($delete === 'true' ? '0' : '1')));
}
/**
* List mini programs.
*
* @access public
* @return void
*/
public function miniPrograms($category = '', $status = '', $orderBy = 'createdDate_desc', $recTotal = 0, $recPerPage = 15, $pageID = 1)
{
$this->app->loadClass('pager', true);
$pager = new pager($recTotal, $recPerPage, $pageID);
$order = common::appendOrder($orderBy);
$categoryList = array_merge($this->lang->ai->miniPrograms->categoryList, $this->ai->getCustomCategories());
$programs = $this->ai->getMiniPrograms($category, $status, $order, $pager);
foreach($programs as $program)
{
$program->canPublish = empty($program->published) && $this->ai->canPublishMiniProgram($program);
$program->createdByLabel = $this->loadModel('user')->getById($program->createdBy, 'account')->realname;
$program->categoryLabel = $categoryList[$program->category];
$program->publishedLabel = $program->published === '1'
? $this->lang->ai->miniPrograms->statuses['active']
: $this->lang->ai->miniPrograms->statuses['draft'];
}
$this->view->title = $this->lang->ai->miniPrograms->common;
$this->view->miniPrograms = $programs;
$this->view->category = $category;
$this->view->categoryList = $categoryList;
$this->view->status = $status;
$this->view->orderBy = $orderBy;
$this->view->pager = $pager;
$this->display();
}
/**
* Edit mini program category.
*
* @access public
* @return void
*/
public function editMiniProgramCategory()
{
if(!empty($_POST))
{
$this->ai->updateCustomCategories();
}
$this->view->usedCustomCategories = $this->ai->getUsedCustomCategories();
$this->view->categoryList = $this->ai->getCustomCategories();
$this->view->title = $this->lang->ai->miniPrograms->common;
$this->display();
}
/**
* Change mini program `deleted` value.
*
* @param string $appID
* @param string $deleted
* @access public
* @return void
*/
public function deleteMiniProgram($appID, $deleted)
{
$result = $this->ai->deleteMiniProgram($appID, $deleted);
if($result) return $this->send(array('result' => 'success', 'locate' => $this->server->http_referer));
$this->sendError(dao::getError());
}
/**
* Change mini program `publish` value.
*
* @param string $appID
* @param string $published
* @access public
* @return void
*/
public function publishMiniProgram($appID, $published)
{
$result = $this->ai->publishMiniProgram($appID, $published);
if($result) return $this->send(array('result' => 'success', 'locate' => $this->server->http_referer));
$this->sendError(dao::getError());
}
/**
* Test mini program.
*
* @param string $appID
* @access public
* @return void
*/
public function testMiniProgram($appID)
{
if(!empty($_POST))
{
$toPublish = $_POST['toPublish'];
unset($_POST['toPublish']);
$this->ai->createTmpMiniProgram($appID);
if($toPublish === '1') $this->ai->publishMiniProgram($appID, '1');
return print(js::closeModal('parent', 'this'));
}
$this->view->currentFields = $this->ai->getMiniProgramFields($appID);
$this->view->currentPrompt = $this->ai->getMiniProgramByID($appID)->prompt;
$this->view->appID = $appID;
$this->display();
}
/**
* Create a mini program.
*
* @param string|null $appID
* @access public
* @return void
*/
public function createMiniProgram($appID = null)
{
if(!empty($_POST))
{
$toNext = $_POST['toNext'];
unset($_POST['toNext']);
$requiredFields = array('category' => $this->lang->ai->miniPrograms->category, 'model' => $this->lang->prompt->model, 'name' => $this->lang->prompt->name, 'desc' => $this->lang->ai->miniPrograms->desc);
$errors = $this->ai->verifyRequiredFields($requiredFields);
if($errors !== false) return $this->sendError($errors);
$isDuplicated = $this->ai->checkDuplicatedAppName($_POST['name'], $appID);
if($isDuplicated) return $this->sendError(array('name' => $this->lang->ai->miniPrograms->field->duplicatedNameTip));
$id = $this->ai->createTmpMiniProgram($appID);
if($id === false) return $this->sendError(array('message' => $this->lang->fail));
if($toNext === '1') return $this->sendSuccess(array('message' => $this->lang->saveSuccess, 'locate' => $this->createLink('ai', 'configuredMiniProgram', "appID=$id")));
return $this->sendSuccess(array('message' => $this->lang->saveSuccess, 'locate' => $this->createLink('ai', 'createMiniProgram', "appID=$id")));
}
if(!empty($appID))
{
$miniProgram = $this->ai->getMiniProgramByID($appID);
$this->view->name = $miniProgram->name;
$this->view->desc = $miniProgram->desc;
$this->view->model = $miniProgram->model;
$this->view->category = $miniProgram->category;
list($this->view->iconName, $this->view->iconTheme) = explode('-', $miniProgram->icon);
}
else
{
$this->view->iconName = 'writinghand';
$this->view->iconTheme = 7;
}
$this->view->categoryList = $this->ai->getCustomCategories();
$this->view->title = $this->lang->ai->miniPrograms->common;
$this->display();
}
/**
* Mini program view.
*
* @param string $id
* @access public
* @return void
*/
public function miniProgramView($id)
{
$miniProgram = $this->ai->getMiniProgramByID($id);
$this->view->miniProgram = $miniProgram;
$this->view->categoryList = array_merge($this->lang->ai->miniPrograms->categoryList, $this->ai->getCustomCategories());
$this->view->preAndNext = $this->loadModel('common')->getPreAndNextObject('miniProgram', $id);
$this->view->actions = $this->loadModel('action')->getList('miniProgram', $id);
$this->view->fields = $this->ai->getMiniProgramFields($id);
$this->view->users = $this->loadModel('user')->getPairs('noletter');
$this->view->title = "{$this->lang->ai->miniPrograms->common}#{$miniProgram->id} $miniProgram->name";
$this->display();
}
/**
* Configured a mini program.
*
* @param string $appID
* @access public
* @return void
*/
public function configuredMiniProgram($appID)
{
if(!empty($_POST))
{
$toPublish = $_POST['toPublish'];
unset($_POST['toPublish']);
$this->ai->saveMiniProgramFields($appID);
if($toPublish === '1') $this->ai->publishMiniProgram($appID, '1');
}
$program = $this->ai->getMiniProgramByID($appID);
$this->view->currentFields = $this->ai->getMiniProgramFields($appID);
$this->view->currentPrompt = $program->prompt;
$this->view->title = $this->lang->ai->miniPrograms->common;
$this->view->appID = $appID;
$this->display();
}
/**
* Test connection to API endpoint.
*
* @access public
* @return void
*/
public function testConnection()
{
$modelConfig = fixer::input('post')->get();
$currentVendor = empty($modelConfig->vendor) ? key($this->lang->ai->models->vendorList->{empty($modelConfig->type) ? key($this->lang->ai->models->typeList) : $modelConfig->type}) : $modelConfig->vendor;
$vendorRequiredFields = $this->config->ai->vendorList[$currentVendor]['requiredFields'];
$errors = array();
if(empty($modelConfig->type)) $errors[] = sprintf($this->lang->ai->validate->noEmpty, $this->lang->ai->models->type);
foreach($vendorRequiredFields as $field)
{
if(empty($modelConfig->$field)) $errors[] = sprintf($this->lang->ai->validate->noEmpty, $this->lang->ai->models->$field);
}
if(!empty($modelConfig->proxyType) && empty($modelConfig->proxyAddr))
{
$errors[] = sprintf($this->lang->ai->validate->noEmpty, $this->lang->ai->models->proxyAddr);
}
if(!empty($errors)) return $this->send(array('result' => 'fail', 'message' => implode(' ', $errors)));
$this->ai->setConfig($modelConfig);
if($this->config->ai->models[$modelConfig->type] == 'ernie' || $currentVendor == 'azure' || $modelConfig->type == 'openai-gpt4')
{
$messages = array((object)array('role' => 'user', 'content' => 'test'));
$result = $this->ai->converse($messages, array('maxTokens' => 1));
}
else
{
$result = $this->ai->complete('test', 1); // Test completing 'test' with length of 1.
}
if($result === false)
{
return $this->send(array('result' => 'fail', 'message' => empty($this->ai->errors) ? $this->lang->ai->models->testConnectionResult->fail : sprintf($this->lang->ai->models->testConnectionResult->failFormat, implode(', ', $this->ai->errors))));
}
return $this->send(array('result' => 'success', 'message' => $this->lang->ai->models->testConnectionResult->success));
}
/**
* List prompts.
*
* @param string $module
* @param string $status
* @access public
* @return void
*/
public function prompts($module = '', $status = '', $orderBy = 'id_desc', $recTotal = 0, $recPerPage = 15, $pageID = 1)
{
$this->loadModel('user');
$users = $this->user->getPairs('noletter');
if($_POST)
{
$data = fixer::input('post')->get();
if(isset($data->togglePromptStatus) && isset($data->promptId))
{
if(!$this->ai->isModelConfigured()) return $this->send(array('result' => 'fail', 'message' => $this->lang->ai->models->noModelError, 'locate' => $this->inlink('models') . '#app=admin'));
$this->ai->togglePromptStatus($data->promptId);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
return $this->send(array('result' => 'success'));
}
}
/* Set pager and order. */
$this->app->loadClass('pager', $static = true);
$pager = new pager($recTotal, $recPerPage, $pageID);
$order = common::appendOrder($orderBy);
$this->view->prompts = $this->ai->getPrompts($module, $status, $order, $pager);
$this->view->module = $module;
$this->view->status = $status;
$this->view->orderBy = $orderBy;
$this->view->pager = $pager;
$this->view->title = $this->lang->ai->prompts->common;
$this->view->users = $users;
if($this->config->edition == 'open')
{
$this->view->promptModules = array_map(function ($prompt)
{
return $prompt->module;
}, $this->ai->getPrompts());
}
$this->display();
}
/**
* View prompt details.
*
* @param int $id
* @access public
* @return void
*/
public function promptView($id)
{
$prompt = $this->ai->getPromptById($id);
$this->view->prompt = $prompt;
$this->view->preAndNext = $this->loadModel('common')->getPreAndNextObject('prompt', $id);
$this->view->actions = $this->loadModel('action')->getList('prompt', $id);
$this->view->dataPreview = $this->ai->generateDemoDataPrompt($prompt->module, $prompt->source);
$this->view->users = $this->loadModel('user')->getPairs('noletter');
$this->view->title = "{$this->lang->ai->prompts->common}#{$prompt->id} $prompt->name";
$this->display();
}
/**
* Create a prompt.
*
* @access public
* @return void
*/
public function createPrompt()
{
if(strtolower($this->server->request_method) == 'post')
{
$prompt = fixer::input('post')->get();
$promptID = $this->ai->createPrompt($prompt);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
return $this->send(array('result' => 'success', 'callback' => "gotoPrompt($promptID)"));
}
$this->view->title = $this->lang->ai->prompts->create;
$this->display();
}
/**
* Edit a prompt.
*
* @param int $id
* @access public
* @return void
*/
public function promptEdit($id)
{
$prompt = $this->ai->getPromptByID($id);
if(strtolower($this->server->request_method) == 'post')
{
$data = fixer::input('post')->get();
$prompt->name = $data->name;
$prompt->desc = $data->desc;
$this->ai->updatePrompt($prompt);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
return $this->send(array('result' => 'success', 'locate' => $this->inlink('promptView', "id={$prompt->id}") . '#app=admin'));
}
$this->view->prompt = $prompt;
$this->view->title = $this->lang->ai->prompts->edit;
$this->display();
}
/**
* Delete a prompt.
*
* @param int $prompt
* @access public
* @return void
*/
public function promptDelete($prompt)
{
$result = $this->ai->deletePrompt($prompt);
if(dao::isError() || $result === false) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
return $this->send(array('result' => 'success'));
}
/**
* Edit role of prompt, prompt editing step 2.
*
* @param int $promptID
* @access public
* @return void
*/
public function promptAssignRole($promptID)
{
$prompt = $this->ai->getPromptByID($promptID);
if($_POST)
{
$data = fixer::input('post')->get();
$originalPrompt = clone $prompt;
$prompt->model = 0; // TODO: use actual $data->model;
$prompt->role = $data->role;
$prompt->characterization = $data->characterization;
if(!empty($data->saveTemplate) && $data->saveTemplate == 'save')
{
$this->ai->createRoleTemplate($prompt->role, $prompt->characterization);
}
$this->ai->updatePrompt($prompt, $originalPrompt);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
if(!empty($data->jumpToNext)) return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('promptSelectDataSource', "promptID=$promptID") . '#app=admin'));
return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('promptAssignRole', "promptID=$promptID") . '#app=admin'));
}
$this->view->prompt = $prompt;
$this->view->promptID = $promptID;
$this->view->lastActiveStep = $this->ai->getLastActiveStep($prompt);
$this->view->title = "{$this->lang->ai->prompts->common}#{$prompt->id} $prompt->name {$this->lang->colon} " . $this->lang->ai->prompts->assignRole . " {$this->lang->colon} " . $this->lang->ai->prompts->common;
$this->view->roleTemplates = $this->ai->getRoleTemplates();
$this->display();
}
/**
* Set data source of prompt, prompt editing step 3.
*
* @param int $promptID
* @access public
* @return void
*/
public function promptSelectDataSource($promptID)
{
$prompt = $this->ai->getPromptByID($promptID);
if($_POST)
{
$data = fixer::input('post')->get();
$originalPrompt = clone $prompt;
$prompt->module = $data->datagroup;
$prompt->source = ",$data->datasource,";
$this->ai->updatePrompt($prompt, $originalPrompt);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
if(!empty($data->jumpToNext)) return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('promptSetPurpose', "promptID=$promptID") . '#app=admin'));
return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('promptSelectDataSource', "promptID=$promptID") . '#app=admin'));
}
$this->view->activeDataSource = empty($prompt->module) ? current(array_keys($this->config->ai->dataSource)) : $prompt->module;
$this->view->dataSource = $this->config->ai->dataSource;
$this->view->prompt = $prompt;
$this->view->promptID = $promptID;
$this->view->lastActiveStep = $this->ai->getLastActiveStep($prompt);
$this->view->title = "{$this->lang->ai->prompts->common}#{$prompt->id} $prompt->name {$this->lang->colon} " . $this->lang->ai->prompts->selectDataSource . " {$this->lang->colon} " . $this->lang->ai->prompts->common;
$this->display();
}
/**
* Edit purpose and purpose elaboration of prompt, prompt editing step 4.
*
* @param int $promptID
* @access public
* @return void
*/
public function promptSetPurpose($promptID)
{
$prompt = $this->ai->getPromptByID($promptID);
if($_POST)
{
$data = fixer::input('post')->get();
$originalPrompt = clone $prompt;
$prompt->purpose = $data->purpose;
$prompt->elaboration = $data->elaboration;
$this->ai->updatePrompt($prompt, $originalPrompt);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
if(!empty($data->jumpToNext)) return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('promptSetTargetForm', "promptID=$promptID") . '#app=admin'));
return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('promptSetPurpose', "promptID=$promptID") . '#app=admin'));
}
$this->view->dataPreview = $this->ai->generateDemoDataPrompt($prompt->module, $prompt->source);
$this->view->prompt = $prompt;
$this->view->promptID = $promptID;
$this->view->lastActiveStep = $this->ai->getLastActiveStep($prompt);
$this->view->title = "{$this->lang->ai->prompts->common}#{$prompt->id} $prompt->name {$this->lang->colon} " . $this->lang->ai->prompts->setPurpose . " {$this->lang->colon} " . $this->lang->ai->prompts->common;
$this->display();
}
/**
* Set target form of prompt, prompt editing step 5.
*
* @param int $promptID
* @access public
* @return void
*/
public function promptSetTargetForm($promptID)
{
$prompt = $this->ai->getPromptByID($promptID);
if($_POST)
{
$data = fixer::input('post')->get();
$originalPrompt = clone $prompt;
$prompt->targetForm = $data->targetForm;
$this->ai->updatePrompt($prompt, $originalPrompt);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
if(!empty($data->goTesting)) // Go to testing object view.
{
$location = $this->ai->getTestingLocation($prompt);
return $this->send(empty($location) ? array('result' => 'fail', 'target' => '#go-test-btn', 'message' => $this->lang->ai->prompts->goingTestingFail) : array('result' => 'success', 'target' => '#go-test-btn', 'msg' => $this->lang->ai->prompts->goingTesting, 'locate' => $location));
}
if(!empty($data->jumpToNext)) $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('promptFinalize', "promptID=$promptID") . '#app=admin'));
return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('promptSetTargetForm', "promptID=$promptID") . '#app=admin'));
}
$this->view->dataPreview = $this->ai->generateDemoDataPrompt($prompt->module, $prompt->source);
$this->view->prompt = $prompt;
$this->view->promptID = $promptID;
$this->view->lastActiveStep = $this->ai->getLastActiveStep($prompt);
$this->view->title = "{$this->lang->ai->prompts->common}#{$prompt->id} $prompt->name {$this->lang->colon} " . $this->lang->ai->prompts->setTargetForm . " {$this->lang->colon} " . $this->lang->ai->prompts->common;
$this->display();
}
/**
* Edit additional information of prompt, final prompt editing step.
*
* @param int $promptID
* @access public
* @return void
*/
public function promptFinalize($promptID)
{
$prompt = $this->ai->getPromptByID($promptID);
if($_POST)
{
$data = fixer::input('post')->get();
$originalPrompt = clone $prompt;
$prompt->name = $data->name;
$prompt->desc = $data->desc;
$this->ai->updatePrompt($prompt, $originalPrompt);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
if(!empty($data->jumpToNext)) $this->ai->togglePromptStatus($prompt, 'active');
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
if(!empty($data->jumpToNext)) return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('prompts') . '#app=admin'));
return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->inlink('promptFinalize', "promptID=$promptID") . '#app=admin'));
}
$this->view->prompt = $prompt;
$this->view->promptID = $promptID;
$this->view->lastActiveStep = $this->ai->getLastActiveStep($prompt);
$this->view->title = "{$this->lang->ai->prompts->common}#{$prompt->id} $prompt->name {$this->lang->colon} " . $this->lang->ai->prompts->finalize . " {$this->lang->colon} " . $this->lang->ai->prompts->common;
$this->display();
}
/**
* Execute prompt on obejct, and redirect to target form page.
*
* @param int $promptId
* @param int $objectId
* @access public
* @return void
*/
public function promptExecute($promptId, $objectId)
{
if(!$this->ai->isModelConfigured()) return $this->send(array('result' => 'fail', 'message' => $this->lang->ai->models->noModelError, 'locate' => $this->inlink('models') . '#app=admin'));
$prompt = $this->ai->getPromptByID($promptId);
if(empty($prompt) || !$this->ai->isExecutable($prompt)) return $this->send(array('result' => 'fail', 'message' => sprintf($this->lang->ai->execute->failFormat, $this->lang->ai->execute->failReasons['noPrompt'])));
$object = $this->ai->getObjectForPromptById($prompt, $objectId);
if(empty($object)) return $this->send(array('result' => 'fail', 'message' => sprintf($this->lang->ai->execute->failFormat, $this->lang->ai->execute->failReasons['noObjectData'])));
list($objectData, $rawObject) = $object;
list($location, $stop) = $this->ai->getTargetFormLocation($prompt, $rawObject);
if(empty($location)) return $this->send(array('result' => 'fail', 'message' => sprintf($this->lang->ai->execute->failFormat, $this->lang->ai->execute->failReasons['noTargetForm'])));
if(!empty($stop)) return header("location: $location", true, 302);
/* Execute prompt and catch exceptions. */
try
{
$response = $this->ai->executePrompt($prompt, $object);
}
catch (AIResponseException $e)
{
$output = array('result' => 'fail', 'message' => sprintf($this->lang->ai->execute->failFormat, $e->getMessage()));
/* Audition shall quit on such exception. */
if(isset($_SESSION['auditPrompt']) && time() - $_SESSION['auditPrompt']['time'] < 10 * 60)
{
$output['locate'] = $this->inlink('promptAudit', "promptID=$promptId&objectId=$objectId&exit=true");
}
return $this->send($output);
}
if(is_int($response)) return $this->send(array('result' => 'fail', 'message' => sprintf($this->lang->ai->execute->failFormat, $this->lang->ai->execute->executeErrors["$response"]) . (empty($this->ai->errors) ? '' : implode(', ', $this->ai->errors))));
if(empty($response)) return $this->send(array('result' => 'fail', 'message' => sprintf($this->lang->ai->execute->failFormat, $this->lang->ai->execute->failReasons['noResponse'])));
$this->ai->setInjectData($prompt->targetForm, $response);
$_SESSION['aiPrompt']['prompt'] = $prompt;
$_SESSION['aiPrompt']['objectId'] = $objectId;
if($prompt->status == 'draft') $_SESSION['auditPrompt']['time'] = time();
$promptViewLink = $this->inlink('promptview', "promptID=$promptId");
echo js::start() . "const currentApp = parent.$.apps.getLastApp().code; parent.$.apps.open('$location'); if(currentApp == 'admin') {location.href = '$promptViewLink';} else if(currentApp != parent.$.apps.getLastApp().code) {parent.$.apps.close(currentApp);}" . js::end();
}
/**
* Reset prompt execution.
*
* @param bool $failed Whether the execution failed, triggers error message.
* @access public
* @return void
*/
public function promptExecutionReset($failed = false)
{
/* Reset session. */
unset($_SESSION['aiPrompt']);
unset($_SESSION['auditPrompt']);
if($failed) return $this->send(array('result' => 'fail', 'message' => sprintf($this->lang->ai->execute->failFormat, $this->lang->ai->execute->failReasons['noTargetForm']), 'locate' => isset($_SERVER['HTTP_REFERER']) ? $_SERVER['HTTP_REFERER'] : ''));
return $this->send(array('result' => 'success'));
}
/**
* Audit prompt modal.
*
* @param int $promptId
* @param int $objectId
* @param bool $exit
* @access public
* @return void|int
*/
public function promptAudit($promptId, $objectId, $exit = false)
{
if(!empty($exit))
{
unset($_SESSION['auditPrompt']);
return $this->send(array('result' => 'success', 'locate' => $this->inlink('promptview', "promptID=$promptId") . '#app=admin'));
}
$prompt = $this->ai->getPromptByID($promptId);
if($_POST)
{
$data = fixer::input('post')->get();
$originalPrompt = clone $prompt;
$prompt->role = $data->role;
$prompt->characterization = $data->characterization;
$prompt->purpose = $data->purpose;
$prompt->elaboration = $data->elaboration;
$this->ai->updatePrompt($prompt, $originalPrompt);
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
if(empty($data->backLocation))
{
$response = array('result' => 'success', 'message' => $this->lang->saveSuccess, 'closeModal' => true);
}
else
{
$response = array('result' => 'success', 'message' => $this->lang->saveSuccess, 'callback' => "reloadPrompt($promptId, $objectId)");
}
$this->sendSuccess($response);
}
$objectForPrompt = $this->ai->getObjectForPromptById($prompt, $objectId);
if(empty($objectForPrompt)) return $this->send(array('result' => 'fail', 'message' => $this->lang->ai->execute->fail));
list($objectData, $object) = $objectForPrompt;
$this->view->prompt = $prompt;
$this->view->object = $object;
$this->view->dataPrompt = $this->ai->serializeDataToPrompt($prompt->module, $prompt->source, $objectData);
$this->display();
}
/**
* Publish prompt, set status to active.
*
* @param int $id
* @param bool $backToTestingLocation
* @access public
* @return void
*/
public function promptPublish($id, $backToTestingLocation = false)
{
if(!$this->ai->isModelConfigured()) return $this->send(array('result' => 'fail', 'message' => $this->lang->ai->models->noModelError, 'locate' => $this->inlink('models') . '#app=admin'));
unset($_SESSION['auditPrompt']);
$this->ai->togglePromptStatus($id, 'active');
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
if($backToTestingLocation)
{
return $this->send(array('result' => 'success', 'message' => $this->lang->ai->prompts->action->publishSuccess, 'locate' => $this->inlink('promptview', "id=$id") . '#app=admin'));
}
return $this->send(array('result' => 'success'));
}
/**
* Unpublish prompt, set status to draft.
*
* @param int $id
* @access public
* @return void
*/
public function promptUnpublish($id)
{
$this->ai->togglePromptStatus($id, 'draft');
if(dao::isError()) return $this->send(array('result' => 'fail', 'message' => dao::getError()));
return $this->send(array('result' => 'success'));
}
/**
* Return html element for the role template list.
* This is used in prompt designing step 2.
* Return html will deserialize in the front end js and replace the original role template list.
*
* @access public
* @return void
*/
public function roleTemplates()
{
if($_POST)
{
$data = fixer::input('post')->get();
$method = $data->method;
switch ($method)
{
case 'create':
$this->ai->createRoleTemplate($data->role, $data->characterization);
break;
case 'delete':
$this->ai->deleteRoleTemplate($data->id);
break;
case 'edit':
$this->ai->updateRoleTemplate($data->id, $data->role, $data->characterization);
break;
}
}
$this->view->roleTemplates = $this->ai->getRoleTemplates();
$this->display();
}
/**
* Chat with LLMs.
*
* @access public
* @return void
*/
public function chat()
{
$messages = array();
if(!empty($_POST))
{
$history = $this->post->history;
$message = $this->post->message;
$isRetry = $this->post->retry == 'true';
$messages = json_decode($history);
if(empty($messages)) $messages[] = (object)array('role' => 'system', 'content' => $this->lang->ai->chatSystemMessage);
if(!$isRetry) $messages[] = (object)array('role' => 'user', 'content' => $message);
if($this->ai->isModelConfigured())
{
$response = $this->ai->converse($messages);
if(empty($response))
{
$this->view->error = $this->lang->ai->chatNoResponse;
}
else
{
$messages[] = (object)array('role' => 'assistant', 'content' => is_array($response) ? current($response) : $response);
}
}
else
{
$this->view->error = $this->lang->ai->models->noModelError;
}
}
$this->view->title = $this->lang->ai->chat;
$this->view->messages = $messages;
$this->display();
}
}