Skip to content

Commit 95b9b58

Browse files
Inference: Improved support for mid session TTS updates (#4412)
1 parent bfc0ce0 commit 95b9b58

File tree

1 file changed

+8
-0
lines changed
  • livekit-agents/livekit/agents/inference

1 file changed

+8
-0
lines changed

livekit-agents/livekit/agents/inference/tts.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -502,6 +502,14 @@ async def _sentence_stream_task(ws: aiohttp.ClientWebSocketResponse) -> None:
502502
async for ev in sent_tokenizer_stream:
503503
token_pkt = base_pkt.copy()
504504
token_pkt["transcript"] = ev.token + " "
505+
generation_config: dict[str, Any] = {}
506+
if self._opts.voice:
507+
generation_config["voice"] = self._opts.voice
508+
if self._opts.model:
509+
generation_config["model"] = self._opts.model
510+
if self._opts.language:
511+
generation_config["language"] = self._opts.language
512+
token_pkt["generation_config"] = generation_config
505513
token_pkt["extra"] = self._opts.extra_kwargs if self._opts.extra_kwargs else {}
506514
self._mark_started()
507515
await ws.send_str(json.dumps(token_pkt))

0 commit comments

Comments
 (0)