try/catch import of groq SDK

This commit is contained in:
Kwindla Hultman Kramer
2025-03-26 09:13:03 -07:00
committed by Mark Backman
parent e087f6ec5d
commit f5d49fea81

View File

@@ -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.