107 lines
3.5 KiB
PHP
107 lines
3.5 KiB
PHP
<?php
|
||
|
||
namespace app\api\lists\chat;
|
||
|
||
use app\api\lists\BaseApiDataLists;
|
||
use app\common\lists\ListsSearchInterface;
|
||
use app\common\model\chat\Assistants;
|
||
use app\common\model\chat\ChatLog;
|
||
use app\common\enum\user\AccountLogEnum;
|
||
use app\common\model\chat\Scene;
|
||
|
||
class ChatLists extends BaseApiDataLists implements ListsSearchInterface
|
||
{
|
||
public function setSearch(): array
|
||
{
|
||
return [];
|
||
}
|
||
|
||
/**
|
||
* @notes 获取列表
|
||
* @return array
|
||
* @throws \think\db\exception\DataNotFoundException
|
||
* @throws \think\db\exception\DbException
|
||
* @throws \think\db\exception\ModelNotFoundException
|
||
* @author 段誉
|
||
* @date 2023/2/23 18:43
|
||
*/
|
||
public function lists(): array
|
||
{
|
||
$this->searchWhere[] = ['user_id', '=', $this->userId];
|
||
$sceneId = $this->request->get('scene_id');
|
||
if ($sceneId || $sceneId == 0) {
|
||
// 获取所有子集场景ID
|
||
$sceneIds = Scene::where('pid', $sceneId)->column('id');
|
||
if($sceneId == 0){
|
||
$assistantIds = [0];
|
||
}else{
|
||
$assistantIds = Assistants::whereIn('scene_id', $sceneIds)->column('id');
|
||
}
|
||
$this->searchWhere[] = ['assistant_id', 'in', $assistantIds];
|
||
}
|
||
|
||
// 首先获取所有不同的 task_id（按最新记录排序）
|
||
$taskIds = ChatLog::where($this->searchWhere)
|
||
->whereIn('chat_type', [AccountLogEnum::TOKENS_DEC_COMMON_CHAT, AccountLogEnum::TOKENS_DEC_SCENE_CHAT, AccountLogEnum::TOKENS_DEC_KNOWLEDGE_CHAT])
|
||
->group('task_id')
|
||
->order('update_time', 'desc')
|
||
->limit($this->limitOffset, $this->limitLength)
|
||
->column('task_id');
|
||
|
||
if (empty($taskIds)) {
|
||
return [];
|
||
}
|
||
|
||
$logList = [];
|
||
|
||
// 对每个 task_id 处理
|
||
foreach ($taskIds as $taskId) {
|
||
$logInfo = ChatLog::where('task_id', $taskId)
|
||
->order('id', 'asc')
|
||
->field('task_id,message,assistant_id,create_time,update_time')
|
||
->find()
|
||
->toArray();
|
||
|
||
// 场景名
|
||
if ($logInfo['assistant_id'] == 0) {
|
||
$logInfo['scene_name'] = Assistants::where('id', 1)->value('name');
|
||
} else {
|
||
$logInfo['scene_name'] = Assistants::where('id', $logInfo['assistant_id'])->value('name');
|
||
}
|
||
|
||
$logList[] = $logInfo;
|
||
}
|
||
|
||
return $logList;
|
||
}
|
||
|
||
/**
|
||
* @notes 获取数量
|
||
* @return int
|
||
* @author 段誉
|
||
* @date 2023/2/23 18:43
|
||
*/
|
||
public function count(): int
|
||
{
|
||
|
||
$this->searchWhere[] = ['user_id', '=', $this->userId];
|
||
$sceneId = $this->request->get('scene_id');
|
||
if ($sceneId || $sceneId == 0) {
|
||
// 获取所有子集场景ID
|
||
$sceneIds = Scene::where('pid', $sceneId)->column('id');
|
||
if($sceneId == 0){
|
||
$assistantIds = [0];
|
||
}else{
|
||
$assistantIds = Assistants::whereIn('scene_id', $sceneIds)->column('id');
|
||
}
|
||
$this->searchWhere[] = ['assistant_id', 'in', $assistantIds];
|
||
}
|
||
|
||
// 首先获取所有不同的 task_id（按最新记录排序）
|
||
return ChatLog::where($this->searchWhere)
|
||
->whereIn('chat_type', [AccountLogEnum::TOKENS_DEC_COMMON_CHAT, AccountLogEnum::TOKENS_DEC_SCENE_CHAT, AccountLogEnum::TOKENS_DEC_KNOWLEDGE_CHAT])
|
||
->group('task_id')
|
||
->count();
|
||
}
|
||
}
|