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

add workflow_run_id and workflow_run_block_id in create/update script_blocks code (#3904)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-11-05 08:46:03 +08:00
committed by GitHub
parent 8288c973bd
commit 16f61af6cf
6 changed files with 35 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -4499,6 +4499,8 @@ class AgentDB:
script_block_label: str,
script_file_id: str | None = None,
run_signature: str | None = None,
workflow_run_id: str | None = None,
workflow_run_block_id: str | None = None,
) -> ScriptBlock:
"""Create a script block."""
async with self.Session() as session:
@@ -4509,6 +4511,8 @@ class AgentDB:
script_block_label=script_block_label,
script_file_id=script_file_id,
run_signature=run_signature,
workflow_run_id=workflow_run_id,
workflow_run_block_id=workflow_run_block_id,
)
session.add(script_block)
await session.commit()
@@ -4521,6 +4525,8 @@ class AgentDB:
organization_id: str,
script_file_id: str | None = None,
run_signature: str | None = None,
workflow_run_id: str | None = None,
workflow_run_block_id: str | None = None,
) -> ScriptBlock:
async with self.Session() as session:
script_block = (
@@ -4535,6 +4541,10 @@ class AgentDB:
script_block.script_file_id = script_file_id
if run_signature is not None:
script_block.run_signature = run_signature
if workflow_run_id is not None:
script_block.workflow_run_id = workflow_run_id
if workflow_run_block_id is not None:
script_block.workflow_run_block_id = workflow_run_block_id
await session.commit()
await session.refresh(script_block)
return convert_to_script_block(script_block)

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

@@ -584,6 +584,8 @@ def convert_to_script_block(script_block_model: ScriptBlockModel) -> ScriptBlock
script_block_label=script_block_model.script_block_label,
script_file_id=script_block_model.script_file_id,
run_signature=script_block_model.run_signature,
workflow_run_id=script_block_model.workflow_run_id,
workflow_run_block_id=script_block_model.workflow_run_block_id,
created_at=script_block_model.created_at,
modified_at=script_block_model.modified_at,
deleted_at=script_block_model.deleted_at,