k_online_backend/service/robot.class.php

<?php

if (!defined('IN_ONLINE')) {
    exit('Access Denied');
}

/**
 * Description of robot
 *
 * @author Administrator
 */
class robot{

    private static $_robot;

    function __construct() {  
       
    }

    public static function getInstance() {
        global $memory_limit;
        $memory_limit = ini_get("memory_limit");
        ini_set("memory_limit", '2048M');
        if (!self::$_robot) {
            $c = __CLASS__;
            self::$_robot = new $c;
        }
        return self::$_robot;
    }
    public function robot_chat() {
        $operate = utils::_get('operate');
        $operate = strtolower($operate);

        if (empty($operate)) {
            $column = column::getInstance()->getDisplayColumn('Robot_Chat_Search');
            $OperationTableColumns = column::getInstance()->tableColumns('Robot_Chat_Search',$column);
            $data['OperationTableColumns'] = $OperationTableColumns;
            common::echo_json_encode(200,$data);
            exit();
        }

        if ($operate == "search") {
            //$this->_operation_search();
        }

        if ($operate == "excel") {
            $sql = common::deCode($_POST ['tmp_search'], 'D');
            $sql = substr($sql, 0, strripos($sql, " limit"));
            if(!empty($sql)){
                $rss = common::excuteListSql($sql);
            }
            foreach($rss as $key =>$val){
                $rss[$key]['Operation Time'] = $rss[$key]['_operation_time'];
            }
            common::echo_json_encode(200,array("msg"=>"success","Data" => $rss));
            exit;
        }
        
    }

    public function robot_prompt_configuration() {
        $operate = utils::_get('operate');
        $operate = strtolower($operate);

        if (empty($operate)) {
           //查询所有配置信息 目前只有一个表一个数据
            $configuration = common::excuteObjectSql("select id, character_name,professional_field,main_tasks,table_name,table_description,
                table_structure_configuration,response_rule_configuration,output_format_configuration,fixed_problem_configuration
            from public.kln_robot_prompt_configuration where id = 1");

            $PromptAndVue = $this->dealPromptFormatAndVue($configuration);
            //tableDat-VUE 
            $tableData = $PromptAndVue["vuedata"]["tableData"];
            //stepData -VUE
            $stepData = $PromptAndVue["vuedata"]["stepData"]; 
            //formatList -VUE
            $formatListType  = $PromptAndVue["vuedata"]["formatListType"];
            $formatList = $PromptAndVue["vuedata"]["formatList"];
            //promptFormat
            $promptFormat =$PromptAndVue["promptFormat"];

            //获取历史变更记录
            $old_configuration = $this->getChangedLog($configuration["id"]);

            $id = $configuration['id'];
            $character_name = $configuration['character_name'];
            $professional_field = $configuration['professional_field'];
            $main_tasks =  $configuration['main_tasks'];
            $table_name = $configuration['table_name'];
            $table_description = $configuration['table_description'];

            $prompt_summary = array("您是专門从事".$professional_field."的".$character_name."...",
                "表名：{".$table_name."}...",
                "响应规则，请按以下步骤处理...");

            $data = array(
                "id"=>$id,
                "character_name"=>$character_name,
                "professional_field"=>$professional_field,
                "main_tasks"=>$main_tasks,
                "table_name"=>$table_name,
                "table_description"=>$table_description,
                "tableData"=>$tableData,
                "stepData"=>$stepData,
                "formatList"=>$formatList,
                "formatListType"=>$formatListType,
                "complete_prompt"=>$promptFormat,
                "prompt_summary"=>$prompt_summary,
                "prompt_log_record"=>$old_configuration
            );
            $data = utils::arrayRemoveNull($data);
            common::echo_json_encode(200,$data);                
            exit(); 
        }
        if ($operate == "save") {
            $id =common::check_input($_POST['id']);
            $character_name = common::check_input($_POST['role_name']);
            $professional_field = common::check_input($_POST['professional_field']);
            $main_tasks = common::check_input($_POST['main_tasks']);
            $table_name = common::check_input($_POST['table_name']);
            $table_description = common::check_input($_POST['table_description']);
            $outputvalue = common::check_input($_POST['outputvalue']);

            $tableDataList = $_POST['tableDataList'];
            $stepData = $_POST['stepData'];
            $formatList = $_POST['formatList'];


            $table_structure_configuration = common::check_input(json_encode($tableDataList));
            $response_rule_configuration = common::check_input(json_encode($stepData));

            $output_data = array("output_type"=>$outputvalue,"data"=>$formatList);
            $output_format_configuration = common::check_input(json_encode($output_data));

            if(empty($id)){
                $sql = "INSERT INTO public.kln_robot_prompt_configuration(
                        character_name, professional_field, main_tasks, table_name, 
                        table_description, table_structure_configuration, response_rule_configuration, 
                        output_format_configuration, created_by, created_time)
                VALUES ('$character_name', '$professional_field', '$main_tasks', '$table_name', 
                        '$table_description', '$table_structure_configuration', '$response_rule_configuration', 
                        '$output_format_configuration', '"._getLoginName()."', now());";
            }else{
                $sql = "update public.kln_robot_prompt_configuration 
                    set character_name = '$character_name',professional_field ='$professional_field', main_tasks = '$main_tasks',
                        table_name = '$table_name',table_description = '$table_description',
                        table_structure_configuration = '$table_structure_configuration',
                        response_rule_configuration = '$response_rule_configuration',
                        output_format_configuration = '$output_format_configuration'
                where id = '$id';";  

                //插入变更日志记录
                $sql .= "INSERT INTO public.kln_robot_prompt_configuration_log(ref_id, character_name, professional_field, main_tasks, table_name, 
                        table_description, table_structure_configuration, response_rule_configuration, 
                        output_format_configuration, created_by, created_time)
                VALUES ('$id','$character_name', '$professional_field', '$main_tasks', '$table_name', 
                        '$table_description', '$table_structure_configuration', '$response_rule_configuration', 
                        '$output_format_configuration', '"._getLoginName()."', now());";

            }
            error_log($sql);
            $rs = common::excuteUpdateSql($sql);
            if (!$rs) {
                $data = array("msg" => "save Error");
            } else {
                $data = array("msg" => "save Successful");
            }
            $data = array("msg" => "save Successful");
            common::echo_json_encode(200,$data);
            exit();
        }

