From 09dad78a5dce35a40c3e56ff4331e46986fb3cd0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=9D=9E=E6=B3=95=E6=93=8D=E4=BD=9C?= <hjlarry@163.com>
Date: Mon, 16 Mar 2026 10:10:03 +0800
Subject: [PATCH] chore: add indexes for human_input_forms query patterns
 (#32849)

Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com>
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Bond Zhu <37842169+MRZHUH@users.noreply.github.com>
---
 ...55790_add_indexes_for_human_input_forms.py | 68 +++++++++++++++++++
 api/models/human_input.py                     | 19 ++++++
 2 files changed, 87 insertions(+)
 create mode 100644 api/migrations/versions/2026_03_02_1805-0ec65df55790_add_indexes_for_human_input_forms.py

diff --git a/api/migrations/versions/2026_03_02_1805-0ec65df55790_add_indexes_for_human_input_forms.py b/api/migrations/versions/2026_03_02_1805-0ec65df55790_add_indexes_for_human_input_forms.py
new file mode 100644
index 0000000000..63fd58b1bf
--- /dev/null
+++ b/api/migrations/versions/2026_03_02_1805-0ec65df55790_add_indexes_for_human_input_forms.py
@@ -0,0 +1,68 @@
+"""add indexes for human_input_forms query patterns
+
+Revision ID: 0ec65df55790
+Revises: e288952f2994
+Create Date: 2026-03-02 18:05:00.000000
+
+"""
+
+from alembic import op
+
+
+# revision identifiers, used by Alembic.
+revision = "0ec65df55790"
+down_revision = "e288952f2994"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    with op.batch_alter_table("human_input_forms", schema=None) as batch_op:
+        batch_op.create_index(
+            "human_input_forms_workflow_run_id_node_id_idx",
+            ["workflow_run_id", "node_id"],
+            unique=False,
+        )
+        batch_op.create_index(
+            "human_input_forms_status_created_at_idx",
+            ["status", "created_at"],
+            unique=False,
+        )
+        batch_op.create_index(
+            "human_input_forms_status_expiration_time_idx",
+            ["status", "expiration_time"],
+            unique=False,
+        )
+
+    with op.batch_alter_table("human_input_form_deliveries", schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f("human_input_form_deliveries_form_id_idx"),
+            ["form_id"],
+            unique=False,
+        )
+
+    with op.batch_alter_table("human_input_form_recipients", schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f("human_input_form_recipients_delivery_id_idx"),
+            ["delivery_id"],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f("human_input_form_recipients_form_id_idx"),
+            ["form_id"],
+            unique=False,
+        )
+
+
+def downgrade():
+    with op.batch_alter_table("human_input_forms", schema=None) as batch_op:
+        batch_op.drop_index("human_input_forms_workflow_run_id_node_id_idx")
+        batch_op.drop_index("human_input_forms_status_expiration_time_idx")
+        batch_op.drop_index("human_input_forms_status_created_at_idx")
+
+    with op.batch_alter_table("human_input_form_recipients", schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f("human_input_form_recipients_form_id_idx"))
+        batch_op.drop_index(batch_op.f("human_input_form_recipients_delivery_id_idx"))
+
+    with op.batch_alter_table("human_input_form_deliveries", schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f("human_input_form_deliveries_form_id_idx"))
diff --git a/api/models/human_input.py b/api/models/human_input.py
index 709cc8fe61..48e7fbb9ea 100644
--- a/api/models/human_input.py
+++ b/api/models/human_input.py
@@ -30,6 +30,15 @@ def _generate_token() -> str:
 
 class HumanInputForm(DefaultFieldsMixin, Base):
     __tablename__ = "human_input_forms"
+    __table_args__ = (
+        sa.Index(
+            "human_input_forms_workflow_run_id_node_id_idx",
+            "workflow_run_id",
+            "node_id",
+        ),
+        sa.Index("human_input_forms_status_expiration_time_idx", "status", "expiration_time"),
+        sa.Index("human_input_forms_status_created_at_idx", "status", "created_at"),
+    )
 
     tenant_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
     app_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
@@ -84,6 +93,12 @@ class HumanInputForm(DefaultFieldsMixin, Base):
 
 class HumanInputDelivery(DefaultFieldsMixin, Base):
     __tablename__ = "human_input_form_deliveries"
+    __table_args__ = (
+        sa.Index(
+            None,
+            "form_id",
+        ),
+    )
 
     form_id: Mapped[str] = mapped_column(
         StringUUID,
@@ -181,6 +196,10 @@ RecipientPayload = Annotated[
 
 class HumanInputFormRecipient(DefaultFieldsMixin, Base):
     __tablename__ = "human_input_form_recipients"
+    __table_args__ = (
+        sa.Index(None, "form_id"),
+        sa.Index(None, "delivery_id"),
+    )
 
     form_id: Mapped[str] = mapped_column(
         StringUUID,