mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-07 07:55:16 +02:00
chore: Fix lint formatting
This commit is contained in:
parent
02646e0a7a
commit
03d8f30ba2
14 changed files with 29 additions and 26 deletions
|
|
@ -1,10 +1,11 @@
|
|||
import os, sys
|
||||
import os
|
||||
import sys
|
||||
|
||||
import loguru
|
||||
from pipecat.utils.context import run_id_var, turn_var
|
||||
|
||||
from api.enums import Environment
|
||||
from api.utils.worker import get_worker_id, is_worker_process
|
||||
from pipecat.utils.context import run_id_var, turn_var
|
||||
|
||||
ENVIRONMENT = os.getenv("ENVIRONMENT", Environment.LOCAL.value)
|
||||
ENABLE_TURN_LOGGING = os.getenv("ENABLE_TURN_LOGGING", "false").lower() == "true"
|
||||
|
|
|
|||
|
|
@ -2,13 +2,13 @@ from typing import Dict
|
|||
|
||||
from fastapi import APIRouter, BackgroundTasks, Depends
|
||||
from loguru import logger
|
||||
from pipecat.transports.smallwebrtc.connection import SmallWebRTCConnection
|
||||
from pipecat.utils.context import set_current_run_id
|
||||
from pydantic import BaseModel
|
||||
|
||||
from api.db.models import UserModel
|
||||
from api.services.auth.depends import get_user
|
||||
from api.services.pipecat.run_pipeline import run_pipeline_smallwebrtc
|
||||
from pipecat.transports.smallwebrtc.connection import SmallWebRTCConnection
|
||||
from pipecat.utils.context import set_current_run_id
|
||||
|
||||
router = APIRouter(prefix="/pipecat")
|
||||
|
||||
|
|
|
|||
|
|
@ -5,12 +5,12 @@ from typing import Dict
|
|||
|
||||
from fastapi import APIRouter, Depends, WebSocket, WebSocketDisconnect
|
||||
from loguru import logger
|
||||
from pipecat.transports.smallwebrtc.connection import SmallWebRTCConnection
|
||||
from pipecat.utils.context import set_current_run_id
|
||||
|
||||
from api.db.models import UserModel
|
||||
from api.services.auth.depends import get_user_ws
|
||||
from api.services.pipecat.run_pipeline import run_pipeline_smallwebrtc
|
||||
from pipecat.transports.smallwebrtc.connection import SmallWebRTCConnection
|
||||
from pipecat.utils.context import set_current_run_id
|
||||
|
||||
router = APIRouter(prefix="/ws")
|
||||
|
||||
|
|
|
|||
|
|
@ -3,12 +3,6 @@
|
|||
from typing import Any, Dict
|
||||
|
||||
from loguru import logger
|
||||
from pipecat.pipeline.pipeline import Pipeline
|
||||
from pipecat.processors.filters.stt_mute_filter import (
|
||||
STTMuteConfig,
|
||||
STTMuteFilter,
|
||||
STTMuteStrategy,
|
||||
)
|
||||
|
||||
from api.db.db_client import DBClient
|
||||
from api.services.looptalk.audio_streamer import get_or_create_audio_streamer
|
||||
|
|
@ -29,6 +23,12 @@ from api.services.pipecat.service_factory import (
|
|||
from api.services.workflow.dto import ReactFlowDTO
|
||||
from api.services.workflow.pipecat_engine import PipecatEngine
|
||||
from api.services.workflow.workflow import WorkflowGraph
|
||||
from pipecat.pipeline.pipeline import Pipeline
|
||||
from pipecat.processors.filters.stt_mute_filter import (
|
||||
STTMuteConfig,
|
||||
STTMuteFilter,
|
||||
STTMuteStrategy,
|
||||
)
|
||||
|
||||
|
||||
class LoopTalkPipelineBuilder:
|
||||
|
|
|
|||
|
|
@ -7,6 +7,7 @@
|
|||
"""Internal frame serializer for agent-to-agent communication."""
|
||||
|
||||
from loguru import logger
|
||||
|
||||
from pipecat.frames.frames import (
|
||||
Frame,
|
||||
InputAudioRawFrame,
|
||||
|
|
|
|||
|
|
@ -11,6 +11,8 @@ import time
|
|||
from typing import Dict, Optional, Tuple
|
||||
|
||||
from loguru import logger
|
||||
|
||||
from api.services.looptalk.internal_serializer import InternalFrameSerializer
|
||||
from pipecat.frames.frames import (
|
||||
CancelFrame,
|
||||
EndFrame,
|
||||
|
|
@ -27,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 api.services.looptalk.internal_serializer import InternalFrameSerializer
|
||||
|
||||
|
||||
class InternalInputTransport(BaseInputTransport):
|
||||
"""Input side of internal transport for agent-to-agent communication."""
|
||||
|
|
|
|||
|
|
@ -6,8 +6,6 @@ from pathlib import Path
|
|||
from typing import Any, Dict, Optional
|
||||
|
||||
from loguru import logger
|
||||
from pipecat.pipeline.task import PipelineTask
|
||||
from pipecat.utils.context import set_current_run_id
|
||||
|
||||
from api.db.db_client import DBClient
|
||||
from api.services.looptalk.internal_transport import (
|
||||
|
|
@ -15,6 +13,8 @@ from api.services.looptalk.internal_transport import (
|
|||
InternalTransportManager,
|
||||
)
|
||||
from api.services.pipecat.transport_setup import create_internal_transport
|
||||
from pipecat.pipeline.task import PipelineTask
|
||||
from pipecat.utils.context import set_current_run_id
|
||||
|
||||
from .core.pipeline_builder import LoopTalkPipelineBuilder
|
||||
from .core.recording_manager import RecordingManager
|
||||
|
|
|
|||
|
|
@ -20,6 +20,7 @@ from typing import Any, Dict, Optional
|
|||
import numpy as np
|
||||
import websockets
|
||||
from loguru import logger
|
||||
|
||||
from pipecat.audio.turn.smart_turn.base_smart_turn import (
|
||||
BaseSmartTurn,
|
||||
SmartTurnTimeoutException,
|
||||
|
|
|
|||
|
|
@ -37,7 +37,6 @@ logging_queue_listener = setup_logging()
|
|||
import redis.asyncio as aioredis
|
||||
import redis.exceptions
|
||||
from loguru import logger
|
||||
from pipecat.utils.enums import EndTaskReason
|
||||
|
||||
from api.services.telephony.ari_client import Channel
|
||||
from api.services.telephony.ari_client_manager import (
|
||||
|
|
@ -45,6 +44,7 @@ from api.services.telephony.ari_client_manager import (
|
|||
setup_ari_client_supervisor,
|
||||
)
|
||||
from api.services.telephony.ari_manager_connection import ARIManagerConnection
|
||||
from pipecat.utils.enums import EndTaskReason
|
||||
|
||||
|
||||
class ARIManager:
|
||||
|
|
|
|||
|
|
@ -11,10 +11,10 @@ from typing import Optional
|
|||
|
||||
import httpx
|
||||
from loguru import logger
|
||||
from pipecat.utils.base_object import BaseObject
|
||||
|
||||
from api.services.telephony.ari_client import AsyncARIClient, Bridge, Channel
|
||||
from api.services.telephony.ari_client_singleton import ari_client_singleton
|
||||
from pipecat.utils.base_object import BaseObject
|
||||
|
||||
|
||||
class ARIManagerConnection(BaseObject):
|
||||
|
|
|
|||
|
|
@ -8,7 +8,6 @@ from typing import Optional, Tuple
|
|||
|
||||
import redis.asyncio as aioredis
|
||||
from loguru import logger
|
||||
from pipecat.utils.base_object import BaseObject
|
||||
|
||||
from api.services.telephony.stasis_event_protocol import (
|
||||
DisconnectCommand,
|
||||
|
|
@ -16,6 +15,7 @@ from api.services.telephony.stasis_event_protocol import (
|
|||
SocketClosedCommand,
|
||||
TransferCommand,
|
||||
)
|
||||
from pipecat.utils.base_object import BaseObject
|
||||
|
||||
|
||||
class StasisRTPConnection(BaseObject):
|
||||
|
|
|
|||
|
|
@ -7,6 +7,10 @@ import time
|
|||
from typing import Awaitable, Callable, Optional
|
||||
|
||||
from loguru import logger
|
||||
from pydantic import BaseModel
|
||||
|
||||
from api.services.telephony.stasis_rtp_client import StasisRTPClient
|
||||
from api.services.telephony.stasis_rtp_connection import StasisRTPConnection
|
||||
from pipecat.frames.frames import (
|
||||
CancelFrame,
|
||||
EndFrame,
|
||||
|
|
@ -23,10 +27,6 @@ from pipecat.transports.base_output import (
|
|||
TransportClientNotConnectedException,
|
||||
)
|
||||
from pipecat.transports.base_transport import BaseTransport, TransportParams
|
||||
from pydantic import BaseModel
|
||||
|
||||
from api.services.telephony.stasis_rtp_client import StasisRTPClient
|
||||
from api.services.telephony.stasis_rtp_connection import StasisRTPConnection
|
||||
|
||||
|
||||
class StasisRTPTransportParams(TransportParams):
|
||||
|
|
|
|||
|
|
@ -12,7 +12,6 @@ from typing import Awaitable, Callable, Optional
|
|||
|
||||
import redis.asyncio as aioredis
|
||||
from loguru import logger
|
||||
from pipecat.utils.context import set_current_run_id
|
||||
|
||||
from api.routes.stasis_rtp import on_stasis_call
|
||||
from api.services.telephony.stasis_event_protocol import (
|
||||
|
|
@ -24,6 +23,7 @@ from api.services.telephony.stasis_event_protocol import (
|
|||
parse_event,
|
||||
)
|
||||
from api.services.telephony.stasis_rtp_connection import StasisRTPConnection
|
||||
from pipecat.utils.context import set_current_run_id
|
||||
|
||||
|
||||
class WorkerEventSubscriber:
|
||||
|
|
|
|||
|
|
@ -12,13 +12,13 @@ from langfuse import get_client
|
|||
from loguru import logger
|
||||
from openai import AsyncOpenAI
|
||||
from opentelemetry import context as otel_context
|
||||
from pipecat.utils.enums import EndTaskReason
|
||||
from pipecat.utils.tracing.context_registry import get_current_turn_context
|
||||
|
||||
from api.db import db_client
|
||||
from api.services.pipecat.tracing_config import is_tracing_enabled
|
||||
from api.tasks.arq import enqueue_job
|
||||
from api.tasks.function_names import FunctionNames
|
||||
from pipecat.utils.enums import EndTaskReason
|
||||
from pipecat.utils.tracing.context_registry import get_current_turn_context
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from api.services.workflow.pipecat_engine import PipecatEngine
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue