diff --git a/inference/core/entities/requests/workflows.py b/inference/core/entities/requests/workflows.py index d9605c9e6f..4f1668980a 100644 --- a/inference/core/entities/requests/workflows.py +++ b/inference/core/entities/requests/workflows.py @@ -25,6 +25,9 @@ class WorkflowInferenceRequest(BaseModel): "allow profiling traces to be exported to clients. Only applies for Workflows definitions saved " "on Roboflow platform.", ) + workflow_id: Optional[str] = Field( + default=None, description="Optional identifier of workflow" + ) class PredefinedWorkflowInferenceRequest(WorkflowInferenceRequest): diff --git a/inference/core/interfaces/http/http_api.py b/inference/core/interfaces/http/http_api.py index 40870fd424..60ce559233 100644 --- a/inference/core/interfaces/http/http_api.py +++ b/inference/core/interfaces/http/http_api.py @@ -724,6 +724,7 @@ def process_workflow_inference_request( max_concurrent_steps=WORKFLOWS_MAX_CONCURRENT_STEPS, prevent_local_images_loading=True, profiler=profiler, + workflow_id=workflow_request.workflow_id, ) is_preview = False if hasattr(workflow_request, "is_preview"):