mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-05-31 10:10:21 +02:00
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
"""Added input prompts
|
|
|
|
Revision ID: e1392f05e840
|
|
Revises: 08a1eda20fe1
|
|
Create Date: 2024-07-13 19:09:22.556224
|
|
|
|
"""
|
|
|
|
import fastapi_users_db_sqlalchemy
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "e1392f05e840"
|
|
down_revision = "08a1eda20fe1"
|
|
branch_labels: None = None
|
|
depends_on: None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"inputprompt",
|
|
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column("prompt", sa.String(), nullable=False),
|
|
sa.Column("content", sa.String(), nullable=False),
|
|
sa.Column("active", sa.Boolean(), nullable=False),
|
|
sa.Column("is_public", sa.Boolean(), nullable=False),
|
|
sa.Column(
|
|
"user_id",
|
|
fastapi_users_db_sqlalchemy.generics.GUID(),
|
|
nullable=True,
|
|
),
|
|
sa.ForeignKeyConstraint(
|
|
["user_id"],
|
|
["user.id"],
|
|
),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_table(
|
|
"inputprompt__user",
|
|
sa.Column("input_prompt_id", sa.Integer(), nullable=False),
|
|
sa.Column("user_id", sa.Integer(), nullable=False),
|
|
sa.ForeignKeyConstraint(
|
|
["input_prompt_id"],
|
|
["inputprompt.id"],
|
|
),
|
|
sa.ForeignKeyConstraint(
|
|
["user_id"],
|
|
["inputprompt.id"],
|
|
),
|
|
sa.PrimaryKeyConstraint("input_prompt_id", "user_id"),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table("inputprompt__user")
|
|
op.drop_table("inputprompt")
|