wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixing ruff format.

Browse Source
This commit is contained in:
filipi87
2026-05-21 10:05:33 -03:00
parent 6ef7f6446a
commit 144a1ece7b
2 changed files with 6 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/daily/test_tavus_transport.py
View File

@@ -1,7 +1,6 @@
import array import array
import asyncio import asyncio
import datetime import datetime
import io
import os import os
import signal import signal
import wave import wave
@@ -58,7 +57,6 @@ class DailyProxyApp(EventHandler):
self._buffer = bytearray() self._buffer = bytearray()
self._audio_task: asyncio.Task | None = None self._audio_task: asyncio.Task | None = None
self._wav_file: wave.Wave_write | 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: CallClient = CallClient(event_handler=self)
self._client.update_subscription_profiles( self._client.update_subscription_profiles(
@@ -79,8 +77,7 @@ class DailyProxyApp(EventHandler):
os.makedirs("recordings", exist_ok=True) os.makedirs("recordings", exist_ok=True)
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
path = f"recordings/received_pos_speed_{timestamp}.wav" path = f"recordings/received_pos_speed_{timestamp}.wav"
self._wav_io = open(path, "wb") self._wav_file = wave.open(path, "wb")
self._wav_file = wave.open(self._wav_io, "wb")
self._wav_file.setnchannels(1) self._wav_file.setnchannels(1)
self._wav_file.setsampwidth(2) self._wav_file.setsampwidth(2)
# Declare TRUE_SAMPLE_RATE so timestamps match bot_*.wav for comparison. # Declare TRUE_SAMPLE_RATE so timestamps match bot_*.wav for comparison.
@@ -93,9 +90,6 @@ class DailyProxyApp(EventHandler):
if self._wav_file: if self._wav_file:
self._wav_file.close() self._wav_file.close()
self._wav_file = None self._wav_file = None
if self._wav_io:
self._wav_io.close()
self._wav_io = None
def run(self, meeting_url: str): def run(self, meeting_url: str):
asyncio.set_event_loop(self._loop) asyncio.set_event_loop(self._loop)

17
src/pipecat/services/tavus/video.py
View File

@@ -12,7 +12,6 @@ avatar functionality through Tavus's streaming API.
import asyncio import asyncio
import datetime import datetime
import io
import os import os
import wave import wave
from dataclasses import dataclass from dataclasses import dataclass
@@ -105,10 +104,9 @@ class TavusVideoService(AIService):
self._audio_buffer = bytearray() self._audio_buffer = bytearray()
self._send_task: asyncio.Task | None = None self._send_task: asyncio.Task | None = None
# This is the custom track destination expected by Tavus # 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._transport_ready = False
self._wav_file: wave.Wave_write | None = None self._wav_file: wave.Wave_write | None = None
self._wav_io: io.FileIO | None = None
async def setup(self, setup: FrameProcessorSetup): async def setup(self, setup: FrameProcessorSetup):
"""Set up the Tavus video service. """Set up the Tavus video service.
@@ -214,8 +212,7 @@ class TavusVideoService(AIService):
os.makedirs("recordings", exist_ok=True) os.makedirs("recordings", exist_ok=True)
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
path = f"recordings/bot_pre_speed_{timestamp}.wav" path = f"recordings/bot_pre_speed_{timestamp}.wav"
self._wav_io = open(path, "wb") self._wav_file = wave.open(path, "wb")
self._wav_file = wave.open(self._wav_io, "wb")
self._wav_file.setnchannels(1) self._wav_file.setnchannels(1)
self._wav_file.setsampwidth(2) self._wav_file.setsampwidth(2)
self._wav_file.setframerate(sample_rate) self._wav_file.setframerate(sample_rate)
@@ -225,9 +222,6 @@ class TavusVideoService(AIService):
if self._wav_file: if self._wav_file:
self._wav_file.close() self._wav_file.close()
self._wav_file = None self._wav_file = None
if self._wav_io:
self._wav_io.close()
self._wav_io = None
async def start(self, frame: StartFrame): async def start(self, frame: StartFrame):
"""Start the Tavus video service. """Start the Tavus video service.
@@ -237,10 +231,9 @@ class TavusVideoService(AIService):
""" """
await super().start(frame) await super().start(frame)
await self._client.start(frame) await self._client.start(frame)
if self._transport_destination: await self._client.register_audio_destination(
await self._client.register_audio_destination( self._transport_destination, auto_silence=False
self._transport_destination, auto_silence=False )
)
self._open_wav(self._client.out_sample_rate) self._open_wav(self._client.out_sample_rate)
await self._create_send_task() await self._create_send_task()