From 7a8974f6c70361b34e12d904739824df3d365c56 Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Thu, 11 Sep 2025 15:23:37 -0700
Subject: [PATCH] use the run_id as the primary key for input text context
 feature flag (#3412)

---
 skyvern/services/task_v2_service.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/skyvern/services/task_v2_service.py b/skyvern/services/task_v2_service.py
index f588745a..d371fb62 100644
--- a/skyvern/services/task_v2_service.py
+++ b/skyvern/services/task_v2_service.py
@@ -468,7 +468,7 @@ async def run_task_v2_helper(
     current_run_id = context.run_id if context and context.run_id else task_v2_id
     enable_parse_select_in_extract = app.EXPERIMENTATION_PROVIDER.is_feature_enabled_cached(
         "ENABLE_PARSE_SELECT_IN_EXTRACT",
-        organization_id,
+        current_run_id,
         properties={"organization_id": organization_id, "task_url": task_v2.url},
     )
     skyvern_context.set(