243 lines
8.4 KiB
Python
243 lines
8.4 KiB
Python
from sqlalchemy import (
|
||
Column, Integer, String, Text, DateTime, Boolean, Float, ForeignKey
|
||
)
|
||
from sqlalchemy.orm import relationship, declarative_base
|
||
from models.user import DjangoUser # 确保导入 DjangoUser
|
||
|
||
Base = declarative_base()
|
||
|
||
|
||
# 状态选择类（示例）
|
||
class CommonStatus:
|
||
DISABLED = 0
|
||
ENABLED = 1
|
||
|
||
@staticmethod
|
||
def choices():
|
||
return [(0, '禁用'), (1, '启用')]
|
||
|
||
|
||
# 平台选择类（示例）
|
||
class PlatformChoices:
|
||
OPENAI = 'openai'
|
||
ALIMNS = 'alimns'
|
||
|
||
@staticmethod
|
||
def choices():
|
||
return [('openai', 'OpenAI'), ('alimns', '阿里云MNS')]
|
||
|
||
|
||
# 消息类型选择类（示例）
|
||
class MessageType:
|
||
TEXT = 'text'
|
||
IMAGE = 'image'
|
||
|
||
@staticmethod
|
||
def choices():
|
||
return [('text', '文本'), ('image', '图片')]
|
||
|
||
|
||
# 基础模型类
|
||
class CoreModel(Base):
|
||
__abstract__ = True
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
create_time = Column(DateTime)
|
||
update_time = Column(DateTime)
|
||
is_deleted = Column(Boolean, default=False)
|
||
|
||
|
||
# AI API 密钥表
|
||
class AIApiKey(CoreModel):
|
||
__tablename__ = 'ai_api_key'
|
||
|
||
name = Column(String(255), nullable=False)
|
||
platform = Column(String(100), nullable=False)
|
||
api_key = Column(String(255), nullable=False)
|
||
url = Column(String(255), nullable=True)
|
||
status = Column(Integer, default=CommonStatus.DISABLED)
|
||
|
||
def __str__(self):
|
||
return self.name
|
||
|
||
|
||
# AI 模型表
|
||
class AIModel(CoreModel):
|
||
__tablename__ = 'ai_model'
|
||
|
||
name = Column(String(64), nullable=False)
|
||
sort = Column(Integer, default=0)
|
||
status = Column(Integer, default=CommonStatus.DISABLED)
|
||
key_id = Column(Integer, ForeignKey('ai_api_key.id'), nullable=False)
|
||
model_type = Column(String(32), nullable=True)
|
||
platform = Column(String(32), nullable=False)
|
||
model = Column(String(64), nullable=False)
|
||
temperature = Column(Float, nullable=True)
|
||
max_tokens = Column(Integer, nullable=True)
|
||
max_contexts = Column(Integer, nullable=True)
|
||
|
||
key = relationship('AIApiKey', backref='models')
|
||
|
||
def __str__(self):
|
||
return self.name
|
||
|
||
|
||
# AI 工具表
|
||
# class Tool(CoreModel):
|
||
# __tablename__ = 'ai_tool'
|
||
|
||
# name = Column(String(128), nullable=False)
|
||
# description = Column(String(256), nullable=True)
|
||
# status = Column(Integer, default=0)
|
||
|
||
# def __str__(self):
|
||
# return self.name
|
||
|
||
|
||
# AI 知识库表
|
||
# class Knowledge(CoreModel):
|
||
# __tablename__ = 'ai_knowledge'
|
||
|
||
# name = Column(String(255), nullable=False)
|
||
# description = Column(Text, nullable=True)
|
||
# embedding_model_id = Column(Integer, ForeignKey('ai_model.id'), nullable=False)
|
||
# embedding_model = Column(String(32), nullable=False)
|
||
# top_k = Column(Integer, default=0)
|
||
# similarity_threshold = Column(Float, nullable=False)
|
||
# status = Column(Integer, default=CommonStatus.DISABLED)
|
||
|
||
# embedding_model_rel = relationship('AIModel', backref='knowledges')
|
||
# documents = relationship('KnowledgeDocument', backref='knowledge', cascade='all, delete-orphan')
|
||
# segments = relationship('KnowledgeSegment', backref='knowledge', cascade='all, delete-orphan')
|
||
# roles = relationship('ChatRole', secondary='ai_chat_role_knowledge', backref='knowledges')
|
||
|
||
# def __str__(self):
|
||
# return self.name
|
||
|
||
|
||
# AI 知识库文档表
|
||
# class KnowledgeDocument(CoreModel):
|
||
# __tablename__ = 'ai_knowledge_document'
|
||
|
||
# knowledge_id = Column(Integer, ForeignKey('ai_knowledge.id'), nullable=False)
|
||
# name = Column(String(255), nullable=False)
|
||
# url = Column(String(1024), nullable=False)
|
||
# content = Column(Text, nullable=False)
|
||
# content_length = Column(Integer, nullable=False)
|
||
# tokens = Column(Integer, nullable=False)
|
||
# segment_max_tokens = Column(Integer, nullable=False)
|
||
# retrieval_count = Column(Integer, default=0)
|
||
# status = Column(Integer, default=CommonStatus.DISABLED)
|
||
|
||
# segments = relationship('KnowledgeSegment', backref='document', cascade='all, delete-orphan')
|
||
|
||
# def __str__(self):
|
||
# return self.name
|
||
|
||
|
||
# AI 知识库分段表
|
||
# class KnowledgeSegment(CoreModel):
|
||
# __tablename__ = 'ai_knowledge_segment'
|
||
|
||
# knowledge_id = Column(Integer, ForeignKey('ai_knowledge.id'), nullable=False)
|
||
# document_id = Column(Integer, ForeignKey('ai_knowledge_document.id'), nullable=False)
|
||
# content = Column(Text, nullable=False)
|
||
# content_length = Column(Integer, nullable=False)
|
||
# tokens = Column(Integer, nullable=False)
|
||
# vector_id = Column(String(100), nullable=True)
|
||
# retrieval_count = Column(Integer, default=0)
|
||
# status = Column(Integer, default=CommonStatus.DISABLED)
|
||
|
||
# def __str__(self):
|
||
# return f"Segment {self.id}"
|
||
|
||
|
||
# AI 聊天角色表
|
||
# class ChatRole(CoreModel):
|
||
# __tablename__ = 'ai_chat_role'
|
||
|
||
# name = Column(String(128), nullable=False)
|
||
# avatar = Column(String(256), nullable=False)
|
||
# description = Column(String(256), nullable=True)
|
||
# status = Column(Integer, default=CommonStatus.DISABLED)
|
||
# sort = Column(Integer, default=0)
|
||
# public_status = Column(Boolean, default=False)
|
||
# category = Column(String(32), nullable=True)
|
||
# model_id = Column(Integer, ForeignKey('ai_model.id'), nullable=False)
|
||
# system_message = Column(String(1024), nullable=True)
|
||
# user_id = Column(
|
||
# Integer,
|
||
# ForeignKey('system_users.id'), # 假设DjangoUser表名是system_users
|
||
# nullable=True # 允许为空（如匿名角色）
|
||
# )
|
||
# user = relationship(DjangoUser, backref='chat_roles') # 正确：DjangoUser 已定义并导入
|
||
|
||
# model = relationship('AIModel', backref='chat_roles')
|
||
# tools = relationship('Tool', secondary='ai_chat_role_tool', backref='roles')
|
||
# # conversations = relationship('ChatConversation', backref='role', cascade='all, delete-orphan')
|
||
# # messages = relationship('ChatMessage', backref='role', cascade='all, delete-orphan')
|
||
|
||
# def __str__(self):
|
||
# return self.name
|
||
|
||
|
||
# AI 聊天对话表
|
||
# class ChatConversation(CoreModel):
|
||
# __tablename__ = 'ai_chat_conversation'
|
||
|
||
# title = Column(String(256), nullable=False)
|
||
# pinned = Column(Boolean, default=False)
|
||
# pinned_time = Column(DateTime, nullable=True)
|
||
# # user_id = Column(Integer, ForeignKey('system_users.id'), nullable=True)
|
||
# role_id = Column(Integer, ForeignKey('ai_chat_role.id'), nullable=True)
|
||
# model_id = Column(Integer, ForeignKey('ai_model.id'), nullable=False)
|
||
# model = Column(String(32), nullable=False)
|
||
# system_message = Column(String(1024), nullable=True)
|
||
# temperature = Column(Float, nullable=False)
|
||
# max_tokens = Column(Integer, nullable=False)
|
||
# max_contexts = Column(Integer, nullable=False)
|
||
# # user = relationship(DjangoUser, backref='conversations') # 正确：DjangoUser 已定义并导入
|
||
|
||
# model_rel = relationship('AIModel', backref='conversations')
|
||
# messages = relationship('ChatMessage', backref='conversation', cascade='all, delete-orphan')
|
||
|
||
# def __str__(self):
|
||
# return self.title
|
||
|
||
|
||
# AI 聊天消息表
|
||
# class ChatMessage(CoreModel):
|
||
# __tablename__ = 'ai_chat_message'
|
||
|
||
# conversation_id = Column(Integer, nullable=False)
|
||
# # user_id = Column(Integer, ForeignKey('system_users.id'), nullable=True)
|
||
# role_id = Column(Integer, ForeignKey('ai_chat_role.id'), nullable=True)
|
||
# model = Column(String(32), nullable=False)
|
||
# model_id = Column(Integer, ForeignKey('ai_model.id'), nullable=False)
|
||
# type = Column(String(16), nullable=False)
|
||
# reply_id = Column(Integer, nullable=True)
|
||
# content = Column(String(2048), nullable=False)
|
||
# use_context = Column(Boolean, default=False)
|
||
# segment_ids = Column(String(2048), nullable=True)
|
||
|
||
# # user = relationship(DjangoUser, backref='messages') # 正确：DjangoUser 已定义并导入
|
||
# model_rel = relationship('AIModel', backref='messages')
|
||
|
||
# def __str__(self):
|
||
# return self.content[:30]
|
||
|
||
|
||
# # 聊天角色与知识库的关联表
|
||
# class ChatRoleKnowledge(Base):
|
||
# __tablename__ = 'ai_chat_role_knowledge'
|
||
|
||
# chat_role_id = Column(Integer, ForeignKey('ai_chat_role.id'), primary_key=True)
|
||
# knowledge_id = Column(Integer, ForeignKey('ai_knowledge.id'), primary_key=True)
|
||
|
||
|
||
# # 聊天角色与工具的关联表
|
||
# class ChatRoleTool(Base):
|
||
# __tablename__ = 'ai_chat_role_tool'
|
||
|
||
# chat_role_id = Column(Integer, ForeignKey('ai_chat_role.id'), primary_key=True)
|
||
# tool_id = Column(Integer, ForeignKey('ai_tool.id'), primary_key=True) |