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

This commit is contained in:
Shuchang Zheng 2025-04-29 03:55:52 +08:00 committed by GitHub
parent 7783bb0327
commit 2f10e3c430
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 29 additions and 20 deletions

View file

@ -102,7 +102,7 @@ class WorkflowService:
request_id: str | None,
workflow_request: WorkflowRequestBody,
workflow_permanent_id: str,
organization_id: str,
organization: Organization,
is_template_workflow: bool = False,
version: int | None = None,
max_steps_override: int | None = None,
@ -121,7 +121,7 @@ class WorkflowService:
# Validate the workflow and the organization
workflow = await self.get_workflow_by_permanent_id(
workflow_permanent_id=workflow_permanent_id,
organization_id=None if is_template_workflow else organization_id,
organization_id=None if is_template_workflow else organization.organization_id,
version=version,
)
if workflow is None:
@ -137,7 +137,7 @@ class WorkflowService:
workflow_request=workflow_request,
workflow_permanent_id=workflow_permanent_id,
workflow_id=workflow_id,
organization_id=organization_id,
organization_id=organization.organization_id,
parent_workflow_run_id=parent_workflow_run_id,
)
LOG.info(
@ -151,7 +151,8 @@ class WorkflowService:
)
skyvern_context.set(
SkyvernContext(
organization_id=organization_id,
organization_id=organization.organization_id,
organization_name=organization.organization_name,
request_id=request_id,
workflow_id=workflow_id,
workflow_run_id=workflow_run.workflow_run_id,