Add organization_name and workflow_permanent_id to skyvern context, pass workflow_permanet_id when deciding which identifier to use with which llm (#2245)

2025-04-29 03:55:52 +08:00
parent 7783bb0327
commit 2f10e3c430
11 changed files with 29 additions and 20 deletions
--- a/skyvern/services/task_v2_service.py
+++ b/skyvern/services/task_v2_service.py
@@ -211,7 +211,7 @@ async def initialize_task_v2(
            request_id=None,
            workflow_request=WorkflowRequestBody(),
            workflow_permanent_id=new_workflow.workflow_permanent_id,
-            organization_id=organization.organization_id,
+            organization=organization,
            version=None,
            max_steps_override=max_steps_override,
            parent_workflow_run_id=parent_workflow_run_id,
@@ -1491,7 +1491,6 @@ async def _summarize_task_v2(
    )
    task_v2_summary_resp = await app.LLM_API_HANDLER(
        prompt=task_v2_summary_prompt,
-        screenshots=screenshots,
        thought=thought,
        prompt_name="task_v2_summary",
    )