25d54dd896
* auto fixes * auto fix pt2 and transitive deps and undefined var checking locals() * manual fixes (ignored or letta-code fixed) * fix circular import
35 lines
868 B
Python
35 lines
868 B
Python
"""add_index_on_messages_step_id
|
|
|
|
Revision ID: d798609d65ff
|
|
Revises: 89fd4648866b
|
|
Create Date: 2025-11-07 15:43:59.446292
|
|
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
from letta.settings import settings
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "d798609d65ff"
|
|
down_revision: Union[str, None] = "89fd4648866b"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Skip this migration for SQLite
|
|
if not settings.letta_pg_uri_no_default:
|
|
return
|
|
|
|
op.create_index("idx_messages_step_id", "messages", ["step_id"], if_not_exists=True)
|
|
|
|
|
|
def downgrade() -> None:
|
|
# Skip this migration for SQLite
|
|
if not settings.letta_pg_uri_no_default:
|
|
return
|
|
|
|
op.drop_index("idx_messages_step_id", table_name="messages", if_exists=True)
|