mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-25 08:48:13 +02:00
107 lines
3.5 KiB
Python
107 lines
3.5 KiB
Python
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"]
|