mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-22 08:38:13 +02:00
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
"""add public_access_token
|
|
|
|
Revision ID: 181475b2a1a1
|
|
Revises: dc33eef8dabe
|
|
Create Date: 2026-01-23 17:37:54.449308
|
|
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "181475b2a1a1"
|
|
down_revision: Union[str, None] = "dc33eef8dabe"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_index(op.f("ix_api_keys_key_hash"), table_name="api_keys")
|
|
op.create_index("ix_api_keys_key_hash", "api_keys", ["key_hash"], unique=False)
|
|
op.create_index(
|
|
"ix_kb_chunks_embedding_model",
|
|
"knowledge_base_chunks",
|
|
["embedding_model"],
|
|
unique=False,
|
|
)
|
|
op.add_column(
|
|
"workflow_runs",
|
|
sa.Column("public_access_token", sa.String(length=36), nullable=True),
|
|
)
|
|
op.create_index(
|
|
"idx_workflow_runs_public_access_token",
|
|
"workflow_runs",
|
|
["public_access_token"],
|
|
unique=True,
|
|
postgresql_where=sa.text("public_access_token IS NOT NULL"),
|
|
)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_index(
|
|
"idx_workflow_runs_public_access_token",
|
|
table_name="workflow_runs",
|
|
postgresql_where=sa.text("public_access_token IS NOT NULL"),
|
|
)
|
|
op.drop_column("workflow_runs", "public_access_token")
|
|
op.drop_index("ix_kb_chunks_embedding_model", table_name="knowledge_base_chunks")
|
|
op.drop_index("ix_api_keys_key_hash", table_name="api_keys")
|
|
op.create_index(op.f("ix_api_keys_key_hash"), "api_keys", ["key_hash"], unique=True)
|
|
# ### end Alembic commands ###
|