created_at index for steps table (Skyvern-AI#1301)

alembic/versions/2024_12_02_1609-db41106b9f1a_steps_db_change_add_created_at_index.py

@@ -0,0 +1,29 @@
+"""steps DB change: add created_at index
+
+Revision ID: db41106b9f1a
+Revises: a5feab7712fe
+Create Date: 2024-12-02 16:09:57.679626+00:00
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "db41106b9f1a"
+down_revision: Union[str, None] = "a5feab7712fe"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_index("created_at_org_index", "steps", ["created_at", "organization_id"], unique=False)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index("created_at_org_index", table_name="steps")
+    # ### end Alembic commands ###

skyvern/forge/sdk/db/models.py

@@ -84,7 +84,10 @@ class TaskModel(Base):
 
 class StepModel(Base):
     __tablename__ = "steps"
-    __table_args__ = (Index("org_task_index", "organization_id", "task_id"),)
+    __table_args__ = (
+        Index("org_task_index", "organization_id", "task_id"),
+        Index("created_at_org_index", "created_at", "organization_id"),
+    )
 
     step_id = Column(String, primary_key=True, index=True, default=generate_step_id)
     organization_id = Column(String, ForeignKey("organizations.organization_id"))