695 lines
33 KiB
Python
695 lines
33 KiB
Python
"""SQLAlchemy数据库模型定义
|
||
|
||
本文件只包含纯模型定义，使用SQLAlchemy 2.0的Mapped类型注解风格。
|
||
引擎和会话管理已移至core/engine.py和core/session.py。
|
||
|
||
支持的数据库类型：
|
||
- SQLite: 使用 Text 类型
|
||
- PostgreSQL: 使用 Text 类型（PostgreSQL 的 Text 类型性能与 VARCHAR 相当）
|
||
|
||
所有模型使用统一的类型注解风格：
|
||
field_name: Mapped[PyType] = mapped_column(Type, ...)
|
||
|
||
这样IDE/Pylance能正确推断实例属性类型。
|
||
"""
|
||
|
||
import datetime
|
||
import time
|
||
|
||
from sqlalchemy import Boolean, DateTime, Float, Index, Integer, String, Text
|
||
from sqlalchemy.ext.declarative import declarative_base
|
||
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
||
# 创建基类
|
||
Base = declarative_base()
|
||
|
||
|
||
# 数据库兼容的字段类型辅助函数
|
||
def get_string_field(max_length=255, **kwargs):
|
||
"""
|
||
根据数据库类型返回合适的字符串字段类型
|
||
|
||
对于需要索引的字段：
|
||
- PostgreSQL: 可以使用 Text，但为了兼容性使用 VARCHAR
|
||
- SQLite: 可以使用 Text，无长度限制
|
||
|
||
Args:
|
||
max_length: 最大长度
|
||
**kwargs: 传递给 String/Text 的额外参数
|
||
|
||
Returns:
|
||
SQLAlchemy 类型
|
||
"""
|
||
from src.config.config import global_config
|
||
|
||
assert global_config is not None
|
||
db_type = global_config.database.database_type
|
||
|
||
# PostgreSQL 可以使用 Text，但为了跨数据库迁移兼容性，使用 VARCHAR
|
||
if db_type == "postgresql":
|
||
return String(max_length, **kwargs)
|
||
# SQLite 使用 Text（无长度限制）
|
||
else:
|
||
return Text(**kwargs)
|
||
|
||
|
||
class ChatStreams(Base):
|
||
"""聊天流模型"""
|
||
|
||
__tablename__ = "chat_streams"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
stream_id: Mapped[str] = mapped_column(get_string_field(64), nullable=False, unique=True, index=True)
|
||
create_time: Mapped[float] = mapped_column(Float, nullable=False)
|
||
group_platform: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
group_id: Mapped[str | None] = mapped_column(get_string_field(100), nullable=True, index=True)
|
||
group_name: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
last_active_time: Mapped[float] = mapped_column(Float, nullable=False)
|
||
platform: Mapped[str] = mapped_column(Text, nullable=False)
|
||
user_platform: Mapped[str] = mapped_column(Text, nullable=False)
|
||
user_id: Mapped[str] = mapped_column(get_string_field(100), nullable=False, index=True)
|
||
user_nickname: Mapped[str] = mapped_column(Text, nullable=False)
|
||
user_cardname: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
energy_value: Mapped[float | None] = mapped_column(Float, nullable=True, default=5.0)
|
||
sleep_pressure: Mapped[float | None] = mapped_column(Float, nullable=True, default=0.0)
|
||
focus_energy: Mapped[float | None] = mapped_column(Float, nullable=True, default=0.5)
|
||
# 动态兴趣度系统字段
|
||
base_interest_energy: Mapped[float | None] = mapped_column(Float, nullable=True, default=0.5)
|
||
message_interest_total: Mapped[float | None] = mapped_column(Float, nullable=True, default=0.0)
|
||
message_count: Mapped[int | None] = mapped_column(Integer, nullable=True, default=0)
|
||
action_count: Mapped[int | None] = mapped_column(Integer, nullable=True, default=0)
|
||
reply_count: Mapped[int | None] = mapped_column(Integer, nullable=True, default=0)
|
||
last_interaction_time: Mapped[float | None] = mapped_column(Float, nullable=True, default=None)
|
||
consecutive_no_reply: Mapped[int | None] = mapped_column(Integer, nullable=True, default=0)
|
||
# 消息打断系统字段
|
||
interruption_count: Mapped[int | None] = mapped_column(Integer, nullable=True, default=0)
|
||
# 聊天流印象字段
|
||
stream_impression_text: Mapped[str | None] = mapped_column(Text, nullable=True) # 对聊天流的主观印象描述
|
||
stream_chat_style: Mapped[str | None] = mapped_column(Text, nullable=True) # 聊天流的总体风格
|
||
stream_topic_keywords: Mapped[str | None] = mapped_column(Text, nullable=True) # 话题关键词，逗号分隔
|
||
stream_interest_score: Mapped[float | None] = mapped_column(Float, nullable=True, default=0.5) # 对聊天流的兴趣程度(0-1)
|
||
|
||
__table_args__ = (
|
||
Index("idx_chatstreams_stream_id", "stream_id"),
|
||
Index("idx_chatstreams_user_id", "user_id"),
|
||
Index("idx_chatstreams_group_id", "group_id"),
|
||
)
|
||
|
||
|
||
class LLMUsage(Base):
|
||
"""LLM使用记录模型"""
|
||
|
||
__tablename__ = "llm_usage"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
model_name: Mapped[str] = mapped_column(get_string_field(100), nullable=False, index=True)
|
||
model_assign_name: Mapped[str] = mapped_column(get_string_field(100), index=True)
|
||
model_api_provider: Mapped[str] = mapped_column(get_string_field(100), index=True)
|
||
user_id: Mapped[str] = mapped_column(get_string_field(50), nullable=False, index=True)
|
||
request_type: Mapped[str] = mapped_column(get_string_field(50), nullable=False, index=True)
|
||
endpoint: Mapped[str] = mapped_column(Text, nullable=False)
|
||
prompt_tokens: Mapped[int] = mapped_column(Integer, nullable=False)
|
||
completion_tokens: Mapped[int] = mapped_column(Integer, nullable=False)
|
||
time_cost: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
total_tokens: Mapped[int] = mapped_column(Integer, nullable=False)
|
||
cost: Mapped[float] = mapped_column(Float, nullable=False)
|
||
status: Mapped[str] = mapped_column(Text, nullable=False)
|
||
timestamp: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, index=True, default=datetime.datetime.now)
|
||
|
||
__table_args__ = (
|
||
Index("idx_llmusage_model_name", "model_name"),
|
||
Index("idx_llmusage_model_assign_name", "model_assign_name"),
|
||
Index("idx_llmusage_model_api_provider", "model_api_provider"),
|
||
Index("idx_llmusage_time_cost", "time_cost"),
|
||
Index("idx_llmusage_user_id", "user_id"),
|
||
Index("idx_llmusage_request_type", "request_type"),
|
||
Index("idx_llmusage_timestamp", "timestamp"),
|
||
)
|
||
|
||
|
||
class Emoji(Base):
|
||
"""表情包模型"""
|
||
|
||
__tablename__ = "emoji"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
full_path: Mapped[str] = mapped_column(get_string_field(500), nullable=False, unique=True, index=True)
|
||
format: Mapped[str] = mapped_column(Text, nullable=False)
|
||
emoji_hash: Mapped[str] = mapped_column(get_string_field(64), nullable=False, index=True)
|
||
description: Mapped[str] = mapped_column(Text, nullable=False)
|
||
query_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
is_registered: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
is_banned: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
emotion: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
record_time: Mapped[float] = mapped_column(Float, nullable=False)
|
||
register_time: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
usage_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
last_used_time: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
|
||
__table_args__ = (
|
||
Index("idx_emoji_full_path", "full_path"),
|
||
Index("idx_emoji_hash", "emoji_hash"),
|
||
)
|
||
|
||
|
||
class Messages(Base):
|
||
"""消息模型"""
|
||
|
||
__tablename__ = "messages"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
message_id: Mapped[str] = mapped_column(get_string_field(100), nullable=False, index=True)
|
||
time: Mapped[float] = mapped_column(Float, nullable=False)
|
||
chat_id: Mapped[str] = mapped_column(get_string_field(64), nullable=False, index=True)
|
||
reply_to: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
interest_value: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
key_words: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
key_words_lite: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
is_mentioned: Mapped[bool | None] = mapped_column(Boolean, nullable=True)
|
||
|
||
# 从 chat_info 扁平化而来的字段
|
||
chat_info_stream_id: Mapped[str] = mapped_column(Text, nullable=False)
|
||
chat_info_platform: Mapped[str] = mapped_column(Text, nullable=False)
|
||
chat_info_user_platform: Mapped[str] = mapped_column(Text, nullable=False)
|
||
chat_info_user_id: Mapped[str] = mapped_column(Text, nullable=False)
|
||
chat_info_user_nickname: Mapped[str] = mapped_column(Text, nullable=False)
|
||
chat_info_user_cardname: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
chat_info_group_platform: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
chat_info_group_id: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
chat_info_group_name: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
chat_info_create_time: Mapped[float] = mapped_column(Float, nullable=False)
|
||
chat_info_last_active_time: Mapped[float] = mapped_column(Float, nullable=False)
|
||
|
||
# 从顶层 user_info 扁平化而来的字段
|
||
user_platform: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
user_id: Mapped[str | None] = mapped_column(get_string_field(100), nullable=True, index=True)
|
||
user_nickname: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
user_cardname: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
|
||
processed_plain_text: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
display_message: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
memorized_times: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
priority_mode: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
priority_info: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
additional_config: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
is_emoji: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
is_picid: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
is_command: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
is_notify: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
is_public_notice: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
notice_type: Mapped[str | None] = mapped_column(String(50), nullable=True)
|
||
|
||
# 兴趣度系统字段
|
||
actions: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
should_reply: Mapped[bool | None] = mapped_column(Boolean, nullable=True, default=False)
|
||
should_act: Mapped[bool | None] = mapped_column(Boolean, nullable=True, default=False)
|
||
|
||
__table_args__ = (
|
||
Index("idx_messages_message_id", "message_id"),
|
||
Index("idx_messages_chat_id", "chat_id"),
|
||
Index("idx_messages_time", "time"),
|
||
Index("idx_messages_user_id", "user_id"),
|
||
Index("idx_messages_should_reply", "should_reply"),
|
||
Index("idx_messages_should_act", "should_act"),
|
||
)
|
||
|
||
|
||
class ActionRecords(Base):
|
||
"""动作记录模型"""
|
||
|
||
__tablename__ = "action_records"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
action_id: Mapped[str] = mapped_column(get_string_field(100), nullable=False, index=True)
|
||
time: Mapped[float] = mapped_column(Float, nullable=False)
|
||
action_name: Mapped[str] = mapped_column(Text, nullable=False)
|
||
action_data: Mapped[str] = mapped_column(Text, nullable=False)
|
||
action_done: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
action_build_into_prompt: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
action_prompt_display: Mapped[str] = mapped_column(Text, nullable=False)
|
||
chat_id: Mapped[str] = mapped_column(get_string_field(64), nullable=False, index=True)
|
||
chat_info_stream_id: Mapped[str] = mapped_column(Text, nullable=False)
|
||
chat_info_platform: Mapped[str] = mapped_column(Text, nullable=False)
|
||
|
||
__table_args__ = (
|
||
Index("idx_actionrecords_action_id", "action_id"),
|
||
Index("idx_actionrecords_chat_id", "chat_id"),
|
||
Index("idx_actionrecords_time", "time"),
|
||
)
|
||
|
||
|
||
class Images(Base):
|
||
"""图像信息模型"""
|
||
|
||
__tablename__ = "images"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
image_id: Mapped[str] = mapped_column(Text, nullable=False, default="")
|
||
emoji_hash: Mapped[str] = mapped_column(get_string_field(64), nullable=False, index=True)
|
||
description: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
path: Mapped[str] = mapped_column(get_string_field(500), nullable=False, unique=True)
|
||
count: Mapped[int] = mapped_column(Integer, nullable=False, default=1)
|
||
timestamp: Mapped[float] = mapped_column(Float, nullable=False)
|
||
type: Mapped[str] = mapped_column(Text, nullable=False)
|
||
vlm_processed: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
|
||
__table_args__ = (
|
||
Index("idx_images_emoji_hash", "emoji_hash"),
|
||
Index("idx_images_path", "path"),
|
||
)
|
||
|
||
|
||
class ImageDescriptions(Base):
|
||
"""图像描述信息模型"""
|
||
|
||
__tablename__ = "image_descriptions"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
type: Mapped[str] = mapped_column(Text, nullable=False)
|
||
image_description_hash: Mapped[str] = mapped_column(get_string_field(64), nullable=False, index=True)
|
||
description: Mapped[str] = mapped_column(Text, nullable=False)
|
||
timestamp: Mapped[float] = mapped_column(Float, nullable=False)
|
||
|
||
__table_args__ = (Index("idx_imagedesc_hash", "image_description_hash"),)
|
||
|
||
|
||
class Videos(Base):
|
||
"""视频信息模型"""
|
||
|
||
__tablename__ = "videos"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
video_id: Mapped[str] = mapped_column(Text, nullable=False, default="")
|
||
video_hash: Mapped[str] = mapped_column(get_string_field(64), nullable=False, index=True, unique=True)
|
||
description: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
count: Mapped[int] = mapped_column(Integer, nullable=False, default=1)
|
||
timestamp: Mapped[float] = mapped_column(Float, nullable=False)
|
||
vlm_processed: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
|
||
# 视频特有属性
|
||
duration: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
frame_count: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
||
fps: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
resolution: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
file_size: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
||
|
||
__table_args__ = (
|
||
Index("idx_videos_video_hash", "video_hash"),
|
||
Index("idx_videos_timestamp", "timestamp"),
|
||
)
|
||
|
||
|
||
class OnlineTime(Base):
|
||
"""在线时长记录模型"""
|
||
|
||
__tablename__ = "online_time"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
timestamp: Mapped[str] = mapped_column(Text, nullable=False, default=str(datetime.datetime.now))
|
||
duration: Mapped[int] = mapped_column(Integer, nullable=False)
|
||
start_timestamp: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now)
|
||
end_timestamp: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, index=True)
|
||
|
||
__table_args__ = (Index("idx_onlinetime_end_timestamp", "end_timestamp"),)
|
||
|
||
|
||
class PersonInfo(Base):
|
||
"""人物信息模型"""
|
||
|
||
__tablename__ = "person_info"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
person_id: Mapped[str] = mapped_column(get_string_field(100), nullable=False, unique=True, index=True)
|
||
person_name: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
name_reason: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
platform: Mapped[str] = mapped_column(Text, nullable=False)
|
||
user_id: Mapped[str] = mapped_column(get_string_field(50), nullable=False, index=True)
|
||
nickname: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
impression: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
short_impression: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
points: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
forgotten_points: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
info_list: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
know_times: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
know_since: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
last_know: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
attitude: Mapped[int | None] = mapped_column(Integer, nullable=True, default=50)
|
||
|
||
__table_args__ = (
|
||
Index("idx_personinfo_person_id", "person_id"),
|
||
Index("idx_personinfo_user_id", "user_id"),
|
||
)
|
||
|
||
|
||
class BotPersonalityInterests(Base):
|
||
"""机器人人格兴趣标签模型"""
|
||
|
||
__tablename__ = "bot_personality_interests"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
personality_id: Mapped[str] = mapped_column(get_string_field(100), nullable=False, index=True)
|
||
personality_description: Mapped[str] = mapped_column(Text, nullable=False)
|
||
interest_tags: Mapped[str] = mapped_column(Text, nullable=False)
|
||
embedding_model: Mapped[str] = mapped_column(get_string_field(100), nullable=False, default="text-embedding-ada-002")
|
||
version: Mapped[int] = mapped_column(Integer, nullable=False, default=1)
|
||
last_updated: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now, index=True)
|
||
|
||
__table_args__ = (
|
||
Index("idx_botpersonality_personality_id", "personality_id"),
|
||
Index("idx_botpersonality_version", "version"),
|
||
Index("idx_botpersonality_last_updated", "last_updated"),
|
||
)
|
||
|
||
|
||
class Memory(Base):
|
||
"""记忆模型"""
|
||
|
||
__tablename__ = "memory"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
memory_id: Mapped[str] = mapped_column(get_string_field(64), nullable=False, index=True)
|
||
chat_id: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
memory_text: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
keywords: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
create_time: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
last_view_time: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
|
||
__table_args__ = (Index("idx_memory_memory_id", "memory_id"),)
|
||
|
||
|
||
class Expression(Base):
|
||
"""表达风格模型"""
|
||
|
||
__tablename__ = "expression"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
situation: Mapped[str] = mapped_column(Text, nullable=False)
|
||
style: Mapped[str] = mapped_column(Text, nullable=False)
|
||
count: Mapped[float] = mapped_column(Float, nullable=False)
|
||
last_active_time: Mapped[float] = mapped_column(Float, nullable=False)
|
||
chat_id: Mapped[str] = mapped_column(get_string_field(64), nullable=False, index=True)
|
||
type: Mapped[str] = mapped_column(Text, nullable=False)
|
||
create_date: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||
|
||
__table_args__ = (Index("idx_expression_chat_id", "chat_id"),)
|
||
|
||
|
||
class ThinkingLog(Base):
|
||
"""思考日志模型"""
|
||
|
||
__tablename__ = "thinking_logs"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
chat_id: Mapped[str] = mapped_column(get_string_field(64), nullable=False, index=True)
|
||
trigger_text: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
response_text: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
trigger_info_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
response_info_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
timing_results_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
chat_history_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
chat_history_in_thinking_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
chat_history_after_response_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
heartflow_data_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
reasoning_data_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
created_at: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now)
|
||
|
||
__table_args__ = (Index("idx_thinkinglog_chat_id", "chat_id"),)
|
||
|
||
|
||
class GraphNodes(Base):
|
||
"""记忆图节点模型"""
|
||
|
||
__tablename__ = "graph_nodes"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
concept: Mapped[str] = mapped_column(get_string_field(255), nullable=False, unique=True, index=True)
|
||
memory_items: Mapped[str] = mapped_column(Text, nullable=False)
|
||
hash: Mapped[str] = mapped_column(Text, nullable=False)
|
||
weight: Mapped[float] = mapped_column(Float, nullable=False, default=1.0)
|
||
created_time: Mapped[float] = mapped_column(Float, nullable=False)
|
||
last_modified: Mapped[float] = mapped_column(Float, nullable=False)
|
||
|
||
__table_args__ = (Index("idx_graphnodes_concept", "concept"),)
|
||
|
||
|
||
class GraphEdges(Base):
|
||
"""记忆图边模型"""
|
||
|
||
__tablename__ = "graph_edges"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
source: Mapped[str] = mapped_column(get_string_field(255), nullable=False, index=True)
|
||
target: Mapped[str] = mapped_column(get_string_field(255), nullable=False, index=True)
|
||
strength: Mapped[int] = mapped_column(Integer, nullable=False)
|
||
hash: Mapped[str] = mapped_column(Text, nullable=False)
|
||
created_time: Mapped[float] = mapped_column(Float, nullable=False)
|
||
last_modified: Mapped[float] = mapped_column(Float, nullable=False)
|
||
|
||
__table_args__ = (
|
||
Index("idx_graphedges_source", "source"),
|
||
Index("idx_graphedges_target", "target"),
|
||
)
|
||
|
||
|
||
class Schedule(Base):
|
||
"""日程模型"""
|
||
|
||
__tablename__ = "schedule"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
date: Mapped[str] = mapped_column(get_string_field(10), nullable=False, unique=True, index=True)
|
||
schedule_data: Mapped[str] = mapped_column(Text, nullable=False)
|
||
created_at: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now)
|
||
updated_at: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now, onupdate=datetime.datetime.now)
|
||
|
||
__table_args__ = (Index("idx_schedule_date", "date"),)
|
||
|
||
|
||
class MaiZoneScheduleStatus(Base):
|
||
"""麦麦空间日程处理状态模型"""
|
||
|
||
__tablename__ = "maizone_schedule_status"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
datetime_hour: Mapped[str] = mapped_column(get_string_field(13), nullable=False, unique=True, index=True)
|
||
activity: Mapped[str] = mapped_column(Text, nullable=False)
|
||
is_processed: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
processed_at: Mapped[datetime.datetime | None] = mapped_column(DateTime, nullable=True)
|
||
story_content: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
send_success: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
created_at: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now)
|
||
updated_at: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now, onupdate=datetime.datetime.now)
|
||
|
||
__table_args__ = (
|
||
Index("idx_maizone_datetime_hour", "datetime_hour"),
|
||
Index("idx_maizone_is_processed", "is_processed"),
|
||
)
|
||
|
||
|
||
class BanUser(Base):
|
||
"""被禁用用户模型
|
||
|
||
使用 SQLAlchemy 2.0 类型标注写法，方便静态类型检查器识别实际字段类型，
|
||
避免在业务代码中对属性赋值时报 `Column[...]` 不可赋值的告警。
|
||
"""
|
||
|
||
__tablename__ = "ban_users"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
platform: Mapped[str] = mapped_column(get_string_field(50), nullable=False) # 使用有限长度，以便创建索引
|
||
user_id: Mapped[str] = mapped_column(get_string_field(50), nullable=False, index=True)
|
||
violation_num: Mapped[int] = mapped_column(Integer, nullable=False, default=0, index=True)
|
||
reason: Mapped[str] = mapped_column(Text, nullable=False)
|
||
created_at: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now)
|
||
|
||
__table_args__ = (
|
||
Index("idx_violation_num", "violation_num"),
|
||
Index("idx_banuser_user_id", "user_id"),
|
||
Index("idx_banuser_platform", "platform"),
|
||
Index("idx_banuser_platform_user_id", "platform", "user_id"),
|
||
)
|
||
|
||
|
||
class AntiInjectionStats(Base):
|
||
"""反注入系统统计模型"""
|
||
|
||
__tablename__ = "anti_injection_stats"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
total_messages: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
"""总处理消息数"""
|
||
|
||
detected_injections: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
"""检测到的注入攻击数"""
|
||
|
||
blocked_messages: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
"""被阻止的消息数"""
|
||
|
||
shielded_messages: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
"""被加盾的消息数"""
|
||
|
||
processing_time_total: Mapped[float] = mapped_column(Float, nullable=False, default=0.0)
|
||
"""总处理时间"""
|
||
|
||
total_process_time: Mapped[float] = mapped_column(Float, nullable=False, default=0.0)
|
||
"""累计总处理时间"""
|
||
|
||
last_process_time: Mapped[float] = mapped_column(Float, nullable=False, default=0.0)
|
||
"""最近一次处理时间"""
|
||
|
||
error_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
"""错误计数"""
|
||
|
||
start_time: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now)
|
||
"""统计开始时间"""
|
||
|
||
created_at: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now)
|
||
"""记录创建时间"""
|
||
|
||
updated_at: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now, onupdate=datetime.datetime.now)
|
||
"""记录更新时间"""
|
||
|
||
__table_args__ = (
|
||
Index("idx_anti_injection_stats_created_at", "created_at"),
|
||
Index("idx_anti_injection_stats_updated_at", "updated_at"),
|
||
)
|
||
|
||
|
||
class CacheEntries(Base):
|
||
"""工具缓存条目模型"""
|
||
|
||
__tablename__ = "cache_entries"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
cache_key: Mapped[str] = mapped_column(get_string_field(500), nullable=False, unique=True, index=True)
|
||
"""缓存键，包含工具名、参数和代码哈希"""
|
||
|
||
cache_value: Mapped[str] = mapped_column(Text, nullable=False)
|
||
"""缓存的数据，JSON格式"""
|
||
|
||
expires_at: Mapped[float] = mapped_column(Float, nullable=False, index=True)
|
||
"""过期时间戳"""
|
||
|
||
tool_name: Mapped[str] = mapped_column(get_string_field(100), nullable=False, index=True)
|
||
"""工具名称"""
|
||
|
||
created_at: Mapped[float] = mapped_column(Float, nullable=False, default=lambda: time.time())
|
||
"""创建时间戳"""
|
||
|
||
last_accessed: Mapped[float] = mapped_column(Float, nullable=False, default=lambda: time.time())
|
||
"""最后访问时间戳"""
|
||
|
||
access_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
"""访问次数"""
|
||
|
||
__table_args__ = (
|
||
Index("idx_cache_entries_key", "cache_key"),
|
||
Index("idx_cache_entries_expires_at", "expires_at"),
|
||
Index("idx_cache_entries_tool_name", "tool_name"),
|
||
Index("idx_cache_entries_created_at", "created_at"),
|
||
)
|
||
|
||
|
||
class MonthlyPlan(Base):
|
||
"""月度计划模型"""
|
||
|
||
__tablename__ = "monthly_plans"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
plan_text: Mapped[str] = mapped_column(Text, nullable=False)
|
||
target_month: Mapped[str] = mapped_column(String(7), nullable=False, index=True)
|
||
status: Mapped[str] = mapped_column(get_string_field(20), nullable=False, default="active", index=True)
|
||
usage_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
last_used_date: Mapped[str | None] = mapped_column(String(10), nullable=True, index=True)
|
||
created_at: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False, default=datetime.datetime.now)
|
||
is_deleted: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False, index=True)
|
||
|
||
__table_args__ = (
|
||
Index("idx_monthlyplan_target_month_status", "target_month", "status"),
|
||
Index("idx_monthlyplan_last_used_date", "last_used_date"),
|
||
Index("idx_monthlyplan_usage_count", "usage_count"),
|
||
)
|
||
|
||
|
||
class PermissionNodes(Base):
|
||
"""权限节点模型"""
|
||
|
||
__tablename__ = "permission_nodes"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
node_name: Mapped[str] = mapped_column(get_string_field(255), nullable=False, unique=True, index=True)
|
||
description: Mapped[str] = mapped_column(Text, nullable=False)
|
||
plugin_name: Mapped[str] = mapped_column(get_string_field(100), nullable=False, index=True)
|
||
default_granted: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)
|
||
created_at: Mapped[datetime.datetime] = mapped_column(DateTime, default=datetime.datetime.utcnow, nullable=False)
|
||
|
||
__table_args__ = (
|
||
Index("idx_permission_plugin", "plugin_name"),
|
||
Index("idx_permission_node", "node_name"),
|
||
)
|
||
|
||
|
||
class UserPermissions(Base):
|
||
"""用户权限模型"""
|
||
|
||
__tablename__ = "user_permissions"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
platform: Mapped[str] = mapped_column(get_string_field(50), nullable=False, index=True)
|
||
user_id: Mapped[str] = mapped_column(get_string_field(100), nullable=False, index=True)
|
||
permission_node: Mapped[str] = mapped_column(get_string_field(255), nullable=False, index=True)
|
||
granted: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
||
granted_at: Mapped[datetime.datetime] = mapped_column(DateTime, default=datetime.datetime.utcnow, nullable=False)
|
||
granted_by: Mapped[str | None] = mapped_column(get_string_field(100), nullable=True)
|
||
|
||
__table_args__ = (
|
||
Index("idx_user_platform_id", "platform", "user_id"),
|
||
Index("idx_user_permission", "platform", "user_id", "permission_node"),
|
||
Index("idx_permission_granted", "permission_node", "granted"),
|
||
)
|
||
|
||
|
||
class UserRelationships(Base):
|
||
"""用户关系模型 - 存储用户与bot的关系数据
|
||
|
||
核心字段：
|
||
- relationship_text: 当前印象描述（用于兼容旧系统，逐步迁移到 impression_text）
|
||
- impression_text: 长期印象（新字段，自然叙事风格）
|
||
- preference_keywords: 用户偏好关键词
|
||
- relationship_score: 好感度分数(0-1)
|
||
- key_facts: 关键信息JSON（生日、职业、理想等）
|
||
- relationship_stage: 关系阶段（stranger/acquaintance/friend/close_friend/bestie）
|
||
"""
|
||
|
||
__tablename__ = "user_relationships"
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
user_id: Mapped[str] = mapped_column(get_string_field(100), nullable=False, unique=True, index=True)
|
||
user_name: Mapped[str | None] = mapped_column(get_string_field(100), nullable=True)
|
||
user_aliases: Mapped[str | None] = mapped_column(Text, nullable=True) # 用户别名，逗号分隔
|
||
|
||
# 印象相关（新旧兼容）
|
||
relationship_text: Mapped[str | None] = mapped_column(Text, nullable=True) # 旧字段，保持兼容
|
||
impression_text: Mapped[str | None] = mapped_column(Text, nullable=True) # 新字段：长期印象（自然叙事）
|
||
|
||
# 用户信息
|
||
preference_keywords: Mapped[str | None] = mapped_column(Text, nullable=True) # 用户偏好关键词，逗号分隔
|
||
key_facts: Mapped[str | None] = mapped_column(Text, nullable=True) # 关键信息JSON（生日、职业等）
|
||
|
||
# 关系状态
|
||
relationship_score: Mapped[float] = mapped_column(Float, nullable=False, default=0.3) # 好感度(0-1)
|
||
relationship_stage: Mapped[str | None] = mapped_column(get_string_field(50), nullable=True, default="stranger") # 关系阶段
|
||
|
||
# 时间记录
|
||
first_met_time: Mapped[float | None] = mapped_column(Float, nullable=True) # 首次认识时间戳
|
||
last_impression_update: Mapped[float | None] = mapped_column(Float, nullable=True) # 上次更新印象时间
|
||
last_updated: Mapped[float] = mapped_column(Float, nullable=False, default=time.time)
|
||
created_at: Mapped[datetime.datetime] = mapped_column(DateTime, default=datetime.datetime.utcnow, nullable=False)
|
||
|
||
__table_args__ = (
|
||
Index("idx_user_relationship_id", "user_id"),
|
||
Index("idx_relationship_score", "relationship_score"),
|
||
Index("idx_relationship_updated", "last_updated"),
|
||
Index("idx_relationship_stage", "relationship_stage"),
|
||
)
|