Merge 0781f643c3 into 569f25098a
neil from camb.ai
7 days ago
committed by
GitHub
commit
e42f52fabc
@ -0,0 +1,50 @@
|
||||
import random
|
||||
|
||||
from camb.client import CambAI, save_stream_to_file
|
||||
from camb.types import StreamTtsOutputConfiguration
|
||||
|
||||
from utils import settings
|
||||
|
||||
|
||||
class CambAITTS:
|
||||
def __init__(self):
|
||||
self.max_chars = 5000
|
||||
self.client: CambAI = None
|
||||
|
||||
def run(self, text, filepath, random_voice: bool = False):
|
||||
if self.client is None:
|
||||
self.initialize()
|
||||
if random_voice:
|
||||
voice_id = self.randomvoice()
|
||||
else:
|
||||
voice_id = int(settings.config["settings"]["tts"]["cambai_voice_id"])
|
||||
|
||||
language = str(
|
||||
settings.config["settings"]["tts"].get("cambai_language", "en-us")
|
||||
)
|
||||
speech_model = str(
|
||||
settings.config["settings"]["tts"].get("cambai_speech_model", "mars-flash")
|
||||
)
|
||||
|
||||
stream = self.client.text_to_speech.tts(
|
||||
text=text,
|
||||
language=language,
|
||||
voice_id=voice_id,
|
||||
speech_model=speech_model,
|
||||
output_configuration=StreamTtsOutputConfiguration(format="mp3"),
|
||||
)
|
||||
save_stream_to_file(stream, filepath)
|
||||
|
||||
def initialize(self):
|
||||
api_key = settings.config["settings"]["tts"].get("cambai_api_key", "")
|
||||
if not api_key:
|
||||
raise ValueError(
|
||||
"You didn't set a CAMB AI API key! Please set the config variable cambai_api_key to a valid API key."
|
||||
)
|
||||
self.client = CambAI(api_key=api_key)
|
||||
|
||||
def randomvoice(self):
|
||||
if self.client is None:
|
||||
self.initialize()
|
||||
voices = self.client.voice_cloning.list_voices()
|
||||
return random.choice(voices)["id"]
|
||||
@ -0,0 +1,145 @@
|
||||
import os
|
||||
import tempfile
|
||||
import pytest
|
||||
from unittest.mock import patch, Mock, MagicMock
|
||||
|
||||
|
||||
class TestCambAITTSUnit:
|
||||
"""Unit tests for CambAITTS — all mocked, no API key needed."""
|
||||
|
||||
def test_max_chars_is_5000(self):
|
||||
with patch("TTS.cambai.settings") as mock_settings:
|
||||
from TTS.cambai import CambAITTS
|
||||
tts = CambAITTS()
|
||||
assert tts.max_chars == 5000
|
||||
|
||||
def test_initialize_raises_without_api_key(self):
|
||||
with patch("TTS.cambai.settings") as mock_settings:
|
||||
mock_settings.config = {"settings": {"tts": {"cambai_api_key": ""}}}
|
||||
from TTS.cambai import CambAITTS
|
||||
tts = CambAITTS()
|
||||
with pytest.raises(ValueError, match="CAMB AI API key"):
|
||||
tts.initialize()
|
||||
|
||||
def test_initialize_creates_client(self):
|
||||
with patch("TTS.cambai.settings") as mock_settings, \
|
||||
patch("TTS.cambai.CambAI") as mock_camb:
|
||||
mock_settings.config = {"settings": {"tts": {"cambai_api_key": "test-key"}}}
|
||||
from TTS.cambai import CambAITTS
|
||||
tts = CambAITTS()
|
||||
tts.initialize()
|
||||
mock_camb.assert_called_once_with(api_key="test-key")
|
||||
assert tts.client is not None
|
||||
|
||||
def test_run_generates_mp3(self):
|
||||
with patch("TTS.cambai.settings") as mock_settings, \
|
||||
patch("TTS.cambai.CambAI") as mock_camb_cls, \
|
||||
patch("TTS.cambai.save_stream_to_file") as mock_save, \
|
||||
patch("TTS.cambai.StreamTtsOutputConfiguration") as mock_config:
|
||||
mock_settings.config = {
|
||||
"settings": {"tts": {
|
||||
"cambai_api_key": "test-key",
|
||||
"cambai_voice_id": "147320",
|
||||
"cambai_language": "en-us",
|
||||
"cambai_speech_model": "mars-flash",
|
||||
}}
|
||||
}
|
||||
mock_client = Mock()
|
||||
mock_client.text_to_speech.tts.return_value = iter([b"fake-audio"])
|
||||
mock_camb_cls.return_value = mock_client
|
||||
|
||||
from TTS.cambai import CambAITTS
|
||||
tts = CambAITTS()
|
||||
tts.run("Hello world", "/tmp/test.mp3", random_voice=False)
|
||||
|
||||
mock_client.text_to_speech.tts.assert_called_once()
|
||||
call_kwargs = mock_client.text_to_speech.tts.call_args
|
||||
assert call_kwargs.kwargs["text"] == "Hello world"
|
||||
assert call_kwargs.kwargs["voice_id"] == 147320
|
||||
assert call_kwargs.kwargs["language"] == "en-us"
|
||||
assert call_kwargs.kwargs["speech_model"] == "mars-flash"
|
||||
mock_save.assert_called_once()
|
||||
assert mock_save.call_args[0][1] == "/tmp/test.mp3"
|
||||
|
||||
def test_run_reads_config_voice_id(self):
|
||||
with patch("TTS.cambai.settings") as mock_settings, \
|
||||
patch("TTS.cambai.CambAI") as mock_camb_cls, \
|
||||
patch("TTS.cambai.save_stream_to_file"), \
|
||||
patch("TTS.cambai.StreamTtsOutputConfiguration"):
|
||||
mock_settings.config = {
|
||||
"settings": {"tts": {
|
||||
"cambai_api_key": "test-key",
|
||||
"cambai_voice_id": "99999",
|
||||
"cambai_language": "es-es",
|
||||
"cambai_speech_model": "mars-pro",
|
||||
}}
|
||||
}
|
||||
|
||||
mock_client = Mock()
|
||||
mock_client.text_to_speech.tts.return_value = iter([b"audio"])
|
||||
mock_camb_cls.return_value = mock_client
|
||||
|
||||
from TTS.cambai import CambAITTS
|
||||
tts = CambAITTS()
|
||||
tts.run("test", "/tmp/out.mp3")
|
||||
|
||||
call_kwargs = mock_client.text_to_speech.tts.call_args.kwargs
|
||||
assert call_kwargs["voice_id"] == 99999
|
||||
assert call_kwargs["language"] == "es-es"
|
||||
assert call_kwargs["speech_model"] == "mars-pro"
|
||||
|
||||
def test_random_voice_picks_from_list(self):
|
||||
with patch("TTS.cambai.settings") as mock_settings, \
|
||||
patch("TTS.cambai.CambAI") as mock_camb_cls, \
|
||||
patch("TTS.cambai.save_stream_to_file"), \
|
||||
patch("TTS.cambai.StreamTtsOutputConfiguration"), \
|
||||
patch("TTS.cambai.random") as mock_random:
|
||||
mock_settings.config = {"settings": {"tts": {"cambai_api_key": "test-key"}}}
|
||||
mock_client = Mock()
|
||||
mock_client.voice_cloning.list_voices.return_value = [
|
||||
{"id": 111, "voice_name": "Voice A"},
|
||||
{"id": 222, "voice_name": "Voice B"},
|
||||
]
|
||||
mock_client.text_to_speech.tts.return_value = iter([b"audio"])
|
||||
mock_camb_cls.return_value = mock_client
|
||||
mock_random.choice.return_value = {"id": 222, "voice_name": "Voice B"}
|
||||
|
||||
from TTS.cambai import CambAITTS
|
||||
tts = CambAITTS()
|
||||
tts.run("test", "/tmp/out.mp3", random_voice=True)
|
||||
|
||||
mock_client.voice_cloning.list_voices.assert_called_once()
|
||||
mock_random.choice.assert_called_once()
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
class TestCambAITTSIntegration:
|
||||
"""Integration tests — require CAMB_API_KEY env var."""
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def skip_without_key(self):
|
||||
if not os.environ.get("CAMB_API_KEY"):
|
||||
pytest.skip("CAMB_API_KEY not set")
|
||||
|
||||
def test_real_api_generates_audio(self):
|
||||
from camb.client import CambAI, save_stream_to_file
|
||||
from camb.types import StreamTtsOutputConfiguration
|
||||
|
||||
client = CambAI(api_key=os.environ["CAMB_API_KEY"])
|
||||
with tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as f:
|
||||
filepath = f.name
|
||||
|
||||
try:
|
||||
stream = client.text_to_speech.tts(
|
||||
text="Integration test for RedditVideoMakerBot.",
|
||||
language="en-us",
|
||||
voice_id=147320,
|
||||
speech_model="mars-flash",
|
||||
output_configuration=StreamTtsOutputConfiguration(format="mp3"),
|
||||
)
|
||||
save_stream_to_file(stream, filepath)
|
||||
assert os.path.exists(filepath)
|
||||
assert os.path.getsize(filepath) > 0
|
||||
finally:
|
||||
if os.path.exists(filepath):
|
||||
os.remove(filepath)
|
||||
Loading…
Reference in new issue