diff --git a/src/pipecat/services/groq.py b/src/pipecat/services/groq.py index 9345f9be1..9d4b09455 100644 --- a/src/pipecat/services/groq.py +++ b/src/pipecat/services/groq.py @@ -7,7 +7,6 @@ from typing import AsyncGenerator, Optional -from groq import AsyncGroq from loguru import logger from pydantic import BaseModel @@ -17,6 +16,15 @@ from pipecat.services.base_whisper import BaseWhisperSTTService, Transcription from pipecat.services.openai import OpenAILLMService from pipecat.transcriptions.language import Language +try: + from groq import AsyncGroq +except ModuleNotFoundError as e: + logger.error(f"Exception: {e}") + logger.error( + "In order to use Groq, you need to `pip install pipecat-ai[groq]`. Also, set a `GROQ_API_KEY` environment variable." + ) + raise Exception(f"Missing module: {e}") + class GroqLLMService(OpenAILLMService): """A service for interacting with Groq's API using the OpenAI-compatible interface.