118 lines
4.0 KiB
Python
118 lines
4.0 KiB
Python
|
# apps/common/services/chat_service.py
|
|||
|
|
import logging
|
|||
|
|
import json
|
|||
|
|
from uuid import uuid4
|
|||
|
|
from django.db import transaction
|
|||
|
|
from apps.user.models import User
|
|||
|
|
from apps.chat.models import ChatHistory
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
class ChatService:
|
|||
|
|
@transaction.atomic
|
|||
|
|
def create_chat_record(self, user, data, conversation_id=None):
|
|||
|
|
"""创建聊天记录，供chat、gmail、feishu模块复用"""
|
|||
|
|
try:
|
|||
|
|
# 验证必填字段
|
|||
|
|
if 'question' not in data:
|
|||
|
|
raise ValueError("缺少必填字段: question")
|
|||
|
|
|
|||
|
|
# 如果未提供conversation_id，生成新的
|
|||
|
|
if not conversation_id:
|
|||
|
|
conversation_id = str(uuid4())
|
|||
|
|
logger.info(f"生成新的会话ID: {conversation_id}")
|
|||
|
|
|
|||
|
|
# 创建metadata
|
|||
|
|
metadata = {
|
|||
|
|
'model_id': data.get('model_id', '7a214d0e-e65e-11ef-9f4a-0242ac120006'),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 设置标题
|
|||
|
|
title = data.get('title', 'New chat')
|
|||
|
|
|
|||
|
|
# 创建用户问题记录
|
|||
|
|
question_record = ChatHistory.objects.create(
|
|||
|
|
user=user,
|
|||
|
|
knowledge_base_id="b680a4fa-37be-11f0-a7cb-0242ac120002", # 使用默认知识库ID
|
|||
|
|
conversation_id=conversation_id,
|
|||
|
|
title=title,
|
|||
|
|
role='user',
|
|||
|
|
content=data['question'],
|
|||
|
|
metadata=metadata
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
return question_record, conversation_id, metadata, [], []
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"创建聊天记录失败: {str(e)}")
|
|||
|
|
raise
|
|||
|
|
|
|||
|
|
def get_conversation_detail(self, user, conversation_id):
|
|||
|
|
"""获取会话详情，供chat、gmail、feishu模块复用"""
|
|||
|
|
try:
|
|||
|
|
# 查询会话记录
|
|||
|
|
messages = ChatHistory.objects.filter(
|
|||
|
|
conversation_id=conversation_id,
|
|||
|
|
user=user,
|
|||
|
|
is_deleted=False
|
|||
|
|
).order_by('created_at')
|
|||
|
|
|
|||
|
|
if not messages.exists():
|
|||
|
|
raise ValueError("对话不存在或无权限")
|
|||
|
|
|
|||
|
|
# 构建消息列表
|
|||
|
|
message_list = [
|
|||
|
|
{
|
|||
|
|
'id': str(msg.id),
|
|||
|
|
'parent_id': msg.parent_id,
|
|||
|
|
'role': msg.role,
|
|||
|
|
'content': msg.content,
|
|||
|
|
'created_at': msg.created_at.strftime('%Y-%m-%d %H:%M:%S'),
|
|||
|
|
'metadata': msg.metadata
|
|||
|
|
}
|
|||
|
|
for msg in messages
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
'conversation_id': conversation_id,
|
|||
|
|
'messages': message_list
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"获取会话详情失败: {str(e)}")
|
|||
|
|
raise
|
|||
|
|
|
|||
|
|
def can_user_access_conversation(self, user, conversation_id):
|
|||
|
|
"""检查用户是否有权限访问指定的对话"""
|
|||
|
|
return ChatHistory.objects.filter(
|
|||
|
|
conversation_id=conversation_id,
|
|||
|
|
user=user,
|
|||
|
|
is_deleted=False
|
|||
|
|
).exists()
|
|||
|
|
|
|||
|
|
def format_chat_response(self, chat_history_list):
|
|||
|
|
"""格式化聊天历史记录为前端需要的格式"""
|
|||
|
|
try:
|
|||
|
|
result = []
|
|||
|
|
|
|||
|
|
for item in chat_history_list:
|
|||
|
|
formatted_item = {
|
|||
|
|
"id": str(item.id),
|
|||
|
|
"conversation_id": str(item.conversation_id),
|
|||
|
|
"role": item.role,
|
|||
|
|
"content": item.content,
|
|||
|
|
"created_at": item.created_at.strftime('%Y-%m-%d %H:%M:%S'),
|
|||
|
|
"title": item.title
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 添加parent_id如果存在
|
|||
|
|
if item.parent_id:
|
|||
|
|
formatted_item["parent_id"] = str(item.parent_id)
|
|||
|
|
|
|||
|
|
result.append(formatted_item)
|
|||
|
|
|
|||
|
|
return result
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"格式化聊天响应失败: {e}")
|
|||
|
|
return []
|
|||
|
|
|