Merge pull request #954 from roboflow/feat/handle-workflow-id-if-pass…

…ed-with-specification Handle optional workflow_id if passed as part of request to /workflows/run
roboflow · Jan 16, 2025 · b9d29cc · b9d29cc
2 parents d42c522 + e55f50c
commit b9d29cc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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"):