From 3c5029f801b861631d353843259b983ebb957894 Mon Sep 17 00:00:00 2001 From: blakkd Date: Mon, 24 Feb 2025 02:10:21 +0100 Subject: [PATCH 1/3] converted CRLF ending lines to LF ones in api/src/structures/custom_responses.py let ruff organise the imports --- api/src/structures/custom_responses.py | 100 ++++++++++++------------- 1 file changed, 50 insertions(+), 50 deletions(-) diff --git a/api/src/structures/custom_responses.py b/api/src/structures/custom_responses.py index 3d3a987..4163c85 100644 --- a/api/src/structures/custom_responses.py +++ b/api/src/structures/custom_responses.py @@ -1,51 +1,51 @@ -from collections.abc import AsyncIterable, Iterable - -import json -import typing -from pydantic import BaseModel -from starlette.background import BackgroundTask -from starlette.concurrency import iterate_in_threadpool -from starlette.responses import JSONResponse, StreamingResponse - - -class JSONStreamingResponse(StreamingResponse, JSONResponse): - """StreamingResponse that also render with JSON.""" - - def __init__( - self, - content: Iterable | AsyncIterable, - status_code: int = 200, - headers: dict[str, str] | None = None, - media_type: str | None = None, - background: BackgroundTask | None = None, - ) -> None: - if isinstance(content, AsyncIterable): - self._content_iterable: AsyncIterable = content - else: - self._content_iterable = iterate_in_threadpool(content) - - - - async def body_iterator() -> AsyncIterable[bytes]: - async for content_ in self._content_iterable: - if isinstance(content_, BaseModel): - content_ = content_.model_dump() - yield self.render(content_) - - - - self.body_iterator = body_iterator() - self.status_code = status_code - if media_type is not None: - self.media_type = media_type - self.background = background - self.init_headers(headers) - - def render(self, content: typing.Any) -> bytes: - return (json.dumps( - content, - ensure_ascii=False, - allow_nan=False, - indent=None, - separators=(",", ":"), +import json +import typing +from collections.abc import AsyncIterable, Iterable + +from pydantic import BaseModel +from starlette.background import BackgroundTask +from starlette.concurrency import iterate_in_threadpool +from starlette.responses import JSONResponse, StreamingResponse + + +class JSONStreamingResponse(StreamingResponse, JSONResponse): + """StreamingResponse that also render with JSON.""" + + def __init__( + self, + content: Iterable | AsyncIterable, + status_code: int = 200, + headers: dict[str, str] | None = None, + media_type: str | None = None, + background: BackgroundTask | None = None, + ) -> None: + if isinstance(content, AsyncIterable): + self._content_iterable: AsyncIterable = content + else: + self._content_iterable = iterate_in_threadpool(content) + + + + async def body_iterator() -> AsyncIterable[bytes]: + async for content_ in self._content_iterable: + if isinstance(content_, BaseModel): + content_ = content_.model_dump() + yield self.render(content_) + + + + self.body_iterator = body_iterator() + self.status_code = status_code + if media_type is not None: + self.media_type = media_type + self.background = background + self.init_headers(headers) + + def render(self, content: typing.Any) -> bytes: + return (json.dumps( + content, + ensure_ascii=False, + allow_nan=False, + indent=None, + separators=(",", ":"), ) + "\n").encode("utf-8") \ No newline at end of file From 664451e11c8bcba976a93b8d1c0ad76dd999c820 Mon Sep 17 00:00:00 2001 From: blakkd Date: Mon, 24 Feb 2025 02:16:07 +0100 Subject: [PATCH 2/3] added docker to video group --- docker/gpu/docker-compose.yml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docker/gpu/docker-compose.yml b/docker/gpu/docker-compose.yml index 762aca6..d15fe85 100644 --- a/docker/gpu/docker-compose.yml +++ b/docker/gpu/docker-compose.yml @@ -1,6 +1,9 @@ name: kokoro-tts-gpu services: kokoro-tts: + user: "${UID}:${GID}" + group_add: + - "video" # image: ghcr.io/remsky/kokoro-fastapi-gpu:v0.2.0 build: context: ../.. From 6edc44edf32d9b5a6c515e9122c6f535a1dd85f7 Mon Sep 17 00:00:00 2001 From: Fireblade2534 <77405729+fireblade2534@users.noreply.github.com> Date: Mon, 10 Mar 2025 18:12:52 -0400 Subject: [PATCH 3/3] Update docker-compose.yml --- docker/gpu/docker-compose.yml | 3 --- 1 file changed, 3 deletions(-) diff --git a/docker/gpu/docker-compose.yml b/docker/gpu/docker-compose.yml index d15fe85..762aca6 100644 --- a/docker/gpu/docker-compose.yml +++ b/docker/gpu/docker-compose.yml @@ -1,9 +1,6 @@ name: kokoro-tts-gpu services: kokoro-tts: - user: "${UID}:${GID}" - group_add: - - "video" # image: ghcr.io/remsky/kokoro-fastapi-gpu:v0.2.0 build: context: ../..