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

14
skyvern/core/script_generations/generate_script.py
View File

@@ -1781,6 +1781,8 @@ async def generate_workflow_script_python_code(
block_label=block_name,
update=pending,
run_signature=run_signature,
workflow_run_id=task.get("workflow_run_id"),
workflow_run_block_id=task.get("workflow_run_block_id"),
)
except Exception as e:
LOG.error("Failed to create script block", error=str(e), exc_info=True)
@@ -1837,6 +1839,8 @@ async def generate_workflow_script_python_code(
block_label=block_name,
update=pending,
run_signature=run_signature,
workflow_run_id=task_v2.get("workflow_run_id"),
workflow_run_block_id=task_v2.get("workflow_run_block_id"),
)
except Exception as e:
LOG.error("Failed to create task_v2 script block", error=str(e), exc_info=True)
@@ -1926,6 +1930,8 @@ async def create_or_update_script_block(
block_label: str,
update: bool = False,
run_signature: str | None = None,
workflow_run_id: str | None = None,
workflow_run_block_id: str | None = None,
) -> None:
"""
Create a script block in the database and save the block code to a script file.
@@ -1939,6 +1945,8 @@ async def create_or_update_script_block(
block_label: Optional custom name for the block (defaults to function name)
update: Whether to update the script block instead of creating a new one
run_signature: The function call code to execute this block (e.g., "await skyvern.action(...)")
workflow_run_id: The workflow run that generated this cached block
workflow_run_block_id: The workflow run block that generated this cached block
"""
block_code_bytes = block_code if isinstance(block_code, bytes) else block_code.encode("utf-8")
try:
@@ -1955,6 +1963,8 @@ async def create_or_update_script_block(
organization_id=organization_id,
script_block_label=block_label,
run_signature=run_signature,
workflow_run_id=workflow_run_id,
workflow_run_block_id=workflow_run_block_id,
)
elif run_signature:
# Update the run_signature if provided
@@ -1962,6 +1972,8 @@ async def create_or_update_script_block(
script_block_id=script_block.script_block_id,
organization_id=organization_id,
run_signature=run_signature,
workflow_run_id=workflow_run_id,
workflow_run_block_id=workflow_run_block_id,
)
# Step 4: Create script file for the block
@@ -2011,6 +2023,8 @@ async def create_or_update_script_block(
script_block_id=script_block.script_block_id,
organization_id=organization_id,
script_file_id=script_file.file_id,
workflow_run_id=workflow_run_id,
workflow_run_block_id=workflow_run_block_id,
)
except Exception as e:

5
skyvern/core/script_generations/transform_workflow_run.py
View File

@@ -155,6 +155,11 @@ async def transform_workflow_run_to_code_gen_input(workflow_run_id: str, organiz
else:
LOG.warning(f"Task v2 block {run_block.label} does not have a child workflow run id")
final_dump["workflow_run_id"] = workflow_run_id
if run_block:
final_dump["workflow_run_block_id"] = run_block.workflow_run_block_id
else:
final_dump["workflow_run_block_id"] = None
workflow_block_dump.append(final_dump)
return CodeGenInput(