From 11b6ea742de2fb1f0501f2412173395cbfc220aa Mon Sep 17 00:00:00 2001 From: Stream Date: Fri, 22 Aug 2025 20:43:49 +0800 Subject: [PATCH] feat: add index for data tables --- api/models/chatflow_memory.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/api/models/chatflow_memory.py b/api/models/chatflow_memory.py index f0b36d0d69..2c964c4766 100644 --- a/api/models/chatflow_memory.py +++ b/api/models/chatflow_memory.py @@ -10,6 +10,10 @@ from .types import StringUUID class ChatflowMemoryVariable(Base): __tablename__ = "chatflow_memory_variables" + __table_args__ = ( + sa.PrimaryKeyConstraint("id", name="chatflow_memory_variables_pkey"), + sa.Index("chatflow_memory_variables_memory_id_idx", "tenant_id", "app_id", "node_id", "memory_id"), + ) id: Mapped[str] = mapped_column(StringUUID, primary_key=True, server_default=sa.text("uuid_generate_v4()")) tenant_id: Mapped[str] = mapped_column(StringUUID, nullable=False) @@ -30,6 +34,13 @@ class ChatflowMemoryVariable(Base): class ChatflowConversation(Base): __tablename__ = "chatflow_conversations" + __table_args__ = ( + sa.PrimaryKeyConstraint("id", name="chatflow_conversations_pkey"), + sa.Index( + "chatflow_conversations_original_conversation_id_idx", + "tenant_id", "app_id", "node_id", "original_conversation_id" + ), + ) id: Mapped[str] = mapped_column(StringUUID, primary_key=True, server_default=sa.text("uuid_generate_v4()")) tenant_id: Mapped[str] = mapped_column(StringUUID, nullable=False) @@ -46,6 +57,10 @@ class ChatflowConversation(Base): class ChatflowMessage(Base): __tablename__ = "chatflow_messages" + __table_args__ = ( + sa.PrimaryKeyConstraint("id", name="chatflow_messages_pkey"), + sa.Index("chatflow_messages_version_idx", "conversation_id", "index", "version"), + ) id: Mapped[str] = mapped_column(StringUUID, primary_key=True, server_default=sa.text("uuid_generate_v4()")) conversation_id: Mapped[str] = mapped_column(StringUUID, nullable=False)