Change extra_headers to additional_headers, update websocket version support
This commit is contained in:
@@ -192,7 +192,7 @@ class AssemblyAISTTService(STTService):
|
||||
}
|
||||
self._websocket = await websockets.connect(
|
||||
ws_url,
|
||||
extra_headers=headers,
|
||||
additional_headers=headers,
|
||||
)
|
||||
self._connected = True
|
||||
self._receive_task = self.create_task(self._receive_task_handler())
|
||||
|
||||
@@ -238,7 +238,7 @@ class AWSTranscribeSTTService(STTService):
|
||||
)
|
||||
|
||||
# Add required headers
|
||||
extra_headers = {
|
||||
additional_headers = {
|
||||
"Origin": "https://localhost",
|
||||
"Sec-WebSocket-Key": websocket_key,
|
||||
"Sec-WebSocket-Version": "13",
|
||||
@@ -270,7 +270,7 @@ class AWSTranscribeSTTService(STTService):
|
||||
# Connect with the required headers and settings
|
||||
self._ws_client = await websockets.connect(
|
||||
presigned_url,
|
||||
extra_headers=extra_headers,
|
||||
additional_headers=additional_headers,
|
||||
subprotocols=["mqtt"],
|
||||
ping_interval=None,
|
||||
ping_timeout=None,
|
||||
|
||||
@@ -229,7 +229,7 @@ class CartesiaSTTService(STTService):
|
||||
headers = {"Cartesia-Version": "2025-04-16", "X-API-Key": self._api_key}
|
||||
|
||||
try:
|
||||
self._connection = await websockets.connect(ws_url, extra_headers=headers)
|
||||
self._connection = await websockets.connect(ws_url, additional_headers=headers)
|
||||
# Setup the receiver task to handle the incoming messages from the Cartesia server
|
||||
if self._receiver_task is None or self._receiver_task.done():
|
||||
self._receiver_task = asyncio.create_task(self._receive_messages())
|
||||
|
||||
@@ -475,7 +475,7 @@ class ElevenLabsTTSService(AudioContextWordTTSService):
|
||||
|
||||
# Set max websocket message size to 16MB for large audio responses
|
||||
self._websocket = await websockets.connect(
|
||||
url, max_size=16 * 1024 * 1024, extra_headers={"xi-api-key": self._api_key}
|
||||
url, max_size=16 * 1024 * 1024, additional_headers={"xi-api-key": self._api_key}
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
|
||||
@@ -216,7 +216,7 @@ class FishAudioTTSService(InterruptibleTTSService):
|
||||
logger.debug("Connecting to Fish Audio")
|
||||
headers = {"Authorization": f"Bearer {self._api_key}"}
|
||||
headers["model"] = self.model_name
|
||||
self._websocket = await websockets.connect(self._base_url, extra_headers=headers)
|
||||
self._websocket = await websockets.connect(self._base_url, additional_headers=headers)
|
||||
|
||||
# Send initial start message with ormsgpack
|
||||
start_message = {"event": "start", "request": {"text": "", **self._settings}}
|
||||
|
||||
@@ -292,7 +292,7 @@ class NeuphonicTTSService(InterruptibleTTSService):
|
||||
|
||||
headers = {"x-api-key": self._api_key}
|
||||
|
||||
self._websocket = await websockets.connect(url, extra_headers=headers)
|
||||
self._websocket = await websockets.connect(url, additional_headers=headers)
|
||||
except Exception as e:
|
||||
logger.error(f"{self} initialization error: {e}")
|
||||
self._websocket = None
|
||||
|
||||
@@ -57,7 +57,7 @@ class AzureRealtimeBetaLLMService(OpenAIRealtimeBetaLLMService):
|
||||
logger.info(f"Connecting to {self.base_url}, api key: {self.api_key}")
|
||||
self._websocket = await websockets.connect(
|
||||
uri=self.base_url,
|
||||
extra_headers={
|
||||
additional_headers={
|
||||
"api-key": self.api_key,
|
||||
},
|
||||
)
|
||||
|
||||
@@ -389,7 +389,7 @@ class OpenAIRealtimeBetaLLMService(LLMService):
|
||||
return
|
||||
self._websocket = await websockets.connect(
|
||||
uri=self.base_url,
|
||||
extra_headers={
|
||||
additional_headers={
|
||||
"Authorization": f"Bearer {self.api_key}",
|
||||
"OpenAI-Beta": "realtime=v1",
|
||||
},
|
||||
|
||||
@@ -244,7 +244,7 @@ class RimeTTSService(AudioContextWordTTSService):
|
||||
params = "&".join(f"{k}={v}" for k, v in self._settings.items())
|
||||
url = f"{self._url}?{params}"
|
||||
headers = {"Authorization": f"Bearer {self._api_key}"}
|
||||
self._websocket = await websockets.connect(url, extra_headers=headers)
|
||||
self._websocket = await websockets.connect(url, additional_headers=headers)
|
||||
except Exception as e:
|
||||
logger.error(f"{self} initialization error: {e}")
|
||||
self._websocket = None
|
||||
|
||||
Reference in New Issue
Block a user