refactor(#5430): reuse shared TTS transport scaffolding

This commit is contained in:
Rod Boev committed 2026-07-02 15:43:42 -04:00
1 parent 74724b944d
commit 4993eed120
2 files changed
+1 -21

No files matched your search

+1 -9
View File
@@ -16670,26 +16670,19 @@ def _handle_tts(handler, parsed):
"voice_settings": {"stability": 0.5, "similarity_boost": 0.75}, "voice_settings": {"stability": 0.5, "similarity_boost": 0.75},
}).encode("utf-8") }).encode("utf-8")
from urllib.request import Request, build_opener
req = Request(url, data=req_body, headers={ req = Request(url, data=req_body, headers={
"xi-api-key": api_key, "xi-api-key": api_key,
"Content-Type": "application/json", "Content-Type": "application/json",
"Accept": "audio/mpeg", "Accept": "audio/mpeg",
}) })
class _NoRedirectElevenlabsHandler(HTTPRedirectHandler):
"""Refuse to follow redirects on the ElevenLabs call."""
def redirect_request(self, req, fp, code, msg, headers, newurl):
raise ValueError("ElevenLabs TTS upstream attempted a redirect")
# Buffer the full response before sending first byte. # Buffer the full response before sending first byte.
# The streaming endpoint is designed for chunked delivery, but urllib's # The streaming endpoint is designed for chunked delivery, but urllib's
# chunked-read path adds per-chunk overhead that dominates short TTS # chunked-read path adds per-chunk overhead that dominates short TTS
# payloads. A hard cap keeps the buffered path bounded even if the # payloads. A hard cap keeps the buffered path bounded even if the
# upstream misbehaves. # upstream misbehaves.
try: try:
with _tts_open(req, timeout=30, opener_factory=lambda: build_opener(ProxyHandler({}), _NoRedirectElevenlabsHandler())) as resp: with _tts_open(req, timeout=30, opener_factory=lambda: build_opener(ProxyHandler({}), _NoRedirectTtsHandler())) as resp:
audio_data = _buffer_tts_audio_response(resp) audio_data = _buffer_tts_audio_response(resp)
except ValueError: except ValueError:
logger.warning("ElevenLabs TTS rejected an invalid upstream response", exc_info=True) logger.warning("ElevenLabs TTS rejected an invalid upstream response", exc_info=True)
@@ -16759,7 +16752,6 @@ def _handle_tts(handler, parsed):
"voice": oai_voice, "voice": oai_voice,
}).encode("utf-8") }).encode("utf-8")
from urllib.request import Request, build_opener, urlopen as _urlopen
req = Request(url, data=req_body, headers={ req = Request(url, data=req_body, headers={
"Authorization": f"Bearer {api_key}", "Authorization": f"Bearer {api_key}",
"Content-Type": "application/json", "Content-Type": "application/json",
-12
View File
@@ -176,7 +176,6 @@ def test_elevenlabs_rejects_oversized_upstream_audio(monkeypatch):
def test_elevenlabs_tts_does_not_follow_upstream_redirect(monkeypatch): def test_elevenlabs_tts_does_not_follow_upstream_redirect(monkeypatch):
# Regression proof: a fake redirect response should fail at _tts_open, not be followed. # Regression proof: a fake redirect response should fail at _tts_open, not be followed.
import urllib.request as _ur
from urllib.request import HTTPRedirectHandler from urllib.request import HTTPRedirectHandler
class _Resp: class _Resp:
@@ -194,9 +193,6 @@ def test_elevenlabs_tts_does_not_follow_upstream_redirect(monkeypatch):
def __exit__(self, *a): def __exit__(self, *a):
return False return False
def _fake_urlopen(req, timeout=30):
return _Resp()
def _fake_tts_open(req, timeout=30, opener_factory=None, **_kw): def _fake_tts_open(req, timeout=30, opener_factory=None, **_kw):
assert opener_factory is not None assert opener_factory is not None
opener = opener_factory() opener = opener_factory()
@@ -216,12 +212,9 @@ def test_elevenlabs_tts_does_not_follow_upstream_redirect(monkeypatch):
) )
raise ValueError("redirect blocked 302") raise ValueError("redirect blocked 302")
# Keep the legacy path patched as fake audio to keep base and head checks
# behaviorally distinct.
monkeypatch.setenv("ELEVENLABS_API_KEY", "sk-test") monkeypatch.setenv("ELEVENLABS_API_KEY", "sk-test")
import api.config as _cfg import api.config as _cfg
monkeypatch.setattr(_cfg, "get_config", lambda: {"tts": {"elevenlabs": {"voice_id": "pNInz6obpgDQGcFmaJgB", "model": "eleven_multilingual_v2"}}}) monkeypatch.setattr(_cfg, "get_config", lambda: {"tts": {"elevenlabs": {"voice_id": "pNInz6obpgDQGcFmaJgB", "model": "eleven_multilingual_v2"}}})
monkeypatch.setattr(_ur, "urlopen", _fake_urlopen)
monkeypatch.setattr(routes, "_tts_open", _fake_tts_open) monkeypatch.setattr(routes, "_tts_open", _fake_tts_open)
h = _post({"text": "hello world", "engine": "elevenlabs"}, client="9.9.9.6") h = _post({"text": "hello world", "engine": "elevenlabs"}, client="9.9.9.6")
routes._handle_tts(h, None) routes._handle_tts(h, None)
@@ -232,7 +225,6 @@ def test_elevenlabs_tts_does_not_follow_upstream_redirect(monkeypatch):
def test_elevenlabs_tts_uses_no_proxy_opener(monkeypatch): def test_elevenlabs_tts_uses_no_proxy_opener(monkeypatch):
captured = {} captured = {}
import urllib.request as _ur
class _Resp: class _Resp:
def __init__(self): def __init__(self):
@@ -248,9 +240,6 @@ def test_elevenlabs_tts_uses_no_proxy_opener(monkeypatch):
def __exit__(self, *a): def __exit__(self, *a):
return False return False
def _fake_urlopen(req, timeout=30):
return _Resp()
def _fake_tts_open(req, timeout=30, opener_factory=None, **_kw): def _fake_tts_open(req, timeout=30, opener_factory=None, **_kw):
captured["opener_factory"] = opener_factory captured["opener_factory"] = opener_factory
opener = opener_factory() opener = opener_factory()
@@ -270,7 +259,6 @@ def test_elevenlabs_tts_uses_no_proxy_opener(monkeypatch):
monkeypatch.setenv("ELEVENLABS_API_KEY", "sk-test") monkeypatch.setenv("ELEVENLABS_API_KEY", "sk-test")
import api.config as _cfg import api.config as _cfg
monkeypatch.setattr(_cfg, "get_config", lambda: {"tts": {"elevenlabs": {"voice_id": "pNInz6obpgDQGcFmaJgB", "model": "eleven_multilingual_v2"}}}) monkeypatch.setattr(_cfg, "get_config", lambda: {"tts": {"elevenlabs": {"voice_id": "pNInz6obpgDQGcFmaJgB", "model": "eleven_multilingual_v2"}}})
monkeypatch.setattr(_ur, "urlopen", _fake_urlopen)
monkeypatch.setattr(routes, "_tts_open", _fake_tts_open) monkeypatch.setattr(routes, "_tts_open", _fake_tts_open)
h = _post({"text": "hello world", "engine": "elevenlabs"}, client="9.9.9.7") h = _post({"text": "hello world", "engine": "elevenlabs"}, client="9.9.9.7")