fix ruff linter import organization

src/pipecat/audio/filters/krisp_filter.py

@@ -4,11 +4,12 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
-import numpy as np
 import os
 
-from pipecat.audio.filters.base_audio_filter import BaseAudioFilter
+import numpy as np
 from loguru import logger
+
+from pipecat.audio.filters.base_audio_filter import BaseAudioFilter
 from pipecat.frames.frames import FilterControlFrame, FilterEnableFrame
 
 try:

src/pipecat/audio/filters/noisereduce_filter.py

@@ -5,11 +5,9 @@
 #
 
 import numpy as np
-
-from pipecat.audio.filters.base_audio_filter import BaseAudioFilter
-
 from loguru import logger
 
+from pipecat.audio.filters.base_audio_filter import BaseAudioFilter
 from pipecat.frames.frames import FilterControlFrame, FilterEnableFrame
 
 try:

src/pipecat/audio/utils.py

@@ -5,6 +5,7 @@
 #
 
 import audioop
+
 import numpy as np
 import pyloudnorm as pyln
 import resampy

src/pipecat/audio/vad/silero.py

@@ -7,11 +7,10 @@
 import time
 
 import numpy as np
+from loguru import logger
 
 from pipecat.audio.vad.vad_analyzer import VADAnalyzer, VADParams
 
-from loguru import logger
-
 # How often should we reset internal model state
 _MODEL_RESET_STATES_TIME = 5.0
 

src/pipecat/metrics/metrics.py

@@ -1,4 +1,5 @@
 from typing import Optional
+
 from pydantic import BaseModel
 
 

src/pipecat/pipeline/base_pipeline.py

@@ -5,7 +5,6 @@
 #
 
 from abc import abstractmethod
-
 from typing import List
 
 from pipecat.processors.frame_processor import FrameProcessor

src/pipecat/pipeline/parallel_pipeline.py

@@ -5,16 +5,15 @@
 #
 
 import asyncio
-
 from itertools import chain
 from typing import Awaitable, Callable, List
 
+from loguru import logger
+
+from pipecat.frames.frames import CancelFrame, EndFrame, Frame, StartFrame, SystemFrame
 from pipecat.pipeline.base_pipeline import BasePipeline
 from pipecat.pipeline.pipeline import Pipeline
 from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
-from pipecat.frames.frames import CancelFrame, EndFrame, Frame, StartFrame, SystemFrame
-
-from loguru import logger
 
 
 class Source(FrameProcessor):

src/pipecat/pipeline/runner.py

@@ -7,11 +7,11 @@
 import asyncio
 import signal
 
+from loguru import logger
+
 from pipecat.pipeline.task import PipelineTask
 from pipecat.utils.utils import obj_count, obj_id
 
-from loguru import logger
-
 
 class PipelineRunner:
     def __init__(self, *, name: str | None = None, handle_sigint: bool = True):

src/pipecat/pipeline/sync_parallel_pipeline.py

@@ -5,18 +5,17 @@
 #
 
 import asyncio
-
 from dataclasses import dataclass
 from itertools import chain
 from typing import List
 
+from loguru import logger
+
 from pipecat.frames.frames import ControlFrame, EndFrame, Frame, SystemFrame
 from pipecat.pipeline.base_pipeline import BasePipeline
 from pipecat.pipeline.pipeline import Pipeline
 from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 
-from loguru import logger
-
 
 @dataclass
 class SyncFrame(ControlFrame):

src/pipecat/pipeline/task.py

@@ -5,9 +5,9 @@
 #
 
 import asyncio
-
 from typing import AsyncIterable, Iterable
 
+from loguru import logger
 from pydantic import BaseModel
 
 from pipecat.clocks.base_clock import BaseClock
@@ -23,13 +23,11 @@ from pipecat.frames.frames import (
     StartFrame,
     StopTaskFrame,
 )
-from pipecat.metrics.metrics import TTFBMetricsData, ProcessingMetricsData
+from pipecat.metrics.metrics import ProcessingMetricsData, TTFBMetricsData
 from pipecat.pipeline.base_pipeline import BasePipeline
 from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 from pipecat.utils.utils import obj_count, obj_id
 
-from loguru import logger
-
 
 class PipelineParams(BaseModel):
     allow_interruptions: bool = False

src/pipecat/pipeline/to_be_updated/merge_pipeline.py

@@ -1,4 +1,5 @@
 from typing import List
+
 from pipecat.frames.frames import EndFrame, EndPipeFrame
 from pipecat.pipeline.pipeline import Pipeline
 

src/pipecat/processors/aggregators/gated.py

