From 428cee75c53a0b53a69aa4bfc3bb140206ca051e Mon Sep 17 00:00:00 2001 From: Martin Schweiger Date: Mon, 26 May 2025 09:10:55 +0800 Subject: [PATCH] Add User-Agent header to AssemblyAI websocket connection --- src/pipecat/services/assemblyai/stt.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/pipecat/services/assemblyai/stt.py b/src/pipecat/services/assemblyai/stt.py index 2e8ad3058..bdc2bafd1 100644 --- a/src/pipecat/services/assemblyai/stt.py +++ b/src/pipecat/services/assemblyai/stt.py @@ -11,6 +11,7 @@ from urllib.parse import urlencode from loguru import logger +from pipecat import __version__ as pipecat_version from pipecat.frames.frames import ( CancelFrame, EndFrame, @@ -108,9 +109,13 @@ class AssemblyAISTTService(STTService): async def _connect(self): try: ws_url = self._build_ws_url() + headers = { + "Authorization": self._api_key, + "User-Agent": f"AssemblyAI/1.0 (integration=Pipecat/{pipecat_version})", + } self._websocket = await websockets.connect( ws_url, - extra_headers={"Authorization": self._api_key}, + extra_headers=headers, ) self._connected = True self._receive_task = asyncio.create_task(self._receive_task_handler())