diff --git a/dashscope/audio/qwen_omni/omni_realtime.py b/dashscope/audio/qwen_omni/omni_realtime.py index e08c6f3..ecf27e5 100644 --- a/dashscope/audio/qwen_omni/omni_realtime.py +++ b/dashscope/audio/qwen_omni/omni_realtime.py @@ -5,7 +5,7 @@ import threading import time from dataclasses import field, dataclass -from typing import List +from typing import List, Any import uuid from enum import Enum, unique @@ -35,7 +35,13 @@ class TranslationParams: """ TranslationParams """ + + @dataclass + class Corpus: + phrases: dict[str, Any] = field(default=None) + language: str = field(default=None) + corpus: Corpus = field(default=None) @dataclass @@ -262,8 +268,13 @@ def update_session(self, self.config['turn_detection'] = None if translation_params is not None: self.config['translation'] = { - 'language': translation_params.language + 'language': translation_params.language, } + if translation_params.corpus is not None: + if translation_params.corpus and translation_params.corpus.phrases is not None: + self.config['translation']['corpus'] = { + 'phrases': translation_params.corpus.phrases + } if transcription_params is not None: self.config['language'] = transcription_params.language if transcription_params.corpus is not None: