From b1130b76ae3cb8d4a8c618f90cacd51c27c3d6d5 Mon Sep 17 00:00:00 2001 From: Andrei Markin Date: Tue, 2 Jun 2026 21:39:49 +0400 Subject: [PATCH] fix(executors): propagate simulate option to celery and server --- libs/executors/garf/executors/entrypoints/server.py | 2 ++ libs/executors/garf/executors/entrypoints/tasks.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/executors/garf/executors/entrypoints/server.py b/libs/executors/garf/executors/entrypoints/server.py index cc73858..8290cb8 100644 --- a/libs/executors/garf/executors/entrypoints/server.py +++ b/libs/executors/garf/executors/entrypoints/server.py @@ -202,6 +202,7 @@ async def execute_workflow_task( config_path: str | pathlib.Path = None, selected_aliases: Optional[list[str]] = None, skipped_aliases: Optional[list[str]] = None, + simulate: bool = False, ) -> dict[str, str]: """Creates a single operation for running garf workflow.""" telemetry.workflow_requested.add(1) @@ -217,6 +218,7 @@ async def execute_workflow_task( execution_workflow=execution_workflow.model_dump(), selected_aliases=selected_aliases, skipped_aliases=skipped_aliases, + simulate=simulate, ) span.set_attribute('garf.operation.id', task.id) return {'operation_id': task.id, 'status': 'PENDING'} diff --git a/libs/executors/garf/executors/entrypoints/tasks.py b/libs/executors/garf/executors/entrypoints/tasks.py index 2162111..1169fd7 100644 --- a/libs/executors/garf/executors/entrypoints/tasks.py +++ b/libs/executors/garf/executors/entrypoints/tasks.py @@ -142,7 +142,7 @@ def execute_workflow( execution_workflow: workflow.Workflow, selected_aliases: list[str], skipped_aliases: list[str], - simulate: bool, + simulate: bool = False, ): """Executes a batch of queries.""" runner = workflow_runner.WorkflowRunner(execution_workflow=execution_workflow)