diff --git a/runner/app/pipelines/live_video_to_video.py b/runner/app/pipelines/live_video_to_video.py index 3b29a96dc..bc8772f30 100644 --- a/runner/app/pipelines/live_video_to_video.py +++ b/runner/app/pipelines/live_video_to_video.py @@ -30,7 +30,7 @@ def __init__(self, model_id: str): self.start_process() def __call__( # type: ignore - self, *, subscribe_url: str, publish_url: str, control_url: str, events_url: str, params: dict, request_id: str, manifest_id: str, stream_id: str, **kwargs + self, *, subscribe_url: str, publish_url: str, control_url: str, events_url: str, data_url: str, params: dict, request_id: str, manifest_id: str, stream_id: str, **kwargs ): if not self.process: raise RuntimeError("Pipeline process not running") @@ -51,6 +51,7 @@ def __call__( # type: ignore "publish_url": publish_url, "control_url": control_url, "events_url": events_url, + "data_url": data_url, "params": params, "request_id": request_id or "", "manifest_id": manifest_id or "", diff --git a/runner/app/routes/live_video_to_video.py b/runner/app/routes/live_video_to_video.py index 5614ba898..2967c3f17 100644 --- a/runner/app/routes/live_video_to_video.py +++ b/runner/app/routes/live_video_to_video.py @@ -58,6 +58,13 @@ class LiveVideoToVideoParams(BaseModel): description="URL for publishing events via Trickle protocol for pipeline status and logs.", ), ] + data_url: Annotated[ + str, + Field( + default="", + description="URL for publishing data via Trickle protocol for pipeline status and logs.", + ), + ] model_id: Annotated[ str, Field( diff --git a/runner/app/routes/utils.py b/runner/app/routes/utils.py index 5ed185f8b..b9f9cdacb 100644 --- a/runner/app/routes/utils.py +++ b/runner/app/routes/utils.py @@ -131,6 +131,10 @@ class LiveVideoToVideoResponse(BaseModel): default='', description="URL for subscribing to events for pipeline status and logs", ) + data_url: str = Field( + default='', + description="URL for publishing data for pipeline", + ) request_id: str = Field( default='', description="The ID generated for this request", diff --git a/runner/gateway.openapi.yaml b/runner/gateway.openapi.yaml index 177813ef3..573ad163c 100644 --- a/runner/gateway.openapi.yaml +++ b/runner/gateway.openapi.yaml @@ -1000,6 +1000,12 @@ components: description: URL for publishing events via Trickle protocol for pipeline status and logs. default: '' + data_url: + type: string + title: Data Url + description: URL for publishing data via Trickle protocol for pipeline status + and logs. + default: '' model_id: type: string title: Model Id @@ -1055,6 +1061,11 @@ components: title: Events Url description: URL for subscribing to events for pipeline status and logs default: '' + data_url: + type: string + title: Data Url + description: URL for publishing data for pipeline + default: '' request_id: type: string title: Request Id diff --git a/runner/openapi.yaml b/runner/openapi.yaml index 28756adf8..45297e527 100644 --- a/runner/openapi.yaml +++ b/runner/openapi.yaml @@ -1158,6 +1158,12 @@ components: description: URL for publishing events via Trickle protocol for pipeline status and logs. default: '' + data_url: + type: string + title: Data Url + description: URL for publishing data via Trickle protocol for pipeline status + and logs. + default: '' model_id: type: string title: Model Id @@ -1212,6 +1218,11 @@ components: title: Events Url description: URL for subscribing to events for pipeline status and logs default: '' + data_url: + type: string + title: Data Url + description: URL for publishing data for pipeline + default: '' request_id: type: string title: Request Id