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

Add TaskRunResponse data to task v2 webhook (#2444)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-05-23 18:39:25 -07:00
committed by GitHub
parent 2fe2b53690
commit d277a03b80
3 changed files with 21 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
skyvern/forge/sdk/routes/agent_protocol.py
View File

@@ -72,6 +72,7 @@ from skyvern.forge.sdk.workflow.models.workflow import (
WorkflowStatus,
)
from skyvern.forge.sdk.workflow.models.yaml import WorkflowCreateYAMLRequest
from skyvern.schemas.artifacts import EntityType, entity_type_to_param
from skyvern.schemas.runs import (
CUA_ENGINES,
RunEngine,
@@ -89,23 +90,6 @@ from skyvern.webeye.actions.actions import Action
LOG = structlog.get_logger()
class EntityType(str, Enum):
STEP = "step"
TASK = "task"
WORKFLOW_RUN = "workflow_run"
WORKFLOW_RUN_BLOCK = "workflow_run_block"
THOUGHT = "thought"
entity_type_to_param = {
EntityType.STEP: "step_id",
EntityType.TASK: "task_id",
EntityType.WORKFLOW_RUN: "workflow_run_id",
EntityType.WORKFLOW_RUN_BLOCK: "workflow_run_block_id",
EntityType.THOUGHT: "thought_id",
}
class AISuggestionType(str, Enum):
DATA_SCHEMA = "data_schema"