From 6cf20221e6b980d2698d8dd03609953df0ff48b0 Mon Sep 17 00:00:00 2001 From: Kerem Yilmaz Date: Wed, 9 Oct 2024 09:55:47 -0700 Subject: [PATCH] Add tasks.workflow_run_id index (#946) --- ...e685_add_workflow_run_id_index_to_tasks.py | 29 +++++++++++++++++++ skyvern/forge/sdk/db/models.py | 2 +- 2 files changed, 30 insertions(+), 1 deletion(-) create mode 100644 alembic/versions/2024_10_09_1652-12fb2dede685_add_workflow_run_id_index_to_tasks.py diff --git a/alembic/versions/2024_10_09_1652-12fb2dede685_add_workflow_run_id_index_to_tasks.py b/alembic/versions/2024_10_09_1652-12fb2dede685_add_workflow_run_id_index_to_tasks.py new file mode 100644 index 00000000..5bdbca69 --- /dev/null +++ b/alembic/versions/2024_10_09_1652-12fb2dede685_add_workflow_run_id_index_to_tasks.py @@ -0,0 +1,29 @@ +"""Add workflow_run_id index to tasks + +Revision ID: 12fb2dede685 +Revises: a575628e1965 +Create Date: 2024-10-09 16:52:36.095562+00:00 + +""" + +from typing import Sequence, Union + +from alembic import op + +# revision identifiers, used by Alembic. +revision: str = "12fb2dede685" +down_revision: Union[str, None] = "a575628e1965" +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(op.f("ix_tasks_workflow_run_id"), "tasks", ["workflow_run_id"], unique=False) + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.drop_index(op.f("ix_tasks_workflow_run_id"), table_name="tasks") + # ### end Alembic commands ### diff --git a/skyvern/forge/sdk/db/models.py b/skyvern/forge/sdk/db/models.py index 9c857905..9e245989 100644 --- a/skyvern/forge/sdk/db/models.py +++ b/skyvern/forge/sdk/db/models.py @@ -61,7 +61,7 @@ class TaskModel(Base): failure_reason = Column(String) proxy_location = Column(Enum(ProxyLocation)) extracted_information_schema = Column(JSON) - workflow_run_id = Column(String, ForeignKey("workflow_runs.workflow_run_id")) + workflow_run_id = Column(String, ForeignKey("workflow_runs.workflow_run_id"), index=True) order = Column(Integer, nullable=True) retry = Column(Integer, nullable=True) error_code_mapping = Column(JSON, nullable=True)