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

[WIP] AWS Nova Sonic service

Browse Source
This commit is contained in:
Paul Kompfner
2025-04-28 10:37:11 -04:00
parent 5b64613f65
commit 68c1069548
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/pipecat/services/aws_nova_sonic/aws.py
View File

@@ -412,7 +412,9 @@ class AWSNovaSonicService(LLMService):
) )
self._content_being_received = content self._content_being_received = content
# print(f"[pk] content start: {self._content_being_received}") # print(f"[pk] content start: {content}")
if content.role == Role.ASSISTANT:
print(f"[pk] assistant content start: {content}")
if content.role == Role.ASSISTANT: if content.role == Role.ASSISTANT:
if content.type == ContentType.AUDIO: if content.type == ContentType.AUDIO:
@@ -425,13 +427,15 @@ class AWSNovaSonicService(LLMService):
# This should never happen # This should never happen
if not self._content_being_received: if not self._content_being_received:
return return
content = self._content_being_received
text_content = event_json["textOutput"]["content"] text_content = event_json["textOutput"]["content"]
# print(f"[pk] text output. content: {text_content}") # print(f"[pk] text output. content: {text_content}")
if content.role == Role.ASSISTANT:
print(f"[pk] assistant text output. content: {text_content}")
# Bookkeeping: augment the current content being received with text # Bookkeeping: augment the current content being received with text
# Assumption: only one text content per content block # Assumption: only one text content per content block
content = self._content_being_received
content.text_content = text_content content.text_content = text_content
async def _handle_audio_output_event(self, event_json): async def _handle_audio_output_event(self, event_json):
@@ -457,13 +461,15 @@ class AWSNovaSonicService(LLMService):
# This should never happen # This should never happen
if not self._content_being_received: if not self._content_being_received:
return return
content = self._content_being_received
content_end = event_json["contentEnd"] content_end = event_json["contentEnd"]
stop_reason = content_end["stopReason"] stop_reason = content_end["stopReason"]
# print(f"[pk] content end: {self._content_being_received}.\n stop_reason: {stop_reason}") # print(f"[pk] content end: {content}.\n stop_reason: {stop_reason}")
if content.role == Role.ASSISTANT:
print(f"[pk] assistant content end: {content}.\n stop_reason: {stop_reason}")
# Bookkeeping: clear current content being received # Bookkeeping: clear current content being received
content = self._content_being_received
self._content_being_received = None self._content_being_received = None
if content.role == Role.ASSISTANT: if content.role == Role.ASSISTANT: