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
index 11bcfb82f..c41935b8c 100644
--- 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
@@ -15,20 +15,8 @@ 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$$;
-        """
-    )
+    op.execute("COMMIT")
+    op.execute("ALTER TYPE litellmprovider ADD VALUE IF NOT EXISTS 'GITHUB_MODELS'")
 
 
 def downgrade() -> None:
diff --git a/surfsense_backend/app/agents/new_chat/llm_config.py b/surfsense_backend/app/agents/new_chat/llm_config.py
index b7305cb4b..4ffc6e623 100644
--- a/surfsense_backend/app/agents/new_chat/llm_config.py
+++ b/surfsense_backend/app/agents/new_chat/llm_config.py
@@ -45,7 +45,7 @@ PROVIDER_MAP = {
     "ALIBABA_QWEN": "openai",
     "MOONSHOT": "openai",
     "ZHIPU": "openai",
-    "GITHUB_MODELS": "openai",
+    "GITHUB_MODELS": "github",
     "REPLICATE": "replicate",
     "PERPLEXITY": "perplexity",
     "ANYSCALE": "anyscale",
diff --git a/surfsense_backend/app/services/llm_router_service.py b/surfsense_backend/app/services/llm_router_service.py
index 0e7d0b1a1..58a75cf54 100644
--- a/surfsense_backend/app/services/llm_router_service.py
+++ b/surfsense_backend/app/services/llm_router_service.py
@@ -56,7 +56,7 @@ PROVIDER_MAP = {
     "ALIBABA_QWEN": "openai",
     "MOONSHOT": "openai",
     "ZHIPU": "openai",
-    "GITHUB_MODELS": "openai",
+    "GITHUB_MODELS": "github",
     "HUGGINGFACE": "huggingface",
     "CUSTOM": "custom",
 }
diff --git a/surfsense_backend/app/services/llm_service.py b/surfsense_backend/app/services/llm_service.py
index 8bd4da57b..0a805b31f 100644
--- a/surfsense_backend/app/services/llm_service.py
+++ b/surfsense_backend/app/services/llm_service.py
@@ -119,7 +119,7 @@ async def validate_llm_config(
                 "ALIBABA_QWEN": "openai",
                 "MOONSHOT": "openai",
                 "ZHIPU": "openai",  # GLM needs special handling
-                "GITHUB_MODELS": "openai",
+                "GITHUB_MODELS": "github",
             }
             provider_prefix = provider_map.get(provider, provider.lower())
             model_string = f"{provider_prefix}/{model_name}"
@@ -336,7 +336,7 @@ async def get_search_space_llm_instance(
                 "ALIBABA_QWEN": "openai",
                 "MOONSHOT": "openai",
                 "ZHIPU": "openai",
-                "GITHUB_MODELS": "openai",
+                "GITHUB_MODELS": "github",
             }
             provider_prefix = provider_map.get(
                 llm_config.provider.value, llm_config.provider.value.lower()
diff --git a/surfsense_web/contracts/types/new-llm-config.types.ts b/surfsense_web/contracts/types/new-llm-config.types.ts
index b99df1022..8239505a1 100644
--- a/surfsense_web/contracts/types/new-llm-config.types.ts
+++ b/surfsense_web/contracts/types/new-llm-config.types.ts
@@ -33,6 +33,7 @@ export const liteLLMProviderEnum = z.enum([
 	"DATABRICKS",
 	"COMETAPI",
 	"HUGGINGFACE",
+	"GITHUB_MODELS",
 	"CUSTOM",
 ]);