221 lines
8.0 KiB
Python
221 lines
8.0 KiB
Python
import uuid
|
||
|
||
from sqlalchemy import Column, String, Integer, Float, DateTime, Text, JSON, Boolean
|
||
from sqlalchemy.sql import func
|
||
|
||
from database import Base
|
||
|
||
|
||
def _iso(dt):
|
||
return dt.isoformat() if dt else None
|
||
|
||
|
||
class Avatar(Base):
|
||
__tablename__ = "avatars"
|
||
id = Column(String, primary_key=True, default=lambda: uuid.uuid4().hex)
|
||
owner_id = Column(String, default="", index=True) # 归属用户（会会 huihui_user_id）；空=未归属/种子数据
|
||
name = Column(String, nullable=False)
|
||
display_name = Column(String, default="")
|
||
description = Column(Text, default="")
|
||
photo_url = Column(String, default="")
|
||
emoji = Column(String, default="🤖")
|
||
status = Column(String, default="active") # active | inactive | training
|
||
token_balance = Column(Integer, default=0)
|
||
config = Column(JSON, default=dict)
|
||
created_at = Column(DateTime, server_default=func.now())
|
||
updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now())
|
||
|
||
def to_dict(self):
|
||
return {
|
||
"id": self.id,
|
||
"ownerId": self.owner_id,
|
||
"name": self.name,
|
||
"displayName": self.display_name,
|
||
"description": self.description,
|
||
"photoUrl": self.photo_url,
|
||
"emoji": self.emoji,
|
||
"status": self.status,
|
||
"tokenBalance": self.token_balance,
|
||
"config": self.config or {},
|
||
"createdAt": _iso(self.created_at),
|
||
"updatedAt": _iso(self.updated_at),
|
||
}
|
||
|
||
|
||
class Authorization(Base):
|
||
__tablename__ = "authorizations"
|
||
id = Column(String, primary_key=True, default=lambda: uuid.uuid4().hex)
|
||
avatar_id = Column(String, nullable=False, default="")
|
||
target_type = Column(String, default="user") # user | organization | application
|
||
target_id = Column(String, default="")
|
||
target_name = Column(String, default="")
|
||
permissions = Column(JSON, default=list)
|
||
status = Column(String, default="active") # active | inactive
|
||
created_at = Column(DateTime, server_default=func.now())
|
||
|
||
def to_dict(self):
|
||
return {
|
||
"id": self.id,
|
||
"avatarId": self.avatar_id,
|
||
"targetType": self.target_type,
|
||
"targetId": self.target_id,
|
||
"targetName": self.target_name,
|
||
"permissions": self.permissions or [],
|
||
"status": self.status,
|
||
"createdAt": _iso(self.created_at),
|
||
}
|
||
|
||
|
||
class Organization(Base):
|
||
__tablename__ = "organizations"
|
||
id = Column(String, primary_key=True, default=lambda: uuid.uuid4().hex)
|
||
name = Column(String, nullable=False)
|
||
description = Column(Text, default="")
|
||
emoji = Column(String, default="🏢")
|
||
org_type = Column(String, default="team") # team | company | community
|
||
role = Column(String, default="admin") # admin | member | viewer
|
||
member_count = Column(Integer, default=1)
|
||
created_at = Column(DateTime, server_default=func.now())
|
||
|
||
def to_dict(self):
|
||
return {
|
||
"id": self.id,
|
||
"name": self.name,
|
||
"description": self.description,
|
||
"emoji": self.emoji,
|
||
"type": self.org_type,
|
||
"role": self.role,
|
||
"memberCount": self.member_count,
|
||
"createdAt": _iso(self.created_at),
|
||
}
|
||
|
||
|
||
class KnowledgeDoc(Base):
|
||
__tablename__ = "knowledge_docs"
|
||
id = Column(String, primary_key=True, default=lambda: uuid.uuid4().hex)
|
||
avatar_id = Column(String, nullable=False, default="")
|
||
filename = Column(String, default="")
|
||
file_type = Column(String, default="") # pdf | doc | docx | xlsx
|
||
file_size = Column(Integer, default=0)
|
||
file_url = Column(String, default="")
|
||
status = Column(String, default="uploaded") # uploaded | parsing | ready
|
||
vectorized = Column(Boolean, default=False) # 是否已向量化
|
||
embedding_model = Column(String, default="") # 向量模型标识
|
||
chunk_count = Column(Integer, default=0) # 切片数量
|
||
vectorized_at = Column(DateTime) # 向量化时间
|
||
created_at = Column(DateTime, server_default=func.now())
|
||
|
||
def to_dict(self):
|
||
return {
|
||
"id": self.id,
|
||
"avatarId": self.avatar_id,
|
||
"filename": self.filename,
|
||
"fileType": self.file_type,
|
||
"fileSize": self.file_size,
|
||
"fileUrl": self.file_url,
|
||
"status": self.status,
|
||
"vectorized": bool(self.vectorized),
|
||
"embeddingModel": self.embedding_model,
|
||
"chunkCount": self.chunk_count,
|
||
"vectorizedAt": _iso(self.vectorized_at),
|
||
"createdAt": _iso(self.created_at),
|
||
}
|
||
|
||
|
||
class QAPair(Base):
|
||
__tablename__ = "qa_pairs"
|
||
id = Column(String, primary_key=True, default=lambda: uuid.uuid4().hex)
|
||
avatar_id = Column(String, nullable=False, default="")
|
||
question = Column(Text, default="")
|
||
answer = Column(Text, default="")
|
||
enabled = Column(Boolean, default=True) # 是否启用（关闭后不参与作答）
|
||
created_at = Column(DateTime, server_default=func.now())
|
||
updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now())
|
||
|
||
def to_dict(self):
|
||
return {
|
||
"id": self.id,
|
||
"avatarId": self.avatar_id,
|
||
"question": self.question,
|
||
"answer": self.answer,
|
||
"enabled": bool(self.enabled),
|
||
"createdAt": _iso(self.created_at),
|
||
"updatedAt": _iso(self.updated_at),
|
||
}
|
||
|
||
|
||
class KnowledgeChunk(Base):
|
||
__tablename__ = "knowledge_chunks"
|
||
id = Column(String, primary_key=True, default=lambda: uuid.uuid4().hex)
|
||
doc_id = Column(String, default="") # 关联 KnowledgeDoc.id
|
||
avatar_id = Column(String, default="")
|
||
content = Column(Text, default="") # 切片文本
|
||
vector = Column(Text, default="") # JSON 编码的向量
|
||
chunk_index = Column(Integer, default=0)
|
||
embedding_model = Column(String, default="")
|
||
created_at = Column(DateTime, server_default=func.now())
|
||
|
||
def to_dict(self):
|
||
return {
|
||
"id": self.id,
|
||
"docId": self.doc_id,
|
||
"avatarId": self.avatar_id,
|
||
"content": self.content,
|
||
"chunkIndex": self.chunk_index,
|
||
"embeddingModel": self.embedding_model,
|
||
"createdAt": _iso(self.created_at),
|
||
}
|
||
|
||
|
||
class TokenAccount(Base):
|
||
__tablename__ = "token_account"
|
||
id = Column(Integer, primary_key=True)
|
||
balance = Column(Integer, default=1250)
|
||
|
||
|
||
class TokenPlan(Base):
|
||
__tablename__ = "token_plans"
|
||
id = Column(String, primary_key=True)
|
||
name = Column(String, default="")
|
||
amount = Column(Integer, default=0)
|
||
price = Column(Float, default=0)
|
||
badge = Column(String, default="")
|
||
desc = Column(String, default="")
|
||
|
||
def to_dict(self):
|
||
return {
|
||
"id": self.id,
|
||
"name": self.name,
|
||
"amount": self.amount,
|
||
"price": self.price,
|
||
"badge": self.badge,
|
||
"desc": self.desc,
|
||
}
|
||
|
||
|
||
class User(Base):
|
||
"""会会用户 ↔ 本地用户体系映射（短信验证码登录落库）"""
|
||
|
||
__tablename__ = "users"
|
||
id = Column(String, primary_key=True, default=lambda: uuid.uuid4().hex)
|
||
huihui_user_id = Column(String, default="", index=True) # 会会 userId（唯一标识）
|
||
phone = Column(String, default="", index=True)
|
||
nickname = Column(String, default="")
|
||
avatar_url = Column(String, default="")
|
||
huihui_token = Column(String, default="") # 会会 access_token
|
||
app_token = Column(String, default="") # 本系统会话 token
|
||
last_login_at = Column(DateTime)
|
||
created_at = Column(DateTime, server_default=func.now())
|
||
updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now())
|
||
|
||
def to_dict(self):
|
||
return {
|
||
"id": self.id,
|
||
"huihuiUserId": self.huihui_user_id,
|
||
"phone": self.phone,
|
||
"nickname": self.nickname,
|
||
"avatarUrl": self.avatar_url,
|
||
"createdAt": _iso(self.created_at),
|
||
"lastLoginAt": _iso(self.last_login_at),
|
||
}
|