From 20c495089a7b1164f9f1624f1dea02971c136ec8 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Mon, 23 Jun 2025 11:16:45 +0800 Subject: [PATCH] add taskv2 and task id to artifacts by default if present in context (#2770) --- skyvern/forge/sdk/artifact/manager.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/skyvern/forge/sdk/artifact/manager.py b/skyvern/forge/sdk/artifact/manager.py index 38cd225c..95662778 100644 --- a/skyvern/forge/sdk/artifact/manager.py +++ b/skyvern/forge/sdk/artifact/manager.py @@ -42,11 +42,13 @@ class ArtifactManager: if data and path: raise ValueError("Both data and path cannot be provided to create an artifact.") - if not workflow_run_id: - context = skyvern_context.current() - - if context: - workflow_run_id = context.workflow_run_id + context = skyvern_context.current() + if not workflow_run_id and context: + workflow_run_id = context.workflow_run_id + if not task_v2_id and context: + task_v2_id = context.task_v2_id + if not task_id and context: + task_id = context.task_id artifact = await app.DATABASE.create_artifact( artifact_id,