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

improve block output jinja reference (#2006)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-03-23 18:20:20 -07:00
committed by GitHub
parent 70adf8b9c6
commit 008cc26a15
2 changed files with 38 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
skyvern/forge/sdk/workflow/models/block.py
View File

@@ -184,8 +184,18 @@ class Block(BaseModel, abc.ABC):
return potential_template
template = Template(potential_template)
block_reference_data: dict[str, Any] = workflow_run_context.get_block_metadata(self.label)
template_data = workflow_run_context.values.copy()
template_data[self.label] = workflow_run_context.get_block_metadata(self.label)
if self.label in template_data:
current_value = template_data[self.label]
if isinstance(current_value, dict):
block_reference_data.update(current_value)
else:
LOG.warning(
f"Parameter {self.label} has a registered reference value, going to overwrite it by block metadata"
)
template_data[self.label] = block_reference_data
return template.render(template_data)
@classmethod