Skip to content

Commit

Permalink
runner: Make control_url and events_url optional (#304)
Browse files Browse the repository at this point in the history
* worker: Make control_url and events_url optional

* openapi: Generate

* runner.gen.go
  • Loading branch information
victorges authored Dec 3, 2024
1 parent 00ee73b commit ff220bb
Show file tree
Hide file tree
Showing 5 changed files with 96 additions and 91 deletions.
9 changes: 6 additions & 3 deletions runner/app/routes/live_video_to_video.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,19 +47,22 @@ class LiveVideoToVideoParams(BaseModel):
control_url: Annotated[
str,
Field(
default="",description="URL for subscribing via Trickle protocol for updates in the live video-to-video generation params.",
default="",
description="URL for subscribing via Trickle protocol for updates in the live video-to-video generation params.",
),
]
events_url: Annotated[
str,
Field(
default="", description="EventsUrl URL for publishing events via Trickle protocol for pipeline status and logs.",
default="",
description="EventsUrl URL for publishing events via Trickle protocol for pipeline status and logs.",
),
]
model_id: Annotated[
str,
Field(
default="", description="Name of the pipeline to run in the live video to video job. Notice that this is named model_id for consistency with other routes, but it does not refer to a Hugging Face model ID. The exact model(s) depends on the pipeline implementation and might be configurable via the `params` argument."
default="",
description="Name of the pipeline to run in the live video to video job. Notice that this is named model_id for consistency with other routes, but it does not refer to a Hugging Face model ID. The exact model(s) depends on the pipeline implementation and might be configurable via the `params` argument."
),
]
params: Annotated[
Expand Down
6 changes: 4 additions & 2 deletions runner/app/routes/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,10 +93,12 @@ class LiveVideoToVideoResponse(BaseModel):
..., description="Destination URL of the outgoing stream to publish to"
)
control_url: str = Field(
..., description="URL for updating the live video-to-video generation"
default='',
description="URL for updating the live video-to-video generation",
)
events_url: str = Field(
..., description="URL for subscribing to events for pipeline status and logs"
default='',
description="URL for subscribing to events for pipeline status and logs",
)

class APIError(BaseModel):
Expand Down
4 changes: 2 additions & 2 deletions runner/gateway.openapi.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -971,16 +971,16 @@ components:
type: string
title: Control Url
description: URL for updating the live video-to-video generation
default: ''
events_url:
type: string
title: Events Url
description: URL for subscribing to events for pipeline status and logs
default: ''
type: object
required:
- subscribe_url
- publish_url
- control_url
- events_url
title: LiveVideoToVideoResponse
description: Response model for live video-to-video generation.
MasksResponse:
Expand Down
4 changes: 2 additions & 2 deletions runner/openapi.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -1109,16 +1109,16 @@ components:
type: string
title: Control Url
description: URL for updating the live video-to-video generation
default: ''
events_url:
type: string
title: Events Url
description: URL for subscribing to events for pipeline status and logs
default: ''
type: object
required:
- subscribe_url
- publish_url
- control_url
- events_url
title: LiveVideoToVideoResponse
description: Response model for live video-to-video generation.
MasksResponse:
Expand Down
164 changes: 82 additions & 82 deletions worker/runner.gen.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit ff220bb

Please sign in to comment.