Fix tests and mypy issues

Author: PaulAsjes

.fernignore

@@ -6,6 +6,7 @@ src/elevenlabs/conversational_ai/conversation.py
 src/elevenlabs/conversational_ai/default_audio_interface.py
 src/elevenlabs/realtime_tts.py
 src/elevenlabs/play.py
+src/elevenlabs/realtime_tts.py
 src/elevenlabs/webhooks.py
 
 # Ignore CI files

README.md

@@ -83,7 +83,7 @@ client = ElevenLabs(
   api_key="YOUR_API_KEY",
 )
 
-response = client.voices.get_all()
+response = client.voices.search()
 print(response.voices)
 ```
 
@@ -106,7 +106,7 @@ client = ElevenLabs(
   api_key="YOUR_API_KEY",
 )
 
-voice = client.voices.add(
+voice = client.voices.ivc.create(
     name="Alex",
     description="An old American male voice with a slight hoarseness in his throat. Perfect for news", # Optional
     files=["./sample_0.mp3", "./sample_1.mp3", "./sample_2.mp3"],
@@ -123,7 +123,7 @@ from elevenlabs.client import ElevenLabs
 
 client = ElevenLabs()
 
-audio_stream = client.text_to_speech.convert_as_stream(
+audio_stream = client.text_to_speech.stream(
     text="This is a test",
     voice_id="JBFqnCBsd6RMkjVDRZzb",
     model_id="eleven_multilingual_v2"
@@ -153,7 +153,7 @@ eleven = AsyncElevenLabs(
 )
 
 async def print_models() -> None:
-    models = await eleven.models.get_all()
+    models = await eleven.models.list()
     print(models)
 
 asyncio.run(print_models())

src/elevenlabs/realtime_tts.py

@@ -42,6 +42,7 @@ def text_chunker(chunks: typing.Iterator[str]) -> typing.Iterator[str]:
 class RealtimeTextToSpeechClient(TextToSpeechClient):
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         super().__init__(client_wrapper=client_wrapper)
+        self._client_wrapper = client_wrapper
         self._ws_base_url = urllib.parse.urlparse(self._client_wrapper.get_environment().base)._replace(scheme="wss").geturl()
 
     def convert_realtime(

tests/test_audio_isolation.py

@@ -9,7 +9,7 @@ def test_audio_isolation() -> None:
     client = ElevenLabs()
     audio_file = open(DEFAULT_VOICE_FILE, "rb")
     try:
-        audio_stream = client.audio_isolation.audio_isolation(audio=audio_file)
+        audio_stream = client.audio_isolation.convert(audio=audio_file)
         audio = b"".join(chunk for chunk in audio_stream)
         assert isinstance(audio, bytes), "Combined audio should be bytes"
         if not IN_GITHUB:
@@ -23,7 +23,7 @@ def test_audio_isolation_as_stream():
     client = ElevenLabs()
     audio_file = open(DEFAULT_VOICE_FILE, "rb")
     try:
-        audio_stream = client.audio_isolation.audio_isolation_stream(audio=audio_file)
+        audio_stream = client.audio_isolation.stream(audio=audio_file)
         audio = b"".join(chunk for chunk in audio_stream)
         assert isinstance(audio, bytes), "Combined audio should be bytes"
         if not IN_GITHUB:

tests/test_convai.py

@@ -91,7 +91,7 @@ def test_conversation_basic_flow():
 def test_conversation_with_auth():
     # Mock setup
     mock_client = MagicMock()
-    mock_client.conversational_ai.get_signed_url.return_value.signed_url = "wss://signed.url"
+    mock_client.conversational_ai.conversations.get_signed_url.return_value.signed_url = "wss://signed.url"
     mock_ws = create_mock_websocket(
         [
             {
@@ -116,7 +116,7 @@ def test_conversation_with_auth():
         conversation.wait_for_session_end()
 
     # Assertions
-    mock_client.conversational_ai.get_signed_url.assert_called_once_with(agent_id=TEST_AGENT_ID)
+    mock_client.conversational_ai.conversations.get_signed_url.assert_called_once_with(agent_id=TEST_AGENT_ID)
 
 def test_conversation_with_dynamic_variables():
     # Mock setup

tests/test_history.py

@@ -4,5 +4,5 @@
 def test_history():
     client = ElevenLabs()
     page_size = 5
-    history = client.history.get_all(page_size=page_size)
+    history = client.history.list(page_size=page_size)
     assert isinstance(history, GetSpeechHistoryResponse)

tests/test_models.py

@@ -4,6 +4,6 @@
 
 def test_models_get_all():
     client = ElevenLabs()
-    models = client.models.get_all()
+    models = client.models.list()
     assert len(models) > 0
     assert isinstance(models[0], Model)

tests/test_sts.py

@@ -22,7 +22,7 @@ def test_sts_as_stream():
     client = ElevenLabs()
     audio_file = open(DEFAULT_VOICE_FILE, "rb")
     try:
-        audio_stream = client.speech_to_speech.convert_as_stream(voice_id=DEFAULT_VOICE, audio=audio_file)
+        audio_stream = client.speech_to_speech.stream(voice_id=DEFAULT_VOICE, audio=audio_file)
         audio = b"".join(chunk for chunk in audio_stream)
         assert isinstance(audio, bytes), "Combined audio should be bytes"
         if not IN_GITHUB:

tests/test_tts.py

@@ -35,7 +35,7 @@ def test_tts_convert_with_voice_settings() -> None:
 def test_tts_convert_as_stream():
     async def main():
         async_client = AsyncElevenLabs()
-        results = async_client.text_to_speech.convert_as_stream(
+        results = async_client.text_to_speech.stream(
             text=DEFAULT_TEXT, voice_id=DEFAULT_VOICE, model_id=DEFAULT_MODEL
         )
         out = b""

tests/test_voices.py

@@ -16,7 +16,7 @@ def test_get_voice():
 
 def test_get_voices():
     client = ElevenLabs()
-    response = client.voices.get_all()
+    response = client.voices.search()
 
     assert len(response.voices) > 0