try/catch import of groq SDK
This commit is contained in:
committed by
Mark Backman
parent
e087f6ec5d
commit
f5d49fea81
@@ -7,7 +7,6 @@
|
|||||||
|
|
||||||
from typing import AsyncGenerator, Optional
|
from typing import AsyncGenerator, Optional
|
||||||
|
|
||||||
from groq import AsyncGroq
|
|
||||||
from loguru import logger
|
from loguru import logger
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
@@ -17,6 +16,15 @@ from pipecat.services.base_whisper import BaseWhisperSTTService, Transcription
|
|||||||
from pipecat.services.openai import OpenAILLMService
|
from pipecat.services.openai import OpenAILLMService
|
||||||
from pipecat.transcriptions.language import Language
|
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):
|
class GroqLLMService(OpenAILLMService):
|
||||||
"""A service for interacting with Groq's API using the OpenAI-compatible interface.
|
"""A service for interacting with Groq's API using the OpenAI-compatible interface.
|
||||||
|
|||||||
Reference in New Issue
Block a user