use conditional imports and show help errors if modules not found

src/dailyai/services/anthropic_llm_service.py

@@ -1,9 +1,16 @@
 from typing import AsyncGenerator
-from anthropic import AsyncAnthropic
 from dailyai.pipeline.frames import Frame, LLMMessagesFrame, TextFrame
 
 from dailyai.services.ai_services import LLMService
 
+try:
+    from anthropic import AsyncAnthropic
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use Anthropic, you need to `pip install dailyai[anthropic]`. Also, set `ANTHROPIC_API_KEY` environment variable.")
+    raise Exception(f"Missing module: {e}")
+
 
 class AnthropicLLMService(LLMService):
 

src/dailyai/services/azure_ai_services.py

@@ -9,12 +9,18 @@ from dailyai.services.ai_services import TTSService, ImageGenService
 from PIL import Image
 
 # See .env.example for Azure configuration needed
-from azure.cognitiveservices.speech import (
-    SpeechSynthesizer,
-    SpeechConfig,
-    ResultReason,
-    CancellationReason,
-)
+try:
+    from azure.cognitiveservices.speech import (
+        SpeechSynthesizer,
+        SpeechConfig,
+        ResultReason,
+        CancellationReason,
+    )
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use Azure TTS, you need to `pip install dailyai[azure]`. Also, set `SPEECH_KEY` and `SPEECH_REGION` environment variables.")
+    raise Exception(f"Missing module: {e}")
 
 from dailyai.services.openai_api_llm_service import BaseOpenAILLMService
 

src/dailyai/services/fal_ai_services.py

@@ -1,4 +1,3 @@
-import fal
 import aiohttp
 import asyncio
 import io
@@ -10,7 +9,13 @@ from dailyai.services.ai_services import ImageGenService
 
 from dailyai.services.ai_services import ImageGenService
 
-# Fal expects FAL_KEY_ID and FAL_KEY_SECRET to be set in the env
+try:
+    import fal
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use Fal, you need to `pip install dailyai[fal]`. Also, set `FAL_KEY_ID` and `FAL_KEY_SECRET` environment variables.")
+    raise Exception(f"Missing module: {e}")
 
 
 class FalImageGenService(ImageGenService):

src/dailyai/services/open_ai_services.py

@@ -1,12 +1,20 @@
 import aiohttp
 from PIL import Image
 import io
-from openai import AsyncOpenAI
 
 from dailyai.services.ai_services import ImageGenService
 from dailyai.services.openai_api_llm_service import BaseOpenAILLMService
 
 
+try:
+    from openai import AsyncOpenAI
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use OpenAI, you need to `pip install dailyai[openai]`. Also, set `OPENAI_API_KEY` environment variable.")
+    raise Exception(f"Missing module: {e}")
+
+
 class OpenAILLMService(BaseOpenAILLMService):
 
     def __init__(self, model="gpt-4", * args, **kwargs):

src/dailyai/services/openai_api_llm_service.py

@@ -1,7 +1,6 @@
 import json
 import time
 from typing import AsyncGenerator, List
-from openai import AsyncOpenAI, AsyncStream
 from dailyai.pipeline.frames import (
     Frame,
     LLMFunctionCallFrame,
@@ -9,17 +8,25 @@ from dailyai.pipeline.frames import (
     LLMMessagesFrame,
     LLMResponseEndFrame,
     LLMResponseStartFrame,
-    OpenAILLMContextFrame,
     TextFrame,
 )
 from dailyai.services.ai_services import LLMService
+from dailyai.pipeline.openai_frames import OpenAILLMContextFrame
 from dailyai.services.openai_llm_context import OpenAILLMContext
 
-from openai.types.chat import (
-    ChatCompletion,
-    ChatCompletionChunk,
-    ChatCompletionMessageParam,
-)
+try:
+    from openai import AsyncOpenAI, AsyncStream
+
+    from openai.types.chat import (
+        ChatCompletion,
+        ChatCompletionChunk,
+        ChatCompletionMessageParam,
+    )
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use OpenAI, you need to `pip install dailyai[openai]`. Also, set `OPENAI_API_KEY` environment variable.")
+    raise Exception(f"Missing module: {e}")
 
 
 class BaseOpenAILLMService(LLMService):

src/dailyai/services/openai_llm_context.py

@@ -1,11 +1,18 @@
 from typing import List
-from openai._types import NOT_GIVEN, NotGiven
 
-from openai.types.chat import (
-    ChatCompletionToolParam,
-    ChatCompletionToolChoiceOptionParam,
-    ChatCompletionMessageParam,
-)
+try:
+    from openai._types import NOT_GIVEN, NotGiven
+
+    from openai.types.chat import (
+        ChatCompletionToolParam,
+        ChatCompletionToolChoiceOptionParam,
+        ChatCompletionMessageParam,
+    )
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use OpenAI, you need to `pip install dailyai[openai]`. Also, set `OPENAI_API_KEY` environment variable.")
+    raise Exception(f"Missing module: {e}")
 
 
 class OpenAILLMContext:

src/dailyai/services/playht_ai_service.py

@@ -1,11 +1,18 @@
 import io
 import struct
-from pyht import Client
-from pyht.client import TTSOptions
-from pyht.protos.api_pb2 import Format
 
 from dailyai.services.ai_services import TTSService
 
+try:
+    from pyht import Client
+    from pyht.client import TTSOptions
+    from pyht.protos.api_pb2 import Format
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use PlayHT, you need to `pip install dailyai[playht]`. Also, set `PLAY_HT_USER_ID` and `PLAY_HT_API_KEY` environment variables.")
+    raise Exception(f"Missing module: {e}")
+
 
 class PlayHTAIService(TTSService):
 

src/dailyai/services/whisper_ai_services.py

@@ -3,10 +3,18 @@ import asyncio
 from enum import Enum
 import logging
 from typing import BinaryIO
-from faster_whisper import WhisperModel
 from dailyai.services.local_stt_service import LocalSTTService
 
 
+try:
+    from faster_whisper import WhisperModel
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use Whisper, you need to `pip install dailyai[whisper]`.")
+    raise Exception(f"Missing module: {e}")
+
+
 class Model(Enum):
     """Class of basic Whisper model selection options"""
     TINY = "tiny"