From fc946ee1e5195220fa1e40c77fd728380dac098e Mon Sep 17 00:00:00 2001
From: CREDO23 <thierrybakera12@gmail.com>
Date: Mon, 9 Feb 2026 17:30:52 +0200
Subject: [PATCH] Add migration 96 for GITHUB_MODELS enum

---
 ...d_github_models_to_litellmprovider_enum.py | 35 +++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 surfsense_backend/alembic/versions/96_add_github_models_to_litellmprovider_enum.py

diff --git a/surfsense_backend/alembic/versions/96_add_github_models_to_litellmprovider_enum.py b/surfsense_backend/alembic/versions/96_add_github_models_to_litellmprovider_enum.py
new file mode 100644
index 000000000..11bcfb82f
--- /dev/null
+++ b/surfsense_backend/alembic/versions/96_add_github_models_to_litellmprovider_enum.py
@@ -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