"""MongoDB中的数据结构"""
import uuid
from datetime import UTC, datetime
from pydantic import BaseModel, Field
from apps.common.config import Config
from apps.constants import NEW_CHAT
from apps.templates.generate_llm_operator_config import llm_provider_dict
class Blacklist(BaseModel):
"""
黑名单
Collection: blacklist
"""
id: str = Field(default_factory=lambda: str(uuid.uuid4()), alias="_id")
question: str
answer: str
is_audited: bool = False
reason_type: str = ""
reason: str | None = None
updated_at: float = Field(default_factory=lambda: round(datetime.now(tz=UTC).timestamp(), 3))
class UserDomainData(BaseModel):
"""用户领域数据"""
name: str
count: int
class AppUsageData(BaseModel):
"""User表子项：应用使用情况数据"""
count: int = 0
last_used: float = Field(default_factory=lambda: round(datetime.now(tz=UTC).timestamp(), 3))
class User(BaseModel):
"""
用户信息
Collection: user
外键：user - conversation
"""
id: str = Field(alias="_id")
last_login: float = Field(default_factory=lambda: round(datetime.now(tz=UTC).timestamp(), 3))
is_active: bool = False
is_whitelisted: bool = False
credit: int = 100
api_key: str | None = None
conversations: list[str] = []
domains: list[UserDomainData] = []
app_usage: dict[str, AppUsageData] = {}
fav_apps: list[str] = []
fav_services: list[str] = []
is_admin: bool = Field(default=False, description="是否为管理员")
auto_execute: bool = Field(default=True, description="是否自动执行任务")
class LLM(BaseModel):
"""
大模型信息
Collection: llm
"""
id: str = Field(default_factory=lambda: str(uuid.uuid4()), alias="_id")
user_sub: str = Field(default="", description="用户ID")
icon: str = Field(default=llm_provider_dict["ollama"]["icon"], description="图标")
openai_base_url: str = Field(default=Config().get_config().llm.endpoint)
openai_api_key: str = Field(default=Config().get_config().llm.key)
model_name: str = Field(default=Config().get_config().llm.model)
max_tokens: int | None = Field(default=Config().get_config().llm.max_tokens)
created_at: float = Field(default_factory=lambda: round(datetime.now(tz=UTC).timestamp(), 3))
class LLMItem(BaseModel):
"""大模型信息"""
llm_id: str = Field(default="empty")
model_name: str = Field(default=Config().get_config().llm.model)
icon: str = Field(default=llm_provider_dict["ollama"]["icon"])
class KnowledgeBaseItem(BaseModel):
"""知识库信息"""
kb_id: str
kb_name: str
class Conversation(BaseModel):
"""
对话信息
Collection: conversation
外键：conversation - task, document, record_group
"""
id: str = Field(default_factory=lambda: str(uuid.uuid4()), alias="_id")
user_sub: str
title: str = Field(default=NEW_CHAT)
created_at: float = Field(default_factory=lambda: round(datetime.now(tz=UTC).timestamp(), 3))
app_id: str | None = Field(default="")
tasks: list[str] = []
unused_docs: list[str] = []
record_groups: list[str] = []
debug: bool = Field(default=False)
llm: LLMItem | None = None
kb_list: list[KnowledgeBaseItem] = Field(default=[])
class Document(BaseModel):
"""
文件信息
Collection: document
"""
id: str = Field(default_factory=lambda: str(uuid.uuid4()), alias="_id")
user_sub: str
name: str
type: str
size: float
created_at: float = Field(default_factory=lambda: round(datetime.now(tz=UTC).timestamp(), 3))
conversation_id: str | None = Field(default=None)
class Audit(BaseModel):
"""
审计日志
Collection: audit
"""
id: str = Field(default_factory=lambda: str(uuid.uuid4()), alias="_id")
user_sub: str | None = None
http_method: str
created_at: float = Field(default_factory=lambda: round(datetime.now(tz=UTC).timestamp(), 3))
module: str
client_ip: str | None = None
message: str
class Domain(BaseModel):
"""
领域信息
Collection: domain
"""
id: str = Field(default_factory=lambda: str(uuid.uuid4()), alias="_id")
name: str
definition: str
updated_at: float = Field(default_factory=lambda: round(datetime.now(tz=UTC).timestamp(), 3))