vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add org_task_index for steps table (#258)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-05-04 22:52:18 -04:00
committed by GitHub
parent b26e2c9699
commit d9ddb1bdbe
2 changed files with 29 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
alembic/versions/2024_05_05_0249-c4dca14a5e69_add_org_task_index_for_steps_table.py Normal file
View File

@@ -0,0 +1,28 @@
"""Add org_task_index for steps table
Revision ID: c4dca14a5e69
Revises: 68d78072fdb5
Create Date: 2024-05-05 02:49:34.719311+00:00
"""
from typing import Sequence, Union
from alembic import op
# revision identifiers, used by Alembic.
revision: str = "c4dca14a5e69"
down_revision: Union[str, None] = "68d78072fdb5"
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("org_task_index", "steps", ["organization_id", "task_id"], unique=False)
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_index("org_task_index", table_name="steps")
# ### end Alembic commands ###

1
skyvern/forge/sdk/db/models.py
View File

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