From 7ce6493caa79a5cb0038c0a9a1caa4fac717bbf9 Mon Sep 17 00:00:00 2001
From: "DESKTOP-RTLN3BA\\$punk" <vermarohanfinal@gmail.com>
Date: Thu, 5 Feb 2026 18:16:21 -0800
Subject: [PATCH] fix: update ENUM import for PostgreSQL compatibility in image
 generations table migration

---
 .../alembic/versions/93_add_image_generations_table.py         | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/surfsense_backend/alembic/versions/93_add_image_generations_table.py b/surfsense_backend/alembic/versions/93_add_image_generations_table.py
index f24adc68f..eba9d7c86 100644
--- a/surfsense_backend/alembic/versions/93_add_image_generations_table.py
+++ b/surfsense_backend/alembic/versions/93_add_image_generations_table.py
@@ -13,6 +13,7 @@ Changes:
 from collections.abc import Sequence
 
 import sqlalchemy as sa
+from sqlalchemy.dialects.postgresql import ENUM as PG_ENUM
 from sqlalchemy.dialects.postgresql import JSONB, UUID
 
 from alembic import op
@@ -59,7 +60,7 @@ def upgrade() -> None:
             sa.Column("description", sa.String(500), nullable=True),
             sa.Column(
                 "provider",
-                sa.Enum(
+                PG_ENUM(
                     "OPENAI",
                     "AZURE_OPENAI",
                     "GOOGLE",