        if($operate == "preview_propmpt_witout_save"){
            $character_name = common::check_input($_POST['role_name']);
            $professional_field = common::check_input($_POST['professional_field']);
            $main_tasks = common::check_input($_POST['main_tasks']);
            $table_name = common::check_input($_POST['table_name']);
            $table_description = common::check_input($_POST['table_description']);
            $outputvalue = common::check_input($_POST['outputvalue']);

            $tableDataList = $_POST['tableDataList'];
            $stepData = $_POST['stepData'];
            $formatList = $_POST['formatList'];
            $table_structure_configuration = common::check_input(json_encode($tableDataList));
            $response_rule_configuration = common::check_input(json_encode($stepData));
            $output_data = array("output_type"=>$outputvalue,"data"=>$formatList);
            $output_format_configuration = common::check_input(json_encode($output_data));

            //组合拼接需要的数据
            $configuration = array("character_name"=>$character_name,
                "professional_field"=>$professional_field,
                "main_tasks"=>$main_tasks,
                "table_name"=>$table_name,
                "table_description"=>$table_description,
                "professional_field"=>$professional_field,
                "table_structure_configuration"=>$table_structure_configuration,
                "response_rule_configuration" =>$response_rule_configuration,
                "output_format_configuration" =>$output_format_configuration
             );
            $PromptAndVue = $this->dealPromptFormatAndVue($configuration);
            //promptFormat
            $promptFormat =$PromptAndVue["promptFormat"];
            $character_name = empty($configuration['character_name']) ? "" :$configuration['character_name'];
            $professional_field = empty($configuration['professional_field']) ? "" :$configuration['professional_field'];
            $main_tasks =  empty($configuration['main_tasks']) ? "" :$configuration['main_tasks'];
            $table_name = empty($configuration['table_name']) ? "" :$configuration['table_name'];
            $table_description = empty($configuration['table_description']) ? "" :$configuration['table_description'];
            $prompt_summary = array("您是专門从事".$professional_field."的".$character_name."...",
                "表名：{".$table_name."}...",
                "响应规则，请按以下步骤处理...");

            $data = array(
                "complete_prompt"=>$promptFormat,
                "prompt_summary"=>$prompt_summary
            );
            common::echo_json_encode(200,$data);                
            exit();
        }

