372 lines
13 KiB
Python
372 lines
13 KiB
Python
|
"""
|
|||
|
|
对话列表管理服务
|
|||
|
|
负责管理用户的对话列表（会话列表）
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
from typing import Optional, Dict, Any, List
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
from database import query, insert, update, get_db_type
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ConversationService:
|
|||
|
|
"""对话列表管理服务"""
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def generate_id() -> int:
|
|||
|
|
"""生成唯一ID（使用时间戳+随机数）"""
|
|||
|
|
import time
|
|||
|
|
import random
|
|||
|
|
return int(time.time() * 1000000) + random.randint(1000, 9999)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
async def create_conversation(
|
|||
|
|
system_user_id: int,
|
|||
|
|
context_id: str,
|
|||
|
|
title: str = "新对话",
|
|||
|
|
description: Optional[str] = None
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
创建新对话
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
system_user_id: 系统用户ID
|
|||
|
|
context_id: 会话上下文ID
|
|||
|
|
title: 对话标题
|
|||
|
|
description: 对话描述（可选）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
创建的对话信息
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conv_id = ConversationService.generate_id()
|
|||
|
|
|
|||
|
|
sql = """
|
|||
|
|
INSERT INTO conversation_list
|
|||
|
|
(id, context_id, system_user_id, title, description,
|
|||
|
|
message_count, created_at, updated_at, is_active)
|
|||
|
|
VALUES (%s, %s, %s, %s, %s, 0, NOW(), NOW(), TRUE)
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
params = (conv_id, context_id, system_user_id, title, description)
|
|||
|
|
|
|||
|
|
insert(sql, params)
|
|||
|
|
logger.info(f"✅ 创建对话成功: user={system_user_id}, context={context_id}, title={title}")
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
"id": conv_id,
|
|||
|
|
"context_id": context_id,
|
|||
|
|
"system_user_id": system_user_id,
|
|||
|
|
"title": title,
|
|||
|
|
"description": description,
|
|||
|
|
"message_count": 0,
|
|||
|
|
"created_at": datetime.now().isoformat(),
|
|||
|
|
"updated_at": datetime.now().isoformat(),
|
|||
|
|
"is_active": True
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"❌ 创建对话失败: {e}", exc_info=True)
|
|||
|
|
raise Exception(f"创建对话失败: {str(e)}")
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
async def get_conversation_by_context(
|
|||
|
|
context_id: str
|
|||
|
|
) -> Optional[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
根据context_id获取对话信息
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
context_id: 会话上下文ID
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
对话信息，不存在则返回None
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
sql = """
|
|||
|
|
SELECT id, context_id, system_user_id, title, description,
|
|||
|
|
message_count, last_message, created_at, updated_at, is_active
|
|||
|
|
FROM conversation_list
|
|||
|
|
WHERE context_id = %s
|
|||
|
|
LIMIT 1
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
results = query(sql, (context_id,))
|
|||
|
|
|
|||
|
|
if results:
|
|||
|
|
return results[0]
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"❌ 获取对话信息失败: {e}", exc_info=True)
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
async def update_conversation(
|
|||
|
|
context_id: str,
|
|||
|
|
title: Optional[str] = None,
|
|||
|
|
last_message: Optional[str] = None,
|
|||
|
|
increment_message_count: bool = False
|
|||
|
|
) -> bool:
|
|||
|
|
"""
|
|||
|
|
更新对话信息
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
context_id: 会话上下文ID
|
|||
|
|
title: 新标题（可选）
|
|||
|
|
last_message: 最后一条消息（可选）
|
|||
|
|
increment_message_count: 是否增加消息计数
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
是否成功更新
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
db_type = get_db_type()
|
|||
|
|
|
|||
|
|
# 对于StarRocks，使用DELETE+INSERT策略
|
|||
|
|
if db_type == 'starrocks':
|
|||
|
|
# 先获取当前记录
|
|||
|
|
conv = await ConversationService.get_conversation_by_context(context_id)
|
|||
|
|
if not conv:
|
|||
|
|
logger.warning(f"⚠️ 对话不存在: context={context_id}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 更新字段
|
|||
|
|
if title is not None:
|
|||
|
|
conv['title'] = title
|
|||
|
|
if last_message is not None:
|
|||
|
|
preview = last_message[:200] if len(last_message) > 200 else last_message
|
|||
|
|
conv['last_message'] = preview
|
|||
|
|
if increment_message_count:
|
|||
|
|
conv['message_count'] = conv.get('message_count', 0) + 1
|
|||
|
|
|
|||
|
|
# 删除旧记录
|
|||
|
|
delete_sql = "DELETE FROM conversation_list WHERE context_id = %s"
|
|||
|
|
update(delete_sql, (context_id,))
|
|||
|
|
|
|||
|
|
# 插入新记录
|
|||
|
|
insert_sql = """
|
|||
|
|
INSERT INTO conversation_list
|
|||
|
|
(id, system_user_id, updated_at, context_id, title, description,
|
|||
|
|
message_count, last_message, created_at, is_active)
|
|||
|
|
VALUES (%s, %s, NOW(), %s, %s, %s, %s, %s, %s, %s)
|
|||
|
|
"""
|
|||
|
|
insert(insert_sql, (
|
|||
|
|
conv['id'],
|
|||
|
|
conv['system_user_id'],
|
|||
|
|
conv['context_id'],
|
|||
|
|
conv['title'],
|
|||
|
|
conv.get('description'),
|
|||
|
|
conv['message_count'],
|
|||
|
|
conv.get('last_message'),
|
|||
|
|
conv['created_at'],
|
|||
|
|
conv.get('is_active', True)
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
logger.info(f"✅ 更新对话成功(StarRocks): context={context_id}")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
# 对于MySQL，使用正常的UPDATE
|
|||
|
|
else:
|
|||
|
|
# 构建动态更新SQL
|
|||
|
|
update_fields = []
|
|||
|
|
params = []
|
|||
|
|
|
|||
|
|
if title is not None:
|
|||
|
|
update_fields.append("title = %s")
|
|||
|
|
params.append(title)
|
|||
|
|
|
|||
|
|
if last_message is not None:
|
|||
|
|
# 限制预览长度为200字符
|
|||
|
|
preview = last_message[:200] if len(last_message) > 200 else last_message
|
|||
|
|
update_fields.append("last_message = %s")
|
|||
|
|
params.append(preview)
|
|||
|
|
|
|||
|
|
if increment_message_count:
|
|||
|
|
update_fields.append("message_count = message_count + 1")
|
|||
|
|
|
|||
|
|
# 总是更新 updated_at
|
|||
|
|
update_fields.append("updated_at = NOW()")
|
|||
|
|
|
|||
|
|
if not update_fields:
|
|||
|
|
return True # 没有字段需要更新
|
|||
|
|
|
|||
|
|
params.append(context_id)
|
|||
|
|
|
|||
|
|
sql = f"""
|
|||
|
|
UPDATE conversation_list
|
|||
|
|
SET {', '.join(update_fields)}
|
|||
|
|
WHERE context_id = %s
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
affected = update(sql, tuple(params))
|
|||
|
|
|
|||
|
|
if affected > 0:
|
|||
|
|
logger.info(f"✅ 更新对话成功: context={context_id}")
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
logger.warning(f"⚠️ 对话不存在: context={context_id}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"❌ 更新对话失败: {e}", exc_info=True)
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
async def get_user_conversations(
|
|||
|
|
system_user_id: int,
|
|||
|
|
limit: int = 50,
|
|||
|
|
only_active: bool = True
|
|||
|
|
) -> List[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
获取用户的对话列表
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
system_user_id: 系统用户ID
|
|||
|
|
limit: 返回数量限制
|
|||
|
|
only_active: 是否只返回活跃对话
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
对话列表（按更新时间倒序）
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
if only_active:
|
|||
|
|
sql = """
|
|||
|
|
SELECT id, context_id, system_user_id, title, description,
|
|||
|
|
message_count, last_message, created_at, updated_at, is_active
|
|||
|
|
FROM conversation_list
|
|||
|
|
WHERE system_user_id = %s AND is_active = TRUE
|
|||
|
|
ORDER BY updated_at DESC
|
|||
|
|
LIMIT %s
|
|||
|
|
"""
|
|||
|
|
else:
|
|||
|
|
sql = """
|
|||
|
|
SELECT id, context_id, system_user_id, title, description,
|
|||
|
|
message_count, last_message, created_at, updated_at, is_active
|
|||
|
|
FROM conversation_list
|
|||
|
|
WHERE system_user_id = %s
|
|||
|
|
ORDER BY updated_at DESC
|
|||
|
|
LIMIT %s
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
results = query(sql, (system_user_id, limit))
|
|||
|
|
return results
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"❌ 获取对话列表失败: {e}", exc_info=True)
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
async def delete_conversation(
|
|||
|
|
context_id: str,
|
|||
|
|
system_user_id: int,
|
|||
|
|
soft_delete: bool = True
|
|||
|
|
) -> bool:
|
|||
|
|
"""
|
|||
|
|
删除对话
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
context_id: 会话上下文ID
|
|||
|
|
system_user_id: 系统用户ID（用于权限验证）
|
|||
|
|
soft_delete: 是否软删除（仅标记为不活跃）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
是否成功删除
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
db_type = get_db_type()
|
|||
|
|
|
|||
|
|
# StarRocks只支持硬删除（DELETE），不支持UPDATE
|
|||
|
|
if db_type == 'starrocks' or not soft_delete:
|
|||
|
|
# 硬删除：物理删除记录
|
|||
|
|
sql = """
|
|||
|
|
DELETE FROM conversation_list
|
|||
|
|
WHERE context_id = %s AND system_user_id = %s
|
|||
|
|
"""
|
|||
|
|
affected = update(sql, (context_id, system_user_id))
|
|||
|
|
|
|||
|
|
if affected > 0:
|
|||
|
|
logger.info(f"✅ 删除对话成功: context={context_id}")
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
logger.warning(f"⚠️ 对话不存在或无权限: context={context_id}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# MySQL支持软删除
|
|||
|
|
else:
|
|||
|
|
# 软删除：仅标记为不活跃（需要先获取记录再重新插入）
|
|||
|
|
conv = await ConversationService.get_conversation_by_context(context_id)
|
|||
|
|
if not conv or conv['system_user_id'] != system_user_id:
|
|||
|
|
logger.warning(f"⚠️ 对话不存在或无权限: context={context_id}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 删除旧记录
|
|||
|
|
delete_sql = """
|
|||
|
|
DELETE FROM conversation_list
|
|||
|
|
WHERE context_id = %s AND system_user_id = %s
|
|||
|
|
"""
|
|||
|
|
update(delete_sql, (context_id, system_user_id))
|
|||
|
|
|
|||
|
|
# 插入标记为不活跃的记录
|
|||
|
|
insert_sql = """
|
|||
|
|
INSERT INTO conversation_list
|
|||
|
|
(id, system_user_id, updated_at, context_id, title, description,
|
|||
|
|
message_count, last_message, created_at, is_active)
|
|||
|
|
VALUES (%s, %s, NOW(), %s, %s, %s, %s, %s, %s, FALSE)
|
|||
|
|
"""
|
|||
|
|
insert(insert_sql, (
|
|||
|
|
conv['id'],
|
|||
|
|
conv['system_user_id'],
|
|||
|
|
conv['context_id'],
|
|||
|
|
conv['title'],
|
|||
|
|
conv.get('description'),
|
|||
|
|
conv['message_count'],
|
|||
|
|
conv.get('last_message'),
|
|||
|
|
conv['created_at']
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
logger.info(f"✅ 软删除对话成功: context={context_id}")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"❌ 删除对话失败: {e}", exc_info=True)
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
async def ensure_conversation_exists(
|
|||
|
|
system_user_id: int,
|
|||
|
|
context_id: str,
|
|||
|
|
title: str = "新对话"
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
确保对话存在，不存在则创建
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
system_user_id: 系统用户ID
|
|||
|
|
context_id: 会话上下文ID
|
|||
|
|
title: 对话标题
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
对话信息
|
|||
|
|
"""
|
|||
|
|
# 先尝试获取
|
|||
|
|
conv = await ConversationService.get_conversation_by_context(context_id)
|
|||
|
|
|
|||
|
|
if conv:
|
|||
|
|
return conv
|
|||
|
|
|
|||
|
|
# 不存在则创建
|
|||
|
|
return await ConversationService.create_conversation(
|
|||
|
|
system_user_id=system_user_id,
|
|||
|
|
context_id=context_id,
|
|||
|
|
title=title
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 创建全局服务实例
|
|||
|
|
conversation_service = ConversationService()
|
|||
|
|
|