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

add missing queue task_done()

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-05-15 23:35:15 -07:00
parent 7384b63b1d
commit dc9377fb92
3 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/pipecat/frames/frames.py
View File

@@ -119,7 +119,7 @@ class TextFrame(DataFrame):
text: str text: str
def __str__(self): def __str__(self):
return f'{self.name}: "{self.text}"' return f"{self.name}(text: {self.text})"
@dataclass @dataclass
@@ -132,7 +132,7 @@ class TranscriptionFrame(TextFrame):
timestamp: str timestamp: str
def __str__(self): def __str__(self):
return f"{self.name}(user: {self.user_id}, timestamp: {self.timestamp})" return f"{self.name}(user: {self.user_id}, text: {self.text}, timestamp: {self.timestamp})"
@dataclass @dataclass
@@ -143,7 +143,7 @@ class InterimTranscriptionFrame(TextFrame):
timestamp: str timestamp: str
def __str__(self): def __str__(self):
return f"{self.name}(user: {self.user_id}, timestamp: {self.timestamp})" return f"{self.name}(user: {self.user_id}, text: {self.text}, timestamp: {self.timestamp})"
@dataclass @dataclass

2
src/pipecat/transports/base_input.py
View File

@@ -185,6 +185,8 @@ class BaseInputTransport(FrameProcessor):
future = asyncio.run_coroutine_threadsafe( future = asyncio.run_coroutine_threadsafe(
self._internal_push_frame(frame), self.get_event_loop()) self._internal_push_frame(frame), self.get_event_loop())
future.result() future.result()
self._audio_in_queue.task_done()
except queue.Empty: except queue.Empty:
pass pass
except BaseException as e: except BaseException as e:

3
src/pipecat/transports/base_output.py
View File

@@ -169,6 +169,8 @@ class BaseOutputTransport(FrameProcessor):
# Send any remaining audio # Send any remaining audio
self._send_audio_truncated(buffer, bytes_size_10ms) self._send_audio_truncated(buffer, bytes_size_10ms)
buffer = bytearray() buffer = bytearray()
self._sink_queue.task_done()
except queue.Empty: except queue.Empty:
pass pass
except BaseException as e: except BaseException as e:
@@ -208,6 +210,7 @@ class BaseOutputTransport(FrameProcessor):
if self._params.camera_out_is_live: if self._params.camera_out_is_live:
image = self._camera_out_queue.get(timeout=1) image = self._camera_out_queue.get(timeout=1)
self._draw_image(image) self._draw_image(image)
self._camera_out_queue.task_done()
elif self._camera_images: elif self._camera_images:
image = next(self._camera_images) image = next(self._camera_images)
self._draw_image(image) self._draw_image(image)