mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-06-01 18:49:27 +02:00
28 lines
721 B
Python
28 lines
721 B
Python
"""Add persona to chat_session
|
|
|
|
Revision ID: e86866a9c78a
|
|
Revises: 80696cf850ae
|
|
Create Date: 2023-11-26 02:51:47.657357
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "e86866a9c78a"
|
|
down_revision = "80696cf850ae"
|
|
branch_labels: None = None
|
|
depends_on: None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.add_column("chat_session", sa.Column("persona_id", sa.Integer(), nullable=True))
|
|
op.create_foreign_key(
|
|
"fk_chat_session_persona_id", "chat_session", "persona", ["persona_id"], ["id"]
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_constraint("fk_chat_session_persona_id", "chat_session", type_="foreignkey")
|
|
op.drop_column("chat_session", "persona_id")
|