fix: support vision prompt modes

2026-08-12 19:13:04 +08:00
parent 09b07751b6
commit 94a02d7e2b
29 changed files with 676 additions and 439 deletions
@@ -0,0 +1,47 @@
+"""replace vision focus with explicit prompt modes
+
+Revision ID: 20260812_0020
+Revises: 20260812_0019
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+import sqlalchemy as sa
+
+
+revision: str = "20260812_0020"
+down_revision: str | Sequence[str] | None = "20260812_0019"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.alter_column(
+        "assistants",
+        "vision_focus_prompt",
+        new_column_name="vision_prompt",
+    )
+    op.add_column(
+        "assistants",
+        sa.Column(
+            "vision_prompt_mode",
+            sa.String(length=16),
+            server_default="runtime",
+            nullable=False,
+        ),
+    )
+    # The previous single-field implementation behaved most like guided mode.
+    op.execute(
+        "UPDATE assistants SET vision_prompt_mode = 'guided' "
+        "WHERE TRIM(vision_prompt) <> ''"
+    )
+
+
+def downgrade() -> None:
+    op.drop_column("assistants", "vision_prompt_mode")
+    op.alter_column(
+        "assistants",
+        "vision_prompt",
+        new_column_name="vision_focus_prompt",
+    )