zentao/module/zai/control.php

<?php
/**
 * The control file of zai module of ZenTaoPMS.
 *
 * @copyright   Copyright 2009-2025 禅道软件（青岛）有限公司(ZenTao Software (Qingdao) Co., Ltd. www.cnezsoft.com)
 * @license     ZPL(http://zpl.pub/page/zplv12.html) or AGPL(https://www.gnu.org/licenses/agpl-3.0.en.html)
 * @author      tenghuaian <tenghuaian@chandao.com>
 * @link        https://www.zentao.net
 */
class zai extends control
{
    /**
     * 配置ZAI。
     * Configure ZAI.
     *
     * @param string $mode
     * @access public
     * @return void
     */
    public function setting($mode = 'view')
    {
        if(!empty($_POST))
        {
            $setting = new stdClass();
            $setting->appID      = trim($_POST['appID']);
            $setting->host       = trim($_POST['host']);
            $setting->port       = trim($_POST['port']);
            $setting->token      = trim($_POST['token']);
            $setting->adminToken = trim($_POST['adminToken']);

            if(empty($setting->host)) $setting = null;
            $this->zai->setSetting($setting);

            if(dao::isError()) return $this->sendError(dao::getError());
            return $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'load' => $this->createLink('zai', 'setting')));
        }

        $setting = $this->zai->getSetting(true);
        if($mode == 'view')
        {
            if(!empty($setting->token)) $setting->token = str_repeat('*', strlen($setting->token));
            if(!empty($setting->adminToken)) $setting->adminToken = str_repeat('*', strlen($setting->adminToken));
        }

        $this->view->title   = $this->lang->zai->setting;
        $this->view->setting = $setting;
        $this->view->mode    = $mode;
        $this->display();
    }

    /**
     * Ajax: 获取当前用户的ZAI Authorization token。
     * Ajax: Get ZAI Authorization Token of current user.
     *
     * @access public
     * @return void
     */
    public function ajaxGetToken()
    {
        return $this->send($this->zai->getToken());
    }

    /**
     * 禅道数据向量化。
     * Vectorized data of ZenTao.
     *
     * @access public
     * @return void
     */
    public function vectorized()
    {
        $info = $this->zai->getVectorizedInfo();

        $this->view->info       = $info;
        $this->view->title      = $this->lang->zai->vectorized;
        $this->view->zaiSetting = $this->zai->getSetting();
        $this->view->syncTypes  = zaiModel::getSyncTypes();
        $this->display();
    }

    /**
     * Ajax: 启用数据向量化。
     * Ajax: Enable data vectorization.
     *
     * @access public
     * @return void
     */
    public function ajaxEnableVectorization()
    {
        if($_SERVER['REQUEST_METHOD'] !== 'POST')
        {
            return $this->send(array('result' => 'fail', 'message' => $this->lang->zai->onlyPostRequest));
        }

        $force  = isset($_POST['force']) && $_POST['force'] === 'true';
        $result = $this->zai->enableVectorization($force);
        return $this->send($result);
    }

    /**
     * Ajax: 同步禅道向量化数据到 ZAI 知识库。
     * Ajax: Sync vectorized data of ZenTao to ZAI knowledge base.
     *
     * @access public
     * @return void
     */
    public function ajaxSyncVectorization()
    {
        $info = $this->zai->getVectorizedInfo();
        if($_SERVER['REQUEST_METHOD'] !== 'POST')
        {
            return $this->send(array('result' => 'success', 'data' => $info));
        }

        if(empty($info->key)) return $this->send(array('result' => 'fail', 'message' => $this->lang->zai->vectorizedUnavailableHint, 'data' => $info));

        $force = isset($_POST['force']) && $_POST['force'] === 'true';
        if($info->status === 'synced' && $force)
        {
            $info->status          = 'wait';
            $info->syncedTime      = 0;
            $info->syncedCount     = 0;
            $info->syncFailedCount = 0;
            $info->syncTime        = 0;
            $info->syncingType     = zaiModel::getNextSyncType();
            $info->syncingID       = 0;
            $info->syncDetails     = new stdClass();
        }

        if($info->status !== 'wait' && $info->status !== 'syncing')
        {
            return $this->send(array('result' => 'success', 'data' => $info));
        }

        if($info->status !== 'syncing')
        {
            $info->status = 'syncing';
            $info->synced = 0;
        }

        $startSyncTime  = microtime(true) * 1000;
        $info->syncTime = time();
        $result = $this->zai->syncNextTarget($info->key, $info->syncingType, $info->syncingID);

        $syncingType = $info->syncingType;
        if(!isset($info->syncDetails->$syncingType))
        {
            $syncDetail = new stdClass();
            $syncDetail->failed = 0;
            $syncDetail->synced = 0;
            $info->syncDetails->$syncingType = $syncDetail;
        }
        if($result)
        {
            if(isset($result['fatal']) && $result['fatal'])
            {
                $info->status = 'synced';
                $info->syncFailedCount++;
                $info->syncDetails->$syncingType->failed++;
                $this->zai->setVectorizedInfo($info);
                return $this->send(array('result' => 'fail', 'message' => $result['message'], 'data' => $info, 'request' => $this->app->config->debug > 5 ? $result : null));
            }
            if($result['result'] == 'success')
            {
                $info->syncedCount++;
                $info->syncDetails->$syncingType->synced++;
            }
            else
            {
                $info->syncFailedCount++;
                $info->syncDetails->$syncingType->failed++;
            }
            $info->syncingID = (isset($result['id']) ? $result['id'] : $info->syncingID) + 1;
            $info->lastSync  = ['time' => (microtime(true) * 1000) - $startSyncTime, 'contentLength' => isset($result['syncedData']) ? strlen($result['syncedData']['content']) : 0, 'type' => $info->syncingType, 'id' => $info->syncingID];
        }
        else
        {
            $nextSyncType = zaiModel::getNextSyncType($syncingType);
            if(empty($nextSyncType))
            {
                $info->status      = 'synced';
                $info->syncedTime  = $info->syncTime;
                $info->syncingID   = 0;
                $info->syncingType = zaiModel::getNextSyncType();
            }
            else
            {
                $info->syncingType = $nextSyncType;
                $info->syncingID   = 0;
            }
        }
        $this->zai->setVectorizedInfo($info);

        unset($info->key);
        return $this->send(array('result' => 'success', 'data' => $info));
    }

    /**
     * Ajax: 搜索知识库。
     * Ajax: Search knowledge base.
     *
     * @param string $type 'chunk'（块） | 'content'（内容）
     * @param int    $limit
     * @access public
     * @return void
     */
    public function ajaxSearchKnowledges($type = 'chunk', $limit = 5)
    {
        if($_SERVER['REQUEST_METHOD'] !== 'POST')
        {
            return $this->send(array('result' => 'fail', 'message' => $this->lang->zai->onlyPostRequest));
        }

        $userPrompt = zget($_POST, 'userPrompt', '');
        $filters    = json_decode(zget($_POST, 'filters', '{}'), true);

        if(empty($userPrompt) || empty($filters)) return $this->send(array('result' => 'fail', 'message' => $this->lang->fail));

        $knowledges = $this->zai->searchKnowledgesInCollections($userPrompt, $filters, $type, $limit, 0.5);
        $results    = [];
        foreach($knowledges as $knowledge)
        {
            if($type === 'chunk')
            {
                $results[] = ['key' => $knowledge['content_key'], 'similarity' => $knowledge['similarity'], 'id' => $knowledge['chunk_id'], 'content' => $knowledge['chunk_content'], 'attrs' => $knowledge['content_attrs']];
            }
            else
            {
                $results[] = ['key' => $knowledge['key'], 'similarity' => $knowledge['similarity'], 'id' => $knowledge['id'], 'content' => $knowledge['content'], 'attrs' => $knowledge['attrs']];
            }
            if(count($results) >= $limit) break;
        }

        return $this->send(array('result' => 'success', 'data' => $results));
    }
}