Change extra_headers to additional_headers, update websocket version support

This commit is contained in:
Mark Backman
2025-07-22 15:25:03 -04:00
parent 994e82c1ef
commit 7955080da2
11 changed files with 28 additions and 24 deletions

View File

@@ -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())

View File

@@ -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,

View File

@@ -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())

View File

@@ -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:

View File

@@ -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}}

View File

@@ -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

View File

@@ -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,
},
)

View File

@@ -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",
},

View File

@@ -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