mirror of
https://github.com/YFGaia/dify-plus.git
synced 2026-06-04 10:14:00 +08:00
npc0-hue
51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
from extensions.ext_database import db
|
|
|
|
from .types import StringUUID
|
|
|
|
|
|
class EndUserAccountJoinsExtend(db.Model):
|
|
__tablename__ = "end_user_account_joins_extend"
|
|
__table_args__ = (
|
|
db.PrimaryKeyConstraint("id", name="end_user_account_joins_pkey"),
|
|
db.Index("end_user_account_joins_account_id_idx", "account_id"),
|
|
db.Index("end_user_account_joins_end_user_id_idx", "end_user_id"), # 单独索引，用于计费查询优化
|
|
db.Index("end_user_account_joins_end_user_id_app_id_idx", "end_user_id", "app_id"),
|
|
)
|
|
|
|
id = db.Column(StringUUID, server_default=db.text("uuid_generate_v4()"))
|
|
end_user_id = db.Column(StringUUID, nullable=False)
|
|
account_id = db.Column(StringUUID, nullable=False)
|
|
app_id = db.Column(StringUUID, nullable=False)
|
|
created_at = db.Column(db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)"))
|
|
updated_at = db.Column(db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)"))
|
|
|
|
|
|
# Extend: 记忆上下文功能
|
|
class AppExtend(db.Model):
|
|
__tablename__ = "app_extend"
|
|
__table_args__ = (
|
|
db.PrimaryKeyConstraint("id", name="app_extend_joins_pkey"),
|
|
db.Index("app_extend_id_app_id_idx", "app_id"),
|
|
)
|
|
|
|
id = db.Column(StringUUID, server_default=db.text("uuid_generate_v4()"))
|
|
app_id = db.Column(StringUUID, nullable=False)
|
|
retention_number = db.Column(db.Integer, nullable=True)
|
|
# Extend: 记忆上下文功能
|
|
|
|
|
|
# Extend: 消息上下文分割功能
|
|
class MessageContextExtend(db.Model):
|
|
__tablename__ = "message_context_extend"
|
|
__table_args__ = (
|
|
db.PrimaryKeyConstraint("id", name="message_context_extend_joins_pkey"),
|
|
db.Index("message_context_conversation_id_idx", "conversation_id"),
|
|
db.Index("message_context_created_at_idx", "created_at"),
|
|
)
|
|
|
|
id = db.Column(StringUUID, server_default=db.text("uuid_generate_v4()"))
|
|
created_at = db.Column(db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)"))
|
|
conversation_id = db.Column(db.String(36), nullable=True)
|
|
message_id = db.Column(db.String(36), nullable=False)
|
|
# Extend: 消息上下文分割功能
|