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

Ykeremy/workflow prompt block (#124)

Browse source
This commit is contained in:
Kerem Yilmaz 2024-03-25 00:57:37 -07:00 • committed by GitHub
parent 0b5456a4c6
commit c58aaba4bb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 129 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

20
skyvern/forge/sdk/workflow/service.py
View file

@ -21,7 +21,14 @@ from skyvern.forge.sdk.core.skyvern_context import SkyvernContext
from skyvern.forge.sdk.models import Step
from skyvern.forge.sdk.schemas.tasks import Task, TaskStatus
from skyvern.forge.sdk.workflow.exceptions import WorkflowDefinitionHasDuplicateParameterKeys
from skyvern.forge.sdk.workflow.models.block import BlockType, BlockTypeVar, CodeBlock, ForLoopBlock, TaskBlock
from skyvern.forge.sdk.workflow.models.block import (
BlockType,
BlockTypeVar,
CodeBlock,
ForLoopBlock,
TaskBlock,
TextPromptBlock,
)
from skyvern.forge.sdk.workflow.models.parameter import (
AWSSecretParameter,
OutputParameter,
@ -714,4 +721,15 @@ class WorkflowService:
else [],
output_parameter=output_parameter,
)
elif block_yaml.block_type == BlockType.TEXT_PROMPT:
return TextPromptBlock(
label=block_yaml.label,
llm_key=block_yaml.llm_key,
prompt=block_yaml.prompt,
parameters=[parameters[parameter_key] for parameter_key in block_yaml.parameter_keys]
if block_yaml.parameter_keys
else [],
json_schema=block_yaml.json_schema,
output_parameter=output_parameter,
)
raise ValueError(f"Invalid block type {block_yaml.block_type}")