From c92b2a5c1dff6eef75b78674c5c8cf028d8340cd Mon Sep 17 00:00:00 2001
From: QuantumGhost <obelisk.reg+git@gmail.com>
Date: Fri, 6 Jun 2025 17:19:52 +0800
Subject: [PATCH] fix(api): Update Index declaration functions from SQLALchemy
 in WorkflowNodeExecutionModel

---
 api/models/workflow.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/api/models/workflow.py b/api/models/workflow.py
index c0ca780e3e..49f5819a30 100644
--- a/api/models/workflow.py
+++ b/api/models/workflow.py
@@ -16,7 +16,7 @@ if TYPE_CHECKING:
     from models.model import AppMode
 
 import sqlalchemy as sa
-from sqlalchemy import Index, UniqueConstraint, func
+from sqlalchemy import Index, PrimaryKeyConstraint, UniqueConstraint, func
 from sqlalchemy.orm import Mapped, declared_attr, mapped_column
 
 from constants import DEFAULT_FILE_NUMBER_LIMITS, HIDDEN_VALUE
@@ -594,8 +594,8 @@ class WorkflowNodeExecutionModel(Base):
     @declared_attr
     def __table_args__(cls):  # noqa
         return (
-            db.PrimaryKeyConstraint("id", name="workflow_node_execution_pkey"),
-            db.Index(
+            PrimaryKeyConstraint("id", name="workflow_node_execution_pkey"),
+            Index(
                 "workflow_node_execution_workflow_run_idx",
                 "tenant_id",
                 "app_id",
@@ -603,7 +603,7 @@ class WorkflowNodeExecutionModel(Base):
                 "triggered_from",
                 "workflow_run_id",
             ),
-            db.Index(
+            Index(
                 "workflow_node_execution_node_run_idx",
                 "tenant_id",
                 "app_id",
@@ -611,7 +611,7 @@ class WorkflowNodeExecutionModel(Base):
                 "triggered_from",
                 "node_id",
             ),
-            db.Index(
+            Index(
                 "workflow_node_execution_id_idx",
                 "tenant_id",
                 "app_id",