Pass Websocket to the context if available (#329)

* Add code

* Code

* Fix

* Add code

backend/fastrtc/utils.py

@@ -16,6 +16,7 @@ from typing import Any, Literal, Protocol, TypedDict, cast
 import av
 import librosa
 import numpy as np
+from fastapi import WebSocket
 from numpy.typing import NDArray
 from pydub import AudioSegment
 
@@ -67,6 +68,7 @@ current_channel: ContextVar[DataChannel | None] = ContextVar(
 @dataclass
 class Context:
     webrtc_id: str
+    websocket: WebSocket | None = None
 
 
 current_context: ContextVar[Context | None] = ContextVar(