codex-api/src/endpoint/realtime_websocket/methods_common.rs
95 lines
use crate::endpoint::realtime_websocket::methods_v1::conversation_handoff_append_message as v1_conversation_handoff_append_message;use crate::endpoint::realtime_websocket::methods_v1::conversation_item_create_message as v1_conversation_item_create_message;use crate::endpoint::realtime_websocket::methods_v1::session_update_session as v1_session_update_session;use crate::endpoint::realtime_websocket::methods_v1::websocket_intent as v1_websocket_intent;use crate::endpoint::realtime_websocket::methods_v2::conversation_function_call_output_message as v2_conversation_function_call_output_message;use crate::endpoint::realtime_websocket::methods_v2::conversation_item_create_message as v2_conversation_item_create_message;use crate::endpoint::realtime_websocket::methods_v2::session_update_session as v2_session_update_session;use crate::endpoint::realtime_websocket::methods_v2::websocket_intent as v2_websocket_intent;use crate::endpoint::realtime_websocket::protocol::RealtimeEventParser;use crate::endpoint::realtime_websocket::protocol::RealtimeOutboundMessage;use crate::endpoint::realtime_websocket::protocol::RealtimeOutputModality;use crate::endpoint::realtime_websocket::protocol::RealtimeSessionConfig;use crate::endpoint::realtime_websocket::protocol::RealtimeSessionMode;use crate::endpoint::realtime_websocket::protocol::RealtimeVoice;use crate::endpoint::realtime_websocket::protocol::SessionUpdateSession;use codex_protocol::protocol::ConversationTextRole;use serde_json::Result as JsonResult;use serde_json::Value;use serde_json::to_value;pub(super) const REALTIME_AUDIO_SAMPLE_RATE: u32 = 24_000;const AGENT_FINAL_MESSAGE_PREFIX: &str = "\"Agent Final Message\":\n\n";pub(super) fn normalized_session_mode( event_parser: RealtimeEventParser, session_mode: RealtimeSessionMode,) -> RealtimeSessionMode { match event_parser { RealtimeEventParser::V1 => RealtimeSessionMode::Conversational, RealtimeEventParser::RealtimeV2 => session_mode, }}pub(super) fn conversation_item_create_message( event_parser: RealtimeEventParser, text: String, role: ConversationTextRole,) -> RealtimeOutboundMessage { match event_parser { RealtimeEventParser::V1 => v1_conversation_item_create_message(text, role), RealtimeEventParser::RealtimeV2 => v2_conversation_item_create_message(text, role), }}pub(super) fn conversation_function_call_output_message( event_parser: RealtimeEventParser, call_id: String, output_text: String,) -> RealtimeOutboundMessage { match event_parser { RealtimeEventParser::V1 => v1_conversation_handoff_append_message( call_id, format!("{AGENT_FINAL_MESSAGE_PREFIX}{output_text}"), ), RealtimeEventParser::RealtimeV2 => { v2_conversation_function_call_output_message(call_id, output_text) } }}pub(super) fn session_update_session( event_parser: RealtimeEventParser, instructions: String, session_mode: RealtimeSessionMode, output_modality: RealtimeOutputModality, voice: RealtimeVoice,) -> SessionUpdateSession { let session_mode = normalized_session_mode(event_parser, session_mode); match event_parser { RealtimeEventParser::V1 => v1_session_update_session(instructions, voice), RealtimeEventParser::RealtimeV2 => { v2_session_update_session(instructions, session_mode, output_modality, voice) } }}pub fn session_update_session_json(config: RealtimeSessionConfig) -> JsonResult<Value> { let mut session = session_update_session( config.event_parser, config.instructions, config.session_mode, config.output_modality, config.voice, ); session.id = config.session_id; session.model = config.model; to_value(session)}pub(super) fn websocket_intent(event_parser: RealtimeEventParser) -> Option<&'static str> { match event_parser { RealtimeEventParser::V1 => v1_websocket_intent(), RealtimeEventParser::RealtimeV2 => v2_websocket_intent(), }}