Add websockets dependency to assemblyai extra

pyproject.toml

@@ -40,7 +40,7 @@ Website = "https://pipecat.ai"
 
 [project.optional-dependencies]
 anthropic = [ "anthropic~=0.49.0" ]
-assemblyai = [ "assemblyai~=0.37.0" ]
+assemblyai = [ "websockets~=13.1" ]
 aws = [ "boto3~=1.37.16", "websockets~=13.1" ]
 aws-nova-sonic = [ "aws_sdk_bedrock_runtime~=0.0.2" ]
 azure = [ "azure-cognitiveservices-speech~=1.42.0"]

src/pipecat/services/assemblyai/stt.py

@@ -38,7 +38,7 @@ try:
     import websockets
 except ModuleNotFoundError as e:
     logger.error(f"Exception: {e}")
-    logger.error("In order to use AssemblyAI, you need to `pip install websockets`.")
+    logger.error('In order to use AssemblyAI, you need to `pip install "pipecat-ai[assemblyai]"`.')
     raise Exception(f"Missing module: {e}")