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

add observer_thought_id to artifacts table (#1341)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-12-06 10:25:20 -08:00
committed by GitHub
parent 18762f574b
commit 3467382845
2 changed files with 46 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
alembic/versions/2024_12_06_1819-fe49b59d836c_add_observer_thought_id_column_to_.py Normal file
View File

@@ -0,0 +1,43 @@
"""Add observer_thought_id column to artifacts table. Add user_input, observation to ObserverThoughts
Revision ID: fe49b59d836c
Revises: 4d51ed4719d5
Create Date: 2024-12-06 18:19:23.286827+00:00
"""
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
# revision identifiers, used by Alembic.
revision: str = "fe49b59d836c"
down_revision: Union[str, None] = "4d51ed4719d5"
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.add_column("artifacts", sa.Column("observer_thought_id", sa.String(), nullable=True))
op.create_foreign_key(
"artifacts_observer_thought_id_fkey",
"artifacts",
"observer_thoughts",
["observer_thought_id"],
["observer_thought_id"],
)
op.add_column("observer_thoughts", sa.Column("user_input", sa.UnicodeText(), nullable=True))
op.add_column("observer_thoughts", sa.Column("observation", sa.String(), nullable=True))
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_column("observer_thoughts", "observation")
op.drop_column("observer_thoughts", "user_input")
op.drop_constraint("artifacts_observer_thought_id_fkey", "artifacts", type_="foreignkey")
op.drop_column("artifacts", "observer_thought_id")
# ### end Alembic commands ###

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

@@ -169,6 +169,7 @@ class ArtifactModel(Base):
workflow_run_id = Column(String, ForeignKey("workflow_runs.workflow_run_id"))
workflow_run_block_id = Column(String, ForeignKey("workflow_run_blocks.workflow_run_block_id"))
observer_cruise_id = Column(String, ForeignKey("observer_cruises.observer_cruise_id"))
observer_thought_id = Column(String, ForeignKey("observer_thoughts.observer_thought_id"))
task_id = Column(String, ForeignKey("tasks.task_id"))
step_id = Column(String, ForeignKey("steps.step_id"), index=True)
artifact_type = Column(String)
@@ -524,5 +525,7 @@ class ObserverThought(Base):
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)
user_input = Column(UnicodeText, nullable=True)
observation = Column(String, nullable=True)
thought = Column(String, nullable=True)
answer = Column(String, nullable=True)