fix(deepgram): forward enable_diarization and tags to open streams

livekit-plugins/livekit-plugins-deepgram/livekit/plugins/deepgram/stt.py

@@ -413,6 +413,7 @@ def update_options(
                 sample_rate=sample_rate,
                 no_delay=no_delay,
                 endpointing_ms=endpointing_ms,
+                enable_diarization=enable_diarization,
                 filler_words=filler_words,
                 keywords=keywords,
                 keyterm=keyterm,
@@ -421,6 +422,7 @@ def update_options(
                 numerals=numerals,
                 mip_opt_out=mip_opt_out,
                 vad_events=vad_events,
+                tags=tags,
                 endpoint_url=endpoint_url,
                 utterance_end_ms=utterance_end_ms,
                 dictation=dictation,

tests/test_plugin_deepgram_stt.py

@@ -391,6 +391,43 @@ async def ws_connect(self, url: str, **kwargs):
         return self._ws
 
 
+class _URLRecordingSession:
+    """Hands out a fresh socket per connect and keeps every URL it was asked to open."""
+
+    def __init__(self) -> None:
+        self.closed = False
+        self.urls: list[str] = []
+
+    async def ws_connect(self, url: str, **kwargs):
+        self.urls.append(url)
+        ws = _LiveWS()
+        ws._response = SimpleNamespace(headers={})  # read for the dg- response headers
+        return ws
+
+
+async def test_stt_update_options_reaches_open_streams_diarization_and_tags():
+    from urllib.parse import parse_qs, urlparse
+
+    from livekit.plugins.deepgram import STT
+
+    session = _URLRecordingSession()
+    instance = STT(api_key="test-key", language="en-US", http_session=session)
+    stream = instance.stream()
+    try:
+        await _wait_until(lambda: len(session.urls) == 1)
+        assert "diarize" not in parse_qs(urlparse(session.urls[0]).query)
+
+        instance.update_options(enable_diarization=True, tags=["prod"])
+
+        await _wait_until(lambda: len(session.urls) == 2)
+        query = parse_qs(urlparse(session.urls[1]).query)
+        assert query["diarize"] == ["true"]
+        assert query["tag"] == ["prod"]
+    finally:
+        await stream.aclose()
+        await instance.aclose()
+
+
 class _DeadSocket:
     """A socket that has silently gone away: writes fail, the read side never notices.