|
|
@@ -0,0 +1,73 @@
|
|
|
+"""init memory models
|
|
|
+
|
|
|
+Revision ID: 20260425_0001
|
|
|
+Revises:
|
|
|
+Create Date: 2026-04-25 13:20:00
|
|
|
+"""
|
|
|
+
|
|
|
+from collections.abc import Sequence
|
|
|
+
|
|
|
+from alembic import op
|
|
|
+import sqlalchemy as sa
|
|
|
+
|
|
|
+
|
|
|
+revision: str = "20260425_0001"
|
|
|
+down_revision: str | None = None
|
|
|
+branch_labels: Sequence[str] | None = None
|
|
|
+depends_on: Sequence[str] | None = None
|
|
|
+
|
|
|
+
|
|
|
+def upgrade() -> None:
|
|
|
+ op.create_table(
|
|
|
+ "memory_item",
|
|
|
+ sa.Column("scope_type", sa.String(length=32), nullable=False),
|
|
|
+ sa.Column("scope_id", sa.String(length=64), nullable=False),
|
|
|
+ sa.Column("memory_type", sa.String(length=64), nullable=False),
|
|
|
+ sa.Column("content_text", sa.Text(), nullable=False),
|
|
|
+ sa.Column("content_json", sa.JSON(), nullable=True),
|
|
|
+ sa.Column("metadata_json", sa.JSON(), nullable=False),
|
|
|
+ sa.Column("owner_agent_id", sa.String(length=36), nullable=True),
|
|
|
+ sa.Column("user_id", sa.String(length=36), nullable=True),
|
|
|
+ sa.Column("session_id", sa.String(length=36), nullable=True),
|
|
|
+ sa.Column("source_ref", sa.String(length=256), nullable=True),
|
|
|
+ sa.Column("importance_score", sa.Integer(), nullable=False),
|
|
|
+ sa.Column("status", sa.String(length=32), nullable=False),
|
|
|
+ sa.Column("last_accessed_time", sa.DateTime(), nullable=True),
|
|
|
+ sa.Column("expires_time", sa.DateTime(), nullable=True),
|
|
|
+ sa.Column("id", sa.String(length=36), nullable=False),
|
|
|
+ sa.Column("tenant_id", sa.String(length=36), nullable=False),
|
|
|
+ sa.Column("created_by", sa.String(length=36), nullable=True),
|
|
|
+ sa.Column("updated_by", sa.String(length=36), nullable=True),
|
|
|
+ sa.Column("created_time", sa.DateTime(), nullable=False),
|
|
|
+ sa.Column("updated_time", sa.DateTime(), nullable=False),
|
|
|
+ sa.Column("deleted_time", sa.DateTime(), nullable=True),
|
|
|
+ sa.Column("version", sa.Integer(), nullable=False),
|
|
|
+ sa.PrimaryKeyConstraint("id"),
|
|
|
+ )
|
|
|
+ op.create_index("ix_memory_item_memory_type", "memory_item", ["memory_type"], unique=False)
|
|
|
+ op.create_index("ix_memory_item_owner_agent_id", "memory_item", ["owner_agent_id"], unique=False)
|
|
|
+ op.create_index("ix_memory_item_scope_id", "memory_item", ["scope_id"], unique=False)
|
|
|
+ op.create_index("ix_memory_item_scope_type", "memory_item", ["scope_type"], unique=False)
|
|
|
+ op.create_index("ix_memory_item_session_id", "memory_item", ["session_id"], unique=False)
|
|
|
+ op.create_index("ix_memory_item_status", "memory_item", ["status"], unique=False)
|
|
|
+ op.create_index("ix_memory_item_tenant_id", "memory_item", ["tenant_id"], unique=False)
|
|
|
+ op.create_index("ix_memory_item_user_id", "memory_item", ["user_id"], unique=False)
|
|
|
+ op.create_index(
|
|
|
+ "ix_memory_item_search_scope",
|
|
|
+ "memory_item",
|
|
|
+ ["tenant_id", "scope_type", "scope_id", "status"],
|
|
|
+ unique=False,
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def downgrade() -> None:
|
|
|
+ op.drop_index("ix_memory_item_search_scope", table_name="memory_item")
|
|
|
+ op.drop_index("ix_memory_item_user_id", table_name="memory_item")
|
|
|
+ op.drop_index("ix_memory_item_tenant_id", table_name="memory_item")
|
|
|
+ op.drop_index("ix_memory_item_status", table_name="memory_item")
|
|
|
+ op.drop_index("ix_memory_item_session_id", table_name="memory_item")
|
|
|
+ op.drop_index("ix_memory_item_scope_type", table_name="memory_item")
|
|
|
+ op.drop_index("ix_memory_item_scope_id", table_name="memory_item")
|
|
|
+ op.drop_index("ix_memory_item_owner_agent_id", table_name="memory_item")
|
|
|
+ op.drop_index("ix_memory_item_memory_type", table_name="memory_item")
|
|
|
+ op.drop_table("memory_item")
|