111 lines
4.0 KiB
Python
111 lines
4.0 KiB
Python
|
"""
|
|||
|
|
会话管理服务
|
|||
|
|
"""
|
|||
|
|
from typing import List, Optional
|
|||
|
|
from datetime import datetime
|
|||
|
|
from sqlalchemy.orm import Session
|
|||
|
|
import uuid
|
|||
|
|
|
|||
|
|
from models import User, Conversation, Message
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ConversationService:
|
|||
|
|
def __init__(self, db: Session):
|
|||
|
|
self.db = db
|
|||
|
|
|
|||
|
|
def get_or_create_user(self, user_id: str, display_name: str = None, user_type: str = 'web', matrix_user_id: str = None) -> User:
|
|||
|
|
"""获取或创建用户"""
|
|||
|
|
user = self.db.query(User).filter(User.user_id == user_id).first()
|
|||
|
|
if not user:
|
|||
|
|
user = User(
|
|||
|
|
user_id=user_id,
|
|||
|
|
display_name=display_name or user_id,
|
|||
|
|
user_type=user_type,
|
|||
|
|
matrix_user_id=matrix_user_id
|
|||
|
|
)
|
|||
|
|
self.db.add(user)
|
|||
|
|
self.db.commit()
|
|||
|
|
self.db.refresh(user)
|
|||
|
|
else:
|
|||
|
|
user.last_active_at = datetime.utcnow()
|
|||
|
|
self.db.commit()
|
|||
|
|
return user
|
|||
|
|
|
|||
|
|
def create_conversation(self, user_id: int, title: str = None) -> Conversation:
|
|||
|
|
"""创建新会话"""
|
|||
|
|
conversation_id = f"conv_{uuid.uuid4().hex[:12]}"
|
|||
|
|
conversation = Conversation(
|
|||
|
|
conversation_id=conversation_id,
|
|||
|
|
user_id=user_id,
|
|||
|
|
title=title
|
|||
|
|
)
|
|||
|
|
self.db.add(conversation)
|
|||
|
|
self.db.commit()
|
|||
|
|
self.db.refresh(conversation)
|
|||
|
|
return conversation
|
|||
|
|
|
|||
|
|
def get_conversation(self, conversation_id: str) -> Optional[Conversation]:
|
|||
|
|
"""获取会话"""
|
|||
|
|
return self.db.query(Conversation).filter(
|
|||
|
|
Conversation.conversation_id == conversation_id
|
|||
|
|
).first()
|
|||
|
|
|
|||
|
|
def get_user_conversations(self, user_id: int) -> List[Conversation]:
|
|||
|
|
"""获取用户的所有会话"""
|
|||
|
|
return self.db.query(Conversation).filter(
|
|||
|
|
Conversation.user_id == user_id,
|
|||
|
|
Conversation.is_active == True
|
|||
|
|
).order_by(Conversation.updated_at.desc()).all()
|
|||
|
|
|
|||
|
|
def add_message(self, conversation_id: int, role: str, content: str, source: str = 'web', extra_data: dict = None) -> Message:
|
|||
|
|
"""添加消息"""
|
|||
|
|
message = Message(
|
|||
|
|
conversation_id=conversation_id,
|
|||
|
|
role=role,
|
|||
|
|
content=content,
|
|||
|
|
source=source,
|
|||
|
|
extra_data=extra_data
|
|||
|
|
)
|
|||
|
|
self.db.add(message)
|
|||
|
|
|
|||
|
|
# 更新会话时间
|
|||
|
|
conversation = self.db.query(Conversation).filter(Conversation.id == conversation_id).first()
|
|||
|
|
if conversation:
|
|||
|
|
conversation.updated_at = datetime.utcnow()
|
|||
|
|
# 如果没有标题，用第一条用户消息作为标题
|
|||
|
|
if not conversation.title and role == 'user':
|
|||
|
|
conversation.title = content[:50] + ('...' if len(content) > 50 else '')
|
|||
|
|
|
|||
|
|
self.db.commit()
|
|||
|
|
self.db.refresh(message)
|
|||
|
|
return message
|
|||
|
|
|
|||
|
|
def get_messages(self, conversation_id: int, limit: int = 50) -> List[Message]:
|
|||
|
|
"""获取会话消息"""
|
|||
|
|
return self.db.query(Message).filter(
|
|||
|
|
Message.conversation_id == conversation_id
|
|||
|
|
).order_by(Message.created_at.asc()).limit(limit).all()
|
|||
|
|
|
|||
|
|
def get_conversation_history(self, conversation_id: str, limit: int = 20) -> List[dict]:
|
|||
|
|
"""获取会话历史（用于AI上下文）"""
|
|||
|
|
conversation = self.get_conversation(conversation_id)
|
|||
|
|
if not conversation:
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
messages = self.db.query(Message).filter(
|
|||
|
|
Message.conversation_id == conversation.id
|
|||
|
|
).order_by(Message.created_at.desc()).limit(limit).all()
|
|||
|
|
|
|||
|
|
# 反转顺序，最早的在前
|
|||
|
|
messages.reverse()
|
|||
|
|
|
|||
|
|
return [{"role": m.role, "content": m.content} for m in messages]
|
|||
|
|
|
|||
|
|
def delete_conversation(self, conversation_id: str):
|
|||
|
|
"""删除会话（软删除）"""
|
|||
|
|
conversation = self.get_conversation(conversation_id)
|
|||
|
|
if conversation:
|
|||
|
|
conversation.is_active = False
|
|||
|
|
self.db.commit()
|
|||
|
|
return True
|
|||
|
|
return False
|