mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-02 19:55:18 +02:00
Add migration 96 for GITHUB_MODELS enum
This commit is contained in:
parent
515be6b648
commit
fc946ee1e5
1 changed files with 35 additions and 0 deletions
|
|
@ -0,0 +1,35 @@
|
||||||
|
"""Add GITHUB_MODELS to LiteLLMProvider enum
|
||||||
|
|
||||||
|
Revision ID: 96
|
||||||
|
Revises: 95
|
||||||
|
"""
|
||||||
|
|
||||||
|
from collections.abc import Sequence
|
||||||
|
|
||||||
|
from alembic import op
|
||||||
|
|
||||||
|
revision: str = "96"
|
||||||
|
down_revision: str | None = "95"
|
||||||
|
branch_labels: str | Sequence[str] | None = None
|
||||||
|
depends_on: str | Sequence[str] | None = None
|
||||||
|
|
||||||
|
|
||||||
|
def upgrade() -> None:
|
||||||
|
op.execute(
|
||||||
|
"""
|
||||||
|
DO $$
|
||||||
|
BEGIN
|
||||||
|
IF NOT EXISTS (
|
||||||
|
SELECT 1 FROM pg_enum
|
||||||
|
WHERE enumtypid = 'litellmprovider'::regtype
|
||||||
|
AND enumlabel = 'GITHUB_MODELS'
|
||||||
|
) THEN
|
||||||
|
ALTER TYPE litellmprovider ADD VALUE 'GITHUB_MODELS';
|
||||||
|
END IF;
|
||||||
|
END$$;
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def downgrade() -> None:
|
||||||
|
pass
|
||||||
Loading…
Add table
Add a link
Reference in a new issue