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

ObserverThought reproduce migration script (#1338)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-12-06 00:58:30 -08:00
committed by GitHub
parent 95bef7f441
commit bb03891f55
2 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
alembic/versions/2024_12_06_0725-5e266217e9d9_introduce_observercruise_and_.py → alembic/versions/2024_12_06_0852-4d51ed4719d5_introduce_observercruise_and_.py
View File

@@ -1,8 +1,8 @@
"""Introduce ObserverCruise and ObserverThought. Add workflow_run_block_id and observer_cruise_id to artifacts """Introduce ObserverCruise and ObserverThought. Add workflow_run_block_id and observer_cruise_id to artifacts
Revision ID: 5e266217e9d9 Revision ID: 4d51ed4719d5
Revises: de0254717601 Revises: de0254717601
Create Date: 2024-12-06 07:25:50.080125+00:00 Create Date: 2024-12-06 08:52:52.111448+00:00
""" """
@@ -13,7 +13,7 @@ import sqlalchemy as sa
from alembic import op from alembic import op
# revision identifiers, used by Alembic. # revision identifiers, used by Alembic.
revision: str = "5e266217e9d9" revision: str = "4d51ed4719d5"
down_revision: Union[str, None] = "de0254717601" down_revision: Union[str, None] = "de0254717601"
branch_labels: Union[str, Sequence[str], None] = None branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None depends_on: Union[str, Sequence[str], None] = None
@@ -48,6 +48,7 @@ def upgrade() -> None:
sa.Column("organization_id", sa.String(), nullable=True), sa.Column("organization_id", sa.String(), nullable=True),
sa.Column("observer_cruise_id", sa.String(), nullable=False), sa.Column("observer_cruise_id", sa.String(), nullable=False),
sa.Column("workflow_run_id", sa.String(), nullable=True), sa.Column("workflow_run_id", sa.String(), nullable=True),
sa.Column("workflow_run_block_id", sa.String(), nullable=True),
sa.Column("workflow_id", sa.String(), nullable=True), sa.Column("workflow_id", sa.String(), nullable=True),
sa.Column("thought", sa.String(), nullable=True), sa.Column("thought", sa.String(), nullable=True),
sa.Column("answer", sa.String(), nullable=True), sa.Column("answer", sa.String(), nullable=True),
@@ -63,6 +64,10 @@ def upgrade() -> None:
["workflow_id"], ["workflow_id"],
["workflows.workflow_id"], ["workflows.workflow_id"],
), ),
sa.ForeignKeyConstraint(
["workflow_run_block_id"],
["workflow_run_blocks.workflow_run_block_id"],
),
sa.ForeignKeyConstraint( sa.ForeignKeyConstraint(
["workflow_run_id"], ["workflow_run_id"],
["workflow_runs.workflow_run_id"], ["workflow_runs.workflow_run_id"],

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

@@ -522,6 +522,7 @@ class ObserverThought(Base):
organization_id = Column(String, ForeignKey("organizations.organization_id"), nullable=True) organization_id = Column(String, ForeignKey("organizations.organization_id"), nullable=True)
observer_cruise_id = Column(String, ForeignKey("observer_cruises.observer_cruise_id"), nullable=False) observer_cruise_id = Column(String, ForeignKey("observer_cruises.observer_cruise_id"), nullable=False)
workflow_run_id = Column(String, ForeignKey("workflow_runs.workflow_run_id"), nullable=True) workflow_run_id = Column(String, ForeignKey("workflow_runs.workflow_run_id"), nullable=True)
workflow_run_block_id = Column(String, ForeignKey("workflow_run_blocks.workflow_run_block_id"), nullable=True)
workflow_id = Column(String, ForeignKey("workflows.workflow_id"), nullable=True) workflow_id = Column(String, ForeignKey("workflows.workflow_id"), nullable=True)
thought = Column(String, nullable=True) thought = Column(String, nullable=True)
answer = Column(String, nullable=True) answer = Column(String, nullable=True)