@@ -6,11 +6,11 @@
 
 from typing import List, Tuple
 
+from loguru import logger
+
 from pipecat.frames.frames import Frame, SystemFrame
 from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 
-from loguru import logger
-
 
 class GatedAggregator(FrameProcessor):
     """Accumulate frames, with custom functions to start and stop accumulation.

src/pipecat/processors/aggregators/user_response.py

@@ -4,7 +4,6 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
-from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 from pipecat.frames.frames import (
     Frame,
     InterimTranscriptionFrame,
@@ -14,6 +13,7 @@ from pipecat.frames.frames import (
     UserStartedSpeakingFrame,
     UserStoppedSpeakingFrame,
 )
+from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 
 
 class ResponseAggregator(FrameProcessor):

src/pipecat/processors/async_generator.py

@@ -5,7 +5,6 @@
 #
 
 import asyncio
-
 from typing import Any, AsyncGenerator
 
 from pipecat.frames.frames import (
@@ -13,7 +12,7 @@ from pipecat.frames.frames import (
     EndFrame,
     Frame,
 )
-from pipecat.processors.frame_processor import FrameProcessor, FrameDirection
+from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 from pipecat.serializers.base_serializer import FrameSerializer
 
 

src/pipecat/processors/audio/vad/silero.py

@@ -4,6 +4,8 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
+from loguru import logger
+
 from pipecat.audio.vad.silero import SileroVADAnalyzer
 from pipecat.audio.vad.vad_analyzer import VADParams, VADState
 from pipecat.frames.frames import (
@@ -16,8 +18,6 @@ from pipecat.frames.frames import (
 )
 from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 
-from loguru import logger
-
 
 class SileroVAD(FrameProcessor):
     def __init__(

src/pipecat/processors/filters/wake_check_filter.py

@@ -6,14 +6,13 @@
 
 import re
 import time
-
 from enum import Enum
 
+from loguru import logger
+
 from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame
 from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 
-from loguru import logger
-
 
 class WakeCheckFilter(FrameProcessor):
     """

src/pipecat/processors/frameworks/langchain.py

@@ -6,6 +6,8 @@
 
 from typing import Union
 
+from loguru import logger
+
 from pipecat.frames.frames import (
     Frame,
     LLMFullResponseEndFrame,
@@ -15,8 +17,6 @@ from pipecat.frames.frames import (
 )
 from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 
-from loguru import logger
-
 try:
     from langchain_core.messages import AIMessageChunk
     from langchain_core.runnables import Runnable

src/pipecat/processors/gstreamer/pipeline_source.py

@@ -6,6 +6,7 @@
 
 import asyncio
 
+from loguru import logger
 from pydantic import BaseModel
 
 from pipecat.frames.frames import (
@@ -19,8 +20,6 @@ from pipecat.frames.frames import (
 )
 from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 
-from loguru import logger
-
 try:
     import gi
 

src/pipecat/processors/idle_frame_processor.py

@@ -5,7 +5,6 @@
 #
 
 import asyncio
-
 from typing import Awaitable, Callable, List
 
 from pipecat.frames.frames import Frame

src/pipecat/processors/logger.py

@@ -4,11 +4,13 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
-from pipecat.frames.frames import BotSpeakingFrame, Frame, AudioRawFrame, TransportMessageFrame
-from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
-from loguru import logger
 from typing import Optional
 
+from loguru import logger
+
+from pipecat.frames.frames import AudioRawFrame, BotSpeakingFrame, Frame, TransportMessageFrame
+from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
+
 logger = logger.opt(ansi=True)
 
 

src/pipecat/processors/metrics/frame_processor_metrics.py

@@ -6,6 +6,8 @@
 
 import time
 
+from loguru import logger
+
 from pipecat.frames.frames import MetricsFrame
 from pipecat.metrics.metrics import (
     LLMTokenUsage,
@@ -16,8 +18,6 @@ from pipecat.metrics.metrics import (
     TTSUsageMetricsData,
 )
 
-from loguru import logger
-
 
 class FrameProcessorMetrics:
     def __init__(self):

src/pipecat/processors/metrics/sentry.py

@@ -5,6 +5,7 @@
 #
 
 import time
+
 from loguru import logger
 
 try:

src/pipecat/serializers/livekit.py

@@ -7,11 +7,11 @@
 import ctypes
 import pickle
 
+from loguru import logger
+
 from pipecat.frames.frames import Frame, InputAudioRawFrame, OutputAudioRawFrame
 from pipecat.serializers.base_serializer import FrameSerializer, FrameSerializerType
 
-from loguru import logger
-
 try:
     from livekit.rtc import AudioFrame
 except ModuleNotFoundError as e:

src/pipecat/serializers/protobuf.py

@@ -6,8 +6,9 @@
 
 import dataclasses
 
-import pipecat.frames.protobufs.frames_pb2 as frame_protos
+from loguru import logger
 
+import pipecat.frames.protobufs.frames_pb2 as frame_protos
 from pipecat.frames.frames import (
     Frame,
     InputAudioRawFrame,
@@ -17,8 +18,6 @@ from pipecat.frames.frames import (
 )
 from pipecat.serializers.base_serializer import FrameSerializer, FrameSerializerType
 
-from loguru import logger
-
 
 class ProtobufFrameSerializer(FrameSerializer):
     SERIALIZABLE_TYPES = {

src/pipecat/serializers/twilio.py

@@ -9,7 +9,7 @@ import json
 
 from pydantic import BaseModel
 
-from pipecat.audio.utils import ulaw_to_pcm, pcm_to_ulaw
+from pipecat.audio.utils import pcm_to_ulaw, ulaw_to_pcm
 from pipecat.frames.frames import AudioRawFrame, Frame, InputAudioRawFrame, StartInterruptionFrame
 from pipecat.serializers.base_serializer import FrameSerializer, FrameSerializerType
 

src/pipecat/services/canonical.py

@@ -4,23 +4,22 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
-import aiohttp
 import io
 import os
 import uuid
 import wave
-
 from datetime import datetime
 from typing import Dict, List, Tuple
 
+import aiohttp
+from loguru import logger
+
 from pipecat.frames.frames import CancelFrame, EndFrame, Frame
 from pipecat.processors.audio import audio_buffer_processor
 from pipecat.processors.audio.audio_buffer_processor import AudioBufferProcessor
 from pipecat.processors.frame_processor import FrameDirection
 from pipecat.services.ai_services import AIService
 
-from loguru import logger
-
 try:
     import aiofiles
     import aiofiles.os

src/pipecat/services/cartesia.py

@@ -14,7 +14,6 @@ from loguru import logger
 from pydantic import BaseModel
 from tenacity import AsyncRetrying, RetryCallState, stop_after_attempt, wait_exponential
 
-
 from pipecat.frames.frames import (
     BotStoppedSpeakingFrame,
     CancelFrame,

src/pipecat/services/fal.py

@@ -4,20 +4,18 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
-import aiohttp
 import io
 import os
+from typing import AsyncGenerator, Dict, Optional, Union
 
+import aiohttp
+from loguru import logger
+from PIL import Image
 from pydantic import BaseModel
-from typing import AsyncGenerator, Optional, Union, Dict
 
 from pipecat.frames.frames import ErrorFrame, Frame, URLImageRawFrame
 from pipecat.services.ai_services import ImageGenService
 
-from PIL import Image
-
-from loguru import logger
-
 try:
     import fal_client
 except ModuleNotFoundError as e:

src/pipecat/services/gemini_multimodal_live/events.py

@@ -6,13 +6,12 @@
 #
 
 import base64
-import json
 import io
-
-from pydantic import BaseModel, Field
+import json
 from typing import List, Literal, Optional
 
 from PIL import Image
+from pydantic import BaseModel, Field
 
 from pipecat.frames.frames import ImageRawFrame
 

src/pipecat/services/moondream.py

@@ -5,19 +5,16 @@
 #
 
 import asyncio
-
-from PIL import Image
-
 from typing import AsyncGenerator
 
+from loguru import logger
+from PIL import Image
+
 from pipecat.frames.frames import ErrorFrame, Frame, TextFrame, VisionImageRawFrame
 from pipecat.services.ai_services import VisionService
 
-from loguru import logger
-
 try:
     import torch
-
     from transformers import AutoModelForCausalLM, AutoTokenizer
 except ModuleNotFoundError as e:
     logger.error(f"Exception: {e}")

src/pipecat/services/openai_realtime_beta/context.py

@@ -21,7 +21,7 @@ from pipecat.services.openai import (
 )
 
 from . import events
-from .frames import RealtimeMessagesUpdateFrame, RealtimeFunctionCallResultFrame
+from .frames import RealtimeFunctionCallResultFrame, RealtimeMessagesUpdateFrame
 
 
 class OpenAIRealtimeLLMContext(OpenAILLMContext):

src/pipecat/services/openai_realtime_beta/openai.py

@@ -8,10 +8,10 @@ import asyncio
 import base64
 import json
 import time
-
 from dataclasses import dataclass
 
 import websockets
+from loguru import logger
 
 from pipecat.frames.frames import (
     BotStoppedSpeakingFrame,
@@ -48,13 +48,11 @@ from pipecat.utils.time import time_now_iso8601
 
 from . import events
 from .context import (
+    OpenAIRealtimeAssistantContextAggregator,
     OpenAIRealtimeLLMContext,
     OpenAIRealtimeUserContextAggregator,
-    OpenAIRealtimeAssistantContextAggregator,
 )
-from .frames import RealtimeMessagesUpdateFrame, RealtimeFunctionCallResultFrame
-
-from loguru import logger
+from .frames import RealtimeFunctionCallResultFrame, RealtimeMessagesUpdateFrame
 
 
 @dataclass

src/pipecat/services/openpipe.py

@@ -6,14 +6,15 @@
 
 from typing import Dict, List
 
+from loguru import logger
+
 from pipecat.processors.aggregators.openai_llm_context import OpenAILLMContext
 from pipecat.services.openai import OpenAILLMService
 
-from loguru import logger
-
 try:
-    from openpipe import AsyncOpenAI as OpenPipeAI, AsyncStream
-    from openai.types.chat import ChatCompletionMessageParam, ChatCompletionChunk
+    from openai.types.chat import ChatCompletionChunk, ChatCompletionMessageParam
+    from openpipe import AsyncOpenAI as OpenPipeAI
+    from openpipe import AsyncStream
 except ModuleNotFoundError as e:
     logger.error(f"Exception: {e}")
     logger.error(

src/pipecat/services/simli.py

@@ -6,24 +6,22 @@
 
 import asyncio
 
+import numpy as np
+from loguru import logger
+
 from pipecat.frames.frames import (
+    CancelFrame,
+    EndFrame,
     Frame,
     OutputImageRawFrame,
-    TTSAudioRawFrame,
     StartInterruptionFrame,
-    EndFrame,
-    CancelFrame,
+    TTSAudioRawFrame,
 )
 from pipecat.processors.frame_processor import FrameDirection, FrameProcessor, StartFrame
 
-import numpy as np
-
-from loguru import logger
-
 try:
     from av.audio.frame import AudioFrame
     from av.audio.resampler import AudioResampler
-
     from simli import SimliClient, SimliConfig
 except ModuleNotFoundError as e:
     logger.error(f"Exception: {e}")

src/pipecat/services/tavus.py

@@ -7,24 +7,24 @@
 
 """This module implements Tavus as a sink transport layer"""
 
-import aiohttp
 import base64
 
+import aiohttp
+from loguru import logger
+
+from pipecat.audio.utils import resample_audio
 from pipecat.frames.frames import (
+    CancelFrame,
+    EndFrame,
     Frame,
-    TTSAudioRawFrame,
+    StartInterruptionFrame,
     TransportMessageUrgentFrame,
+    TTSAudioRawFrame,
     TTSStartedFrame,
     TTSStoppedFrame,
-    StartInterruptionFrame,
-    EndFrame,
-    CancelFrame,
 )
 from pipecat.processors.frame_processor import FrameDirection
 from pipecat.services.ai_services import AIService
-from pipecat.audio.utils import resample_audio
-
-from loguru import logger
 
 
 class TavusVideoService(AIService):

src/pipecat/services/to_be_updated/cloudflare_ai_service.py

@@ -1,5 +1,6 @@
-import requests
 import os
+
+import requests
 from services.ai_service import AIService
 
 # Note that Cloudflare's AI workers are still in beta.

src/pipecat/services/to_be_updated/google_ai_service.py

@@ -1,11 +1,12 @@
-from services.ai_service import AIService
-import openai
 import os
 
+import openai
+
 # To use Google Cloud's AI products, you'll need to install Google Cloud
 # CLI and enable the TTS and in your project:
 # https://cloud.google.com/sdk/docs/install
 from google.cloud import texttospeech
+from services.ai_service import AIService
 
 
 class GoogleAIService(AIService):

src/pipecat/services/to_be_updated/mock_ai_service.py

@@ -1,6 +1,7 @@
 import io
-import requests
 import time
+
+import requests
 from PIL import Image
 from services.ai_service import AIService
 

src/pipecat/services/whisper.py

@@ -7,18 +7,16 @@
 """This module implements Whisper transcription with a locally-downloaded model."""
 
 import asyncio
-
 from enum import Enum
 from typing import AsyncGenerator
 
 import numpy as np
+from loguru import logger
 
 from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame
 from pipecat.services.ai_services import SegmentedSTTService
 from pipecat.utils.time import time_now_iso8601
 
-from loguru import logger
-
 try:
     from faster_whisper import WhisperModel
 except ModuleNotFoundError as e:

src/pipecat/transports/base_output.py

@@ -29,9 +29,9 @@ from pipecat.frames.frames import (
     StartInterruptionFrame,
     StopInterruptionFrame,
     SystemFrame,
-    TTSAudioRawFrame,
     TransportMessageFrame,
     TransportMessageUrgentFrame,
+    TTSAudioRawFrame,
 )
 from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
 from pipecat.transports.base_transport import TransportParams

src/pipecat/transports/base_transport.py

@@ -6,20 +6,17 @@
 
 import asyncio
 import inspect
-
 from abc import ABC, abstractmethod
 from typing import Optional
 
-from pydantic import ConfigDict
-from pydantic import BaseModel
+from loguru import logger
+from pydantic import BaseModel, ConfigDict
 
 from pipecat.audio.filters.base_audio_filter import BaseAudioFilter
 from pipecat.audio.mixers.base_audio_mixer import BaseAudioMixer
 from pipecat.audio.vad.vad_analyzer import VADAnalyzer
 from pipecat.processors.frame_processor import FrameProcessor
 
-from loguru import logger
-
 
 class TransportParams(BaseModel):
     model_config = ConfigDict(arbitrary_types_allowed=True)

src/pipecat/transports/local/audio.py

@@ -5,17 +5,16 @@
 #
 
 import asyncio
-
 from concurrent.futures import ThreadPoolExecutor
 
+from loguru import logger
+
 from pipecat.frames.frames import InputAudioRawFrame, StartFrame
 from pipecat.processors.frame_processor import FrameProcessor
 from pipecat.transports.base_input import BaseInputTransport
 from pipecat.transports.base_output import BaseOutputTransport
 from pipecat.transports.base_transport import BaseTransport, TransportParams
 
-from loguru import logger
-
 try:
     import pyaudio
 except ModuleNotFoundError as e:

src/pipecat/transports/local/tk.py

@@ -5,19 +5,17 @@
 #
 
 import asyncio
-
+import tkinter as tk
 from concurrent.futures import ThreadPoolExecutor
 
 import numpy as np
-import tkinter as tk
+from loguru import logger
 
 from pipecat.frames.frames import InputAudioRawFrame, OutputImageRawFrame, StartFrame
 from pipecat.transports.base_input import BaseInputTransport
 from pipecat.transports.base_output import BaseOutputTransport
 from pipecat.transports.base_transport import BaseTransport, TransportParams
 
-from loguru import logger
-
 try:
     import pyaudio
 except ModuleNotFoundError as e:

src/pipecat/transports/network/fastapi_websocket.py

@@ -10,8 +10,9 @@ import io
 import time
 import typing
 import wave
-
 from typing import Awaitable, Callable
+
+from loguru import logger
 from pydantic import BaseModel
 
 from pipecat.frames.frames import (
@@ -27,8 +28,6 @@ from pipecat.transports.base_input import BaseInputTransport
 from pipecat.transports.base_output import BaseOutputTransport
 from pipecat.transports.base_transport import BaseTransport, TransportParams
 
-from loguru import logger
-
 try:
     from fastapi import WebSocket
     from starlette.websockets import WebSocketState

src/pipecat/transports/network/websocket_server.py

@@ -8,8 +8,9 @@ import asyncio
 import io
 import time
 import wave
-
 from typing import Awaitable, Callable
+
+from loguru import logger
 from pydantic import BaseModel
 
 from pipecat.frames.frames import (
@@ -28,8 +29,6 @@ from pipecat.transports.base_input import BaseInputTransport
 from pipecat.transports.base_output import BaseOutputTransport
 from pipecat.transports.base_transport import BaseTransport, TransportParams
 
-from loguru import logger
-
 try:
     import websockets
 except ModuleNotFoundError as e:

src/pipecat/transports/services/livekit.py

@@ -8,6 +8,7 @@ import asyncio
 from dataclasses import dataclass
 from typing import Any, Awaitable, Callable, List
 
+from loguru import logger
 from pydantic import BaseModel
 
 from pipecat.audio.utils import resample_audio
@@ -28,8 +29,6 @@ from pipecat.transports.base_input import BaseInputTransport
 from pipecat.transports.base_output import BaseOutputTransport
 from pipecat.transports.base_transport import BaseTransport, TransportParams
 
-from loguru import logger
-
 try:
     from livekit import rtc
     from tenacity import retry, stop_after_attempt, wait_exponential

src/pipecat/utils/test_frame_processor.py

@@ -1,4 +1,5 @@
 from typing import List
+
 from pipecat.processors.frame_processor import FrameProcessor