Files
cunkebao_v3/Server/application/cozeai/controller/ConversationController.php
2025-04-03 15:53:24 +08:00

373 lines
12 KiB
PHP
Executable File

<?php
namespace app\cozeai\controller;
use app\cozeai\model\Conversation as ConversationModel;
use app\cozeai\model\Message as MessageModel;
use think\facade\Env;
/**
* Coze AI 对话控制器
*/
class ConversationController extends BaseController
{
/**
* 保存对话数据到数据库
* @param array $conversation 对话数据
* @param string $bot_id 机器人ID
*/
private function saveConversation($conversation, $bot_id)
{
if (empty($conversation['id'])) {
return false;
}
// 检查是否已存在
$exists = ConversationModel::where('conversation_id', $conversation['id'])->find();
$meta_data = $conversation['meta_data'] ?? [];
if (!$exists) {
// 不存在则插入
$data = [
'conversation_id' => $conversation['id'],
'bot_id' => $bot_id,
'created_at' => $conversation['created_at'],
'meta_data' => $meta_data,
'create_time' => time(),
'update_time' => time()
];
if(isset($meta_data['uid']) && !empty($meta_data['uid'])){
$data['userId'] = $meta_data['uid'];
}
if(isset($meta_data['companyId']) && !empty($meta_data['companyId'])){
$data['companyId'] = $meta_data['companyId'];
}
return ConversationModel::create($data);
} else {
// 存在则更新
return $exists->save([
'meta_data' => json_encode($meta_data),
'update_time' => time()
]);
}
}
/**
* 获取会话列表
*/
public function list()
{
try {
$bot_id = input('bot_id','');
if(empty($bot_id)){
return errorJson('智能体ID不能为空');
}
$page = input('page',1);
$limit = input('limit',20);
$params = [
'bot_id' => $bot_id,
'page_num' => $page,
'page_size' => $limit,
'sort_order' => 'desc'
];
$result = requestCurl($this->apiUrl . "/v1/conversations", $params, 'GET', $this->headers);
$result = json_decode($result, true);
if ($result['code'] != 0) {
return errorJson($result['msg'], $result['code']);
}
// 处理返回的数据并存入数据库
if (!empty($result['data']['conversations'])) {
foreach ($result['data']['conversations'] as $item) {
$this->saveConversation($item, $bot_id);
}
}
return successJson($result['data'], '获取成功');
} catch (\Exception $e) {
return errorJson('获取对话列表失败:' . $e->getMessage());
}
}
/**
* 创建会话
*/
public function create($is_internal = false)
{
try {
$bot_id = Env::get('ai.bot_id');
$userInfo = request()->userInfo;
$uid = $userInfo['id'];
$companyId = $userInfo['companyId'];
if(empty($bot_id)){
return errorJson('智能体ID不能为空');
}
// 构建元数据和消息
$meta_data = [
'uid' => strval($uid),
'companyId' => strval($companyId),
];
$messages[] = [
'role' => 'assistant',
'content' => Env::get('ai.content'),
'type' => 'answer',
'content_type' => 'text',
];
$params = [
'bot_id' => strval($bot_id),
'meta_data' => $meta_data,
'messages' => $messages,
];
$url = $this->apiUrl . '/v1/conversation/create';
$result = $this->httpRequest($url, 'POST', json_encode($params,256), $this->headers);
$result = json_decode($result, true);
if ($result['code'] != 0) {
return errorJson($result['msg'], $result['code']);
}
// 获取返回的对话数据并保存
$conversation = $result['data'] ?? [];
if (!empty($conversation)) {
$this->saveConversation($conversation, $bot_id);
// 保存用户发送的消息
$userMessageData = [
'chat_id' => $conversation['id'],
'conversation_id' => $conversation['id'],
'bot_id' => $bot_id,
'content' => Env::get('ai.content'),
'content_type' => 'text',
'role' => 'assistant',
'type' => 'answer',
'created_at' => time(),
'updated_at' => time()
];
MessageModel::create($userMessageData);
}
if($is_internal){
return json_encode([
'code' => 200,
'data' => $conversation,
'msg' => '创建成功'
]);
}else{
return successJson($conversation, '创建成功');
}
} catch (\Exception $e) {
return errorJson('创建对话失败:' . $e->getMessage());
}
}
/**
* 创建对话
*/
public function createChat()
{
try {
$bot_id = Env::get('ai.bot_id');
$conversation_id = input('conversation_id','');
$question = input('question','');
if(empty($bot_id)){
return errorJson('智能体ID不能为空');
}
if(empty($conversation_id)){
return errorJson('会话ID不能为空');
}
if(empty($question)){
return errorJson('问题不能为空');
}
$userInfo = request()->userInfo;
$uid = $userInfo['id'];
$companyId = $userInfo['companyId'];
// 构建请求数据
$params = [
'bot_id' => strval($bot_id),
'user_id' => strval($uid),
'additional_messages' => [
[
'role' => 'user',
'content' => $question,
'type' => 'question',
'content_type' => 'text'
]
],
'stream' => false,
'auto_save_history' => true
];
$url = $this->apiUrl . '/v3/chat?conversation_id='.$conversation_id;
$result = $this->httpRequest($url, 'POST', json_encode($params,256), $this->headers);
$result = json_decode($result, true);
if ($result['code'] != 0) {
return errorJson($result['msg'], $result['code']);
}
// 保存用户发送的消息
$userMessageData = [
'chat_id' => $result['data']['id'],
'conversation_id' => $conversation_id,
'bot_id' => $bot_id,
'content' => $question,
'content_type' => 'text',
'role' => 'user',
'type' => 'question',
'created_at' => $result['data']['created_at'],
'updated_at' => $result['data']['created_at']
];
MessageModel::create($userMessageData);
return successJson($result['data'], '发送成功');
} catch (\Exception $e) {
return errorJson('创建对话失败:' . $e->getMessage());
}
}
/**
* 查看对话详情
*/
public function chatRetrieve()
{
$conversation_id = input('conversation_id','');
$chat_id = input('chat_id','');
if(empty($conversation_id) && empty($chat_id)){
return errorJson('参数缺失');
}
$conversation = ConversationModel::where('conversation_id', $conversation_id)->find();
if(empty($conversation)){
return errorJson('会话不存在');
}
$params = [
'conversation_id' => $conversation_id,
'chat_id' => $chat_id
];
$url = $this->apiUrl . '/v3/chat/retrieve?' . dataBuild($params);
$result = $this->httpRequest($url, 'GET', [], $this->headers);
$result = json_decode($result, true);
if ($result['code'] != 0) {
return errorJson($result['msg'], $result['code']);
}
$status = [
'created' => '对话已创建',
'in_progress' => '智能体正在处理中',
'completed' => '智能体已完成处理,本次对话结束',
'failed' => '对话失败',
'requires_action' => '对话中断,需要进一步处理',
'canceled' => '对话已取消',
];
$status_msg = $status[$result['data']['status']] ?? '未知状态';
if($result['data']['status'] == 'failed'){
$last_error = $result['data']['last_error'];
$error_msg = $status_msg . ',错误信息:' . $last_error['msg'];
return errorJson($error_msg);
}else{
return successJson($result['data'],$status_msg);
}
}
/**
* 获取对话消息详情
*/
public function chatMessage(){
$conversation_id = input('conversation_id','');
$chat_id = input('chat_id','');
if(empty($conversation_id) && empty($chat_id)){
return errorJson('参数缺失');
}
$conversation = ConversationModel::where('conversation_id', $conversation_id)->find();
if(empty($conversation)){
return errorJson('会话不存在');
}
$params = [
'conversation_id' => $conversation_id,
'chat_id' => $chat_id
];
$url = $this->apiUrl . '/v3/chat/message/list?' . dataBuild($params);
$result = $this->httpRequest($url, 'GET', [], $this->headers);
$result = json_decode($result, true);
if ($result['code'] != 0) {
return errorJson($result['msg'], $result['code']);
}
$data = $result['data'];
$list = [];
foreach($data as $item){
if($item['type'] == 'answer'){
$timestamp = $item['updated_at'];
$now = time();
$today = strtotime(date('Y-m-d'));
$yesterday = strtotime('-1 day', $today);
$thisYear = strtotime(date('Y-01-01'));
// 格式化时间
if($timestamp >= $today) {
$time = date('H:i', $timestamp);
} elseif($timestamp >= $yesterday) {
$time = '昨天 ' . date('H:i', $timestamp);
} elseif($timestamp >= $thisYear) {
$time = date('m-d H:i', $timestamp);
} else {
$time = date('Y-m-d H:i', $timestamp);
}
// 保存消息记录
$messageData = [
'chat_id' => $item['id'],
'conversation_id' => $conversation_id,
'bot_id' => $item['bot_id'],
'content' => $item['content'],
'content_type' => $item['content_type'],
'role' => $item['role'],
'type' => $item['type'],
'created_at' => $item['created_at'],
'updated_at' => $item['updated_at']
];
// 检查消息是否已存在
$exists = MessageModel::where('chat_id', $item['id'])->find();
if (!$exists) {
MessageModel::create($messageData);
}
$list = [
'id' => $item['id'],
'type' => 'assistant',
'content' => $item['content'],
'time' => $time
];
break;
}
}
return successJson($list, '获取成功');
}
}