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

use jinja template for workflow parameter (#1272)

Browse Source
This commit is contained in:
LawyZheng
2024-11-27 15:19:34 +08:00
committed by GitHub
parent e0aadac962
commit 8760b967fd
1 changed files with 30 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -16,6 +16,7 @@ from typing import Annotated, Any, Literal, Union
import filetype import filetype
import structlog import structlog
from email_validator import EmailNotValidError, validate_email from email_validator import EmailNotValidError, validate_email
from jinja2 import Template
from playwright.async_api import Error from playwright.async_api import Error
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
@@ -216,6 +217,15 @@ class BaseTaskBlock(Block):
return parameters return parameters
@staticmethod
def format_task_block_parameter_template_from_workflow_run_context(
potential_template: str | None, workflow_run_context: WorkflowRunContext
) -> str | None:
if not potential_template:
return potential_template
template = Template(potential_template)
return template.render(workflow_run_context.values)
@staticmethod @staticmethod
async def get_task_order(workflow_run_id: str, current_retry: int) -> tuple[int, int]: async def get_task_order(workflow_run_id: str, current_retry: int) -> tuple[int, int]:
""" """
@@ -289,6 +299,26 @@ class BaseTaskBlock(Block):
) )
self.download_suffix = download_suffix_parameter_value self.download_suffix = download_suffix_parameter_value
self.url = self.format_task_block_parameter_template_from_workflow_run_context(self.url, workflow_run_context)
self.totp_identifier = self.format_task_block_parameter_template_from_workflow_run_context(
self.totp_identifier, workflow_run_context
)
self.download_suffix = self.format_task_block_parameter_template_from_workflow_run_context(
self.download_suffix, workflow_run_context
)
self.navigation_goal = self.format_task_block_parameter_template_from_workflow_run_context(
self.navigation_goal, workflow_run_context
)
self.data_extraction_goal = self.format_task_block_parameter_template_from_workflow_run_context(
self.data_extraction_goal, workflow_run_context
)
self.complete_criterion = self.format_task_block_parameter_template_from_workflow_run_context(
self.complete_criterion, workflow_run_context
)
self.terminate_criterion = self.format_task_block_parameter_template_from_workflow_run_context(
self.terminate_criterion, workflow_run_context
)
# TODO (kerem) we should always retry on terminated. We should make a distinction between retriable and # TODO (kerem) we should always retry on terminated. We should make a distinction between retriable and
# non-retryable terminations # non-retryable terminations
while will_retry: while will_retry: