Set organization_id for posthog through event (#277)

2024-05-08 01:33:06 -07:00
parent cc91c1b2b6
commit 42d652f381
2 changed files with 7 additions and 5 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -629,7 +629,9 @@ class ForgeAgent:
        )
        prompt_template = "extract-action"
        if app.EXPERIMENTATION_PROVIDER.is_feature_enabled_cached(
-            "USE_CLAUDE3_SONNET", task.workflow_run_id or task.task_id
+            "USE_CLAUDE3_SONNET",
+            task.workflow_run_id or task.task_id,
+            properties={"organization_id": task.organization_id},
        ):
            LOG.info("Using Claude3 Sonnet prompt template for action extraction")
            prompt_template = "extract-action-claude3-sonnet"