        if($operate == "test_with_ds_claude"){
            //分担查询
            include ONLINE_ROOT . 'libs' . DS . 'map_config.ini.php';

            $ai_method = $_POST["test_ai_method"];
            $systemPrompt = $_POST["prompt"];
            $message = $_POST["test_question"];
            $model = $ai_method == 'DS' ? "deepseek" : "claude";
            //question_date request_time
            $request_question_time = date("Y-m-d H:i:s");
            $question_time = date("H:i:s");

            $item_value = common::excuteOneSql("select item_value from public.config where item = 'AIAPISetting'");
            $config = json_decode($item_value,true);
            $response = AIClientFactory::create($model, $config[$model], $systemPrompt, $message, $history = []);
            $response_time = date("Y-m-d H:i:s");

            $input_token = "";
            $output_token = "";
            $date1 = new DateTime($request_question_time);
            $date2 = new DateTime($response_time);
            $response_duration = $date2->getTimestamp() - $date1->getTimestamp();

            //回答问题类型，在没有超时的情况下，固定:AI回答
            $answer_type = "AI回答";
            if ($response_duration > 120){
                $answer_type = "回答超时";
            }
            //获取自然序列
            $sequence = common::getChatAiSequence();
            $name = $ai_method == 'DS' ? "DS" : "CD";   
            $request_id = "R".$name."".date("Ymd").$sequence; 
            $question_id = "Q"."".date("Ymd").$sequence; 

            //处理一下参数
            $user_name = common::check_input(_getLoginName());
            $user_type = _isApexLogin() ? "employee" : "customer";
            //这里的测试，固定写自由文本
            $question_type = 'free text';
            $question_id = common::check_input($question_id);
            $question_content = common::check_input($message);
            $answer_type = common::check_input($answer_type);
            //记录回答展示给客户的具体内容
            $answer = common::check_input(common::getChatAimessage($response['message']));

            $ai_model = common::check_input($model);
            $input_token = common::check_input($input_token);
            $request_content = common::check_input(json_encode($response['data']));
            $ai_response_content = common::check_input(json_encode($response['full_response']));
            $output_token = common::check_input($output_token);

            $answer_template = "";
            $message = json_decode(common::getChatAimessage($response['message']),true);
            if($message["can_query"] == "true" && !empty($message["sql"])){
                $reference = $message["reference"];
                $answer_template = common::check_input($message["reference"]);
                $sql = $message["sql"];
                //拆分sql 存在多条的情况
                $sqlArr = explode(";", $sql);
                //给所有sql 拼接用户权限
                $sqlWhere = '  ' . common::searchExtendHand_KLN("ocean", $_SESSION["ONLINE_USER"]);
                //先不考虑总数，处理有难度
                foreach($sqlArr as $_sql){
                    if(empty($_sql)){
                        continue;
                    }
                    //根据public.kln_ocean 和 WHERE 的位置关系，带入权限
                    $new_sql = utils::modifyString($_sql,$sqlWhere);
                    //处理limit 超过10 先限制10
                    $new_sql = utils::processLimitClause($new_sql);
                    //$rs = common::excuteListSql($new_sql);
                    $rs =  $mapdb->GetAll($new_sql) or ( (!$mapdb->ErrorMsg()) or error_log(common::dbLog($mapdb, $new_sql), 0));
                    $reference = utils::replacementsMultiline($rs,$reference);
                }
                //有就去掉{{#each hbol_list}}\n  {{/each}}
                $reference = preg_replace([
                    '/\{\{[^}]+\}\}\n/',  // 匹配开始标签及换行
                    '/\{\{\/[^}]+\}\}/'   // 匹配结束标签
                ], '', $reference);

                //$reference = $reference.$message["reference"];
                $answer = $reference;
                //$response['message'] = json_decode(common::getChatAimessage($response['message']),true);
                //$answer = $response;
            }else{
                $answer = $message["response"];
            }

            //最后插入数据
            $sql = "INSERT INTO public.kln_robot_chat_log( question_id, user_name, user_type, question_type, question_content, 
                        answer_type, answer, answer_satisfication, answer_duration,answer_template, question_date, 
                        question_time, request_id, ai_model, request_content, input_token, 
                        ai_response_content, output_token, request_time, response_time, 
                        response_duration)
                VALUES ('$question_id', '$user_name', '$user_type', '$question_type', '$question_content', 
                        '$answer_type', '".common::check_input($answer)."', '', '$response_duration','$answer_template','$request_question_time', 
                        '$question_time', '$request_id', '$ai_model','$request_content',  '$input_token', 
                        '$ai_response_content', '$output_token', '$request_question_time', '$response_time', 
                        '$response_duration');";
            $rs = common::excuteUpdateSql($sql);
            if (!$rs) {
                $data = array("msg" => "AI Chat Save Error");
                common::echo_json_encode(200,$data);
                exit();
            }

