604 lines
18 KiB
PHP
604 lines
18 KiB
PHP
<?php
|
|
|
|
namespace app\api\logic;
|
|
|
|
use app\api\logic\service\TokenLogService;
|
|
use app\common\enum\user\AccountLogEnum;
|
|
use app\common\logic\AccountLogLogic;
|
|
use app\common\model\chat\Assistants;
|
|
use app\common\model\chat\ChatLog;
|
|
use app\common\model\user\User;
|
|
use app\common\model\ChatPrompt;
|
|
use app\common\service\FileService;
|
|
use app\common\model\file\File;
|
|
use app\common\model\mindMap\MindMap;
|
|
|
|
class ChatLogic extends ApiLogic
|
|
{
|
|
|
|
const COMMON_CHAT = 'common_chat'; //通用聊天
|
|
const SCENE_CHAT = 'scene_chat'; //场景聊天
|
|
|
|
|
|
/**
|
|
* @desc 通用聊天
|
|
* @param array $params
|
|
* @return void
|
|
*/
|
|
public static function commonChat(array $params)
|
|
{
|
|
|
|
if (empty($params['message'])) {
|
|
message('参数错误');
|
|
}
|
|
|
|
$request['message'] = $params['message'];
|
|
$request['open_reasoning'] = $params['open_reasoning'] ?? 0;
|
|
$request['stream'] = true;
|
|
|
|
// 存在文件 TODO
|
|
if (isset($params['file_id'])) {
|
|
}
|
|
|
|
$logs = [];
|
|
|
|
if (isset($params['task_id']) && $params['task_id']) {
|
|
$request['task_id'] = $params['task_id'];
|
|
|
|
// 对话记录
|
|
$logs = self::chatLog($request['task_id'], 0, self::$uid);
|
|
|
|
if (!$logs) {
|
|
|
|
message('对话记录ID错误');
|
|
}
|
|
} else {
|
|
|
|
$request['task_id'] = generate_unique_task_id();
|
|
}
|
|
|
|
$messages = array_merge($logs, [
|
|
[
|
|
'role' => 'user',
|
|
'content' => $params['message']
|
|
]
|
|
]);
|
|
|
|
$request['messages'] = $messages;
|
|
|
|
self::requestChatUrl($request, self::COMMON_CHAT, self::$uid);
|
|
|
|
exit;
|
|
}
|
|
|
|
/**
|
|
* @desc 获取通用聊天助手信息
|
|
* @return bool
|
|
*/
|
|
public static function commonChatInfo(): bool
|
|
{
|
|
try {
|
|
$assistant = Assistants::where('id', 1)->findOrEmpty();
|
|
|
|
if ($assistant->isEmpty()) {
|
|
throw new \Exception("助手不存在");
|
|
}
|
|
$preliminary_ask = json_decode($assistant->preliminary_ask, true) ?? [];
|
|
$extra = json_decode($assistant->extra ?? '', true) ?? [];
|
|
|
|
foreach ($preliminary_ask as $key => $value) {
|
|
|
|
if (isset($value['logo'])) {
|
|
|
|
$preliminary_ask[$key]['logo'] = FileService::getFileUrl($value['logo']);
|
|
}
|
|
}
|
|
|
|
if (isset($extra['banner'])) {
|
|
$extra['banner'] = FileService::getFileUrl($extra['banner']);
|
|
}
|
|
|
|
$assistant->preliminary_ask = $preliminary_ask;
|
|
$assistant->logo = FileService::getFileUrl($assistant['logo']);
|
|
$assistant->banner = $extra['banner'] ?? '';
|
|
$assistant->new_chat_prompt = $extra['new_chat_prompt'] ?? '';
|
|
$assistant->file_prompt = $extra['file_prompt'] ?? '';
|
|
$assistant->extra = $extra;
|
|
self::$returnData = $assistant->toArray();
|
|
return true;
|
|
} catch (\Throwable $e) {
|
|
self::$error = $e->getMessage();
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @desc 获取场景聊天 - 助理信息
|
|
* @param array $params
|
|
* @return bool
|
|
*/
|
|
public static function sceneChatInfo(array $params): bool
|
|
{
|
|
try {
|
|
$assistant = Assistants::where('id', $params['assistant_id'])->findOrEmpty();
|
|
|
|
if ($assistant->isEmpty()) {
|
|
throw new \Exception("助手不存在");
|
|
}
|
|
|
|
$assistant->template_info = json_decode($assistant->template_info, true) ?? [];
|
|
$preliminary_ask = json_decode($assistant->preliminary_ask, true) ?? [];
|
|
|
|
foreach ($preliminary_ask as $key => $value) {
|
|
|
|
if (isset($value['logo'])) {
|
|
|
|
$preliminary_ask[$key]['logo'] = FileService::getFileUrl($value['logo']);
|
|
}
|
|
}
|
|
|
|
$assistant->preliminary_ask = $preliminary_ask;
|
|
self::$returnData = $assistant->toArray();
|
|
return true;
|
|
} catch (\Throwable $e) {
|
|
self::$error = $e->getMessage();
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @desc 场景聊天
|
|
* @param array $params
|
|
* @return true
|
|
*/
|
|
public static function sceneChat(array $params): bool
|
|
{
|
|
|
|
if (empty($params['message']) && empty($params['message_ext'])) {
|
|
message('参数错误');
|
|
}
|
|
|
|
// 获取 场景聊天 - 助理信息
|
|
$assistant = Assistants::where('id', $params['assistant_id'])->findOrEmpty();
|
|
|
|
if ($assistant->isEmpty()) {
|
|
message('助手不存在');
|
|
}
|
|
|
|
$message = $params['message'];
|
|
|
|
// 表单变量替换
|
|
$message_ext = $params['message_ext'] ?? '';
|
|
|
|
if ($message_ext) {
|
|
$message_ext_text = self::parseMsg($message_ext, $assistant['form_info']);
|
|
$message = $message_ext_text . $message;
|
|
}
|
|
|
|
$logs = [];
|
|
|
|
if (isset($params['task_id']) && $params['task_id']) {
|
|
$taskId = $params['task_id'];
|
|
|
|
// 对话记录
|
|
$logs = self::chatLog($taskId, $assistant->id, self::$uid);
|
|
|
|
if (!$logs) {
|
|
|
|
message('对话记录ID错误');
|
|
}
|
|
} else {
|
|
|
|
$taskId = generate_unique_task_id();
|
|
}
|
|
|
|
$request = self::assembleAssistantRequest($assistant->toArray(), $message, $logs);
|
|
|
|
$request['message'] = $message;
|
|
$request['task_id'] = $taskId;
|
|
|
|
// 存在文件 TODO
|
|
if (isset($params['file_id'])) {
|
|
}
|
|
|
|
self::requestChatUrl($request, self::SCENE_CHAT, self::$uid);
|
|
|
|
exit;
|
|
}
|
|
|
|
|
|
/**
|
|
* @desc 提示词聊天
|
|
* @param array $params
|
|
* @return true
|
|
*/
|
|
public static function promptChat(array $params): bool
|
|
{
|
|
|
|
if (empty($params['message'])) {
|
|
|
|
message('参数错误');
|
|
}
|
|
|
|
//获取提示词
|
|
$prompt = ChatPrompt::where('id', $params['prompt_id'])->value('prompt_text') ?? '';
|
|
|
|
if (!$prompt) {
|
|
|
|
message("提示词不存在");
|
|
}
|
|
|
|
//获取场景
|
|
switch ($params['prompt_id']) {
|
|
case 1: //数字人口播
|
|
$scene = 'human_prompt';
|
|
$scene_type = AccountLogEnum::TOKENS_DEC_HUMAN_PROMPT;
|
|
break;
|
|
case 2: //思维导图
|
|
$scene = 'mind_map';
|
|
$scene_type = AccountLogEnum::TOKENS_DEC_MIND_MAP;
|
|
break;
|
|
case 3: //AI画图 - 文生图
|
|
case 4: //AI画图 - 图生图
|
|
case 5: //AI画图 - 商品图
|
|
$scene = 'image_prompt';
|
|
$scene_type = AccountLogEnum::TOKENS_DEC_IMAGE_PROMPT;
|
|
break;
|
|
}
|
|
|
|
$request = [
|
|
"messages" => [
|
|
[
|
|
'role' => "system",
|
|
'content' => $prompt
|
|
],
|
|
[
|
|
'role' => "user",
|
|
'content' => $params['message']
|
|
]
|
|
],
|
|
'stream' => false,
|
|
'message' => $params['message'],
|
|
'task_id' => generate_unique_task_id(),
|
|
'user_id' => self::$uid,
|
|
'assistant_id' => 0,
|
|
'chat_type' => $scene_type,
|
|
'now' => time(),
|
|
];
|
|
|
|
$unit = TokenLogService::checkToken(self::$uid, $scene);
|
|
|
|
if($scene == 'human_prompt'){
|
|
$request['open_reasoning'] = 5;
|
|
}
|
|
|
|
$response = \app\common\service\ToolsService::Chat()->message($request);
|
|
|
|
$reply = $response['data']['choices'][0]['message']['content'] ?? '';
|
|
|
|
//计费
|
|
$tokens = $response['data']['usage']['total_tokens'] ?? 0;
|
|
|
|
if (!$reply || $tokens == 0) {
|
|
|
|
message('获取内容失败');
|
|
}
|
|
|
|
$response = [
|
|
'reply' => $reply,
|
|
'usage_tokens' => $response['data']['usage'] ?? [],
|
|
];
|
|
|
|
// 保存聊天记录
|
|
self::saveChatResponseLog($request, $response);
|
|
|
|
//计算消耗tokens
|
|
$points = $unit > 0 ? ceil($tokens / $unit) : 0;
|
|
|
|
//token扣除
|
|
User::userTokensChange(self::$uid, $points);
|
|
|
|
$extra = ['总消耗tokens数' => $tokens, '算力单价' => $unit, '实际消耗算力' => $points];
|
|
|
|
//扣费记录
|
|
AccountLogLogic::recordUserTokensLog(true, self::$uid, $scene_type, $points, $request['task_id'], $extra);
|
|
|
|
if ($scene_type == AccountLogEnum::TOKENS_DEC_MIND_MAP) {
|
|
|
|
self::$returnData = MindMap::create([
|
|
'user_id' => self::$uid,
|
|
'task_id' => $request['task_id'],
|
|
'ask' => $request['message'],
|
|
'reply' => $reply,
|
|
'task_time' => time() - $request['now'],
|
|
])->toArray();
|
|
} else {
|
|
|
|
self::$returnData = [
|
|
'reply' => $reply,
|
|
];
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* @desc 聊天记录
|
|
* @param array $params
|
|
* @return true
|
|
*/
|
|
public static function chatLogs(array $params)
|
|
{
|
|
try {
|
|
|
|
$logList = [];
|
|
|
|
ChatLog::where('user_id', self::$uid)
|
|
->where('assistant_id', $params['assistant_id'])
|
|
->whereIn('chat_type', [AccountLogEnum::TOKENS_DEC_COMMON_CHAT, AccountLogEnum::TOKENS_DEC_SCENE_CHAT, AccountLogEnum::TOKENS_DEC_KNOWLEDGE_CHAT])
|
|
->where('task_id', $params['task_id'])
|
|
->field('id,user_id,task_id,assistant_id,message,reasoning_content,usage_tokens,reply,file_ids,create_time')
|
|
->order('id asc')->select()
|
|
->each(function ($item) use (&$logList) {
|
|
|
|
// 文件处理
|
|
$files = [];
|
|
if (!empty($item['file_ids'])) {
|
|
$ids = json_decode($item['file_ids'], true);
|
|
foreach ($ids as $id) {
|
|
$file = File::where('id', $id)->value('uri') ?? '';
|
|
if ($file) {
|
|
$files[] = FileService::getFileUrl($file);
|
|
}
|
|
}
|
|
}
|
|
|
|
$user_avatar = User::where('id', $item['user_id'])->value('avatar') ?? '';
|
|
$assistants_avatar = Assistants::where('id', $item['assistant_id'] ?: 1)->value('logo') ?? '';
|
|
|
|
$logList[] = [
|
|
'avatar' => FileService::getFileUrl($user_avatar),
|
|
'message' => $item['message'],
|
|
'type' => 1,
|
|
'create_time' => $item['create_time'],
|
|
'file_urls' => $files,
|
|
'tokens_info' => $item['usage_tokens']
|
|
];
|
|
|
|
$logList[] = [
|
|
'avatar' => FileService::getFileUrl($assistants_avatar),
|
|
'reply' => $item['reply'],
|
|
'reasoning_content' => $item['reasoning_content'],
|
|
'type' => 2,
|
|
'create_time' => $item['create_time'],
|
|
'tokens_info' => $item['usage_tokens']
|
|
];
|
|
});
|
|
|
|
self::$returnData = $logList;
|
|
return true;
|
|
} catch (\Throwable $e) {
|
|
self::$error = $e->getMessage();
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @desc 删除聊天记录
|
|
* @param array $params
|
|
* @return true
|
|
*/
|
|
public static function deleteChat(array $params): bool
|
|
{
|
|
try {
|
|
ChatLog::where('task_id', $params['task_id'])->where('user_id', self::$uid)->select()->delete();
|
|
return true;
|
|
} catch (\Throwable $e) {
|
|
self::$error = $e->getMessage();
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @desc 保存聊天记录
|
|
* @return void
|
|
* @date 2024/6/27 9:30
|
|
* @author dagouzi
|
|
*/
|
|
public static function saveChatResponseLog(array $request, array $response = [])
|
|
{
|
|
try {
|
|
$chatLogData = [
|
|
'user_id' => $request['user_id'],
|
|
'task_id' => $request['task_id'],
|
|
'assistant_id' => $request['assistant_id'] ?? 0,
|
|
'message' => $request['message'] ?? ($request['prompt'] ?? ''),
|
|
'reply' => $response['reply'],
|
|
'chat_type' => $request['chat_type'],
|
|
'usage_tokens' => $response['usage_tokens'] ?? [],
|
|
'reasoning_content' => $response['reasoning_content'] ?? null,
|
|
'file_ids' => !empty($request['file_id']) ? json_encode($request['file_id']) : '',
|
|
'task_time' => time() - $request['now'],
|
|
];
|
|
ChatLog::create($chatLogData);
|
|
} catch (\Throwable $e) {
|
|
message($e->getMessage(), 1);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @desc tokens计费
|
|
* @return void
|
|
* @date 2024/12/17 10:46
|
|
* @author dagouzi
|
|
*/
|
|
public static function chatTokensCharge($request, $tokens): void
|
|
{
|
|
|
|
[$tokenScene, $tokenCode] = match ($request['chat_type']) {
|
|
AccountLogEnum::TOKENS_DEC_COMMON_CHAT => ['common_chat', AccountLogEnum::TOKENS_DEC_COMMON_CHAT],
|
|
AccountLogEnum::TOKENS_DEC_SCENE_CHAT => ['scene_chat', AccountLogEnum::TOKENS_DEC_SCENE_CHAT],
|
|
AccountLogEnum::TOKENS_DEC_KNOWLEDGE_CHAT => ['knowledge_chat', AccountLogEnum::TOKENS_DEC_KNOWLEDGE_CHAT],
|
|
};
|
|
|
|
$unit = TokenLogService::getTypeScore($tokenScene);
|
|
|
|
//计算消耗tokens
|
|
$points = $unit > 0 ? ceil($tokens / $unit) : 0;
|
|
|
|
//token扣除
|
|
User::userTokensChange($request['user_id'], $points);
|
|
|
|
if($request['chat_type'] == AccountLogEnum::TOKENS_DEC_KNOWLEDGE_CHAT){
|
|
$extra = ['总消耗tokens数' => $tokens, '知识库消耗tokens数' => $request['knowledge_tokens'], '算力单价' => $unit, '实际消耗算力' => $points];
|
|
}else{
|
|
$extra = ['总消耗tokens数' => $tokens, '算力单价' => $unit, '实际消耗算力' => $points];
|
|
}
|
|
|
|
|
|
//扣费记录
|
|
AccountLogLogic::recordUserTokensLog(true, $request['user_id'], $tokenCode, $points, $request['task_id'], $extra);
|
|
}
|
|
|
|
/**
|
|
* 获取聊天记录
|
|
* @param string $taskId
|
|
* @param int $assistantId
|
|
* @param int $userId
|
|
* @param int $limit
|
|
* @return array
|
|
*/
|
|
private static function chatLog(string $taskId, int $assistantId, int $userId, int $limit = 10): array
|
|
{
|
|
$logs = [];
|
|
|
|
// 获取指定 taskId 的所有记录，按 id 升序排序
|
|
$ids = ChatLog::where('task_id', $taskId)
|
|
->where('assistant_id', $assistantId)
|
|
->where('user_id', $userId)
|
|
->order('id', 'desc')
|
|
->limit($limit)
|
|
->column('id');
|
|
|
|
ChatLog::whereIn('id', $ids)
|
|
->order('id', 'asc')
|
|
->field('message,reply')
|
|
->select()
|
|
->each(function ($item) use (&$logs) {
|
|
$logs[] = [
|
|
'role' => 'user',
|
|
'content' => $item->message
|
|
];
|
|
|
|
$logs[] = [
|
|
'role' => 'assistant',
|
|
'content' => $item->reply
|
|
];
|
|
});
|
|
|
|
return $logs;
|
|
}
|
|
|
|
/**
|
|
* 助手参数
|
|
* @param array $assistant
|
|
* @return array
|
|
*/
|
|
private static function assembleAssistantRequest(array $assistant, string $message, array $logs = []): array
|
|
{
|
|
|
|
// 系统提示词
|
|
$messages = [
|
|
[
|
|
'role' => 'system',
|
|
'content' => $assistant['instructions']
|
|
],
|
|
];
|
|
|
|
// 对话轮数
|
|
$messages = array_merge($messages, $logs, [
|
|
[
|
|
'role' => 'user',
|
|
'content' => $message
|
|
]
|
|
]);
|
|
|
|
return [
|
|
'temperature' => $assistant['temperature'] ?? 1,
|
|
'top_p' => $assistant['top_p'] ?? 1,
|
|
'stream' => true,
|
|
'assistant_id' => $assistant['id'],
|
|
'messages' => $messages,
|
|
];
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
* @desc 解析表单变量
|
|
* @param $message_ext
|
|
* @param $form_info
|
|
* @return array|string|string[]
|
|
* @date 2024/7/2 10:14
|
|
* @author dagouzi
|
|
*/
|
|
private static function parseMsg($message_ext, $form_info)
|
|
{
|
|
$message_ext = json_decode($message_ext, true);
|
|
if (empty($message_ext)) {
|
|
return '';
|
|
}
|
|
preg_match_all('/\${([^\}]+)}/u', $form_info, $matches);
|
|
$keys = $matches[1];
|
|
if (empty($keys)) {
|
|
return '';
|
|
}
|
|
foreach ($message_ext as $key => $value) {
|
|
foreach ($keys as $keyword) {
|
|
if ($keyword == $key) {
|
|
if (!empty($value) && is_array($value)) {
|
|
$value = implode(',', $value);
|
|
}
|
|
$form_info = str_replace('${' . $keyword . '}', $value, $form_info);
|
|
}
|
|
}
|
|
}
|
|
return $form_info;
|
|
}
|
|
|
|
|
|
/**
|
|
* 请求上游接口与计费
|
|
* @param array $request
|
|
* @param string $scene
|
|
* @param int $userId
|
|
* @return void
|
|
* @throws \Exception
|
|
*/
|
|
private static function requestChatUrl(array $request, string $scene, int $userId): void
|
|
{
|
|
|
|
[$tokenScene, $tokenCode] = match ($scene) {
|
|
self::COMMON_CHAT => ['common_chat', AccountLogEnum::TOKENS_DEC_COMMON_CHAT],
|
|
self::SCENE_CHAT => ['scene_chat', AccountLogEnum::TOKENS_DEC_SCENE_CHAT],
|
|
};
|
|
|
|
//检查用户token
|
|
TokenLogService::checkToken($userId, $tokenScene);
|
|
|
|
$requestService = \app\common\service\ToolsService::Chat();
|
|
|
|
|
|
$request['user_id'] = $userId;
|
|
$request['chat_type'] = $tokenCode;
|
|
$request['now'] = time();
|
|
|
|
if ($scene == self::COMMON_CHAT) {
|
|
|
|
$requestService->message($request);
|
|
} else {
|
|
$requestService->sceneMessage($request);
|
|
}
|
|
}
|
|
}
|