vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-17 02:39:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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 structlog
from email_validator import EmailNotValidError, validate_email
from jinja2 import Template
from playwright.async_api import Error
from pydantic import BaseModel, Field
@ -216,6 +217,15 @@ class BaseTaskBlock(Block):
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
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.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
# non-retryable terminations
while will_retry: