chore: Review comments

pipecat-ai · Jun 3, 2024 · d462c03 · d462c03
1 parent 7085b1e
commit d462c03
Showing 1 changed file with 30 additions and 16 deletions.
diff --git a/src/pipecat/services/openai.py b/src/pipecat/services/openai.py
@@ -13,24 +13,39 @@
 from loguru import logger
 from PIL import Image
 
-from pipecat.frames.frames import (AudioRawFrame, ErrorFrame, Frame,
-                                   LLMFullResponseEndFrame,
-                                   LLMFullResponseStartFrame, LLMMessagesFrame,
-                                   LLMResponseEndFrame, LLMResponseStartFrame,
-                                   TextFrame, URLImageRawFrame,
-                                   VisionImageRawFrame)
+from pipecat.frames.frames import (
+    AudioRawFrame,
+    ErrorFrame,
+    Frame,
+    LLMFullResponseEndFrame,
+    LLMFullResponseStartFrame,
+    LLMMessagesFrame,
+    LLMResponseEndFrame,
+    LLMResponseStartFrame,
+    TextFrame,
+    URLImageRawFrame,
+    VisionImageRawFrame
+)
 from pipecat.processors.aggregators.openai_llm_context import (
-    OpenAILLMContext, OpenAILLMContextFrame)
+    OpenAILLMContext,
+    OpenAILLMContextFrame
+)
 from pipecat.processors.frame_processor import FrameDirection
-from pipecat.services.ai_services import (ImageGenService, LLMService,
-                                          TTSService)
+from pipecat.services.ai_services import (
+    ImageGenService,
+    LLMService,
+    TTSService
+)
 
 try:
     from openai import AsyncOpenAI, AsyncStream, BadRequestError
-    from openai.types.chat import (ChatCompletion, ChatCompletionChunk,
-                                   ChatCompletionFunctionMessageParam,
-                                   ChatCompletionMessageParam,
-                                   ChatCompletionToolParam)
+    from openai.types.chat import (
+        ChatCompletion,
+        ChatCompletionChunk,
+        ChatCompletionFunctionMessageParam,
+        ChatCompletionMessageParam,
+        ChatCompletionToolParam
+    )
 except ModuleNotFoundError as e:
     logger.error(f"Exception: {e}")
     logger.error(
@@ -265,7 +280,7 @@ async def run_image_gen(self, prompt: str) -> AsyncGenerator[Frame, None]:
 
 class OpenAITTSService(TTSService):
     """This service uses the OpenAI TTS API to generate audio from text.
-    The returned audio is PCM encoded at 24kHz. When using the DailyTransport, set the sample rate in the DailyParams:
+    The returned audio is PCM encoded at 24kHz. When using the DailyTransport, set the sample rate in the DailyParams accordingly:
     ```
     DailyParams(
         audio_out_enabled=True,
@@ -308,5 +323,4 @@ async def run_tts(self, text: str) -> AsyncGenerator[Frame, None]:
                         frame = AudioRawFrame(chunk, 24_000, 1)
                         yield frame
         except BadRequestError as e:
-            logger.exception(f"Error generating TTS: {e}")
-            raise
+            logger.error(f"Error generating TTS: {e}")