deepgeek/gradio-webrtc
2
0
Fork 0
mirror of https://github.com/HumanAIGC-Engineering/gradio-webrtc.git synced 2026-02-05 18:09:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enforce modern typing (#258)

Browse Source 
* Allow UP

* Upgrade typing

* test smolagents

* Change to contextlib

---------

Co-authored-by: Marcus Valtonen Örnhag <marcus.valtonen.ornhag@ericsson.com>
This commit is contained in:
Marcus Valtonen Örnhag
2025-04-08 22:46:12 +02:00
committed by GitHub
parent a07e9439b6
commit f70b27bd41
15 changed files with 43 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
backend/fastrtc/websocket.py
View File

@@ -2,7 +2,8 @@ import asyncio
import audioop
import base64
import logging
from typing import Any, Awaitable, Callable, Optional, cast
from collections.abc import Awaitable, Callable
from typing import Any, cast
import anyio
import librosa
@@ -57,9 +58,9 @@ class WebSocketHandler:
):
self.stream_handler = stream_handler
self.stream_handler._clear_queue = self._clear_queue
self.websocket: Optional[WebSocket] = None
self._emit_task: Optional[asyncio.Task] = None
self.stream_id: Optional[str] = None
self.websocket: WebSocket | None = None
self._emit_task: asyncio.Task | None = None
self.stream_id: str | None = None
self.set_additional_outputs_factory = additional_outputs_factory
self.set_additional_outputs: Callable[[AdditionalOutputs], None]
self.set_handler = set_handler
@@ -67,8 +68,8 @@ class WebSocketHandler:
self.clean_up = clean_up
self.queue = asyncio.Queue()
self.playing_durations = [] # Track durations of frames being played
self._frame_cleanup_task: Optional[asyncio.Task] = None
self._graceful_shutdown_task: Optional[asyncio.Task] = None
self._frame_cleanup_task: asyncio.Task | None = None
self._graceful_shutdown_task: asyncio.Task | None = None
def _clear_queue(self):
old_queue = self.queue