From 144a1ece7bd7d1cbb1d61fc4a96ff0ead70cfca9 Mon Sep 17 00:00:00 2001 From: filipi87 Date: Thu, 21 May 2026 10:05:33 -0300 Subject: [PATCH] Fixing ruff format. --- scripts/daily/test_tavus_transport.py | 8 +------- src/pipecat/services/tavus/video.py | 17 +++++------------ 2 files changed, 6 insertions(+), 19 deletions(-) diff --git a/scripts/daily/test_tavus_transport.py b/scripts/daily/test_tavus_transport.py index a283e043a..6e1af22c5 100644 --- a/scripts/daily/test_tavus_transport.py +++ b/scripts/daily/test_tavus_transport.py @@ -1,7 +1,6 @@ import array import asyncio import datetime -import io import os import signal import wave @@ -58,7 +57,6 @@ class DailyProxyApp(EventHandler): self._buffer = bytearray() self._audio_task: asyncio.Task | None = None self._wav_file: wave.Wave_write | None = None - self._wav_io: io.FileIO | None = None self._client: CallClient = CallClient(event_handler=self) self._client.update_subscription_profiles( @@ -79,8 +77,7 @@ class DailyProxyApp(EventHandler): os.makedirs("recordings", exist_ok=True) timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") path = f"recordings/received_pos_speed_{timestamp}.wav" - self._wav_io = open(path, "wb") - self._wav_file = wave.open(self._wav_io, "wb") + self._wav_file = wave.open(path, "wb") self._wav_file.setnchannels(1) self._wav_file.setsampwidth(2) # Declare TRUE_SAMPLE_RATE so timestamps match bot_*.wav for comparison. @@ -93,9 +90,6 @@ class DailyProxyApp(EventHandler): if self._wav_file: self._wav_file.close() self._wav_file = None - if self._wav_io: - self._wav_io.close() - self._wav_io = None def run(self, meeting_url: str): asyncio.set_event_loop(self._loop) diff --git a/src/pipecat/services/tavus/video.py b/src/pipecat/services/tavus/video.py index 23d741797..0f36bd9e4 100644 --- a/src/pipecat/services/tavus/video.py +++ b/src/pipecat/services/tavus/video.py @@ -12,7 +12,6 @@ avatar functionality through Tavus's streaming API. import asyncio import datetime -import io import os import wave from dataclasses import dataclass @@ -105,10 +104,9 @@ class TavusVideoService(AIService): self._audio_buffer = bytearray() self._send_task: asyncio.Task | None = None # This is the custom track destination expected by Tavus - self._transport_destination: str | None = "stream" + self._transport_destination: str = "stream" self._transport_ready = False self._wav_file: wave.Wave_write | None = None - self._wav_io: io.FileIO | None = None async def setup(self, setup: FrameProcessorSetup): """Set up the Tavus video service. @@ -214,8 +212,7 @@ class TavusVideoService(AIService): os.makedirs("recordings", exist_ok=True) timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") path = f"recordings/bot_pre_speed_{timestamp}.wav" - self._wav_io = open(path, "wb") - self._wav_file = wave.open(self._wav_io, "wb") + self._wav_file = wave.open(path, "wb") self._wav_file.setnchannels(1) self._wav_file.setsampwidth(2) self._wav_file.setframerate(sample_rate) @@ -225,9 +222,6 @@ class TavusVideoService(AIService): if self._wav_file: self._wav_file.close() self._wav_file = None - if self._wav_io: - self._wav_io.close() - self._wav_io = None async def start(self, frame: StartFrame): """Start the Tavus video service. @@ -237,10 +231,9 @@ class TavusVideoService(AIService): """ await super().start(frame) await self._client.start(frame) - if self._transport_destination: - await self._client.register_audio_destination( - self._transport_destination, auto_silence=False - ) + await self._client.register_audio_destination( + self._transport_destination, auto_silence=False + ) self._open_wav(self._client.out_sample_rate) await self._create_send_task()