intellema-vdk 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

intellema_vdk/__init__.py

@@ -1,11 +1,72 @@
 from typing import Optional, List, Any
 
-from .livekit_lib.client import LiveKitManager
-from .retell_lib.retell_client import RetellManager
-from .speech_lib.stt_client import STTManager
-from .speech_lib.tts_streamer import TTSStreamer
+from .providers import (
+    VoiceProvider,
+    LiveKitManager,
+    RetellManager,
+    # LiveKit Exceptions
+    LiveKitError,
+    LiveKitConfigurationError,
+    LiveKitRoomError,
+    LiveKitSIPError,
+    LiveKitDispatchError,
+    LiveKitEgressError,
+    # Retell Exceptions
+    RetellError,
+    RetellConfigurationError,
+    RetellAPIError,
+    RetellPhoneNumberError,
+    RetellCallError,
+)
+from .stt import (
+    STTManager,
+    STTAgentError,
+    STTConfigurationError,
+    STTError,
+    STTFileError,
+    STTTranscriptionError
+)
+from .tts import (
+    TTSStreamer,
+    TTSError,
+    TTSConfigurationError,
+    TTSStreamError,
+    TTSAPIError,
+)
+from .utils.logger_config import setup_logging
 
-def VoiceClient(provider: str, **kwargs) -> Any:
+__all__ = [
+    "VoiceClient",
+    "start_outbound_call",
+    "VoiceProvider",
+    "LiveKitManager",
+    "RetellManager",
+    "STTManager",
+    "TTSStreamer",
+    "setup_logging",
+    "LiveKitError",
+    "LiveKitConfigurationError",
+    "LiveKitRoomError",
+    "LiveKitSIPError",
+    "LiveKitDispatchError",
+    "LiveKitEgressError",
+    "RetellError",
+    "RetellConfigurationError",
+    "RetellAPIError",
+    "RetellPhoneNumberError",
+    "RetellCallError",
+    "STTAgentError",
+    "STTConfigurationError",
+    "STTError",
+    "STTFileError",
+    "STTTranscriptionError",
+    "TTSError",
+    "TTSConfigurationError",
+    "TTSStreamError",
+    "TTSAPIError",
+]
+
+def VoiceClient(provider: str, **kwargs) -> VoiceProvider:
     """
     Factory function that returns a specific provider client.
     
@@ -28,8 +89,4 @@ async def start_outbound_call(provider: str, *args, **kwargs):
     Convenience wrapper to start an outbound call.
     """
     client = VoiceClient(provider)
-    # Check if the method is async (LiveKit) or sync (Retell)
-    if provider == "livekit":
-        return await client.start_outbound_call(*args, **kwargs)
-    else:
-        return client.start_outbound_call(*args, **kwargs)
+    return await client.start_outbound_call(*args, **kwargs)

intellema_vdk/config.py

@@ -0,0 +1,14 @@
+import os
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv(dotenv_path=".env.local")
+load_dotenv()
+
+def get_env(key: str, default: str = None) -> str:
+    """Get an environment variable."""
+    return os.getenv(key, default)
+
+# TTS Configuration
+TTS_AUDIO_SAMPLE_RATE = 24000
+WAV_HEADER_SIZE = 44

intellema_vdk/providers/__init__.py

@@ -0,0 +1,35 @@
+from .livekit import (
+    LiveKitManager,
+    LiveKitError,
+    LiveKitConfigurationError,
+    LiveKitRoomError,
+    LiveKitSIPError,
+    LiveKitDispatchError,
+    LiveKitEgressError,
+)
+from .retell import (
+    RetellManager,
+    RetellError,
+    RetellConfigurationError,
+    RetellAPIError,
+    RetellPhoneNumberError,
+    RetellCallError,
+)
+from .protocols import VoiceProvider
+
+__all__ = [
+    "LiveKitManager",
+    "LiveKitError",
+    "LiveKitConfigurationError",
+    "LiveKitRoomError",
+    "LiveKitSIPError",
+    "LiveKitDispatchError",
+    "LiveKitEgressError",
+    "RetellManager",
+    "RetellError",
+    "RetellConfigurationError",
+    "RetellAPIError",
+    "RetellPhoneNumberError",
+    "RetellCallError",
+    "VoiceProvider",
+]

intellema_vdk/providers/livekit/__init__.py

@@ -0,0 +1,19 @@
+from .client import LiveKitManager
+from .exceptions import (
+    LiveKitError,
+    LiveKitConfigurationError,
+    LiveKitRoomError,
+    LiveKitSIPError,
+    LiveKitDispatchError,
+    LiveKitEgressError,
+)
+
+__all__ = [
+    "LiveKitManager",
+    "LiveKitError",
+    "LiveKitConfigurationError",
+    "LiveKitRoomError",
+    "LiveKitSIPError",
+    "LiveKitDispatchError",
+    "LiveKitEgressError",
+]