Compare commits
25 Commits
cb/frame-p
...
khk/debugg
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e7ccaed56c | ||
|
|
07124bfafc | ||
|
|
df2ddb4b91 | ||
|
|
0db5c86494 | ||
|
|
d0cdb496e4 | ||
|
|
b640b2d024 | ||
|
|
bd0649e3ed | ||
|
|
711d9a1021 | ||
|
|
e856566c30 | ||
|
|
996c337dd1 | ||
|
|
856a0e321b | ||
|
|
425ad3e90d | ||
|
|
d13137c99f | ||
|
|
687fd97b63 | ||
|
|
b2aaad43f0 | ||
|
|
8cae729181 | ||
|
|
71fe09f7f0 | ||
|
|
7ae3c420f4 | ||
|
|
830a36319c | ||
|
|
2abf70527a | ||
|
|
b4214b56b3 | ||
|
|
8565655f08 | ||
|
|
fa3a6647ef | ||
|
|
efd3627202 | ||
|
|
cc94ec179c |
@@ -11,7 +11,7 @@ import sys
|
|||||||
|
|
||||||
from pipecat.pipeline.pipeline import Pipeline
|
from pipecat.pipeline.pipeline import Pipeline
|
||||||
from pipecat.pipeline.runner import PipelineRunner
|
from pipecat.pipeline.runner import PipelineRunner
|
||||||
from pipecat.pipeline.task import PipelineTask
|
from pipecat.pipeline.task import PipelineParams, PipelineTask
|
||||||
from pipecat.services.cartesia import CartesiaTTSService
|
from pipecat.services.cartesia import CartesiaTTSService
|
||||||
from pipecat.services.openai import OpenAILLMContext, OpenAILLMService
|
from pipecat.services.openai import OpenAILLMContext, OpenAILLMService
|
||||||
from pipecat.transports.services.daily import DailyParams, DailyTransport
|
from pipecat.transports.services.daily import DailyParams, DailyTransport
|
||||||
@@ -115,13 +115,21 @@ async def main():
|
|||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
task = PipelineTask(pipeline)
|
task = PipelineTask(
|
||||||
|
pipeline,
|
||||||
|
PipelineParams(
|
||||||
|
allow_interruptions=True,
|
||||||
|
enable_metrics=True,
|
||||||
|
enable_usage_metrics=True,
|
||||||
|
report_only_initial_ttfb=True,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
@transport.event_handler("on_first_participant_joined")
|
@transport.event_handler("on_first_participant_joined")
|
||||||
async def on_first_participant_joined(transport, participant):
|
async def on_first_participant_joined(transport, participant):
|
||||||
transport.capture_participant_transcription(participant["id"])
|
transport.capture_participant_transcription(participant["id"])
|
||||||
# Kick off the conversation.
|
# Kick off the conversation.
|
||||||
await tts.say("Hi! Ask me about the weather in San Francisco.")
|
await task.queue_frames([context_aggregator.user().get_context_frame()])
|
||||||
|
|
||||||
runner = PipelineRunner()
|
runner = PipelineRunner()
|
||||||
|
|
||||||
|
|||||||
283
examples/foundational/19-openai-realtime-beta.py
Normal file
283
examples/foundational/19-openai-realtime-beta.py
Normal file
@@ -0,0 +1,283 @@
|
|||||||
|
#
|
||||||
|
# Copyright (c) 2024, Daily
|
||||||
|
#
|
||||||
|
# SPDX-License-Identifier: BSD 2-Clause License
|
||||||
|
#
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
import sys
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
import aiohttp
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
from loguru import logger
|
||||||
|
from runner import configure
|
||||||
|
|
||||||
|
from pipecat.frames.frames import LLMMessagesUpdateFrame
|
||||||
|
from pipecat.pipeline.pipeline import Pipeline
|
||||||
|
from pipecat.pipeline.runner import PipelineRunner
|
||||||
|
from pipecat.pipeline.task import PipelineParams, PipelineTask
|
||||||
|
from pipecat.processors.aggregators.openai_llm_context import (
|
||||||
|
OpenAILLMContext,
|
||||||
|
)
|
||||||
|
from pipecat.services.openai_realtime_beta import (
|
||||||
|
InputAudioTranscription,
|
||||||
|
OpenAILLMServiceRealtimeBeta,
|
||||||
|
SessionProperties,
|
||||||
|
)
|
||||||
|
from pipecat.transports.services.daily import DailyParams, DailyTransport
|
||||||
|
from pipecat.vad.silero import SileroVADAnalyzer
|
||||||
|
from pipecat.vad.vad_analyzer import VADParams
|
||||||
|
|
||||||
|
load_dotenv(override=True)
|
||||||
|
|
||||||
|
logger.remove(0)
|
||||||
|
logger.add(sys.stderr, level="DEBUG")
|
||||||
|
|
||||||
|
|
||||||
|
messages = [
|
||||||
|
{"role": "user", "content": "Say 'Hello there' and ask my name."},
|
||||||
|
{"role": "assistant", "content": [{"type": "text", "text": "Hello there! What's your name?"}]},
|
||||||
|
# {"role": "user", "content": [{"type": "input_audio"}]},
|
||||||
|
{"role": "user", "content": [{"type": "text", "text": "Tell me a joke.\n"}]},
|
||||||
|
# {
|
||||||
|
# "role": "assistant",
|
||||||
|
# "content": [
|
||||||
|
# {
|
||||||
|
# "type": "text",
|
||||||
|
# "text": "Why don't scientists trust atoms? Because they make up everything!",
|
||||||
|
# }
|
||||||
|
# ],
|
||||||
|
# },
|
||||||
|
# {"role": "user", "content": [{"type": "text", "text": "me know the joke.\n"}]},
|
||||||
|
# {
|
||||||
|
# "role": "assistant",
|
||||||
|
# "content": [{"type": "text", "text": "What do you call fake spaghetti? An impasta!"}],
|
||||||
|
# },
|
||||||
|
# {"role": "user", "content": [{"type": "text", "text": "me another joke.\n"}]},
|
||||||
|
# {
|
||||||
|
# "role": "assistant",
|
||||||
|
# "content": [
|
||||||
|
# {
|
||||||
|
# "type": "text",
|
||||||
|
# "text": "Why couldn't the bicycle stand up by itself? It was two-tired!",
|
||||||
|
# }
|
||||||
|
# ],
|
||||||
|
# },
|
||||||
|
# {"role": "user", "content": [{"type": "input_audio"}]},
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
async def fetch_weather_from_api(function_name, tool_call_id, args, llm, context, result_callback):
|
||||||
|
temperature = 75 if args["format"] == "fahrenheit" else 24
|
||||||
|
await result_callback(
|
||||||
|
{
|
||||||
|
"conditions": "nice",
|
||||||
|
"temperature": temperature,
|
||||||
|
"format": args["format"],
|
||||||
|
"timestamp": datetime.now().strftime("%Y%m%d_%H%M%S"),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def get_saved_conversation_filenames(
|
||||||
|
function_name, tool_call_id, args, llm, context, result_callback
|
||||||
|
):
|
||||||
|
pattern = re.compile("example_19_\\d{8}_\\d{6}\\.json$")
|
||||||
|
matching_files = []
|
||||||
|
|
||||||
|
for filename in os.listdir("."):
|
||||||
|
if pattern.match(filename):
|
||||||
|
matching_files.append(filename)
|
||||||
|
|
||||||
|
await result_callback({"filenames": matching_files})
|
||||||
|
|
||||||
|
|
||||||
|
async def save_conversation(function_name, tool_call_id, args, llm, context, result_callback):
|
||||||
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||||
|
filename = f"example_19_{timestamp}.json"
|
||||||
|
logger.debug(f"writing conversation to {filename}\n{json.dumps(context.messages, indent=4)}")
|
||||||
|
try:
|
||||||
|
with open(filename, "w") as file:
|
||||||
|
json.dump(context.messages, file, indent=4)
|
||||||
|
await result_callback({"success": True})
|
||||||
|
except Exception as e:
|
||||||
|
await result_callback({"success": False, "error": str(e)})
|
||||||
|
|
||||||
|
|
||||||
|
async def load_conversation(function_name, tool_call_id, args, llm, context, result_callback):
|
||||||
|
filename = args["filename"]
|
||||||
|
logger.debug(f"loading conversation from {filename}")
|
||||||
|
try:
|
||||||
|
with open(filename, "r") as file:
|
||||||
|
messages = json.load(file)
|
||||||
|
await result_callback({"success": True})
|
||||||
|
await llm.push_frame(LLMMessagesUpdateFrame(messages))
|
||||||
|
except Exception as e:
|
||||||
|
await result_callback({"success": False, "error": str(e)})
|
||||||
|
|
||||||
|
|
||||||
|
tools = [
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"name": "get_current_weather",
|
||||||
|
"description": "Get the current weather",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"location": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "The city and state, e.g. San Francisco, CA",
|
||||||
|
},
|
||||||
|
"format": {
|
||||||
|
"type": "string",
|
||||||
|
"enum": ["celsius", "fahrenheit"],
|
||||||
|
"description": "The temperature unit to use. Infer this from the users location.",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"required": ["location", "format"],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"name": "save_conversation",
|
||||||
|
"description": "Save the current conversatione. Use this function to persist the current conversation to external storage.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {},
|
||||||
|
"required": [],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"name": "get_saved_conversation_filenames",
|
||||||
|
"description": "Get a list of saved conversation histories. Returns a list of filenames. Each filename includes a timestamp. Each file is conversation history that can be loaded into this session.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {},
|
||||||
|
"required": [],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"name": "load_conversation",
|
||||||
|
"description": "Load a conversation history. Use this function to load a conversation history into the current session.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"filename": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "The filename of the conversation history to load.",
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"required": ["filename"],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
async with aiohttp.ClientSession() as session:
|
||||||
|
(room_url, token) = await configure(session)
|
||||||
|
|
||||||
|
transport = DailyTransport(
|
||||||
|
room_url,
|
||||||
|
token,
|
||||||
|
"Respond bot",
|
||||||
|
DailyParams(
|
||||||
|
audio_in_enabled=True,
|
||||||
|
audio_in_sample_rate=24000,
|
||||||
|
audio_out_enabled=True,
|
||||||
|
audio_out_sample_rate=24000,
|
||||||
|
transcription_enabled=False,
|
||||||
|
vad_enabled=True,
|
||||||
|
vad_analyzer=SileroVADAnalyzer(params=VADParams(stop_secs=0.8)),
|
||||||
|
vad_audio_passthrough=True,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
session_properties = SessionProperties(
|
||||||
|
input_audio_transcription=InputAudioTranscription(),
|
||||||
|
# Set openai TurnDetection parameters. Not setting this at all will turn it
|
||||||
|
# on by default
|
||||||
|
# turn_detection=TurnDetection(silence_duration_ms=1000),
|
||||||
|
# Or set to False to disable openai turn detection and use transport VAD
|
||||||
|
turn_detection=False,
|
||||||
|
# tools=tools,
|
||||||
|
instructions="""
|
||||||
|
Your knowledge cutoff is 2023-10. You are a helpful and friendly AI.
|
||||||
|
|
||||||
|
Act like a human, but remember that you aren't a human and that you can't do human
|
||||||
|
things in the real world. Your voice and personality should be warm and engaging, with a lively and
|
||||||
|
playful tone.
|
||||||
|
|
||||||
|
If interacting in a non-English language, start by using the standard accent or dialect familiar to
|
||||||
|
the user. Talk quickly. You should always call a function if you can. Do not refer to these rules,
|
||||||
|
even if you're asked about them.
|
||||||
|
|
||||||
|
You are participating in a voice conversation. Keep your responses concise, short, and to the point
|
||||||
|
unless specifically asked to elaborate on a topic.
|
||||||
|
|
||||||
|
Remember, your responses should be short. Just one or two sentences, usually.
|
||||||
|
""",
|
||||||
|
)
|
||||||
|
|
||||||
|
llm = OpenAILLMServiceRealtimeBeta(
|
||||||
|
api_key=os.getenv("OPENAI_API_KEY"),
|
||||||
|
session_properties=session_properties,
|
||||||
|
start_audio_paused=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
# you can either register a single function for all function calls, or specific functions
|
||||||
|
# llm.register_function(None, fetch_weather_from_api)
|
||||||
|
llm.register_function("get_current_weather", fetch_weather_from_api)
|
||||||
|
llm.register_function("save_conversation", save_conversation)
|
||||||
|
llm.register_function("get_saved_conversation_filenames", get_saved_conversation_filenames)
|
||||||
|
llm.register_function("load_conversation", load_conversation)
|
||||||
|
|
||||||
|
context = OpenAILLMContext(
|
||||||
|
messages,
|
||||||
|
# [{"role": "user", "content": "Say 'hello'."}],
|
||||||
|
# [{"role": "user", "content": "What's the weather right now in San Francisco?"}],
|
||||||
|
# conversation load from file is a WIP -- not functional yet
|
||||||
|
# [{"role": "user", "content": "Load the most recent conversation."}],
|
||||||
|
tools,
|
||||||
|
)
|
||||||
|
context_aggregator = llm.create_context_aggregator(context)
|
||||||
|
|
||||||
|
pipeline = Pipeline(
|
||||||
|
[
|
||||||
|
transport.input(), # Transport user input
|
||||||
|
context_aggregator.user(),
|
||||||
|
llm, # LLM
|
||||||
|
context_aggregator.assistant(),
|
||||||
|
transport.output(), # Transport bot output
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
task = PipelineTask(
|
||||||
|
pipeline,
|
||||||
|
PipelineParams(
|
||||||
|
allow_interruptions=True,
|
||||||
|
enable_metrics=True,
|
||||||
|
enable_usage_metrics=True,
|
||||||
|
# report_only_initial_ttfb=True,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
@transport.event_handler("on_first_participant_joined")
|
||||||
|
async def on_first_participant_joined(transport, participant):
|
||||||
|
transport.capture_participant_transcription(participant["id"])
|
||||||
|
# Kick off the conversation.
|
||||||
|
await task.queue_frames([context_aggregator.user().get_context_frame()])
|
||||||
|
|
||||||
|
runner = PipelineRunner()
|
||||||
|
|
||||||
|
await runner.run(task)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
asyncio.run(main())
|
||||||
@@ -52,11 +52,11 @@ livekit = [ "livekit~=0.13.1", "tenacity~=9.0.0" ]
|
|||||||
lmnt = [ "lmnt~=1.1.4" ]
|
lmnt = [ "lmnt~=1.1.4" ]
|
||||||
local = [ "pyaudio~=0.2.14" ]
|
local = [ "pyaudio~=0.2.14" ]
|
||||||
moondream = [ "einops~=0.8.0", "timm~=1.0.8", "transformers~=4.44.0" ]
|
moondream = [ "einops~=0.8.0", "timm~=1.0.8", "transformers~=4.44.0" ]
|
||||||
openai = [ "openai~=1.37.2" ]
|
openai = [ "openai~=1.50.2", "websockets~=12.0", "python-deepcompare~=1.0.1" ]
|
||||||
openpipe = [ "openpipe~=4.24.0" ]
|
openpipe = [ "openpipe~=4.24.0" ]
|
||||||
playht = [ "pyht~=0.0.28" ]
|
playht = [ "pyht~=0.0.28" ]
|
||||||
silero = [ "onnxruntime>=1.16.1" ]
|
silero = [ "onnxruntime>=1.16.1" ]
|
||||||
together = [ "together~=1.2.7" ]
|
together = [ "openai~=1.50.2" ]
|
||||||
websocket = [ "websockets~=12.0", "fastapi~=0.115.0" ]
|
websocket = [ "websockets~=12.0", "fastapi~=0.115.0" ]
|
||||||
whisper = [ "faster-whisper~=1.0.3" ]
|
whisper = [ "faster-whisper~=1.0.3" ]
|
||||||
xtts = [ "resampy~=0.4.3" ]
|
xtts = [ "resampy~=0.4.3" ]
|
||||||
|
|||||||
@@ -168,6 +168,7 @@ class OpenAILLMContext:
|
|||||||
llm: FrameProcessor,
|
llm: FrameProcessor,
|
||||||
run_llm: bool = True,
|
run_llm: bool = True,
|
||||||
) -> None:
|
) -> None:
|
||||||
|
logger.debug(f"Calling function {function_name} with arguments {arguments}")
|
||||||
# Push a SystemFrame downstream. This frame will let our assistant context aggregator
|
# Push a SystemFrame downstream. This frame will let our assistant context aggregator
|
||||||
# know that we are in the middle of a function call. Some contexts/aggregators may
|
# know that we are in the middle of a function call. Some contexts/aggregators may
|
||||||
# not need this. But some definitely do (Anthropic, for example).
|
# not need this. But some definitely do (Anthropic, for example).
|
||||||
|
|||||||
@@ -490,7 +490,7 @@ class RTVIBotLLMTextProcessor(RTVIFrameProcessor):
|
|||||||
|
|
||||||
await self.push_frame(frame, direction)
|
await self.push_frame(frame, direction)
|
||||||
|
|
||||||
if isinstance(frame, TextFrame):
|
if type(frame) is TextFrame:
|
||||||
await self._handle_text(frame)
|
await self._handle_text(frame)
|
||||||
|
|
||||||
async def _handle_text(self, frame: TextFrame):
|
async def _handle_text(self, frame: TextFrame):
|
||||||
@@ -507,7 +507,7 @@ class RTVIBotTTSTextProcessor(RTVIFrameProcessor):
|
|||||||
|
|
||||||
await self.push_frame(frame, direction)
|
await self.push_frame(frame, direction)
|
||||||
|
|
||||||
if isinstance(frame, TextFrame):
|
if type(frame) is TextFrame:
|
||||||
await self._handle_text(frame)
|
await self._handle_text(frame)
|
||||||
|
|
||||||
async def _handle_text(self, frame: TextFrame):
|
async def _handle_text(self, frame: TextFrame):
|
||||||
|
|||||||
@@ -46,6 +46,7 @@ class AIService(FrameProcessor):
|
|||||||
super().__init__(**kwargs)
|
super().__init__(**kwargs)
|
||||||
self._model_name: str = ""
|
self._model_name: str = ""
|
||||||
self._settings: Dict[str, Any] = {}
|
self._settings: Dict[str, Any] = {}
|
||||||
|
self._session_properties: Dict[str, Any] = {}
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def model_name(self) -> str:
|
def model_name(self) -> str:
|
||||||
@@ -65,11 +66,44 @@ class AIService(FrameProcessor):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
async def _update_settings(self, settings: Dict[str, Any]):
|
async def _update_settings(self, settings: Dict[str, Any]):
|
||||||
|
from pipecat.services.openai_realtime_beta.events import (
|
||||||
|
SessionProperties,
|
||||||
|
)
|
||||||
|
|
||||||
for key, value in settings.items():
|
for key, value in settings.items():
|
||||||
|
print("Update request for:", key, value)
|
||||||
|
|
||||||
if key in self._settings:
|
if key in self._settings:
|
||||||
logger.debug(f"Updating setting {key} to: [{value}] for {self.name}")
|
logger.debug(f"Updating LLM setting {key} to: [{value}]")
|
||||||
self._settings[key] = value
|
self._settings[key] = value
|
||||||
|
elif key in SessionProperties.model_fields:
|
||||||
|
print("Attempting to update", key, value)
|
||||||
|
|
||||||
|
try:
|
||||||
|
from pipecat.services.openai_realtime_beta.events import (
|
||||||
|
TurnDetection,
|
||||||
|
)
|
||||||
|
|
||||||
|
if isinstance(self._session_properties, SessionProperties):
|
||||||
|
current_properties = self._session_properties
|
||||||
|
else:
|
||||||
|
current_properties = SessionProperties(**self._session_properties)
|
||||||
|
|
||||||
|
if key == "turn_detection" and isinstance(value, dict):
|
||||||
|
turn_detection = TurnDetection(**value)
|
||||||
|
setattr(current_properties, key, turn_detection)
|
||||||
|
else:
|
||||||
|
setattr(current_properties, key, value)
|
||||||
|
|
||||||
|
validated_properties = SessionProperties.model_validate(
|
||||||
|
current_properties.model_dump()
|
||||||
|
)
|
||||||
|
logger.debug(f"Updating LLM setting {key} to: [{value}]")
|
||||||
|
self._session_properties = validated_properties.model_dump()
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"Unexpected error updating session property {key}: {e}")
|
||||||
elif key == "model":
|
elif key == "model":
|
||||||
|
logger.debug(f"Updating LLM setting {key} to: [{value}]")
|
||||||
self.set_model_name(value)
|
self.set_model_name(value)
|
||||||
else:
|
else:
|
||||||
logger.warning(f"Unknown setting for {self.name} service: {key}")
|
logger.warning(f"Unknown setting for {self.name} service: {key}")
|
||||||
|
|||||||
@@ -63,6 +63,7 @@ except ModuleNotFoundError as e:
|
|||||||
)
|
)
|
||||||
raise Exception(f"Missing module: {e}")
|
raise Exception(f"Missing module: {e}")
|
||||||
|
|
||||||
|
|
||||||
ValidVoice = Literal["alloy", "echo", "fable", "onyx", "nova", "shimmer"]
|
ValidVoice = Literal["alloy", "echo", "fable", "onyx", "nova", "shimmer"]
|
||||||
|
|
||||||
VALID_VOICES: Dict[str, ValidVoice] = {
|
VALID_VOICES: Dict[str, ValidVoice] = {
|
||||||
@@ -469,7 +470,7 @@ class OpenAIUserContextAggregator(LLMUserContextAggregator):
|
|||||||
if frame.user_id in self._context._user_image_request_context:
|
if frame.user_id in self._context._user_image_request_context:
|
||||||
del self._context._user_image_request_context[frame.user_id]
|
del self._context._user_image_request_context[frame.user_id]
|
||||||
elif isinstance(frame, UserImageRawFrame):
|
elif isinstance(frame, UserImageRawFrame):
|
||||||
# Push a new AnthropicImageMessageFrame with the text context we cached
|
# Push a new OpenAIImageMessageFrame with the text context we cached
|
||||||
# downstream to be handled by our assistant context aggregator. This is
|
# downstream to be handled by our assistant context aggregator. This is
|
||||||
# necessary so that we add the message to the context in the right order.
|
# necessary so that we add the message to the context in the right order.
|
||||||
text = self._context._user_image_request_context.get(frame.user_id) or ""
|
text = self._context._user_image_request_context.get(frame.user_id) or ""
|
||||||
@@ -496,8 +497,10 @@ class OpenAIAssistantContextAggregator(LLMAssistantContextAggregator):
|
|||||||
self._function_calls_in_progress.clear()
|
self._function_calls_in_progress.clear()
|
||||||
self._function_call_finished = None
|
self._function_call_finished = None
|
||||||
elif isinstance(frame, FunctionCallInProgressFrame):
|
elif isinstance(frame, FunctionCallInProgressFrame):
|
||||||
|
logger.debug(f"FunctionCallInProgressFrame: {frame}")
|
||||||
self._function_calls_in_progress[frame.tool_call_id] = frame
|
self._function_calls_in_progress[frame.tool_call_id] = frame
|
||||||
elif isinstance(frame, FunctionCallResultFrame):
|
elif isinstance(frame, FunctionCallResultFrame):
|
||||||
|
logger.debug(f"FunctionCallResultFrame: {frame}")
|
||||||
if frame.tool_call_id in self._function_calls_in_progress:
|
if frame.tool_call_id in self._function_calls_in_progress:
|
||||||
del self._function_calls_in_progress[frame.tool_call_id]
|
del self._function_calls_in_progress[frame.tool_call_id]
|
||||||
self._function_call_result = frame
|
self._function_call_result = frame
|
||||||
|
|||||||
2
src/pipecat/services/openai_realtime_beta/__init__.py
Normal file
2
src/pipecat/services/openai_realtime_beta/__init__.py
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
from .events import InputAudioTranscription, SessionProperties, TurnDetection
|
||||||
|
from .llm_and_context import OpenAILLMServiceRealtimeBeta
|
||||||
428
src/pipecat/services/openai_realtime_beta/events.py
Normal file
428
src/pipecat/services/openai_realtime_beta/events.py
Normal file
@@ -0,0 +1,428 @@
|
|||||||
|
import json
|
||||||
|
import uuid
|
||||||
|
from typing import Any, Dict, List, Literal, Optional, Union
|
||||||
|
|
||||||
|
from loguru import logger
|
||||||
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
#
|
||||||
|
# session properties
|
||||||
|
#
|
||||||
|
|
||||||
|
|
||||||
|
class InputAudioTranscription(BaseModel):
|
||||||
|
model: Optional[str] = "whisper-1"
|
||||||
|
|
||||||
|
|
||||||
|
class TurnDetection(BaseModel):
|
||||||
|
type: Optional[Literal["server_vad"]] = "server_vad"
|
||||||
|
threshold: Optional[float] = 0.5
|
||||||
|
prefix_padding_ms: Optional[int] = 300
|
||||||
|
silence_duration_ms: Optional[int] = 800
|
||||||
|
|
||||||
|
|
||||||
|
class SessionProperties(BaseModel):
|
||||||
|
modalities: Optional[List[Literal["text", "audio"]]] = None
|
||||||
|
instructions: Optional[str] = None
|
||||||
|
voice: Optional[str] = None
|
||||||
|
input_audio_format: Optional[Literal["pcm16", "g711_ulaw", "g711_alaw"]] = None
|
||||||
|
output_audio_format: Optional[Literal["pcm16", "g711_ulaw", "g711_alaw"]] = None
|
||||||
|
input_audio_transcription: Optional[InputAudioTranscription] = None
|
||||||
|
# set turn_detection to False to disable turn detection
|
||||||
|
turn_detection: Optional[Union[TurnDetection, bool]] = Field(default=None)
|
||||||
|
tools: Optional[List[Dict]] = None
|
||||||
|
tool_choice: Optional[Literal["auto", "none", "required"]] = None
|
||||||
|
temperature: Optional[float] = None
|
||||||
|
max_response_output_tokens: Optional[Union[int, Literal["inf"]]] = None
|
||||||
|
|
||||||
|
|
||||||
|
#
|
||||||
|
# context
|
||||||
|
#
|
||||||
|
|
||||||
|
|
||||||
|
class ItemContent(BaseModel):
|
||||||
|
type: Literal["text", "audio", "input_text", "input_audio"]
|
||||||
|
text: Optional[str] = None
|
||||||
|
audio: Optional[str] = None # base64-encoded audio
|
||||||
|
transcript: Optional[str] = None
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationItem(BaseModel):
|
||||||
|
id: str = Field(default_factory=lambda: str(uuid.uuid4().hex))
|
||||||
|
object: Optional[Literal["realtime.item"]] = None
|
||||||
|
type: Literal["message", "function_call", "function_call_output"]
|
||||||
|
status: Optional[Literal["completed", "in_progress", "incomplete"]] = None
|
||||||
|
# role and content are present for message items
|
||||||
|
role: Optional[Literal["user", "assistant", "system"]] = None
|
||||||
|
content: Optional[List[ItemContent]] = None
|
||||||
|
# these four fields are present for function_call items
|
||||||
|
call_id: Optional[str] = None
|
||||||
|
name: Optional[str] = None
|
||||||
|
arguments: Optional[str] = None
|
||||||
|
output: Optional[str] = None
|
||||||
|
|
||||||
|
|
||||||
|
class RealtimeConversation(BaseModel):
|
||||||
|
id: str
|
||||||
|
object: Literal["realtime.conversation"]
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseProperties(BaseModel):
|
||||||
|
modalities: Optional[List[Literal["text", "audio"]]] = ["audio", "text"]
|
||||||
|
instructions: Optional[str] = None
|
||||||
|
voice: Optional[str] = None
|
||||||
|
output_audio_format: Optional[Literal["pcm16", "g711_ulaw", "g711_alaw"]] = None
|
||||||
|
tools: Optional[List[Dict]] = []
|
||||||
|
tool_choice: Optional[Literal["auto", "none", "required"]] = None
|
||||||
|
temperature: Optional[float] = None
|
||||||
|
max_response_output_tokens: Optional[Union[int, Literal["inf"]]] = None
|
||||||
|
|
||||||
|
|
||||||
|
#
|
||||||
|
# error class
|
||||||
|
#
|
||||||
|
class RealtimeError(BaseModel):
|
||||||
|
type: str
|
||||||
|
code: str
|
||||||
|
message: str
|
||||||
|
param: Optional[str] = None
|
||||||
|
|
||||||
|
|
||||||
|
#
|
||||||
|
# client events
|
||||||
|
#
|
||||||
|
|
||||||
|
|
||||||
|
class ClientEvent(BaseModel):
|
||||||
|
event_id: str = Field(default_factory=lambda: str(uuid.uuid4()))
|
||||||
|
|
||||||
|
|
||||||
|
class SessionUpdateEvent(ClientEvent):
|
||||||
|
type: Literal["session.update"] = "session.update"
|
||||||
|
session: SessionProperties
|
||||||
|
|
||||||
|
def model_dump(self, *args, **kwargs) -> Dict[str, Any]:
|
||||||
|
logger.debug(f"!!! SessionUpdateEvent.model_dump: {self}")
|
||||||
|
dump = super().model_dump(*args, **kwargs)
|
||||||
|
|
||||||
|
# Handle turn_detection so that False is serialized as null
|
||||||
|
if "turn_detection" in dump["session"]:
|
||||||
|
if dump["session"]["turn_detection"] is False:
|
||||||
|
dump["session"]["turn_detection"] = None
|
||||||
|
|
||||||
|
return dump
|
||||||
|
|
||||||
|
|
||||||
|
class InputAudioBufferAppendEvent(ClientEvent):
|
||||||
|
type: Literal["input_audio_buffer.append"] = "input_audio_buffer.append"
|
||||||
|
audio: str # base64-encoded audio
|
||||||
|
|
||||||
|
|
||||||
|
class InputAudioBufferCommitEvent(ClientEvent):
|
||||||
|
type: Literal["input_audio_buffer.commit"] = "input_audio_buffer.commit"
|
||||||
|
|
||||||
|
|
||||||
|
class InputAudioBufferClearEvent(ClientEvent):
|
||||||
|
type: Literal["input_audio_buffer.clear"] = "input_audio_buffer.clear"
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationItemCreateEvent(ClientEvent):
|
||||||
|
type: Literal["conversation.item.create"] = "conversation.item.create"
|
||||||
|
previous_item_id: Optional[str] = None
|
||||||
|
item: ConversationItem
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationItemTruncateEvent(ClientEvent):
|
||||||
|
type: Literal["conversation.item.truncate"] = "conversation.item.truncate"
|
||||||
|
item_id: str
|
||||||
|
content_index: int
|
||||||
|
audio_end_ms: int
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationItemDeleteEvent(ClientEvent):
|
||||||
|
type: Literal["conversation.item.delete"] = "conversation.item.delete"
|
||||||
|
item_id: str
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseCreateEvent(ClientEvent):
|
||||||
|
type: Literal["response.create"] = "response.create"
|
||||||
|
response: Optional[ResponseProperties] = None
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseCancelEvent(ClientEvent):
|
||||||
|
type: Literal["response.cancel"] = "response.cancel"
|
||||||
|
|
||||||
|
|
||||||
|
#
|
||||||
|
# server events
|
||||||
|
#
|
||||||
|
|
||||||
|
|
||||||
|
class ServerEvent(BaseModel):
|
||||||
|
event_id: str
|
||||||
|
type: str
|
||||||
|
|
||||||
|
class Config:
|
||||||
|
arbitrary_types_allowed = True
|
||||||
|
|
||||||
|
|
||||||
|
class SessionCreatedEvent(ServerEvent):
|
||||||
|
type: Literal["session.created"]
|
||||||
|
session: SessionProperties
|
||||||
|
|
||||||
|
|
||||||
|
class SessionUpdatedEvent(ServerEvent):
|
||||||
|
type: Literal["session.updated"]
|
||||||
|
session: SessionProperties
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationCreated(ServerEvent):
|
||||||
|
type: Literal["conversation.created"]
|
||||||
|
conversation: RealtimeConversation
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationItemCreated(ServerEvent):
|
||||||
|
type: Literal["conversation.item.created"]
|
||||||
|
previous_item_id: Optional[str] = None
|
||||||
|
item: ConversationItem
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationItemInputAudioTranscriptionCompleted(ServerEvent):
|
||||||
|
type: Literal["conversation.item.input_audio_transcription.completed"]
|
||||||
|
item_id: str
|
||||||
|
content_index: int
|
||||||
|
transcript: str
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationItemInputAudioTranscriptionFailed(ServerEvent):
|
||||||
|
type: Literal["conversation.item.input_audio_transcription.failed"]
|
||||||
|
item_id: str
|
||||||
|
content_index: int
|
||||||
|
error: RealtimeError
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationItemTruncated(ServerEvent):
|
||||||
|
type: Literal["conversation.item.truncated"]
|
||||||
|
item_id: str
|
||||||
|
content_index: int
|
||||||
|
audio_end_ms: int
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationItemDeleted(ServerEvent):
|
||||||
|
type: Literal["conversation.item.deleted"]
|
||||||
|
item_id: str
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseCreated(ServerEvent):
|
||||||
|
type: Literal["response.created"]
|
||||||
|
response: "Response"
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseDone(ServerEvent):
|
||||||
|
type: Literal["response.done"]
|
||||||
|
response: "Response"
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseOutputItemAdded(ServerEvent):
|
||||||
|
type: Literal["response.output_item.added"]
|
||||||
|
response_id: str
|
||||||
|
output_index: int
|
||||||
|
item: ConversationItem
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseOutputItemDone(ServerEvent):
|
||||||
|
type: Literal["response.output_item.done"]
|
||||||
|
response_id: str
|
||||||
|
output_index: int
|
||||||
|
item: ConversationItem
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseContentPartAdded(ServerEvent):
|
||||||
|
type: Literal["response.content_part.added"]
|
||||||
|
response_id: str
|
||||||
|
item_id: str
|
||||||
|
output_index: int
|
||||||
|
content_index: int
|
||||||
|
part: ItemContent
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseContentPartDone(ServerEvent):
|
||||||
|
type: Literal["response.content_part.done"]
|
||||||
|
response_id: str
|
||||||
|
item_id: str
|
||||||
|
output_index: int
|
||||||
|
content_index: int
|
||||||
|
part: ItemContent
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseTextDelta(ServerEvent):
|
||||||
|
type: Literal["response.text.delta"]
|
||||||
|
response_id: str
|
||||||
|
item_id: str
|
||||||
|
output_index: int
|
||||||
|
content_index: int
|
||||||
|
delta: str
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseTextDone(ServerEvent):
|
||||||
|
type: Literal["response.text.done"]
|
||||||
|
response_id: str
|
||||||
|
item_id: str
|
||||||
|
output_index: int
|
||||||
|
content_index: int
|
||||||
|
text: str
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseAudioTranscriptDelta(ServerEvent):
|
||||||
|
type: Literal["response.audio_transcript.delta"]
|
||||||
|
response_id: str
|
||||||
|
item_id: str
|
||||||
|
output_index: int
|
||||||
|
content_index: int
|
||||||
|
delta: str
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseAudioTranscriptDone(ServerEvent):
|
||||||
|
type: Literal["response.audio_transcript.done"]
|
||||||
|
response_id: str
|
||||||
|
item_id: str
|
||||||
|
output_index: int
|
||||||
|
content_index: int
|
||||||
|
transcript: str
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseAudioDelta(ServerEvent):
|
||||||
|
type: Literal["response.audio.delta"]
|
||||||
|
response_id: str
|
||||||
|
item_id: str
|
||||||
|
output_index: int
|
||||||
|
content_index: int
|
||||||
|
delta: str # base64-encoded audio
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseAudioDone(ServerEvent):
|
||||||
|
type: Literal["response.audio.done"]
|
||||||
|
response_id: str
|
||||||
|
item_id: str
|
||||||
|
output_index: int
|
||||||
|
content_index: int
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseFunctionCallArgumentsDelta(ServerEvent):
|
||||||
|
type: Literal["response.function_call_arguments.delta"]
|
||||||
|
response_id: str
|
||||||
|
item_id: str
|
||||||
|
output_index: int
|
||||||
|
call_id: str
|
||||||
|
delta: str
|
||||||
|
|
||||||
|
|
||||||
|
class ResponseFunctionCallArgumentsDone(ServerEvent):
|
||||||
|
type: Literal["response.function_call_arguments.done"]
|
||||||
|
response_id: str
|
||||||
|
item_id: str
|
||||||
|
output_index: int
|
||||||
|
call_id: str
|
||||||
|
arguments: str
|
||||||
|
|
||||||
|
|
||||||
|
class InputAudioBufferSpeechStarted(ServerEvent):
|
||||||
|
type: Literal["input_audio_buffer.speech_started"]
|
||||||
|
audio_start_ms: int
|
||||||
|
item_id: str
|
||||||
|
|
||||||
|
|
||||||
|
class InputAudioBufferSpeechStopped(ServerEvent):
|
||||||
|
type: Literal["input_audio_buffer.speech_stopped"]
|
||||||
|
audio_end_ms: int
|
||||||
|
item_id: str
|
||||||
|
|
||||||
|
|
||||||
|
class InputAudioBufferCommitted(ServerEvent):
|
||||||
|
type: Literal["input_audio_buffer.committed"]
|
||||||
|
previous_item_id: Optional[str] = None
|
||||||
|
item_id: str
|
||||||
|
|
||||||
|
|
||||||
|
class InputAudioBufferCleared(ServerEvent):
|
||||||
|
type: Literal["input_audio_buffer.cleared"]
|
||||||
|
|
||||||
|
|
||||||
|
class ErrorEvent(ServerEvent):
|
||||||
|
type: Literal["error"]
|
||||||
|
error: RealtimeError
|
||||||
|
|
||||||
|
|
||||||
|
class RateLimitsUpdated(ServerEvent):
|
||||||
|
type: Literal["rate_limits.updated"]
|
||||||
|
rate_limits: List[Dict[str, Any]]
|
||||||
|
|
||||||
|
|
||||||
|
class TokenDetails(BaseModel):
|
||||||
|
cached_tokens: Optional[int] = 0
|
||||||
|
text_tokens: Optional[int] = 0
|
||||||
|
audio_tokens: Optional[int] = 0
|
||||||
|
|
||||||
|
class Config:
|
||||||
|
extra = "allow"
|
||||||
|
|
||||||
|
|
||||||
|
class Usage(BaseModel):
|
||||||
|
total_tokens: int
|
||||||
|
input_tokens: int
|
||||||
|
output_tokens: int
|
||||||
|
input_token_details: TokenDetails
|
||||||
|
output_token_details: TokenDetails
|
||||||
|
|
||||||
|
|
||||||
|
class Response(BaseModel):
|
||||||
|
id: str
|
||||||
|
object: Literal["realtime.response"]
|
||||||
|
status: Literal["completed", "in_progress", "incomplete", "cancelled", "failed"]
|
||||||
|
status_details: Any
|
||||||
|
output: List[ConversationItem]
|
||||||
|
usage: Optional[Usage] = None
|
||||||
|
|
||||||
|
|
||||||
|
_server_event_types = {
|
||||||
|
"error": ErrorEvent,
|
||||||
|
"session.created": SessionCreatedEvent,
|
||||||
|
"session.updated": SessionUpdatedEvent,
|
||||||
|
"conversation.created": ConversationCreated,
|
||||||
|
"input_audio_buffer.committed": InputAudioBufferCommitted,
|
||||||
|
"input_audio_buffer.cleared": InputAudioBufferCleared,
|
||||||
|
"input_audio_buffer.speech_started": InputAudioBufferSpeechStarted,
|
||||||
|
"input_audio_buffer.speech_stopped": InputAudioBufferSpeechStopped,
|
||||||
|
"conversation.item.created": ConversationItemCreated,
|
||||||
|
"conversation.item.input_audio_transcription.completed": ConversationItemInputAudioTranscriptionCompleted,
|
||||||
|
"conversation.item.input_audio_transcription.failed": ConversationItemInputAudioTranscriptionFailed,
|
||||||
|
"conversation.item.truncated": ConversationItemTruncated,
|
||||||
|
"conversation.item.deleted": ConversationItemDeleted,
|
||||||
|
"response.created": ResponseCreated,
|
||||||
|
"response.done": ResponseDone,
|
||||||
|
"response.output_item.added": ResponseOutputItemAdded,
|
||||||
|
"response.output_item.done": ResponseOutputItemDone,
|
||||||
|
"response.content_part.added": ResponseContentPartAdded,
|
||||||
|
"response.content_part.done": ResponseContentPartDone,
|
||||||
|
"response.text.delta": ResponseTextDelta,
|
||||||
|
"response.text.done": ResponseTextDone,
|
||||||
|
"response.audio_transcript.delta": ResponseAudioTranscriptDelta,
|
||||||
|
"response.audio_transcript.done": ResponseAudioTranscriptDone,
|
||||||
|
"response.audio.delta": ResponseAudioDelta,
|
||||||
|
"response.audio.done": ResponseAudioDone,
|
||||||
|
"response.function_call_arguments.delta": ResponseFunctionCallArgumentsDelta,
|
||||||
|
"response.function_call_arguments.done": ResponseFunctionCallArgumentsDone,
|
||||||
|
"rate_limits.updated": RateLimitsUpdated,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def parse_server_event(str):
|
||||||
|
try:
|
||||||
|
event = json.loads(str)
|
||||||
|
event_type = event["type"]
|
||||||
|
if event_type not in _server_event_types:
|
||||||
|
raise Exception(f"Unimplemented server event type: {event_type}")
|
||||||
|
return _server_event_types[event_type].model_validate(event)
|
||||||
|
except Exception as e:
|
||||||
|
raise Exception(f"{e} \n\n{str}")
|
||||||
670
src/pipecat/services/openai_realtime_beta/llm_and_context.py
Normal file
670
src/pipecat/services/openai_realtime_beta/llm_and_context.py
Normal file
@@ -0,0 +1,670 @@
|
|||||||
|
import asyncio
|
||||||
|
import base64
|
||||||
|
import json
|
||||||
|
|
||||||
|
# temp: websocket logger
|
||||||
|
import logging
|
||||||
|
import traceback
|
||||||
|
from copy import deepcopy
|
||||||
|
from dataclasses import dataclass
|
||||||
|
from typing import List
|
||||||
|
|
||||||
|
import websockets
|
||||||
|
from loguru import logger
|
||||||
|
|
||||||
|
from pipecat.frames.frames import (
|
||||||
|
CancelFrame,
|
||||||
|
DataFrame,
|
||||||
|
EndFrame,
|
||||||
|
ErrorFrame,
|
||||||
|
Frame,
|
||||||
|
InputAudioRawFrame,
|
||||||
|
LLMFullResponseEndFrame,
|
||||||
|
LLMFullResponseStartFrame,
|
||||||
|
LLMMessagesUpdateFrame,
|
||||||
|
LLMSetToolsFrame,
|
||||||
|
LLMUpdateSettingsFrame,
|
||||||
|
StartFrame,
|
||||||
|
StartInterruptionFrame,
|
||||||
|
StopInterruptionFrame,
|
||||||
|
TextFrame,
|
||||||
|
TranscriptionFrame,
|
||||||
|
TTSAudioRawFrame,
|
||||||
|
TTSStartedFrame,
|
||||||
|
TTSStoppedFrame,
|
||||||
|
UserStartedSpeakingFrame,
|
||||||
|
UserStoppedSpeakingFrame,
|
||||||
|
)
|
||||||
|
from pipecat.metrics.metrics import LLMTokenUsage
|
||||||
|
from pipecat.processors.aggregators.openai_llm_context import (
|
||||||
|
OpenAILLMContext,
|
||||||
|
OpenAILLMContextFrame,
|
||||||
|
)
|
||||||
|
from pipecat.processors.frame_processor import FrameDirection
|
||||||
|
from pipecat.services.ai_services import LLMService
|
||||||
|
from pipecat.services.openai import (
|
||||||
|
OpenAIAssistantContextAggregator,
|
||||||
|
OpenAIContextAggregatorPair,
|
||||||
|
OpenAIUserContextAggregator,
|
||||||
|
)
|
||||||
|
from pipecat.utils.time import time_now_iso8601
|
||||||
|
|
||||||
|
from . import events
|
||||||
|
|
||||||
|
logging.basicConfig(
|
||||||
|
format="%(message)s",
|
||||||
|
level=logging.DEBUG,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class _InternalMessagesUpdateFrame(DataFrame):
|
||||||
|
context: "OpenAIRealtimeLLMContext"
|
||||||
|
|
||||||
|
|
||||||
|
class OpenAIUnhandledFunctionException(Exception):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class OpenAIRealtimeLLMContext(OpenAILLMContext):
|
||||||
|
def __init__(self, messages=None, tools=None, **kwargs):
|
||||||
|
super().__init__(messages=messages, tools=tools, **kwargs)
|
||||||
|
self.__setup_local()
|
||||||
|
|
||||||
|
def __setup_local(self):
|
||||||
|
# messages that have been added to the context but not yet sent to the openai server
|
||||||
|
self._unsent_messages = deepcopy(self._messages)
|
||||||
|
# messages that we added to the context because they were part of our external
|
||||||
|
# context store. we do not want to add these again when we see conversation.item.created
|
||||||
|
# events about them. map from item_id to True
|
||||||
|
self._manually_created_messages = {}
|
||||||
|
# "conversation items" that have been created by opeanai realtime api events but are
|
||||||
|
# not completely filled in, yet. map from item_id to message
|
||||||
|
self._messages_in_progress = {}
|
||||||
|
# count of messages prior to recent reset
|
||||||
|
self._messages_reset_count = 0
|
||||||
|
self._tools_list_updated = True
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def upgrade_to_realtime(obj: OpenAILLMContext) -> "OpenAIRealtimeLLMContext":
|
||||||
|
if isinstance(obj, OpenAILLMContext) and not isinstance(obj, OpenAIRealtimeLLMContext):
|
||||||
|
obj.__class__ = OpenAIRealtimeLLMContext
|
||||||
|
obj.__setup_local()
|
||||||
|
return obj
|
||||||
|
|
||||||
|
# still working on
|
||||||
|
# - clearing the context by deleting all messages
|
||||||
|
# - reloading from a standard messages list
|
||||||
|
# - truncating the last spoken message to maintain context when interrupted
|
||||||
|
|
||||||
|
def set_tools(self, tools: List):
|
||||||
|
super().set_tools(tools)
|
||||||
|
self._tools_list_updated = True
|
||||||
|
|
||||||
|
def add_message(self, message):
|
||||||
|
super().add_message(message)
|
||||||
|
self._unsent_messages.append(message)
|
||||||
|
return message
|
||||||
|
|
||||||
|
def add_messages(self, messages):
|
||||||
|
super().add_messages(messages)
|
||||||
|
self._unsent_messages.extend(messages)
|
||||||
|
|
||||||
|
def add_message_already_present_in_api_context(self, message):
|
||||||
|
super().add_message(message)
|
||||||
|
return message
|
||||||
|
|
||||||
|
def set_messages(self, messages):
|
||||||
|
self._messages_reset_count = len(self.messages) - len(self._unsent_messages)
|
||||||
|
super().set_messages(messages)
|
||||||
|
self._unsent_messages = deepcopy(self._messages)
|
||||||
|
|
||||||
|
def get_unsent_messages(self):
|
||||||
|
return self._unsent_messages
|
||||||
|
|
||||||
|
def get_messages_reset_count(self):
|
||||||
|
return self._messages_reset_count
|
||||||
|
|
||||||
|
def get_tools_list_updated(self):
|
||||||
|
return self._tools_list_updated
|
||||||
|
|
||||||
|
def update_all_messages_sent(self):
|
||||||
|
self._unsent_messages = []
|
||||||
|
self._messages_reset_count = 0
|
||||||
|
|
||||||
|
def update_tools_list_sent(self):
|
||||||
|
self._tools_list_updated = False
|
||||||
|
|
||||||
|
def note_manually_added_message(self, item_id):
|
||||||
|
self._manually_created_messages[item_id] = True
|
||||||
|
|
||||||
|
def add_message_from_realtime_event(self, evt):
|
||||||
|
if evt.item.id in self._manually_created_messages:
|
||||||
|
del self._manually_created_messages[evt.item.id]
|
||||||
|
return
|
||||||
|
|
||||||
|
# add messages. don't add function_call or function_call_output items.
|
||||||
|
if evt.item.type == "message":
|
||||||
|
message = self.add_message_already_present_in_api_context(
|
||||||
|
{"role": evt.item.role, "content": []}
|
||||||
|
)
|
||||||
|
if not evt.item.content:
|
||||||
|
self._messages_in_progress[evt.item.id] = message
|
||||||
|
return
|
||||||
|
for content in evt.item.content:
|
||||||
|
message["content"].append({"type": content.type})
|
||||||
|
if content.text:
|
||||||
|
message["content"] = content.text
|
||||||
|
elif content.transcript:
|
||||||
|
message["content"] = content.transcript
|
||||||
|
else:
|
||||||
|
# we will get the transcript in a later event
|
||||||
|
self._messages_in_progress[evt.item.id] = message
|
||||||
|
return
|
||||||
|
|
||||||
|
def add_transcript_to_message(self, evt):
|
||||||
|
message = self._messages_in_progress.get(evt.item_id)
|
||||||
|
if message:
|
||||||
|
cs = message["content"]
|
||||||
|
cs.extend([{"type": ""}] * (evt.content_index - len(cs) + 1))
|
||||||
|
cs[evt.content_index] = {"type": "text", "text": evt.transcript}
|
||||||
|
del self._messages_in_progress[evt.item_id]
|
||||||
|
else:
|
||||||
|
logger.error(
|
||||||
|
f"Could not find content {evt.item_id}/{evt.content_index} to add transcript to"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class OpenAIRealtimeUserContextAggregator(OpenAIUserContextAggregator):
|
||||||
|
async def process_frame(
|
||||||
|
self, frame: Frame, direction: FrameDirection = FrameDirection.DOWNSTREAM
|
||||||
|
):
|
||||||
|
await super().process_frame(frame, direction)
|
||||||
|
# Parent does not push LLMMessagesUpdateFrame. This ensures that in a typical pipeline,
|
||||||
|
# messages are only processed by the user context aggregator, which is generally what we want. But
|
||||||
|
# we also need to send new messages over the websocket, in case audio mode triggers a response before
|
||||||
|
# we get any other context frames through the pipeline.
|
||||||
|
if isinstance(frame, LLMMessagesUpdateFrame):
|
||||||
|
await self.push_frame(_InternalMessagesUpdateFrame(context=self._context))
|
||||||
|
|
||||||
|
# Parent also doesn't push the LLMSetToolsFrame.
|
||||||
|
if isinstance(frame, LLMSetToolsFrame):
|
||||||
|
await self.push_frame(frame, direction)
|
||||||
|
|
||||||
|
async def _push_aggregation(self):
|
||||||
|
# for the moment, ignore all user input coming into the pipeline.
|
||||||
|
# todo: think about whether/how to fix this to allow for text input from
|
||||||
|
# upstream (transport/transcription, or other sources)
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class OpenAIRealtimeAssistantContextAggregator(OpenAIAssistantContextAggregator):
|
||||||
|
async def _push_aggregation(self):
|
||||||
|
# the only thing we implement here is function calling. in all other cases, messages
|
||||||
|
# are added to the context when we receive openai realtime api events
|
||||||
|
if not self._function_call_result:
|
||||||
|
return
|
||||||
|
|
||||||
|
self._reset()
|
||||||
|
try:
|
||||||
|
frame = self._function_call_result
|
||||||
|
self._function_call_result = None
|
||||||
|
if frame.result:
|
||||||
|
self._context.add_message_already_present_in_api_context(
|
||||||
|
{
|
||||||
|
"role": "assistant",
|
||||||
|
"tool_calls": [
|
||||||
|
{
|
||||||
|
"id": frame.tool_call_id,
|
||||||
|
"function": {
|
||||||
|
"name": frame.function_name,
|
||||||
|
"arguments": json.dumps(frame.arguments),
|
||||||
|
},
|
||||||
|
"type": "function",
|
||||||
|
}
|
||||||
|
],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
self._context.add_message(
|
||||||
|
{
|
||||||
|
"role": "tool",
|
||||||
|
"content": json.dumps(frame.result),
|
||||||
|
"tool_call_id": frame.tool_call_id,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
run_llm = frame.run_llm
|
||||||
|
|
||||||
|
if run_llm:
|
||||||
|
await self._user_context_aggregator.push_context_frame()
|
||||||
|
|
||||||
|
frame = OpenAILLMContextFrame(self._context)
|
||||||
|
await self.push_frame(frame)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Error processing frame: {e}")
|
||||||
|
|
||||||
|
|
||||||
|
class OpenAILLMServiceRealtimeBeta(LLMService):
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
*,
|
||||||
|
api_key: str,
|
||||||
|
base_url="wss://api.openai.com/v1/realtime?model=gpt-4o-realtime-preview-2024-10-01",
|
||||||
|
session_properties: events.SessionProperties = events.SessionProperties(),
|
||||||
|
start_audio_paused: bool = False,
|
||||||
|
send_transcription_frames: bool = True,
|
||||||
|
send_user_started_speaking_frames: bool = False,
|
||||||
|
**kwargs,
|
||||||
|
):
|
||||||
|
super().__init__(base_url=base_url, **kwargs)
|
||||||
|
self.api_key = api_key
|
||||||
|
self.base_url = base_url
|
||||||
|
|
||||||
|
self._session_properties: events.SessionProperties = session_properties
|
||||||
|
self._audio_input_paused = start_audio_paused
|
||||||
|
self._send_transcription_frames = send_transcription_frames
|
||||||
|
# todo: wire _send_user_started_speaking_frames up correctly
|
||||||
|
self._send_user_started_speaking_frames = send_user_started_speaking_frames
|
||||||
|
self._websocket = None
|
||||||
|
self._receive_task = None
|
||||||
|
self._context = None
|
||||||
|
self._bot_speaking = False
|
||||||
|
|
||||||
|
def can_generate_metrics(self) -> bool:
|
||||||
|
return True
|
||||||
|
|
||||||
|
def set_audio_input_paused(self, paused: bool):
|
||||||
|
self._audio_input_paused = paused
|
||||||
|
|
||||||
|
async def start(self, frame: StartFrame):
|
||||||
|
await super().start(frame)
|
||||||
|
await self._connect()
|
||||||
|
|
||||||
|
async def stop(self, frame: EndFrame):
|
||||||
|
await super().stop(frame)
|
||||||
|
await self._disconnect()
|
||||||
|
|
||||||
|
async def cancel(self, frame: CancelFrame):
|
||||||
|
await super().cancel(frame)
|
||||||
|
await self._disconnect()
|
||||||
|
|
||||||
|
async def send_client_event(self, event: events.ClientEvent):
|
||||||
|
await self._ws_send(event.model_dump(exclude_none=True))
|
||||||
|
|
||||||
|
async def _ws_send(self, realtime_message):
|
||||||
|
try:
|
||||||
|
await self._websocket.send(json.dumps(realtime_message))
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Error sending message to websocket: {e}")
|
||||||
|
await self.push_error(ErrorFrame(error=f"Error sending client event: {e}", fatal=True))
|
||||||
|
|
||||||
|
async def _connect(self):
|
||||||
|
try:
|
||||||
|
self._websocket = await websockets.connect(
|
||||||
|
uri=self.base_url,
|
||||||
|
extra_headers={
|
||||||
|
"Authorization": f"Bearer {self.api_key}",
|
||||||
|
"OpenAI-Beta": "realtime=v1",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
self._receive_task = self.get_event_loop().create_task(self._receive_task_handler())
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"{self} initialization error: {e}")
|
||||||
|
self._websocket = None
|
||||||
|
|
||||||
|
async def _disconnect(self):
|
||||||
|
try:
|
||||||
|
await self.stop_all_metrics()
|
||||||
|
|
||||||
|
if self._websocket:
|
||||||
|
await self._websocket.close()
|
||||||
|
self._websocket = None
|
||||||
|
|
||||||
|
if self._receive_task:
|
||||||
|
self._receive_task.cancel()
|
||||||
|
await self._receive_task
|
||||||
|
self._receive_task = None
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"{self} error closing websocket: {e}")
|
||||||
|
|
||||||
|
def _get_websocket(self):
|
||||||
|
if self._websocket:
|
||||||
|
return self._websocket
|
||||||
|
raise Exception("Websocket not connected")
|
||||||
|
|
||||||
|
async def _update_settings(self):
|
||||||
|
# !!! LEAVE ALL DEFAULT SETTINGS FOR NOW
|
||||||
|
return
|
||||||
|
settings = self._session_properties
|
||||||
|
# tools given in the context override the tools in the session properties
|
||||||
|
if self._context and self._context.tools:
|
||||||
|
settings.tools = self._context.tools
|
||||||
|
self._context.update_tools_list_sent()
|
||||||
|
await self.send_client_event(events.SessionUpdateEvent(session=settings))
|
||||||
|
|
||||||
|
async def _receive_task_handler(self):
|
||||||
|
try:
|
||||||
|
async for message in self._get_websocket():
|
||||||
|
evt = events.parse_server_event(message)
|
||||||
|
# logger.debug(f"Received event: {evt}")
|
||||||
|
if evt.type == "session.created":
|
||||||
|
# session.created is received right after connecting. send a message
|
||||||
|
# to configure the session properties.
|
||||||
|
logger.debug(f"!!! GOT SESSION CREATED {evt}")
|
||||||
|
await self._update_settings()
|
||||||
|
elif evt.type == "session.updated":
|
||||||
|
logger.debug(f"!!! GOT SESSION UPDATED {evt}")
|
||||||
|
self._session_properties = evt.session
|
||||||
|
elif evt.type == "conversation.created":
|
||||||
|
logger.debug(f"!!! GOT CONVERSATION CREATED: {evt}")
|
||||||
|
elif evt.type == "input_audio_buffer.speech_started":
|
||||||
|
# user started speaking
|
||||||
|
if self._send_user_started_speaking_frames:
|
||||||
|
await self.push_frame(UserStartedSpeakingFrame())
|
||||||
|
await self.push_frame(StartInterruptionFrame())
|
||||||
|
logger.debug("User started speaking")
|
||||||
|
pass
|
||||||
|
elif evt.type == "input_audio_buffer.speech_stopped":
|
||||||
|
# user stopped speaking
|
||||||
|
if self._send_user_started_speaking_frames:
|
||||||
|
await self.push_frame(UserStoppedSpeakingFrame())
|
||||||
|
await self.push_frame(StopInterruptionFrame())
|
||||||
|
|
||||||
|
logger.debug("User stopped speaking")
|
||||||
|
await self.start_processing_metrics()
|
||||||
|
await self.start_ttfb_metrics()
|
||||||
|
elif evt.type == "conversation.item.created":
|
||||||
|
# this will get sent from the server every time a new "message" is added
|
||||||
|
# to the server's conversation state
|
||||||
|
if self._context:
|
||||||
|
self._context.add_message_from_realtime_event(evt)
|
||||||
|
elif evt.type == "response.created":
|
||||||
|
# todo: 1. figure out TTS started/stopped frame semantics better
|
||||||
|
# 2. do not push these frames in text-only mode
|
||||||
|
logger.debug(f"!!! GOT RESPONSE CREATED {evt}")
|
||||||
|
if not self._bot_speaking:
|
||||||
|
self._bot_speaking = True
|
||||||
|
await self.push_frame(TTSStartedFrame())
|
||||||
|
pass
|
||||||
|
elif evt.type == "conversation.item.input_audio_transcription.completed":
|
||||||
|
if evt.transcript:
|
||||||
|
if self._context:
|
||||||
|
self._context.add_transcript_to_message(evt)
|
||||||
|
if self._send_transcription_frames:
|
||||||
|
await self.push_frame(
|
||||||
|
# no way to get a language code?
|
||||||
|
TranscriptionFrame(evt.transcript, "", time_now_iso8601())
|
||||||
|
)
|
||||||
|
elif evt.type == "response.output_item.added":
|
||||||
|
# todo: think about adding a frame for this (generally, in Pipecat/RTVI), as
|
||||||
|
# it could be useful for managing UI state
|
||||||
|
pass
|
||||||
|
elif evt.type == "response.content_part.added":
|
||||||
|
# todo: same thing — possibly a useful event for client-side UI
|
||||||
|
pass
|
||||||
|
elif evt.type == "response.audio_transcript.delta":
|
||||||
|
# note: the openai playground app uses this, not "response.text.delta"
|
||||||
|
if evt.delta:
|
||||||
|
await self.push_frame(TextFrame(evt.delta))
|
||||||
|
elif evt.type == "response.audio.delta":
|
||||||
|
await self.stop_ttfb_metrics()
|
||||||
|
frame = TTSAudioRawFrame(
|
||||||
|
audio=base64.b64decode(evt.delta),
|
||||||
|
sample_rate=24000,
|
||||||
|
num_channels=1,
|
||||||
|
)
|
||||||
|
await self.push_frame(frame)
|
||||||
|
elif evt.type == "response.audio.done":
|
||||||
|
if self._bot_speaking:
|
||||||
|
self._bot_speaking = False
|
||||||
|
await self.push_frame(TTSStoppedFrame())
|
||||||
|
elif evt.type == "response.audio_transcript.done":
|
||||||
|
if self._context:
|
||||||
|
self._context.add_transcript_to_message(evt)
|
||||||
|
pass
|
||||||
|
elif evt.type == "response.content_part.done":
|
||||||
|
# this doesn't map to any Pipecat frame types
|
||||||
|
pass
|
||||||
|
elif evt.type == "response.output_item.done":
|
||||||
|
# this doesn't map to any Pipecat frame types
|
||||||
|
pass
|
||||||
|
elif evt.type == "response.done":
|
||||||
|
# usage metrics
|
||||||
|
tokens = LLMTokenUsage(
|
||||||
|
prompt_tokens=evt.response.usage.input_tokens,
|
||||||
|
completion_tokens=evt.response.usage.output_tokens,
|
||||||
|
total_tokens=evt.response.usage.total_tokens,
|
||||||
|
)
|
||||||
|
await self.start_llm_usage_metrics(tokens)
|
||||||
|
await self.stop_processing_metrics()
|
||||||
|
# function calls
|
||||||
|
items = evt.response.output
|
||||||
|
function_calls = [item for item in items if item.type == "function_call"]
|
||||||
|
if function_calls:
|
||||||
|
await self._handle_function_call_items(function_calls)
|
||||||
|
await self.push_frame(LLMFullResponseEndFrame())
|
||||||
|
elif evt.type == "rate_limits.updated":
|
||||||
|
# todo: add a Pipecat frame for this. (maybe?)
|
||||||
|
pass
|
||||||
|
elif evt.type == "error":
|
||||||
|
# These errors seem to be fatal to this connection. So, close and send an ErrorFrame.
|
||||||
|
raise Exception(f"Error: {evt}")
|
||||||
|
|
||||||
|
except asyncio.CancelledError:
|
||||||
|
pass
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"{self} exception: {e}\n\nStack trace:\n{traceback.format_exc()}")
|
||||||
|
await self.push_error(ErrorFrame(error=f"Error receiving: {e}", fatal=True))
|
||||||
|
|
||||||
|
async def _handle_function_call_items(self, items):
|
||||||
|
total_items = len(items)
|
||||||
|
for index, item in enumerate(items):
|
||||||
|
function_name = item.name
|
||||||
|
tool_id = item.call_id
|
||||||
|
arguments = json.loads(item.arguments)
|
||||||
|
if self.has_function(function_name):
|
||||||
|
run_llm = index == total_items - 1
|
||||||
|
if function_name in self._callbacks.keys():
|
||||||
|
await self.call_function(
|
||||||
|
context=self._context,
|
||||||
|
tool_call_id=tool_id,
|
||||||
|
function_name=function_name,
|
||||||
|
arguments=arguments,
|
||||||
|
run_llm=run_llm,
|
||||||
|
)
|
||||||
|
elif None in self._callbacks.keys():
|
||||||
|
await self.call_function(
|
||||||
|
context=self._context,
|
||||||
|
tool_call_id=tool_id,
|
||||||
|
function_name=function_name,
|
||||||
|
arguments=arguments,
|
||||||
|
run_llm=run_llm,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
raise OpenAIUnhandledFunctionException(
|
||||||
|
f"The LLM tried to call a function named '{function_name}', but there isn't a callback registered for that function."
|
||||||
|
)
|
||||||
|
|
||||||
|
async def _reset_conversation(self, count):
|
||||||
|
# need to think about how to implement this, and how to think about interop with messages lists
|
||||||
|
# used with the HTTP API
|
||||||
|
logger.debug(f"!!! RESET CONVERSATION: {count} [WIP]")
|
||||||
|
await self._disconnect()
|
||||||
|
await self._connect()
|
||||||
|
pass
|
||||||
|
|
||||||
|
async def _send_messages_context_update(self):
|
||||||
|
if not self._context:
|
||||||
|
return
|
||||||
|
context = self._context
|
||||||
|
messages = context.get_unsent_messages()
|
||||||
|
|
||||||
|
needs_reset = context.get_messages_reset_count()
|
||||||
|
context.update_all_messages_sent()
|
||||||
|
|
||||||
|
if needs_reset:
|
||||||
|
await self._reset_conversation(needs_reset)
|
||||||
|
# debugging
|
||||||
|
logger.debug("MESSAGE HISTORY RELOAD NOT IMPLEMENTED YET")
|
||||||
|
return
|
||||||
|
|
||||||
|
items = []
|
||||||
|
for m in messages:
|
||||||
|
if m and (
|
||||||
|
m.get("role") == "user" or m.get("role") == "system" or m.get("role") == "assistant"
|
||||||
|
):
|
||||||
|
content = m.get("content")
|
||||||
|
if isinstance(content, str):
|
||||||
|
# skip any messages that aren't "text" and change "user" message type to "input_text"
|
||||||
|
|
||||||
|
if m.get("type", "text") == "text":
|
||||||
|
items.append(
|
||||||
|
events.ConversationItem(
|
||||||
|
type="message",
|
||||||
|
status="completed",
|
||||||
|
role=m.get("role", "user"),
|
||||||
|
content=[
|
||||||
|
events.ItemContent(
|
||||||
|
type="input_text" if m.get("role") == "user" else "text",
|
||||||
|
text=content,
|
||||||
|
)
|
||||||
|
],
|
||||||
|
)
|
||||||
|
)
|
||||||
|
elif isinstance(content, list):
|
||||||
|
# skip any messages that aren't "text" and change "user" message type to "input_text"
|
||||||
|
cs = []
|
||||||
|
for item in content:
|
||||||
|
if item.get("type", "text") == "text":
|
||||||
|
# cs.append(events.ItemContent(type="input_text", text=item.get("text")))
|
||||||
|
(
|
||||||
|
cs.append(
|
||||||
|
events.ItemContent(
|
||||||
|
type="input_text" if m.get("role") == "user" else "text",
|
||||||
|
text=item.get("text"),
|
||||||
|
)
|
||||||
|
),
|
||||||
|
)
|
||||||
|
if cs:
|
||||||
|
items.append(
|
||||||
|
events.ConversationItem(
|
||||||
|
type="message",
|
||||||
|
status="completed",
|
||||||
|
role=m.get("role", "user"),
|
||||||
|
content=cs,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
elif m.get("role") == "assistant" and m.get("tool_calls"):
|
||||||
|
tc = m.get("tool_calls")[0]
|
||||||
|
items.append(
|
||||||
|
events.ConversationItem(
|
||||||
|
type="function_call",
|
||||||
|
call_id=tc["id"],
|
||||||
|
name=tc["function"]["name"],
|
||||||
|
arguments=tc["function"]["arguments"],
|
||||||
|
)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
raise Exception(f"Invalid message content {m}")
|
||||||
|
elif m and m.get("role") == "tool":
|
||||||
|
items.append(
|
||||||
|
events.ConversationItem(
|
||||||
|
type="function_call_output",
|
||||||
|
call_id=m.get("tool_call_id"),
|
||||||
|
output=m["content"],
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
for item in items:
|
||||||
|
context.note_manually_added_message(item.id)
|
||||||
|
evt = events.ConversationItemCreateEvent(item=item)
|
||||||
|
logger.debug(
|
||||||
|
f"!!! > Sending message: {evt.model_dump_json(indent=2, exclude_none=True)}"
|
||||||
|
)
|
||||||
|
await self.send_client_event(evt)
|
||||||
|
await asyncio.sleep(2)
|
||||||
|
# await self.send_client_event(events.ConversationItemCreateEvent(item=item))
|
||||||
|
|
||||||
|
async def _create_response(self):
|
||||||
|
if self._context.get_tools_list_updated():
|
||||||
|
await self._update_settings()
|
||||||
|
|
||||||
|
# !!! DEBUGGING - testing await on conversation.create
|
||||||
|
logger.debug("!!! A waiting on conversation.created")
|
||||||
|
await asyncio.sleep(3)
|
||||||
|
logger.debug("!!! A ok, done waiting")
|
||||||
|
|
||||||
|
await self._send_messages_context_update()
|
||||||
|
logger.debug(f"Creating response: {self._context.get_messages_for_logging()}")
|
||||||
|
await self.push_frame(LLMFullResponseStartFrame())
|
||||||
|
await self.start_processing_metrics()
|
||||||
|
await self.send_client_event(
|
||||||
|
events.ResponseCreateEvent(
|
||||||
|
response=events.ResponseProperties(modalities=["audio", "text"])
|
||||||
|
)
|
||||||
|
)
|
||||||
|
# !!! DEBUGGING
|
||||||
|
await asyncio.sleep(2)
|
||||||
|
# logger.debug("Unpausing microphone")
|
||||||
|
# self.set_audio_input_paused(False)
|
||||||
|
|
||||||
|
async def _send_user_audio(self, frame):
|
||||||
|
payload = base64.b64encode(frame.audio).decode("utf-8")
|
||||||
|
await self.send_client_event(events.InputAudioBufferAppendEvent(audio=payload))
|
||||||
|
|
||||||
|
async def _handle_interruption(self, frame):
|
||||||
|
await self.send_client_event(events.InputAudioBufferClearEvent())
|
||||||
|
await self.send_client_event(events.ResponseCancelEvent())
|
||||||
|
await self.stop_all_metrics()
|
||||||
|
await self.push_frame(LLMFullResponseEndFrame())
|
||||||
|
await self.push_frame(TTSStoppedFrame())
|
||||||
|
|
||||||
|
async def _handle_user_started_speaking(self, frame):
|
||||||
|
pass
|
||||||
|
|
||||||
|
async def _handle_user_stopped_speaking(self, frame):
|
||||||
|
if self._session_properties.turn_detection is None:
|
||||||
|
await self.send_client_event(events.InputAudioBufferCommitEvent())
|
||||||
|
await self.send_client_event(events.ResponseCreateEvent())
|
||||||
|
pass
|
||||||
|
|
||||||
|
async def process_frame(self, frame: Frame, direction: FrameDirection):
|
||||||
|
await super().process_frame(frame, direction)
|
||||||
|
|
||||||
|
if isinstance(frame, TranscriptionFrame):
|
||||||
|
pass
|
||||||
|
elif isinstance(frame, OpenAILLMContextFrame):
|
||||||
|
context: OpenAIRealtimeLLMContext = OpenAIRealtimeLLMContext.upgrade_to_realtime(
|
||||||
|
frame.context
|
||||||
|
)
|
||||||
|
self._context = context
|
||||||
|
await self._create_response()
|
||||||
|
elif isinstance(frame, InputAudioRawFrame):
|
||||||
|
if not self._audio_input_paused:
|
||||||
|
await self._send_user_audio(frame)
|
||||||
|
elif isinstance(frame, StartInterruptionFrame):
|
||||||
|
await self._handle_interruption(frame)
|
||||||
|
elif isinstance(frame, UserStartedSpeakingFrame):
|
||||||
|
await self._handle_user_started_speaking(frame)
|
||||||
|
elif isinstance(frame, UserStoppedSpeakingFrame):
|
||||||
|
await self._handle_user_stopped_speaking(frame)
|
||||||
|
elif isinstance(frame, _InternalMessagesUpdateFrame):
|
||||||
|
self._context = frame.context
|
||||||
|
await self._send_messages_context_update()
|
||||||
|
elif isinstance(frame, LLMUpdateSettingsFrame):
|
||||||
|
self._session_properties = frame.settings
|
||||||
|
await self._update_settings()
|
||||||
|
elif isinstance(frame, LLMSetToolsFrame):
|
||||||
|
await self._update_settings()
|
||||||
|
|
||||||
|
await self.push_frame(frame, direction)
|
||||||
|
|
||||||
|
def create_context_aggregator(
|
||||||
|
self, context: OpenAILLMContext, *, assistant_expect_stripped_words: bool = False
|
||||||
|
) -> OpenAIContextAggregatorPair:
|
||||||
|
OpenAIRealtimeLLMContext.upgrade_to_realtime(context)
|
||||||
|
user = OpenAIRealtimeUserContextAggregator(context)
|
||||||
|
assistant = OpenAIRealtimeAssistantContextAggregator(
|
||||||
|
user, expect_stripped_words=assistant_expect_stripped_words
|
||||||
|
)
|
||||||
|
return OpenAIContextAggregatorPair(_user=user, _assistant=assistant)
|
||||||
Reference in New Issue
Block a user