feat: support flux for dograh multi

api/tests/test_dograh_stt_service_factory.py

@@ -0,0 +1,107 @@
+from types import SimpleNamespace
+from unittest.mock import patch
+
+from pipecat.services.settings import NOT_GIVEN
+from pipecat.transcriptions.language import Language
+
+from api.services.configuration.registry import ServiceProviders
+from api.services.pipecat.audio_config import AudioConfig
+from api.services.pipecat.service_factory import (
+    create_stt_service,
+    dograh_stt_uses_flux_language,
+    stt_uses_flux_turns,
+)
+
+
+def _audio_config() -> AudioConfig:
+    return AudioConfig(
+        transport_in_sample_rate=16000,
+        transport_out_sample_rate=16000,
+    )
+
+
+def _dograh_config(language: str | None) -> SimpleNamespace:
+    return SimpleNamespace(
+        stt=SimpleNamespace(
+            provider=ServiceProviders.DOGRAH.value,
+            api_key="mps-key",
+            model="default",
+            language=language,
+        )
+    )
+
+
+def test_dograh_flux_language_predicate_matches_multilingual_support():
+    assert dograh_stt_uses_flux_language(None)
+    assert dograh_stt_uses_flux_language("multi")
+    assert dograh_stt_uses_flux_language("es")
+    assert not dograh_stt_uses_flux_language("ar")
+
+
+def test_stt_uses_flux_turns_only_for_dograh_flux_supported_languages():
+    assert stt_uses_flux_turns(_dograh_config("multi"))
+    assert stt_uses_flux_turns(_dograh_config("es"))
+    assert not stt_uses_flux_turns(_dograh_config("ar"))
+
+
+def test_create_dograh_multi_uses_flux_service_without_language_hint():
+    user_config = _dograh_config("multi")
+
+    with (
+        patch(
+            "api.services.pipecat.service_factory.DograhFluxSTTService"
+        ) as flux_service,
+        patch("api.services.pipecat.service_factory.DograhSTTService") as stt_service,
+    ):
+        create_stt_service(user_config, _audio_config(), correlation_id="corr-123")
+
+    flux_service.assert_called_once()
+    stt_service.assert_not_called()
+    kwargs = flux_service.call_args.kwargs
+    assert kwargs["correlation_id"] == "corr-123"
+    assert kwargs["settings"].model == "flux-general-multi"
+    assert kwargs["settings"].language_hints is NOT_GIVEN
+
+
+def test_create_dograh_supported_language_uses_flux_service_with_hint():
+    user_config = _dograh_config("es")
+
+    with (
+        patch(
+            "api.services.pipecat.service_factory.DograhFluxSTTService"
+        ) as flux_service,
+        patch("api.services.pipecat.service_factory.DograhSTTService") as stt_service,
+    ):
+        create_stt_service(user_config, _audio_config(), keyterms=["Dograh"])
+
+    flux_service.assert_called_once()
+    stt_service.assert_not_called()
+    kwargs = flux_service.call_args.kwargs
+    assert kwargs["settings"].model == "flux-general-multi"
+    assert kwargs["settings"].language_hints == [Language.ES]
+    assert kwargs["settings"].keyterm == ["Dograh"]
+
+
+def test_create_dograh_unsupported_language_falls_back_to_standard_stt_service():
+    user_config = _dograh_config("ar")
+
+    with (
+        patch(
+            "api.services.pipecat.service_factory.DograhFluxSTTService"
+        ) as flux_service,
+        patch("api.services.pipecat.service_factory.DograhSTTService") as stt_service,
+    ):
+        create_stt_service(
+            user_config,
+            _audio_config(),
+            keyterms=["Dograh"],
+            correlation_id="corr-123",
+        )
+
+    flux_service.assert_not_called()
+    stt_service.assert_called_once()
+    kwargs = stt_service.call_args.kwargs
+    assert kwargs["correlation_id"] == "corr-123"
+    assert kwargs["settings"].model == "default"
+    assert kwargs["settings"].language == "ar"
+    assert kwargs["keyterms"] == ["Dograh"]