"""创建客户长期记忆表。"""

from alembic import op

from zbt.infrastructure.mysql.agent_models import AgentBase

revision = "agent_0002"
down_revision = "agent_0001"
branch_labels = None
depends_on = None


def upgrade() -> None:
    bind = op.get_bind()
    AgentBase.metadata.tables["customer_memories"].create(bind=bind)


def downgrade() -> None:
    bind = op.get_bind()
    AgentBase.metadata.tables["customer_memories"].drop(bind=bind)