fix: secret type migration

Admyral-Security · Nov 5, 2024 · 36dec61 · 36dec61
1 parent 9e87566
commit 36dec61
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/admyral/db/alembic/versions/db5cf73c0869_migrate_secrets_to_secret_type.py b/admyral/db/alembic/versions/db5cf73c0869_migrate_secrets_to_secret_type.py
@@ -22,6 +22,8 @@
 def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
 
+    bind = op.get_bind()
+
     integrations = [
         ("anthropic", "Anthropic"),
         ("mistral ai", "Mistral AI"),
@@ -57,17 +59,21 @@ def upgrade() -> None:
 
     for potential_name, secret_type in integrations:
         potential_name: str = potential_name.lower()
-        op.execute(
-            "UPDATE secrets SET secret_type = :new_type WHERE secret_type LIKE :pattern",
-            execution_options={
+        bind.execute(
+            sqlmodel.text(
+                "UPDATE secrets SET secret_type = :new_type WHERE secret_type LIKE :pattern"
+            ),
+            {
                 "new_type": secret_type,
                 "pattern": f"%{potential_name}%",
             },
         )
         potential_name = potential_name.replace(" ", "_")
-        op.execute(
-            "UPDATE secrets SET secret_type = :new_type WHERE secret_type LIKE :pattern",
-            execution_options={
+        bind.execute(
+            sqlmodel.text(
+                "UPDATE secrets SET secret_type = :new_type WHERE secret_type LIKE :pattern"
+            ),
+            {
                 "new_type": secret_type,
                 "pattern": f"%{potential_name}%",
             },