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

add permanent_workflow_id to observer_cruises and observer_thoughts (#1396)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-12-16 14:34:31 -08:00
committed by GitHub
parent e81fdf6512
commit 985f8c9428
4 changed files with 44 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
alembic/versions/2024_12_16_2220-411dd89f3df9_add_workflow_permanent_id_to_observer_.py Normal file
View File

@@ -0,0 +1,33 @@
"""add workflow_permanent_id to observer_cruises and observer_thoughts
Revision ID: 411dd89f3df9
Revises: c502ecf908c6
Create Date: 2024-12-16 22:20:52.174896+00:00
"""
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
# revision identifiers, used by Alembic.
revision: str = "411dd89f3df9"
down_revision: Union[str, None] = "c502ecf908c6"
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("observer_cruises", sa.Column("workflow_permanent_id", sa.String(), nullable=True))
op.add_column("observer_thoughts", sa.Column("workflow_permanent_id", sa.String(), nullable=True))
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_column("observer_thoughts", "workflow_permanent_id")
op.drop_column("observer_cruises", "workflow_permanent_id")
# ### end Alembic commands ###

7
skyvern/forge/sdk/db/client.py
View File

@@ -1793,6 +1793,7 @@ class AgentDB:
self,
workflow_run_id: str | None = None,
workflow_id: str | None = None,
workflow_permanent_id: str | None = None,
prompt: str | None = None,
url: str | None = None,
organization_id: str | None = None,
@@ -1801,6 +1802,7 @@ class AgentDB:
new_observer_cruise = ObserverCruiseModel(
workflow_run_id=workflow_run_id,
workflow_id=workflow_id,
workflow_permanent_id=workflow_permanent_id,
prompt=prompt,
url=url,
organization_id=organization_id,
@@ -1815,6 +1817,7 @@ class AgentDB:
observer_cruise_id: str,
workflow_run_id: str | None = None,
workflow_id: str | None = None,
workflow_permanent_id: str | None = None,
workflow_run_block_id: str | None = None,
user_input: str | None = None,
observation: str | None = None,
@@ -1827,6 +1830,7 @@ class AgentDB:
observer_cruise_id=observer_cruise_id,
workflow_run_id=workflow_run_id,
workflow_id=workflow_id,
workflow_permanent_id=workflow_permanent_id,
workflow_run_block_id=workflow_run_block_id,
user_input=user_input,
observation=observation,
@@ -1845,6 +1849,7 @@ class AgentDB:
status: ObserverCruiseStatus | None = None,
workflow_run_id: str | None = None,
workflow_id: str | None = None,
workflow_permanent_id: str | None = None,
url: str | None = None,
prompt: str | None = None,
organization_id: str | None = None,
@@ -1864,6 +1869,8 @@ class AgentDB:
observer_cruise.workflow_run_id = workflow_run_id
if workflow_id:
observer_cruise.workflow_id = workflow_id
if workflow_permanent_id:
observer_cruise.workflow_permanent_id = workflow_permanent_id
if url:
observer_cruise.url = url
if prompt:

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

@@ -515,6 +515,7 @@ class ObserverCruiseModel(Base):
organization_id = Column(String, ForeignKey("organizations.organization_id"), nullable=True)
workflow_run_id = Column(String, ForeignKey("workflow_runs.workflow_run_id"), nullable=True)
workflow_id = Column(String, ForeignKey("workflows.workflow_id"), nullable=True)
workflow_permanent_id = Column(String, nullable=True)
prompt = Column(UnicodeText, nullable=True)
url = Column(String, nullable=True)
@@ -531,6 +532,7 @@ class ObserverThoughtModel(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)
workflow_permanent_id = Column(String, nullable=True)
user_input = Column(UnicodeText, nullable=True)
observation = Column(String, nullable=True)
thought = Column(String, nullable=True)

2
skyvern/forge/sdk/schemas/observers.py
View File

@@ -23,6 +23,7 @@ class ObserverCruise(BaseModel):
organization_id: str | None = None
workflow_run_id: str | None = None
workflow_id: str | None = None
workflow_permanent_id: str | None = None
prompt: str | None = None
url: HttpUrl | None = None
@@ -39,6 +40,7 @@ class ObserverThought(BaseModel):
workflow_run_id: str | None = None
workflow_run_block_id: str | None = None
workflow_id: str | None = None
workflow_permanent_id: str | None = None
user_input: str | None = None
observation: str | None = None
thought: str | None = None