            $return = array("type"=>"markdown","data" =>$answer);
            common::echo_json_encode(200,$return);            
            exit(); 
        }

        if ($operate == "download"){
            $sql = common::deCode($_REQUEST['tmp_search'], 'D');
            if(!empty($sql)){
                $rs = common::excuteListSql($sql);
            }
            common::echo_json_encode(200,array("msg"=>"success","Data" => $rs));
            exit;
        }
    }

    public function dealPromptFormatAndVue($configuration){
        //tableDat-VUE 
        $tableData = array();
        $table_structure_text="[";
        $table_structure_configuration = json_decode($configuration['table_structure_configuration'],true);
        foreach($table_structure_configuration as $tk =>$tv){
            $tableData[] = $tv;

            $table_structure_text .='{  
                "name": "'.$tv["name"].'",  
                "type": "'.$tv["type"].'",  
                "description": "'.$tv["description"].'", 
                "exampledata": "'.$tv["exampledata"].'", 
            }';
            if(count($table_structure_configuration)-1 > $tk){
                $table_structure_text .=",";
            }
            $table_structure_text .=common::splicedLlineBreaks();
        }
        $table_structure_text.="]";

        //stepData -VUE
        $stepData = array(); 
        $response_rule_text = ""; 
        $response_rule_configuration = json_decode($configuration['response_rule_configuration'],true);
        foreach($response_rule_configuration as $rk =>$rv){
            $tempTable = array();
            $tempTable["description"] = $rv["description"];
            $stepData[] = $tempTable;

            $response_rule_text .=($rk+1).'.'.$rv["description"];
            $response_rule_text .=common::splicedLlineBreaks();
        }

        //formatList -VUE
        $output_format_text = "{"; 
        $output_format_configuration = json_decode($configuration['output_format_configuration'],true);
        $formatListType = $output_format_configuration['output_type'];
        $formatList = empty($output_format_configuration['data']) ? array(): $output_format_configuration['data'];
        foreach($formatList as $ok =>$ov){
            $output_format_text .="\"".$ov['name']."\"".": "."\"".$ov['describe']."\"";
            $output_format_text .=common::splicedLlineBreaks();
        }
        $output_format_text .= "}"; 

        $promptFormat = $this->getCompletePromptFormat();
        $promptFormat = str_replace('<{professional_field}>', $configuration['character_name'], $promptFormat);
        $promptFormat = str_replace('<{character_name}>', $configuration['character_name'], $promptFormat);
        $promptFormat = str_replace('<{main_tasks}>', $configuration['main_tasks'], $promptFormat);
        $promptFormat = str_replace('<{table}>', $configuration['table_name'], $promptFormat);
        $promptFormat = str_replace('<{table_structure_configuration}>', $table_structure_text, $promptFormat);
        $promptFormat = str_replace('<{response_rule_configuration}>', $response_rule_text, $promptFormat);
        $promptFormat = str_replace('<{number}>', count($response_rule_configuration)+1, $promptFormat);
        $promptFormat = str_replace('<{output_type}>', $formatListType, $promptFormat);
        $promptFormat = str_replace('<{output_format_configuration}>', $output_format_text, $promptFormat);

        return array("vuedata"=>array("tableData"=>$tableData,"stepData"=>$stepData,"formatList"=>$formatList,"formatListType"=>$formatListType),
            "promptFormat" =>$promptFormat);
    }

    public function getChangedLog($id){
        $changeLogList = array();
        $configurationArr = common::excuteListSql("select id, character_name,professional_field,main_tasks,table_name,table_description,
             created_by,TO_CHAR(created_time, 'YYYY-mm-dd HH24:MI:SS') as created_time,
            table_structure_configuration,response_rule_configuration,output_format_configuration,fixed_problem_configuration
        from public.kln_robot_prompt_configuration_log where  ref_id = '$id' order by id desc");

        foreach($configurationArr as $ca){
            $data = $this->dealPromptFormatAndVue($ca);
            $temp =array();
            $temp["time"] = $ca['created_time']." Prompt";
            $temp["person"] ="提交人：".$ca['created_by'];
            $temp["text"] = $data['promptFormat'];

            $changeLogList[] = $temp;
        }

        return $changeLogList;
    }

    public function getCompletePromptFormat() {
        $formatTpl = "
您是专门从<{professional_field}>的 <{character_name}>，请始终以<{output_type}>格式响应。您的主要任务是<{main_tasks}>

表名：<{table}>  
表结构：  \"columns\": <{table_structure_configuration}>

 
    响应规则，请按以下步骤处理：  
    <{response_rule_configuration}>  
    <{number}>. 请用以下<{output_type}>格式回复:     
    <{output_format_configuration}>";
        return $formatTpl;
    }
}

?>