33 lines
738 B
Python
33 lines
738 B
Python
|
"""add assistant knowledge retrieval config
|
||
|
|
|
||
|
|
Revision ID: 20260712_0006
|
||
|
|
Revises: 20260712_0005
|
||
|
|
"""
|
||
|
|
from collections.abc import Sequence
|
||
|
|
|
||
|
|
from alembic import op
|
||
|
|
import sqlalchemy as sa
|
||
|
|
|
||
|
|
revision: str = "20260712_0006"
|
||
|
|
down_revision: str | Sequence[str] | None = "20260712_0005"
|
||
|
|
branch_labels = None
|
||
|
|
depends_on = None
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade() -> None:
|
||
|
|
op.add_column(
|
||
|
|
"assistants",
|
||
|
|
sa.Column(
|
||
|
|
"knowledge_retrieval_config",
|
||
|
|
sa.JSON(),
|
||
|
|
server_default=sa.text(
|
||
|
|
"'{\"mode\": \"automatic\", \"top_n\": 5, \"score_threshold\": 0.0}'"
|
||
|
|
),
|
||
|
|
nullable=False,
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade() -> None:
|
||
|
|
op.drop_column("assistants", "knowledge_retrieval_config")
|