mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-30 06:03:37 +00:00
2842 lines
117 KiB
Swift
2842 lines
117 KiB
Swift
import AVFAudio
|
||
import Foundation
|
||
import Observation
|
||
import OpenClawChatUI
|
||
import OpenClawKit
|
||
import OpenClawProtocol
|
||
import OSLog
|
||
import Speech
|
||
|
||
private final class StreamFailureBox: @unchecked Sendable {
|
||
private let lock = NSLock()
|
||
private var valueInternal: Error?
|
||
|
||
func set(_ error: Error) {
|
||
self.lock.lock()
|
||
self.valueInternal = error
|
||
self.lock.unlock()
|
||
}
|
||
|
||
var value: Error? {
|
||
self.lock.lock()
|
||
defer { self.lock.unlock() }
|
||
return self.valueInternal
|
||
}
|
||
}
|
||
|
||
enum TalkGatewayPermissionState: Equatable {
|
||
case unknown
|
||
case ready
|
||
case missingScope(String)
|
||
case requestingUpgrade
|
||
case upgradeRequested(requestId: String?)
|
||
case requestFailed(String)
|
||
case apiKeyMissing
|
||
case loadFailed(String)
|
||
|
||
var statusLabel: String {
|
||
switch self {
|
||
case .unknown:
|
||
"Not checked"
|
||
case .ready:
|
||
"Ready"
|
||
case let .missingScope(scope):
|
||
"Missing \(scope)"
|
||
case .requestingUpgrade:
|
||
"Requesting approval"
|
||
case .upgradeRequested:
|
||
"Approval requested"
|
||
case .requestFailed:
|
||
"Request failed"
|
||
case .apiKeyMissing:
|
||
"API key missing"
|
||
case .loadFailed:
|
||
"Load failed"
|
||
}
|
||
}
|
||
|
||
var requiresTalkPermissionAction: Bool {
|
||
switch self {
|
||
case .missingScope, .requestingUpgrade, .upgradeRequested, .requestFailed:
|
||
true
|
||
default:
|
||
false
|
||
}
|
||
}
|
||
|
||
var isApprovalRequestInProgress: Bool {
|
||
switch self {
|
||
case .requestingUpgrade, .upgradeRequested:
|
||
true
|
||
default:
|
||
false
|
||
}
|
||
}
|
||
|
||
var failureMessage: String? {
|
||
if case let .requestFailed(message) = self {
|
||
return message
|
||
}
|
||
return nil
|
||
}
|
||
|
||
var requestId: String? {
|
||
if case let .upgradeRequested(requestId) = self {
|
||
return requestId
|
||
}
|
||
return nil
|
||
}
|
||
}
|
||
|
||
// This file intentionally centralizes talk mode state + behavior.
|
||
// It's large, and splitting would force `private` -> `fileprivate` across many members.
|
||
// We'll refactor into smaller files when the surface stabilizes.
|
||
// swiftlint:disable type_body_length file_length
|
||
@MainActor
|
||
@Observable
|
||
final class TalkModeManager: NSObject {
|
||
private typealias SpeechRequest = SFSpeechAudioBufferRecognitionRequest
|
||
private static let defaultModelIdFallback = "eleven_v3"
|
||
private static let defaultRealtimeModelIdFallback = "gpt-realtime-2"
|
||
private static let defaultTalkProvider = "elevenlabs"
|
||
private static let defaultSilenceTimeoutMs = TalkDefaults.silenceTimeoutMs
|
||
private static let redactedConfigSentinel = "__OPENCLAW_REDACTED__"
|
||
private static let realtimePrefetchExpiryLeewaySeconds: TimeInterval = 30
|
||
var isEnabled: Bool = false
|
||
var isListening: Bool = false
|
||
var isSpeaking: Bool = false
|
||
var isUserSpeechDetected: Bool = false
|
||
var isPushToTalkActive: Bool = false
|
||
var statusText: String = "Off"
|
||
/// 0..1-ish (not calibrated). Intended for UI feedback only.
|
||
var micLevel: Double = 0
|
||
var gatewayTalkConfigLoaded: Bool = false
|
||
var gatewayTalkApiKeyConfigured: Bool = false
|
||
var gatewayTalkDefaultModelId: String?
|
||
var gatewayTalkDefaultVoiceId: String?
|
||
var gatewayTalkProviderLabel: String = "Not loaded"
|
||
var gatewayTalkTransportLabel: String = "Not loaded"
|
||
var gatewayTalkUsesRealtime: Bool = false
|
||
var gatewayTalkUsesRealtimeRelay: Bool = false
|
||
var gatewayTalkRealtimeProviderLabel: String?
|
||
var gatewayTalkRealtimeModelId: String?
|
||
var gatewayTalkRealtimeVoiceId: String?
|
||
var gatewayTalkVoiceModeTitle: String = "Not loaded"
|
||
var gatewayTalkVoiceModeSubtitle: String?
|
||
var gatewayTalkVoiceModeAccessibilityValue: String = "Not loaded"
|
||
var gatewayTalkPermissionState: TalkGatewayPermissionState = .unknown
|
||
|
||
private enum CaptureMode {
|
||
case idle
|
||
case continuous
|
||
case pushToTalk
|
||
}
|
||
|
||
private var captureMode: CaptureMode = .idle
|
||
private var resumeContinuousAfterPTT: Bool = false
|
||
private var activePTTCaptureId: String?
|
||
private var pttAutoStopEnabled: Bool = false
|
||
private var pttCompletion: CheckedContinuation<OpenClawTalkPTTStopPayload, Never>?
|
||
private var pttTimeoutTask: Task<Void, Never>?
|
||
|
||
private let allowSimulatorCapture: Bool
|
||
|
||
private let audioEngine = AVAudioEngine()
|
||
private var inputTapInstalled = false
|
||
private var audioTapDiagnostics: AudioTapDiagnostics?
|
||
private var speechRecognizer: SFSpeechRecognizer?
|
||
private var recognitionRequest: SFSpeechAudioBufferRecognitionRequest?
|
||
private var recognitionTask: SFSpeechRecognitionTask?
|
||
private var silenceTask: Task<Void, Never>?
|
||
private var realtimeSession: TalkRealtimeWebRTCSession?
|
||
private var realtimeRelaySession: RealtimeTalkRelaySession?
|
||
private var prefetchedRealtimeSession: TalkRealtimeClientSession?
|
||
private var realtimePrefetchTask: Task<Void, Never>?
|
||
|
||
private var lastHeard: Date?
|
||
private var lastTranscript: String = ""
|
||
private var loggedPartialThisCycle: Bool = false
|
||
private var lastSpokenText: String?
|
||
private var lastInterruptedAtSeconds: Double?
|
||
|
||
private var defaultVoiceId: String?
|
||
private var currentVoiceId: String?
|
||
private var defaultModelId: String?
|
||
private var currentModelId: String?
|
||
private var voiceOverrideActive = false
|
||
private var modelOverrideActive = false
|
||
private var defaultOutputFormat: String?
|
||
private var activeTalkProvider: String = TalkModeManager.defaultTalkProvider
|
||
private var executionMode: TalkModeExecutionMode = .native
|
||
private var realtimeWebRTCEnabled: Bool = false
|
||
private var realtimeProvider: String?
|
||
private var realtimeModelId: String?
|
||
private var realtimeVoiceId: String?
|
||
private var configuredVoiceModeDescriptor = TalkVoiceModeDescriptor(
|
||
title: "Not loaded",
|
||
subtitle: nil,
|
||
providerId: nil,
|
||
modelId: nil,
|
||
voiceId: nil,
|
||
transport: nil,
|
||
isRealtime: false)
|
||
private var apiKey: String?
|
||
private var voiceAliases: [String: String] = [:]
|
||
private var interruptOnSpeech: Bool = true
|
||
private var gatewaySpeechLocaleID: String?
|
||
private var mainSessionKey: String = "main"
|
||
private var fallbackVoiceId: String?
|
||
private var lastPlaybackWasPCM: Bool = false
|
||
/// Set when the ElevenLabs API rejects PCM format (e.g. 403 subscription_required).
|
||
/// Once set, all subsequent requests in this session use MP3 instead of re-trying PCM.
|
||
private var pcmFormatUnavailable: Bool = false
|
||
var pcmPlayer: PCMStreamingAudioPlaying = PCMStreamingAudioPlayer.shared
|
||
var mp3Player: StreamingAudioPlaying = StreamingAudioPlayer.shared
|
||
|
||
private var gateway: GatewayNodeSession?
|
||
private var gatewayConnected = false
|
||
private var talkConfigLoadedAt: Date?
|
||
private var silenceWindow: TimeInterval = .init(TalkModeManager.defaultSilenceTimeoutMs) / 1000
|
||
private var lastAudioActivity: Date?
|
||
private var noiseFloorSamples: [Double] = []
|
||
private var noiseFloor: Double?
|
||
private var noiseFloorReady: Bool = false
|
||
|
||
private var chatSubscribedSessionKeys = Set<String>()
|
||
private var incrementalSpeechQueue: [String] = []
|
||
private var incrementalSpeechTask: Task<Void, Never>?
|
||
private var incrementalSpeechActive = false
|
||
private var incrementalSpeechUsed = false
|
||
private var incrementalSpeechLanguage: String?
|
||
private var incrementalSpeechBuffer = IncrementalSpeechBuffer()
|
||
private var incrementalSpeechContext: IncrementalSpeechContext?
|
||
private var incrementalSpeechDirective: TalkDirective?
|
||
private var incrementalSpeechPrefetch: IncrementalSpeechPrefetchState?
|
||
private var incrementalSpeechPrefetchMonitorTask: Task<Void, Never>?
|
||
|
||
private let logger = Logger(subsystem: "ai.openclaw", category: "TalkMode")
|
||
|
||
private static func nowSeconds() -> TimeInterval {
|
||
ProcessInfo.processInfo.systemUptime
|
||
}
|
||
|
||
private static func elapsedMs(since start: TimeInterval) -> Int {
|
||
max(0, Int((self.nowSeconds() - start) * 1000))
|
||
}
|
||
|
||
init(allowSimulatorCapture: Bool = false) {
|
||
self.allowSimulatorCapture = allowSimulatorCapture
|
||
super.init()
|
||
}
|
||
|
||
func attachGateway(_ gateway: GatewayNodeSession) {
|
||
self.gateway = gateway
|
||
}
|
||
|
||
func updateGatewayConnected(_ connected: Bool) {
|
||
self.gatewayConnected = connected
|
||
if connected {
|
||
// If talk mode is enabled before the gateway connects (common on cold start),
|
||
// kick recognition once we're online so the UI doesn’t stay “Offline”.
|
||
if self.isEnabled, !self.isListening, self.captureMode != .pushToTalk {
|
||
Task { await self.start() }
|
||
}
|
||
} else {
|
||
self.stopRealtimeSession()
|
||
if self.isEnabled, !self.isSpeaking {
|
||
self.statusText = "Offline"
|
||
}
|
||
self.realtimePrefetchTask?.cancel()
|
||
self.realtimePrefetchTask = nil
|
||
self.prefetchedRealtimeSession = nil
|
||
}
|
||
}
|
||
|
||
func updateMainSessionKey(_ sessionKey: String?) {
|
||
let trimmed = (sessionKey ?? "").trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !trimmed.isEmpty else { return }
|
||
if trimmed == self.mainSessionKey { return }
|
||
self.mainSessionKey = trimmed
|
||
if self.gatewayConnected, self.isEnabled {
|
||
Task { await self.subscribeChatIfNeeded(sessionKey: trimmed) }
|
||
}
|
||
}
|
||
|
||
func setEnabled(_ enabled: Bool) {
|
||
self.isEnabled = enabled
|
||
if enabled {
|
||
self.logger.info("enabled")
|
||
GatewayDiagnostics.log("talk.timeline manager enabled")
|
||
Task { await self.start() }
|
||
} else {
|
||
self.logger.info("disabled")
|
||
GatewayDiagnostics.log("talk.timeline manager disabled")
|
||
self.stop()
|
||
}
|
||
}
|
||
|
||
func applyProviderSelectionChanged() {
|
||
let shouldRestart = self.isEnabled
|
||
if shouldRestart {
|
||
self.stop()
|
||
self.isEnabled = true
|
||
Task { await self.start() }
|
||
} else {
|
||
Task { await self.reloadConfig() }
|
||
}
|
||
}
|
||
|
||
func start() async {
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline manager start enter enabled=\(self.isEnabled) "
|
||
+ "listening=\(self.isListening) gatewayConnected=\(self.gatewayConnected)")
|
||
guard self.isEnabled else { return }
|
||
guard self.captureMode != .pushToTalk else { return }
|
||
if self.isListening { return }
|
||
guard self.gatewayConnected else {
|
||
self.statusText = "Offline"
|
||
GatewayDiagnostics.log("talk.timeline manager start blocked gateway offline")
|
||
return
|
||
}
|
||
|
||
self.logger.info("start")
|
||
self.statusText = "Requesting permissions…"
|
||
let permissionStartedAt = Self.nowSeconds()
|
||
let micOk = await Self.requestMicrophonePermission()
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline microphone permission ok=\(micOk) "
|
||
+ "elapsedMs=\(Self.elapsedMs(since: permissionStartedAt))")
|
||
guard micOk else {
|
||
self.logger.warning("start blocked: microphone permission denied")
|
||
self.statusText = "Microphone permission denied"
|
||
return
|
||
}
|
||
await self.ensureTalkConfigLoadedForStart()
|
||
if self.gatewayTalkPermissionState.requiresTalkPermissionAction {
|
||
self.statusText = "Gateway permission required"
|
||
GatewayDiagnostics.log("talk.timeline manager start blocked gateway permission")
|
||
return
|
||
}
|
||
if self.realtimeWebRTCEnabled {
|
||
let started = self.executionMode == .realtimeRelay
|
||
? await self.startRealtimeRelayIfAvailable()
|
||
: await self.startRealtimeIfAvailable()
|
||
if started {
|
||
return
|
||
}
|
||
}
|
||
|
||
let speechOk = await Self.requestSpeechPermission()
|
||
guard speechOk else {
|
||
self.logger.warning("start blocked: speech permission denied")
|
||
self.statusText = Self.permissionMessage(
|
||
kind: "Speech recognition",
|
||
status: SFSpeechRecognizer.authorizationStatus())
|
||
return
|
||
}
|
||
|
||
do {
|
||
GatewayDiagnostics.log("talk.timeline fallback speech pipeline start")
|
||
try Self.configureAudioSession()
|
||
// Set this before starting recognition so any early speech errors are classified correctly.
|
||
self.captureMode = .continuous
|
||
try self.startRecognition()
|
||
self.isListening = true
|
||
self.statusText = "Listening"
|
||
self.startSilenceMonitor()
|
||
await self.subscribeChatIfNeeded(sessionKey: self.mainSessionKey)
|
||
self.logger.info("listening")
|
||
} catch {
|
||
self.isListening = false
|
||
self.statusText = "Start failed: \(error.localizedDescription)"
|
||
self.logger.error("start failed: \(error.localizedDescription, privacy: .public)")
|
||
}
|
||
}
|
||
|
||
func stop() {
|
||
self.isEnabled = false
|
||
self.isListening = false
|
||
self.isUserSpeechDetected = false
|
||
self.isPushToTalkActive = false
|
||
self.captureMode = .idle
|
||
self.statusText = "Off"
|
||
self.lastTranscript = ""
|
||
self.lastHeard = nil
|
||
self.silenceTask?.cancel()
|
||
self.silenceTask = nil
|
||
self.stopRealtimeSession()
|
||
self.stopRecognition()
|
||
self.stopSpeaking()
|
||
self.lastInterruptedAtSeconds = nil
|
||
let pendingPTT = self.pttCompletion != nil
|
||
let pendingCaptureId = self.activePTTCaptureId ?? UUID().uuidString
|
||
self.pttTimeoutTask?.cancel()
|
||
self.pttTimeoutTask = nil
|
||
self.pttAutoStopEnabled = false
|
||
if pendingPTT {
|
||
let payload = OpenClawTalkPTTStopPayload(
|
||
captureId: pendingCaptureId,
|
||
transcript: nil,
|
||
status: "cancelled")
|
||
self.finishPTTOnce(payload)
|
||
}
|
||
self.resumeContinuousAfterPTT = false
|
||
self.activePTTCaptureId = nil
|
||
TalkSystemSpeechSynthesizer.shared.stop()
|
||
do {
|
||
try AVAudioSession.sharedInstance().setActive(false, options: [.notifyOthersOnDeactivation])
|
||
} catch {
|
||
self.logger.warning("audio session deactivate failed: \(error.localizedDescription, privacy: .public)")
|
||
}
|
||
Task { await self.unsubscribeAllChats() }
|
||
}
|
||
|
||
/// Suspends microphone usage without disabling Talk Mode.
|
||
/// Used when the app backgrounds (or when we need to temporarily release the mic).
|
||
func suspendForBackground(keepActive: Bool = false) -> Bool {
|
||
guard self.isEnabled else { return false }
|
||
if keepActive {
|
||
self.statusText = self.isListening ? "Listening" : self.statusText
|
||
return false
|
||
}
|
||
let wasActive = self.isListening || self.isSpeaking || self.isPushToTalkActive
|
||
|
||
self.isListening = false
|
||
self.isPushToTalkActive = false
|
||
self.captureMode = .idle
|
||
self.statusText = "Paused"
|
||
self.lastTranscript = ""
|
||
self.lastHeard = nil
|
||
self.silenceTask?.cancel()
|
||
self.silenceTask = nil
|
||
|
||
self.stopRealtimeSession()
|
||
self.stopRecognition()
|
||
self.stopSpeaking()
|
||
self.lastInterruptedAtSeconds = nil
|
||
TalkSystemSpeechSynthesizer.shared.stop()
|
||
|
||
do {
|
||
try AVAudioSession.sharedInstance().setActive(false, options: [.notifyOthersOnDeactivation])
|
||
} catch {
|
||
self.logger.warning("audio session deactivate failed: \(error.localizedDescription, privacy: .public)")
|
||
}
|
||
|
||
Task { await self.unsubscribeAllChats() }
|
||
return wasActive
|
||
}
|
||
|
||
func resumeAfterBackground(wasSuspended: Bool, wasKeptActive: Bool = false) async {
|
||
if wasKeptActive { return }
|
||
guard wasSuspended else { return }
|
||
guard self.isEnabled else { return }
|
||
await self.start()
|
||
}
|
||
|
||
func userTappedOrb() {
|
||
if let realtimeSession {
|
||
realtimeSession.cancelResponse()
|
||
}
|
||
self.realtimeRelaySession?.cancelOutput()
|
||
self.stopSpeaking()
|
||
}
|
||
|
||
func beginPushToTalk() async throws -> OpenClawTalkPTTStartPayload {
|
||
guard self.gatewayConnected else {
|
||
self.statusText = "Offline"
|
||
throw NSError(domain: "TalkMode", code: 7, userInfo: [
|
||
NSLocalizedDescriptionKey: "Gateway not connected",
|
||
])
|
||
}
|
||
if self.isPushToTalkActive, let captureId = activePTTCaptureId {
|
||
return OpenClawTalkPTTStartPayload(captureId: captureId)
|
||
}
|
||
|
||
self.stopSpeaking(storeInterruption: false)
|
||
self.pttTimeoutTask?.cancel()
|
||
self.pttTimeoutTask = nil
|
||
self.pttAutoStopEnabled = false
|
||
|
||
self.resumeContinuousAfterPTT = self.isEnabled && self.captureMode == .continuous
|
||
self.silenceTask?.cancel()
|
||
self.silenceTask = nil
|
||
self.stopRecognition()
|
||
self.isListening = false
|
||
self.isUserSpeechDetected = false
|
||
|
||
let captureId = UUID().uuidString
|
||
self.activePTTCaptureId = captureId
|
||
self.lastTranscript = ""
|
||
self.lastHeard = nil
|
||
|
||
self.statusText = "Requesting permissions…"
|
||
if !self.allowSimulatorCapture {
|
||
let micOk = await Self.requestMicrophonePermission()
|
||
guard micOk else {
|
||
self.statusText = "Microphone permission denied"
|
||
throw NSError(domain: "TalkMode", code: 4, userInfo: [
|
||
NSLocalizedDescriptionKey: "Microphone permission denied",
|
||
])
|
||
}
|
||
let speechOk = await Self.requestSpeechPermission()
|
||
guard speechOk else {
|
||
self.statusText = Self.permissionMessage(
|
||
kind: "Speech recognition",
|
||
status: SFSpeechRecognizer.authorizationStatus())
|
||
throw NSError(domain: "TalkMode", code: 5, userInfo: [
|
||
NSLocalizedDescriptionKey: "Speech recognition permission denied",
|
||
])
|
||
}
|
||
}
|
||
|
||
do {
|
||
try Self.configureAudioSession()
|
||
self.captureMode = .pushToTalk
|
||
try self.startRecognition()
|
||
self.isListening = true
|
||
self.isPushToTalkActive = true
|
||
self.statusText = "Listening (PTT)"
|
||
} catch {
|
||
self.isListening = false
|
||
self.isUserSpeechDetected = false
|
||
self.isPushToTalkActive = false
|
||
self.captureMode = .idle
|
||
self.statusText = "Start failed: \(error.localizedDescription)"
|
||
throw error
|
||
}
|
||
|
||
return OpenClawTalkPTTStartPayload(captureId: captureId)
|
||
}
|
||
|
||
func endPushToTalk() async -> OpenClawTalkPTTStopPayload {
|
||
let captureId = self.activePTTCaptureId ?? UUID().uuidString
|
||
guard self.isPushToTalkActive else {
|
||
let payload = OpenClawTalkPTTStopPayload(
|
||
captureId: captureId,
|
||
transcript: nil,
|
||
status: "idle")
|
||
self.finishPTTOnce(payload)
|
||
return payload
|
||
}
|
||
|
||
self.isPushToTalkActive = false
|
||
self.isListening = false
|
||
self.isUserSpeechDetected = false
|
||
self.captureMode = .idle
|
||
self.stopRecognition()
|
||
self.pttTimeoutTask?.cancel()
|
||
self.pttTimeoutTask = nil
|
||
self.pttAutoStopEnabled = false
|
||
|
||
let transcript = self.lastTranscript.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
self.lastTranscript = ""
|
||
self.lastHeard = nil
|
||
|
||
guard !transcript.isEmpty else {
|
||
self.statusText = "Ready"
|
||
if self.resumeContinuousAfterPTT {
|
||
await self.start()
|
||
}
|
||
self.resumeContinuousAfterPTT = false
|
||
self.activePTTCaptureId = nil
|
||
let payload = OpenClawTalkPTTStopPayload(
|
||
captureId: captureId,
|
||
transcript: nil,
|
||
status: "empty")
|
||
self.finishPTTOnce(payload)
|
||
return payload
|
||
}
|
||
|
||
guard self.gatewayConnected else {
|
||
self.statusText = "Gateway not connected"
|
||
if self.resumeContinuousAfterPTT {
|
||
await self.start()
|
||
}
|
||
self.resumeContinuousAfterPTT = false
|
||
self.activePTTCaptureId = nil
|
||
let payload = OpenClawTalkPTTStopPayload(
|
||
captureId: captureId,
|
||
transcript: transcript,
|
||
status: "offline")
|
||
self.finishPTTOnce(payload)
|
||
return payload
|
||
}
|
||
|
||
self.statusText = "Thinking…"
|
||
Task { @MainActor in
|
||
await self.processTranscript(transcript, restartAfter: self.resumeContinuousAfterPTT)
|
||
}
|
||
self.resumeContinuousAfterPTT = false
|
||
self.activePTTCaptureId = nil
|
||
let payload = OpenClawTalkPTTStopPayload(
|
||
captureId: captureId,
|
||
transcript: transcript,
|
||
status: "queued")
|
||
self.finishPTTOnce(payload)
|
||
return payload
|
||
}
|
||
|
||
func runPushToTalkOnce(maxDurationSeconds: TimeInterval = 12) async throws -> OpenClawTalkPTTStopPayload {
|
||
if self.pttCompletion != nil {
|
||
_ = await self.cancelPushToTalk()
|
||
}
|
||
|
||
if self.isPushToTalkActive {
|
||
let captureId = self.activePTTCaptureId ?? UUID().uuidString
|
||
return OpenClawTalkPTTStopPayload(
|
||
captureId: captureId,
|
||
transcript: nil,
|
||
status: "busy")
|
||
}
|
||
|
||
_ = try await self.beginPushToTalk()
|
||
|
||
return await withCheckedContinuation { cont in
|
||
self.pttCompletion = cont
|
||
self.pttAutoStopEnabled = true
|
||
self.startSilenceMonitor()
|
||
self.schedulePTTTimeout(seconds: maxDurationSeconds)
|
||
}
|
||
}
|
||
|
||
func cancelPushToTalk() async -> OpenClawTalkPTTStopPayload {
|
||
let captureId = self.activePTTCaptureId ?? UUID().uuidString
|
||
guard self.isPushToTalkActive else {
|
||
let payload = OpenClawTalkPTTStopPayload(
|
||
captureId: captureId,
|
||
transcript: nil,
|
||
status: "idle")
|
||
self.finishPTTOnce(payload)
|
||
self.pttAutoStopEnabled = false
|
||
self.pttTimeoutTask?.cancel()
|
||
self.pttTimeoutTask = nil
|
||
self.resumeContinuousAfterPTT = false
|
||
self.activePTTCaptureId = nil
|
||
return payload
|
||
}
|
||
|
||
let shouldResume = self.resumeContinuousAfterPTT
|
||
self.isPushToTalkActive = false
|
||
self.isListening = false
|
||
self.captureMode = .idle
|
||
self.stopRecognition()
|
||
self.lastTranscript = ""
|
||
self.lastHeard = nil
|
||
self.pttAutoStopEnabled = false
|
||
self.pttTimeoutTask?.cancel()
|
||
self.pttTimeoutTask = nil
|
||
self.resumeContinuousAfterPTT = false
|
||
self.activePTTCaptureId = nil
|
||
self.statusText = "Ready"
|
||
|
||
let payload = OpenClawTalkPTTStopPayload(
|
||
captureId: captureId,
|
||
transcript: nil,
|
||
status: "cancelled")
|
||
self.finishPTTOnce(payload)
|
||
|
||
if shouldResume {
|
||
await self.start()
|
||
}
|
||
return payload
|
||
}
|
||
|
||
private func startRecognition() throws {
|
||
#if targetEnvironment(simulator)
|
||
if self.allowSimulatorCapture {
|
||
self.recognitionRequest = SFSpeechAudioBufferRecognitionRequest()
|
||
self.recognitionRequest?.shouldReportPartialResults = true
|
||
return
|
||
}
|
||
if !self.allowSimulatorCapture {
|
||
throw NSError(domain: "TalkMode", code: 2, userInfo: [
|
||
NSLocalizedDescriptionKey: "Talk mode is not supported on the iOS simulator",
|
||
])
|
||
}
|
||
#endif
|
||
|
||
self.stopRecognition()
|
||
let localSpeechLocale = UserDefaults.standard.string(forKey: TalkSpeechLocale.storageKey)
|
||
let resolvedSpeech = TalkSpeechLocale.makeRecognizer(
|
||
localSelection: localSpeechLocale,
|
||
gatewaySelection: self.gatewaySpeechLocaleID)
|
||
self.speechRecognizer = resolvedSpeech.recognizer
|
||
guard let recognizer = speechRecognizer else {
|
||
throw NSError(domain: "TalkMode", code: 1, userInfo: [
|
||
NSLocalizedDescriptionKey: "Speech recognizer unavailable",
|
||
])
|
||
}
|
||
GatewayDiagnostics.log("talk speech: locale=\(resolvedSpeech.localeID ?? "default")")
|
||
|
||
self.recognitionRequest = SFSpeechAudioBufferRecognitionRequest()
|
||
self.recognitionRequest?.shouldReportPartialResults = true
|
||
self.recognitionRequest?.taskHint = .dictation
|
||
guard let request = recognitionRequest else { return }
|
||
|
||
GatewayDiagnostics.log("talk audio: session \(Self.describeAudioSession())")
|
||
|
||
let input = self.audioEngine.inputNode
|
||
let format = input.inputFormat(forBus: 0)
|
||
guard format.sampleRate > 0, format.channelCount > 0 else {
|
||
throw NSError(domain: "TalkMode", code: 3, userInfo: [
|
||
NSLocalizedDescriptionKey: "Invalid audio input format",
|
||
])
|
||
}
|
||
input.removeTap(onBus: 0)
|
||
let tapDiagnostics = AudioTapDiagnostics(label: "talk") { [weak self] level in
|
||
guard let self else { return }
|
||
Task { @MainActor in
|
||
// Smooth + clamp for UI, and keep it cheap.
|
||
let raw = max(0, min(Double(level) * 10.0, 1.0))
|
||
let next = (self.micLevel * 0.80) + (raw * 0.20)
|
||
self.micLevel = next
|
||
|
||
// Dynamic thresholding so background noise doesn’t prevent endpointing.
|
||
if self.isListening, !self.isSpeaking, !self.noiseFloorReady {
|
||
self.noiseFloorSamples.append(raw)
|
||
if self.noiseFloorSamples.count >= 22 {
|
||
let sorted = self.noiseFloorSamples.sorted()
|
||
let take = max(6, sorted.count / 2)
|
||
let slice = sorted.prefix(take)
|
||
let avg = slice.reduce(0.0, +) / Double(slice.count)
|
||
self.noiseFloor = avg
|
||
self.noiseFloorReady = true
|
||
self.noiseFloorSamples.removeAll(keepingCapacity: true)
|
||
let threshold = min(0.35, max(0.12, avg + 0.10))
|
||
GatewayDiagnostics.log(
|
||
"talk audio: noiseFloor=\(String(format: "%.3f", avg)) "
|
||
+ "threshold=\(String(format: "%.3f", threshold))")
|
||
}
|
||
}
|
||
|
||
let threshold: Double = if let floor = self.noiseFloor, self.noiseFloorReady {
|
||
min(0.35, max(0.12, floor + 0.10))
|
||
} else {
|
||
0.18
|
||
}
|
||
if raw >= threshold {
|
||
self.lastAudioActivity = Date()
|
||
}
|
||
}
|
||
}
|
||
self.audioTapDiagnostics = tapDiagnostics
|
||
let tapBlock = Self.makeAudioTapAppendCallback(request: request, diagnostics: tapDiagnostics)
|
||
input.installTap(onBus: 0, bufferSize: 2048, format: format, block: tapBlock)
|
||
self.inputTapInstalled = true
|
||
|
||
self.audioEngine.prepare()
|
||
try self.audioEngine.start()
|
||
self.loggedPartialThisCycle = false
|
||
|
||
GatewayDiagnostics.log(
|
||
"talk speech: recognition started mode=\(String(describing: self.captureMode)) "
|
||
+ "engineRunning=\(self.audioEngine.isRunning)")
|
||
self.recognitionTask = recognizer.recognitionTask(with: request) { [weak self] result, error in
|
||
guard let self else { return }
|
||
if let error {
|
||
let msg = error.localizedDescription
|
||
let lowered = msg.lowercased()
|
||
let isCancellation = lowered.contains("cancelled") || lowered.contains("canceled")
|
||
if isCancellation {
|
||
GatewayDiagnostics.log("talk speech: cancelled")
|
||
if self.captureMode == .continuous, self.isEnabled, !self.isSpeaking {
|
||
self.statusText = "Listening"
|
||
}
|
||
self.logger.debug("speech recognition cancelled")
|
||
return
|
||
}
|
||
GatewayDiagnostics.log("talk speech: error=\(msg)")
|
||
if !self.isSpeaking {
|
||
if msg.localizedCaseInsensitiveContains("no speech detected") {
|
||
// Treat as transient silence. Don't scare users with an error banner.
|
||
self.statusText = self.isEnabled ? "Listening" : "Speech error: \(msg)"
|
||
} else {
|
||
self.statusText = "Speech error: \(msg)"
|
||
}
|
||
}
|
||
self.logger.debug("speech recognition error: \(msg, privacy: .public)")
|
||
// Speech recognition can terminate on transient errors (e.g. no speech detected).
|
||
// If talk mode is enabled and we're in continuous capture, try to restart.
|
||
if self.captureMode == .continuous, self.isEnabled, !self.isSpeaking {
|
||
// Treat the task as terminal on error so we don't get stuck with a dead recognizer.
|
||
self.stopRecognition()
|
||
Task { @MainActor [weak self] in
|
||
await self?.restartRecognitionAfterError()
|
||
}
|
||
}
|
||
}
|
||
guard let result else { return }
|
||
let transcript = result.bestTranscription.formattedString
|
||
if !result.isFinal, !self.loggedPartialThisCycle {
|
||
let trimmed = transcript.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
if !trimmed.isEmpty {
|
||
self.loggedPartialThisCycle = true
|
||
GatewayDiagnostics.log("talk speech: partial chars=\(trimmed.count)")
|
||
}
|
||
}
|
||
Task { @MainActor in
|
||
await self.handleTranscript(transcript: transcript, isFinal: result.isFinal)
|
||
}
|
||
}
|
||
}
|
||
|
||
private func restartRecognitionAfterError() async {
|
||
guard self.isEnabled, self.captureMode == .continuous else { return }
|
||
// Avoid thrashing the audio engine if it’s already running.
|
||
if self.recognitionTask != nil, self.audioEngine.isRunning { return }
|
||
try? await Task.sleep(nanoseconds: 250_000_000)
|
||
guard self.isEnabled, self.captureMode == .continuous else { return }
|
||
do {
|
||
try Self.configureAudioSession()
|
||
try self.startRecognition()
|
||
self.isListening = true
|
||
if self.statusText.localizedCaseInsensitiveContains("speech error") {
|
||
self.statusText = "Listening"
|
||
}
|
||
GatewayDiagnostics.log("talk speech: recognition restarted")
|
||
} catch {
|
||
let msg = error.localizedDescription
|
||
GatewayDiagnostics.log("talk speech: restart failed error=\(msg)")
|
||
}
|
||
}
|
||
|
||
private func stopRecognition() {
|
||
self.recognitionTask?.cancel()
|
||
self.recognitionTask = nil
|
||
self.recognitionRequest?.endAudio()
|
||
self.recognitionRequest = nil
|
||
self.micLevel = 0
|
||
self.lastAudioActivity = nil
|
||
self.noiseFloorSamples.removeAll(keepingCapacity: true)
|
||
self.noiseFloor = nil
|
||
self.noiseFloorReady = false
|
||
self.audioTapDiagnostics = nil
|
||
if self.inputTapInstalled {
|
||
self.audioEngine.inputNode.removeTap(onBus: 0)
|
||
self.inputTapInstalled = false
|
||
}
|
||
self.audioEngine.stop()
|
||
self.speechRecognizer = nil
|
||
}
|
||
|
||
private nonisolated static func makeAudioTapAppendCallback(
|
||
request: SpeechRequest,
|
||
diagnostics: AudioTapDiagnostics) -> AVAudioNodeTapBlock
|
||
{
|
||
{ buffer, _ in
|
||
request.append(buffer)
|
||
diagnostics.onBuffer(buffer)
|
||
}
|
||
}
|
||
|
||
private func handleTranscript(transcript: String, isFinal: Bool) async {
|
||
let trimmed = transcript.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
let ttsActive = self.isSpeechOutputActive
|
||
if ttsActive, self.interruptOnSpeech {
|
||
if self.shouldInterrupt(with: trimmed) {
|
||
self.stopSpeaking()
|
||
}
|
||
return
|
||
}
|
||
|
||
guard self.isListening else { return }
|
||
if !trimmed.isEmpty {
|
||
self.lastTranscript = trimmed
|
||
self.lastHeard = Date()
|
||
}
|
||
if isFinal {
|
||
self.lastTranscript = trimmed
|
||
guard !trimmed.isEmpty else { return }
|
||
GatewayDiagnostics.log("talk speech: final transcript chars=\(trimmed.count)")
|
||
self.loggedPartialThisCycle = false
|
||
if self.captureMode == .pushToTalk, self.pttAutoStopEnabled, self.isPushToTalkActive {
|
||
_ = await self.endPushToTalk()
|
||
return
|
||
}
|
||
if self.captureMode == .continuous, !self.isSpeechOutputActive {
|
||
await self.processTranscript(trimmed, restartAfter: true)
|
||
}
|
||
}
|
||
}
|
||
|
||
private func startSilenceMonitor() {
|
||
self.silenceTask?.cancel()
|
||
self.silenceTask = Task { [weak self] in
|
||
guard let self else { return }
|
||
while self.isEnabled || (self.isPushToTalkActive && self.pttAutoStopEnabled) {
|
||
try? await Task.sleep(nanoseconds: 200_000_000)
|
||
await self.checkSilence()
|
||
}
|
||
}
|
||
}
|
||
|
||
private func checkSilence() async {
|
||
if self.captureMode == .continuous {
|
||
guard self.isListening, !self.isSpeechOutputActive else { return }
|
||
let transcript = self.lastTranscript.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !transcript.isEmpty else { return }
|
||
let lastActivity = [lastHeard, lastAudioActivity].compactMap(\.self).max()
|
||
guard let lastActivity else { return }
|
||
if Date().timeIntervalSince(lastActivity) < self.silenceWindow { return }
|
||
await self.processTranscript(transcript, restartAfter: true)
|
||
return
|
||
}
|
||
|
||
guard self.captureMode == .pushToTalk, self.pttAutoStopEnabled else { return }
|
||
guard self.isListening, !self.isSpeaking, self.isPushToTalkActive else { return }
|
||
let transcript = self.lastTranscript.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !transcript.isEmpty else { return }
|
||
let lastActivity = [lastHeard, lastAudioActivity].compactMap(\.self).max()
|
||
guard let lastActivity else { return }
|
||
if Date().timeIntervalSince(lastActivity) < self.silenceWindow { return }
|
||
_ = await self.endPushToTalk()
|
||
}
|
||
|
||
/// Guardrail for PTT once so we don't stay open indefinitely.
|
||
private func schedulePTTTimeout(seconds: TimeInterval) {
|
||
guard seconds > 0 else { return }
|
||
let nanos = UInt64(seconds * 1_000_000_000)
|
||
self.pttTimeoutTask?.cancel()
|
||
self.pttTimeoutTask = Task { [weak self] in
|
||
try? await Task.sleep(nanoseconds: nanos)
|
||
await self?.handlePTTTimeout()
|
||
}
|
||
}
|
||
|
||
private func handlePTTTimeout() async {
|
||
guard self.pttAutoStopEnabled, self.isPushToTalkActive else { return }
|
||
_ = await self.endPushToTalk()
|
||
}
|
||
|
||
private func finishPTTOnce(_ payload: OpenClawTalkPTTStopPayload) {
|
||
guard let continuation = pttCompletion else { return }
|
||
self.pttCompletion = nil
|
||
continuation.resume(returning: payload)
|
||
}
|
||
|
||
private func processTranscript(_ transcript: String, restartAfter: Bool) async {
|
||
self.isListening = false
|
||
self.isUserSpeechDetected = false
|
||
self.captureMode = .idle
|
||
self.statusText = "Thinking…"
|
||
self.lastTranscript = ""
|
||
self.lastHeard = nil
|
||
self.stopRecognition()
|
||
|
||
GatewayDiagnostics.log("talk: process transcript chars=\(transcript.count) restartAfter=\(restartAfter)")
|
||
await reloadConfig()
|
||
let prompt = self.buildPrompt(transcript: transcript)
|
||
guard self.gatewayConnected, let gateway else {
|
||
self.statusText = "Gateway not connected"
|
||
self.logger.warning("finalize: gateway not connected")
|
||
GatewayDiagnostics.log("talk: abort gateway not connected")
|
||
if restartAfter {
|
||
await self.start()
|
||
}
|
||
return
|
||
}
|
||
|
||
do {
|
||
let startedAt = Date().timeIntervalSince1970
|
||
let sessionKey = self.mainSessionKey
|
||
await self.subscribeChatIfNeeded(sessionKey: sessionKey)
|
||
self.logger.info(
|
||
"chat.send start sessionKey=\(sessionKey, privacy: .public) chars=\(prompt.count, privacy: .public)")
|
||
GatewayDiagnostics.log("talk: chat.send start sessionKey=\(sessionKey) chars=\(prompt.count)")
|
||
let runId = try await sendChat(prompt, gateway: gateway)
|
||
self.logger.info("chat.send ok runId=\(runId, privacy: .public)")
|
||
GatewayDiagnostics.log("talk: chat.send ok runId=\(runId)")
|
||
let shouldIncremental = self.shouldUseIncrementalTTS()
|
||
var streamingTask: Task<Void, Never>?
|
||
if shouldIncremental {
|
||
self.resetIncrementalSpeech()
|
||
streamingTask = Task { @MainActor [weak self] in
|
||
guard let self else { return }
|
||
await self.streamAssistant(runId: runId, gateway: gateway)
|
||
}
|
||
}
|
||
let completion = await waitForChatCompletion(runId: runId, gateway: gateway, timeoutSeconds: 120)
|
||
if completion.state == .timeout {
|
||
self.logger.warning(
|
||
"chat completion timeout runId=\(runId, privacy: .public); attempting history fallback")
|
||
GatewayDiagnostics.log("talk: chat completion timeout runId=\(runId)")
|
||
} else if completion.state == .aborted {
|
||
self.statusText = "Aborted"
|
||
self.logger.warning("chat completion aborted runId=\(runId, privacy: .public)")
|
||
GatewayDiagnostics.log("talk: chat completion aborted runId=\(runId)")
|
||
streamingTask?.cancel()
|
||
await self.finishIncrementalSpeech()
|
||
await self.start()
|
||
return
|
||
} else if completion.state == .error {
|
||
self.statusText = "Chat error"
|
||
self.logger.warning("chat completion error runId=\(runId, privacy: .public)")
|
||
GatewayDiagnostics.log("talk: chat completion error runId=\(runId)")
|
||
streamingTask?.cancel()
|
||
await self.finishIncrementalSpeech()
|
||
await self.start()
|
||
return
|
||
}
|
||
|
||
var assistantText = completion.assistantText
|
||
if assistantText == nil, shouldIncremental {
|
||
let fallback = self.incrementalSpeechBuffer.latestText
|
||
if !fallback.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty {
|
||
assistantText = fallback
|
||
}
|
||
}
|
||
if assistantText == nil {
|
||
assistantText = try await self.waitForAssistantTextFromHistory(
|
||
gateway: gateway,
|
||
since: startedAt,
|
||
timeoutSeconds: completion.state == .final ? 12 : 25)
|
||
}
|
||
guard let assistantText else {
|
||
self.statusText = "No reply"
|
||
self.logger.warning("assistant text timeout runId=\(runId, privacy: .public)")
|
||
GatewayDiagnostics.log("talk: assistant text timeout runId=\(runId)")
|
||
streamingTask?.cancel()
|
||
await self.finishIncrementalSpeech()
|
||
await self.start()
|
||
return
|
||
}
|
||
self.logger.info("assistant text ok chars=\(assistantText.count, privacy: .public)")
|
||
GatewayDiagnostics.log("talk: assistant text ok chars=\(assistantText.count)")
|
||
streamingTask?.cancel()
|
||
if shouldIncremental {
|
||
await self.handleIncrementalAssistantFinal(text: assistantText)
|
||
} else {
|
||
await self.playAssistant(text: assistantText)
|
||
}
|
||
} catch {
|
||
self.statusText = "Talk failed: \(error.localizedDescription)"
|
||
self.logger.error("finalize failed: \(error.localizedDescription, privacy: .public)")
|
||
GatewayDiagnostics.log("talk: failed error=\(error.localizedDescription)")
|
||
}
|
||
|
||
if restartAfter {
|
||
await self.start()
|
||
}
|
||
}
|
||
|
||
private func startRealtimeIfAvailable() async -> Bool {
|
||
guard let gateway else { return false }
|
||
let startedAt = Self.nowSeconds()
|
||
if self.prefetchedRealtimeSession == nil, let prefetchTask = self.realtimePrefetchTask {
|
||
GatewayDiagnostics.log("talk.timeline realtime awaiting in-flight prefetch")
|
||
await prefetchTask.value
|
||
}
|
||
let prefetchedSession = self.consumePrefetchedRealtimeSession()
|
||
GatewayDiagnostics.log("talk.timeline realtime start attempt sessionKey=\(self.mainSessionKey)")
|
||
let session = TalkRealtimeWebRTCSession(
|
||
gateway: gateway,
|
||
sessionKey: mainSessionKey,
|
||
delegate: self)
|
||
self.realtimeSession = session
|
||
do {
|
||
try await session.start(
|
||
provider: self.realtimeProvider,
|
||
model: self.realtimeModelId,
|
||
voice: self.realtimeVoiceId,
|
||
prefetchedSession: prefetchedSession)
|
||
guard self.realtimeSession === session, self.isEnabled else {
|
||
session.stop()
|
||
return true
|
||
}
|
||
self.isListening = true
|
||
self.captureMode = .continuous
|
||
self.statusText = "Listening"
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline realtime start ready elapsedMs=\(Self.elapsedMs(since: startedAt))")
|
||
GatewayDiagnostics.log("talk realtime: started direct OpenAI WebRTC session")
|
||
return true
|
||
} catch {
|
||
guard self.realtimeSession === session, self.isEnabled else {
|
||
session.stop()
|
||
return true
|
||
}
|
||
self.stopRealtimeSession()
|
||
GatewayDiagnostics
|
||
.log("talk realtime: unavailable; falling back to speech pipeline error=\(error.localizedDescription)")
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline realtime start failed elapsedMs=\(Self.elapsedMs(since: startedAt)) "
|
||
+ "error=\(error.localizedDescription)")
|
||
return false
|
||
}
|
||
}
|
||
|
||
private func startRealtimeRelayIfAvailable() async -> Bool {
|
||
guard let gateway else { return false }
|
||
GatewayDiagnostics.log("talk.timeline realtime relay start attempt sessionKey=\(self.mainSessionKey)")
|
||
let startedAt = Self.nowSeconds()
|
||
let relaySession = RealtimeTalkRelaySession(
|
||
gateway: gateway,
|
||
options: RealtimeTalkRelaySession.Options(
|
||
sessionKey: self.mainSessionKey,
|
||
provider: self.realtimeProvider,
|
||
model: self.realtimeModelId,
|
||
voice: self.realtimeVoiceId),
|
||
pcmPlayer: self.pcmPlayer,
|
||
onStatus: { [weak self] status in
|
||
guard let self else { return }
|
||
self.statusText = status
|
||
self.isListening = status.localizedCaseInsensitiveContains("listening")
|
||
if status.localizedCaseInsensitiveContains("thinking") {
|
||
self.isListening = false
|
||
self.isSpeaking = false
|
||
self.isUserSpeechDetected = false
|
||
}
|
||
},
|
||
onSpeakingChanged: { [weak self] speaking in
|
||
guard let self else { return }
|
||
self.isSpeaking = speaking
|
||
if speaking {
|
||
self.isListening = false
|
||
}
|
||
})
|
||
self.realtimeRelaySession = relaySession
|
||
do {
|
||
try Self.configureAudioSession()
|
||
try await relaySession.start()
|
||
guard self.realtimeRelaySession === relaySession, self.isEnabled else {
|
||
relaySession.stop()
|
||
return true
|
||
}
|
||
self.isListening = true
|
||
self.captureMode = .continuous
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline realtime relay start ready elapsedMs=\(Self.elapsedMs(since: startedAt))")
|
||
return true
|
||
} catch {
|
||
guard self.realtimeRelaySession === relaySession, self.isEnabled else {
|
||
relaySession.stop()
|
||
return true
|
||
}
|
||
self.realtimeRelaySession = nil
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline realtime relay start failed elapsedMs=\(Self.elapsedMs(since: startedAt)) "
|
||
+ "error=\(error.localizedDescription)")
|
||
return false
|
||
}
|
||
}
|
||
|
||
func prefetchRealtimeSessionIfReady(reason: String) async {
|
||
guard self.gatewayConnected,
|
||
self.realtimeSession == nil,
|
||
self.realtimeRelaySession == nil,
|
||
!self.isEnabled
|
||
else { return }
|
||
guard self.realtimeWebRTCEnabled, self.executionMode != .realtimeRelay else { return }
|
||
guard self.gatewayTalkPermissionState == .ready else { return }
|
||
guard self.consumePrefetchedRealtimeSession(peekOnly: true) == nil else { return }
|
||
guard self.realtimePrefetchTask == nil else { return }
|
||
|
||
GatewayDiagnostics.log("talk.timeline realtime prefetch scheduled reason=\(reason)")
|
||
self.realtimePrefetchTask = Task { @MainActor [weak self] in
|
||
guard let self else { return }
|
||
let startedAt = Self.nowSeconds()
|
||
do {
|
||
let session = try await self.createRealtimeClientSession(
|
||
provider: self.realtimeProvider,
|
||
model: self.realtimeModelId,
|
||
voice: self.realtimeVoiceId)
|
||
guard !Task.isCancelled else { return }
|
||
self.prefetchedRealtimeSession = session
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline realtime prefetch ready elapsedMs=\(Self.elapsedMs(since: startedAt)) "
|
||
+ "model=\(session.model ?? "unknown") voice=\(session.voice ?? "unknown")")
|
||
} catch {
|
||
guard !Task.isCancelled else { return }
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline realtime prefetch failed elapsedMs=\(Self.elapsedMs(since: startedAt)) "
|
||
+ "error=\(error.localizedDescription)")
|
||
}
|
||
self.realtimePrefetchTask = nil
|
||
}
|
||
}
|
||
|
||
private func createRealtimeClientSession(
|
||
provider: String?,
|
||
model: String?,
|
||
voice: String?) async throws -> TalkRealtimeClientSession
|
||
{
|
||
guard let gateway else {
|
||
throw NSError(domain: "TalkMode", code: 8, userInfo: [
|
||
NSLocalizedDescriptionKey: "Gateway not connected",
|
||
])
|
||
}
|
||
let params = TalkRealtimeClientCreateParams(provider: provider, model: model, voice: voice)
|
||
let data = try JSONEncoder().encode(params)
|
||
let json = String(data: data, encoding: .utf8)
|
||
let res = try await gateway.request(method: "talk.client.create", paramsJSON: json, timeoutSeconds: 12)
|
||
return try JSONDecoder().decode(TalkRealtimeClientSession.self, from: res)
|
||
}
|
||
|
||
private func consumePrefetchedRealtimeSession(peekOnly: Bool = false) -> TalkRealtimeClientSession? {
|
||
guard let session = self.prefetchedRealtimeSession else { return nil }
|
||
if let expiresAt = session.expiresAt {
|
||
let usableUntil = expiresAt - Self.realtimePrefetchExpiryLeewaySeconds
|
||
if Date().timeIntervalSince1970 >= usableUntil {
|
||
GatewayDiagnostics.log("talk.timeline realtime prefetched session expired")
|
||
self.prefetchedRealtimeSession = nil
|
||
return nil
|
||
}
|
||
}
|
||
if !peekOnly {
|
||
self.prefetchedRealtimeSession = nil
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline realtime using prefetched session model=\(session.model ?? "unknown") "
|
||
+ "voice=\(session.voice ?? "unknown")")
|
||
}
|
||
return session
|
||
}
|
||
|
||
private func stopRealtimeSession() {
|
||
self.realtimeSession?.stop()
|
||
self.realtimeSession = nil
|
||
self.realtimeRelaySession?.stop()
|
||
self.realtimeRelaySession = nil
|
||
}
|
||
|
||
private func subscribeChatIfNeeded(sessionKey: String) async {
|
||
let key = sessionKey.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !key.isEmpty else { return }
|
||
guard !self.chatSubscribedSessionKeys.contains(key) else { return }
|
||
|
||
// Operator clients receive chat events without node-style subscriptions.
|
||
self.chatSubscribedSessionKeys.insert(key)
|
||
}
|
||
|
||
private func unsubscribeAllChats() async {
|
||
self.chatSubscribedSessionKeys.removeAll()
|
||
}
|
||
|
||
private func buildPrompt(transcript: String) -> String {
|
||
let interrupted = self.lastInterruptedAtSeconds
|
||
self.lastInterruptedAtSeconds = nil
|
||
return TalkPromptBuilder.build(
|
||
transcript: transcript,
|
||
interruptedAtSeconds: interrupted,
|
||
includeVoiceDirectiveHint: false)
|
||
}
|
||
|
||
private enum ChatCompletionState: CustomStringConvertible {
|
||
case final
|
||
case aborted
|
||
case error
|
||
case timeout
|
||
|
||
var description: String {
|
||
switch self {
|
||
case .final: "final"
|
||
case .aborted: "aborted"
|
||
case .error: "error"
|
||
case .timeout: "timeout"
|
||
}
|
||
}
|
||
}
|
||
|
||
private struct ChatCompletionResult {
|
||
var state: ChatCompletionState
|
||
var assistantText: String?
|
||
}
|
||
|
||
private func sendChat(_ message: String, gateway: GatewayNodeSession) async throws -> String {
|
||
struct SendResponse: Decodable { let runId: String }
|
||
let payload: [String: Any] = [
|
||
"sessionKey": mainSessionKey,
|
||
"message": message,
|
||
"thinking": "low",
|
||
"timeoutMs": 30000,
|
||
"idempotencyKey": UUID().uuidString,
|
||
]
|
||
let data = try JSONSerialization.data(withJSONObject: payload)
|
||
guard let json = String(bytes: data, encoding: .utf8) else {
|
||
throw NSError(
|
||
domain: "TalkModeManager",
|
||
code: 1,
|
||
userInfo: [NSLocalizedDescriptionKey: "Failed to encode chat payload"])
|
||
}
|
||
let res = try await gateway.request(method: "chat.send", paramsJSON: json, timeoutSeconds: 30)
|
||
let decoded = try JSONDecoder().decode(SendResponse.self, from: res)
|
||
return decoded.runId
|
||
}
|
||
|
||
private func waitForChatCompletion(
|
||
runId: String,
|
||
gateway: GatewayNodeSession,
|
||
timeoutSeconds: Int = 120) async -> ChatCompletionResult
|
||
{
|
||
let stream = await gateway.subscribeServerEvents(bufferingNewest: 200)
|
||
return await withTaskGroup(of: ChatCompletionResult.self) { group in
|
||
group.addTask { [runId] in
|
||
var latestAssistantText: String?
|
||
for await evt in stream {
|
||
if Task.isCancelled {
|
||
return ChatCompletionResult(state: .timeout, assistantText: latestAssistantText)
|
||
}
|
||
guard evt.event == "chat", let payload = evt.payload else { continue }
|
||
guard let chatEvent = try? GatewayPayloadDecoding.decode(
|
||
payload,
|
||
as: OpenClawChatEventPayload.self)
|
||
else {
|
||
continue
|
||
}
|
||
guard chatEvent.runId == runId else { continue }
|
||
if let text = OpenClawChatEventText.assistantText(from: chatEvent) {
|
||
latestAssistantText = text
|
||
}
|
||
switch chatEvent.state {
|
||
case "final":
|
||
return ChatCompletionResult(state: .final, assistantText: latestAssistantText)
|
||
case "aborted":
|
||
return ChatCompletionResult(state: .aborted, assistantText: nil)
|
||
case "error":
|
||
return ChatCompletionResult(state: .error, assistantText: nil)
|
||
default:
|
||
break
|
||
}
|
||
}
|
||
return ChatCompletionResult(state: .timeout, assistantText: latestAssistantText)
|
||
}
|
||
group.addTask {
|
||
try? await Task.sleep(nanoseconds: UInt64(timeoutSeconds) * 1_000_000_000)
|
||
return ChatCompletionResult(state: .timeout, assistantText: nil)
|
||
}
|
||
let result = await group.next() ?? ChatCompletionResult(state: .timeout, assistantText: nil)
|
||
group.cancelAll()
|
||
return result
|
||
}
|
||
}
|
||
|
||
private func waitForAssistantTextFromHistory(
|
||
gateway: GatewayNodeSession,
|
||
since: Double,
|
||
timeoutSeconds: Int) async throws -> String?
|
||
{
|
||
let deadline = Date().addingTimeInterval(TimeInterval(timeoutSeconds))
|
||
while Date() < deadline {
|
||
if let text = try await fetchLatestAssistantText(gateway: gateway, since: since) {
|
||
return text
|
||
}
|
||
try? await Task.sleep(nanoseconds: 300_000_000)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
private func fetchLatestAssistantText(gateway: GatewayNodeSession, since: Double? = nil) async throws -> String? {
|
||
let res = try await gateway.request(
|
||
method: "chat.history",
|
||
paramsJSON: "{\"sessionKey\":\"\(self.mainSessionKey)\"}",
|
||
timeoutSeconds: 15)
|
||
guard let json = try JSONSerialization.jsonObject(with: res) as? [String: Any] else { return nil }
|
||
guard let messages = json["messages"] as? [[String: Any]] else { return nil }
|
||
for msg in messages.reversed() {
|
||
guard (msg["role"] as? String) == "assistant" else { continue }
|
||
if let since, let timestamp = msg["timestamp"] as? Double,
|
||
TalkHistoryTimestamp.isAfter(timestamp, sinceSeconds: since) == false
|
||
{
|
||
continue
|
||
}
|
||
guard let content = msg["content"] as? [[String: Any]] else { continue }
|
||
let text = content.compactMap { $0["text"] as? String }.joined(separator: "\n")
|
||
let trimmed = text.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
if !trimmed.isEmpty { return trimmed }
|
||
}
|
||
return nil
|
||
}
|
||
|
||
private func playAssistant(text: String) async {
|
||
let parsed = TalkDirectiveParser.parse(text)
|
||
let directive = parsed.directive
|
||
let cleaned = parsed.stripped.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !cleaned.isEmpty else { return }
|
||
self.applyDirective(directive)
|
||
|
||
self.statusText = "Generating voice…"
|
||
self.isSpeaking = true
|
||
self.lastSpokenText = cleaned
|
||
|
||
do {
|
||
let started = Date()
|
||
let language = ElevenLabsTTSClient.validatedLanguage(directive?.language)
|
||
let requestedVoice = directive?.voiceId?.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
let resolvedVoice = resolveVoiceAlias(requestedVoice)
|
||
if requestedVoice?.isEmpty == false, resolvedVoice == nil {
|
||
self.logger.warning("unknown voice alias \(requestedVoice ?? "?", privacy: .public)")
|
||
}
|
||
|
||
let configuredKey = self.apiKey?
|
||
.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
.isEmpty == false ? self.apiKey : nil
|
||
#if DEBUG
|
||
let resolvedKey = configuredKey ?? ProcessInfo.processInfo.environment["ELEVENLABS_API_KEY"]
|
||
#else
|
||
let resolvedKey = configuredKey
|
||
#endif
|
||
let apiKey = resolvedKey?.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
let preferredVoice = resolvedVoice ?? self.currentVoiceId ?? self.defaultVoiceId
|
||
let voiceId: String? = if let apiKey, !apiKey.isEmpty {
|
||
await resolveVoiceId(preferred: preferredVoice, apiKey: apiKey)
|
||
} else {
|
||
nil
|
||
}
|
||
let canUseElevenLabs = (voiceId?.isEmpty == false) && (apiKey?.isEmpty == false)
|
||
|
||
if canUseElevenLabs, let voiceId, let apiKey {
|
||
GatewayDiagnostics.log("talk tts: provider=elevenlabs voiceId=\(voiceId)")
|
||
let modelId = directive?.modelId ?? self.currentModelId ?? self.defaultModelId
|
||
self.applyVoiceModeDescriptor(TalkVoiceModeDescriptorBuilder.build(
|
||
providerId: "elevenlabs",
|
||
providerLabel: Self.displayName(forProvider: "elevenlabs"),
|
||
modelId: modelId,
|
||
voiceId: voiceId,
|
||
transport: "native",
|
||
isRealtime: false))
|
||
let desiredOutputFormat = (directive?.outputFormat ?? self.defaultOutputFormat)?
|
||
.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
let requestedOutputFormat = (desiredOutputFormat?.isEmpty == false) ? desiredOutputFormat : nil
|
||
let outputFormat = ElevenLabsTTSClient.validatedOutputFormat(
|
||
requestedOutputFormat ?? self.effectiveDefaultOutputFormat)
|
||
if outputFormat == nil, let requestedOutputFormat {
|
||
self.logger.warning(
|
||
"talk output_format unsupported for local playback: \(requestedOutputFormat, privacy: .public)")
|
||
}
|
||
|
||
if let modelId {
|
||
GatewayDiagnostics.log("talk tts: modelId=\(modelId)")
|
||
}
|
||
func makeRequest(outputFormat: String?) -> ElevenLabsTTSRequest {
|
||
ElevenLabsTTSRequest(
|
||
text: cleaned,
|
||
modelId: modelId,
|
||
outputFormat: outputFormat,
|
||
speed: TalkTTSValidation.resolveSpeed(speed: directive?.speed, rateWPM: directive?.rateWPM),
|
||
stability: TalkTTSValidation.validatedStability(directive?.stability, modelId: modelId),
|
||
similarity: TalkTTSValidation.validatedUnit(directive?.similarity),
|
||
style: TalkTTSValidation.validatedUnit(directive?.style),
|
||
speakerBoost: directive?.speakerBoost,
|
||
seed: TalkTTSValidation.validatedSeed(directive?.seed),
|
||
normalize: ElevenLabsTTSClient.validatedNormalize(directive?.normalize),
|
||
language: language,
|
||
latencyTier: TalkTTSValidation.validatedLatencyTier(directive?.latencyTier))
|
||
}
|
||
|
||
let request = makeRequest(outputFormat: outputFormat)
|
||
|
||
let client = ElevenLabsTTSClient(apiKey: apiKey)
|
||
let rawStream = client.streamSynthesize(voiceId: voiceId, request: request)
|
||
|
||
if self.interruptOnSpeech {
|
||
do {
|
||
try self.startRecognition()
|
||
} catch {
|
||
self.logger.warning(
|
||
"startRecognition during speak failed: \(error.localizedDescription, privacy: .public)")
|
||
}
|
||
}
|
||
|
||
self.statusText = "Speaking…"
|
||
let sampleRate = TalkTTSValidation.pcmSampleRate(from: outputFormat)
|
||
let result: StreamingPlaybackResult
|
||
if let sampleRate {
|
||
let streamFailure = StreamFailureBox()
|
||
let stream = Self.monitorStreamFailures(rawStream, failureBox: streamFailure)
|
||
self.lastPlaybackWasPCM = true
|
||
var playback = await pcmPlayer.play(stream: stream, sampleRate: sampleRate)
|
||
if !playback.finished, playback.interruptedAt == nil {
|
||
let mp3Format = ElevenLabsTTSClient.validatedOutputFormat("mp3_44100_128")
|
||
self.logger.warning("pcm playback failed; retrying mp3")
|
||
if Self.isPCMFormatRejectedByAPI(streamFailure.value) {
|
||
self.pcmFormatUnavailable = true
|
||
}
|
||
self.lastPlaybackWasPCM = false
|
||
let mp3Stream = client.streamSynthesize(
|
||
voiceId: voiceId,
|
||
request: makeRequest(outputFormat: mp3Format))
|
||
playback = await self.mp3Player.play(stream: mp3Stream)
|
||
}
|
||
result = playback
|
||
} else {
|
||
self.lastPlaybackWasPCM = false
|
||
result = await self.mp3Player.play(stream: rawStream)
|
||
}
|
||
let duration = Date().timeIntervalSince(started)
|
||
self.logger
|
||
.info(
|
||
"elevenlabs finished=\(result.finished, privacy: .public) dur=\(duration, privacy: .public)s")
|
||
if !result.finished, let interruptedAt = result.interruptedAt {
|
||
self.lastInterruptedAtSeconds = interruptedAt
|
||
}
|
||
} else {
|
||
self.logger.warning("tts unavailable; falling back to system voice (missing key or voiceId)")
|
||
GatewayDiagnostics.log("talk tts: provider=system (missing key or voiceId)")
|
||
self.applyVoiceModeDescriptor(TalkVoiceModeDescriptorBuilder.build(
|
||
providerId: "system",
|
||
providerLabel: Self.displayName(forProvider: "system"),
|
||
modelId: nil,
|
||
voiceId: language,
|
||
transport: "native",
|
||
isRealtime: false))
|
||
if self.interruptOnSpeech {
|
||
do {
|
||
try self.startRecognition()
|
||
} catch {
|
||
self.logger.warning(
|
||
"startRecognition during speak failed: \(error.localizedDescription, privacy: .public)")
|
||
}
|
||
}
|
||
self.statusText = "Speaking (System)…"
|
||
try await TalkSystemSpeechSynthesizer.shared.speak(text: cleaned, language: language)
|
||
}
|
||
} catch {
|
||
self.logger.error(
|
||
"tts failed: \(error.localizedDescription, privacy: .public); falling back to system voice")
|
||
GatewayDiagnostics.log("talk tts: provider=system (error) msg=\(error.localizedDescription)")
|
||
do {
|
||
let language = ElevenLabsTTSClient.validatedLanguage(directive?.language)
|
||
self.applyVoiceModeDescriptor(TalkVoiceModeDescriptorBuilder.build(
|
||
providerId: "system",
|
||
providerLabel: Self.displayName(forProvider: "system"),
|
||
modelId: nil,
|
||
voiceId: language,
|
||
transport: "native",
|
||
isRealtime: false))
|
||
if self.interruptOnSpeech {
|
||
do {
|
||
try self.startRecognition()
|
||
} catch {
|
||
self.logger.warning(
|
||
"startRecognition during speak failed: \(error.localizedDescription, privacy: .public)")
|
||
}
|
||
}
|
||
self.statusText = "Speaking (System)…"
|
||
try await TalkSystemSpeechSynthesizer.shared.speak(text: cleaned, language: language)
|
||
} catch {
|
||
self.statusText = "Speak failed: \(error.localizedDescription)"
|
||
self.logger.error("system voice failed: \(error.localizedDescription, privacy: .public)")
|
||
}
|
||
}
|
||
|
||
self.stopRecognition()
|
||
self.isSpeaking = false
|
||
self.restoreConfiguredVoiceModeDescriptor()
|
||
}
|
||
|
||
private func stopSpeaking(storeInterruption: Bool = true) {
|
||
let hasIncremental = self.incrementalSpeechActive ||
|
||
self.incrementalSpeechTask != nil ||
|
||
!self.incrementalSpeechQueue.isEmpty
|
||
if self.isSpeaking {
|
||
let interruptedAt = self.lastPlaybackWasPCM
|
||
? self.pcmPlayer.stop()
|
||
: self.mp3Player.stop()
|
||
if storeInterruption {
|
||
self.lastInterruptedAtSeconds = interruptedAt
|
||
}
|
||
_ = self.lastPlaybackWasPCM
|
||
? self.mp3Player.stop()
|
||
: self.pcmPlayer.stop()
|
||
} else if !hasIncremental {
|
||
return
|
||
}
|
||
TalkSystemSpeechSynthesizer.shared.stop()
|
||
self.cancelIncrementalSpeech()
|
||
self.isSpeaking = false
|
||
self.restoreConfiguredVoiceModeDescriptor()
|
||
}
|
||
|
||
private func shouldInterrupt(with transcript: String) -> Bool {
|
||
guard self.shouldAllowSpeechInterruptForCurrentRoute() else { return false }
|
||
let trimmed = transcript.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard trimmed.count >= 3 else { return false }
|
||
if let spoken = lastSpokenText?.lowercased(), spoken.contains(trimmed.lowercased()) {
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
private func shouldAllowSpeechInterruptForCurrentRoute() -> Bool {
|
||
let route = AVAudioSession.sharedInstance().currentRoute
|
||
// Built-in speaker/receiver often feeds TTS back into STT, causing false interrupts.
|
||
// Allow barge-in for isolated outputs (headphones/Bluetooth/USB/CarPlay/AirPlay).
|
||
return !route.outputs.contains { output in
|
||
switch output.portType {
|
||
case .builtInSpeaker, .builtInReceiver:
|
||
true
|
||
default:
|
||
false
|
||
}
|
||
}
|
||
}
|
||
|
||
private func shouldUseIncrementalTTS() -> Bool {
|
||
true
|
||
}
|
||
|
||
private var isSpeechOutputActive: Bool {
|
||
self.isSpeaking ||
|
||
self.incrementalSpeechActive ||
|
||
self.incrementalSpeechTask != nil ||
|
||
!self.incrementalSpeechQueue.isEmpty
|
||
}
|
||
|
||
private func applyDirective(_ directive: TalkDirective?) {
|
||
let requestedVoice = directive?.voiceId?.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
let resolvedVoice = resolveVoiceAlias(requestedVoice)
|
||
if requestedVoice?.isEmpty == false, resolvedVoice == nil {
|
||
self.logger.warning("unknown voice alias \(requestedVoice ?? "?", privacy: .public)")
|
||
}
|
||
if let voice = resolvedVoice {
|
||
if directive?.once != true {
|
||
self.currentVoiceId = voice
|
||
self.voiceOverrideActive = true
|
||
}
|
||
}
|
||
if let model = directive?.modelId {
|
||
if directive?.once != true {
|
||
self.currentModelId = model
|
||
self.modelOverrideActive = true
|
||
}
|
||
}
|
||
}
|
||
|
||
private func resetIncrementalSpeech() {
|
||
self.incrementalSpeechQueue.removeAll()
|
||
self.incrementalSpeechTask?.cancel()
|
||
self.incrementalSpeechTask = nil
|
||
self.cancelIncrementalPrefetch()
|
||
self.incrementalSpeechActive = true
|
||
self.incrementalSpeechUsed = false
|
||
self.incrementalSpeechLanguage = nil
|
||
self.incrementalSpeechBuffer = IncrementalSpeechBuffer()
|
||
self.incrementalSpeechContext = nil
|
||
self.incrementalSpeechDirective = nil
|
||
}
|
||
|
||
private func cancelIncrementalSpeech() {
|
||
self.incrementalSpeechQueue.removeAll()
|
||
self.incrementalSpeechTask?.cancel()
|
||
self.incrementalSpeechTask = nil
|
||
self.cancelIncrementalPrefetch()
|
||
self.incrementalSpeechActive = false
|
||
self.incrementalSpeechContext = nil
|
||
self.incrementalSpeechDirective = nil
|
||
}
|
||
|
||
private func enqueueIncrementalSpeech(_ text: String) {
|
||
let trimmed = text.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !trimmed.isEmpty else { return }
|
||
self.incrementalSpeechQueue.append(trimmed)
|
||
self.incrementalSpeechUsed = true
|
||
if self.incrementalSpeechTask == nil {
|
||
self.startIncrementalSpeechTask()
|
||
}
|
||
}
|
||
|
||
private func startIncrementalSpeechTask() {
|
||
if self.interruptOnSpeech {
|
||
do {
|
||
try self.startRecognition()
|
||
} catch {
|
||
self.logger.warning(
|
||
"startRecognition during incremental speak failed: \(error.localizedDescription, privacy: .public)")
|
||
}
|
||
}
|
||
|
||
self.incrementalSpeechTask = Task { @MainActor [weak self] in
|
||
guard let self else { return }
|
||
defer {
|
||
self.cancelIncrementalPrefetch()
|
||
self.isSpeaking = false
|
||
self.stopRecognition()
|
||
self.incrementalSpeechTask = nil
|
||
}
|
||
while !Task.isCancelled {
|
||
guard !self.incrementalSpeechQueue.isEmpty else { break }
|
||
let segment = self.incrementalSpeechQueue.removeFirst()
|
||
self.statusText = "Speaking…"
|
||
self.isSpeaking = true
|
||
self.lastSpokenText = segment
|
||
await self.updateIncrementalContextIfNeeded()
|
||
let context = self.incrementalSpeechContext
|
||
let prefetchedAudio = await self.consumeIncrementalPrefetchedAudioIfAvailable(
|
||
for: segment,
|
||
context: context)
|
||
if let context {
|
||
self.startIncrementalPrefetchMonitor(context: context)
|
||
}
|
||
await self.speakIncrementalSegment(
|
||
segment,
|
||
context: context,
|
||
prefetchedAudio: prefetchedAudio)
|
||
self.cancelIncrementalPrefetchMonitor()
|
||
}
|
||
}
|
||
}
|
||
|
||
private func cancelIncrementalPrefetch() {
|
||
self.cancelIncrementalPrefetchMonitor()
|
||
self.incrementalSpeechPrefetch?.task.cancel()
|
||
self.incrementalSpeechPrefetch = nil
|
||
}
|
||
|
||
private func cancelIncrementalPrefetchMonitor() {
|
||
self.incrementalSpeechPrefetchMonitorTask?.cancel()
|
||
self.incrementalSpeechPrefetchMonitorTask = nil
|
||
}
|
||
|
||
private func startIncrementalPrefetchMonitor(context: IncrementalSpeechContext) {
|
||
self.cancelIncrementalPrefetchMonitor()
|
||
self.incrementalSpeechPrefetchMonitorTask = Task { @MainActor [weak self] in
|
||
guard let self else { return }
|
||
while !Task.isCancelled {
|
||
if self.ensureIncrementalPrefetchForUpcomingSegment(context: context) {
|
||
return
|
||
}
|
||
try? await Task.sleep(nanoseconds: 40_000_000)
|
||
}
|
||
}
|
||
}
|
||
|
||
private func ensureIncrementalPrefetchForUpcomingSegment(context: IncrementalSpeechContext) -> Bool {
|
||
guard context.canUseElevenLabs else {
|
||
self.cancelIncrementalPrefetch()
|
||
return false
|
||
}
|
||
guard let nextSegment = incrementalSpeechQueue.first else { return false }
|
||
if let existing = incrementalSpeechPrefetch {
|
||
if existing.segment == nextSegment, existing.context == context {
|
||
return true
|
||
}
|
||
existing.task.cancel()
|
||
self.incrementalSpeechPrefetch = nil
|
||
}
|
||
self.startIncrementalPrefetch(segment: nextSegment, context: context)
|
||
return self.incrementalSpeechPrefetch != nil
|
||
}
|
||
|
||
private func startIncrementalPrefetch(segment: String, context: IncrementalSpeechContext) {
|
||
guard context.canUseElevenLabs, let apiKey = context.apiKey, let voiceId = context.voiceId else { return }
|
||
let prefetchOutputFormat = self.resolveIncrementalPrefetchOutputFormat(context: context)
|
||
let request = self.makeIncrementalTTSRequest(
|
||
text: segment,
|
||
context: context,
|
||
outputFormat: prefetchOutputFormat)
|
||
let id = UUID()
|
||
let task = Task { [weak self] in
|
||
let stream = ElevenLabsTTSClient(apiKey: apiKey).streamSynthesize(voiceId: voiceId, request: request)
|
||
var chunks: [Data] = []
|
||
do {
|
||
for try await chunk in stream {
|
||
try Task.checkCancellation()
|
||
chunks.append(chunk)
|
||
}
|
||
self?.completeIncrementalPrefetch(id: id, chunks: chunks)
|
||
} catch is CancellationError {
|
||
self?.clearIncrementalPrefetch(id: id)
|
||
} catch {
|
||
self?.failIncrementalPrefetch(id: id, error: error)
|
||
}
|
||
}
|
||
self.incrementalSpeechPrefetch = IncrementalSpeechPrefetchState(
|
||
id: id,
|
||
segment: segment,
|
||
context: context,
|
||
outputFormat: prefetchOutputFormat,
|
||
chunks: nil,
|
||
task: task)
|
||
}
|
||
|
||
private func completeIncrementalPrefetch(id: UUID, chunks: [Data]) {
|
||
guard var prefetch = incrementalSpeechPrefetch, prefetch.id == id else { return }
|
||
prefetch.chunks = chunks
|
||
self.incrementalSpeechPrefetch = prefetch
|
||
}
|
||
|
||
private func clearIncrementalPrefetch(id: UUID) {
|
||
guard let prefetch = incrementalSpeechPrefetch, prefetch.id == id else { return }
|
||
prefetch.task.cancel()
|
||
self.incrementalSpeechPrefetch = nil
|
||
}
|
||
|
||
private func failIncrementalPrefetch(id: UUID, error: any Error) {
|
||
guard let prefetch = incrementalSpeechPrefetch, prefetch.id == id else { return }
|
||
self.logger.debug("incremental prefetch failed: \(error.localizedDescription, privacy: .public)")
|
||
prefetch.task.cancel()
|
||
self.incrementalSpeechPrefetch = nil
|
||
}
|
||
|
||
private func consumeIncrementalPrefetchedAudioIfAvailable(
|
||
for segment: String,
|
||
context: IncrementalSpeechContext?) async -> IncrementalPrefetchedAudio?
|
||
{
|
||
guard let context else {
|
||
self.cancelIncrementalPrefetch()
|
||
return nil
|
||
}
|
||
guard let prefetch = incrementalSpeechPrefetch else {
|
||
return nil
|
||
}
|
||
guard prefetch.context == context else {
|
||
prefetch.task.cancel()
|
||
self.incrementalSpeechPrefetch = nil
|
||
return nil
|
||
}
|
||
guard prefetch.segment == segment else {
|
||
return nil
|
||
}
|
||
if let chunks = prefetch.chunks, !chunks.isEmpty {
|
||
let prefetched = IncrementalPrefetchedAudio(chunks: chunks, outputFormat: prefetch.outputFormat)
|
||
self.incrementalSpeechPrefetch = nil
|
||
return prefetched
|
||
}
|
||
await prefetch.task.value
|
||
guard let completed = incrementalSpeechPrefetch else { return nil }
|
||
guard completed.context == context, completed.segment == segment else { return nil }
|
||
guard let chunks = completed.chunks, !chunks.isEmpty else { return nil }
|
||
let prefetched = IncrementalPrefetchedAudio(chunks: chunks, outputFormat: completed.outputFormat)
|
||
self.incrementalSpeechPrefetch = nil
|
||
return prefetched
|
||
}
|
||
|
||
private func resolveIncrementalPrefetchOutputFormat(context: IncrementalSpeechContext) -> String? {
|
||
if TalkTTSValidation.pcmSampleRate(from: context.outputFormat) != nil {
|
||
return ElevenLabsTTSClient.validatedOutputFormat("mp3_44100_128")
|
||
}
|
||
return context.outputFormat
|
||
}
|
||
|
||
private func finishIncrementalSpeech() async {
|
||
guard self.incrementalSpeechActive else { return }
|
||
let leftover = self.incrementalSpeechBuffer.flush()
|
||
if let leftover {
|
||
self.enqueueIncrementalSpeech(leftover)
|
||
}
|
||
if let task = incrementalSpeechTask {
|
||
_ = await task.result
|
||
}
|
||
self.incrementalSpeechActive = false
|
||
}
|
||
|
||
private func handleIncrementalAssistantFinal(text: String) async {
|
||
let parsed = TalkDirectiveParser.parse(text)
|
||
self.applyDirective(parsed.directive)
|
||
if let lang = parsed.directive?.language {
|
||
self.incrementalSpeechLanguage = ElevenLabsTTSClient.validatedLanguage(lang)
|
||
}
|
||
await self.updateIncrementalContextIfNeeded()
|
||
let segments = self.incrementalSpeechBuffer.ingest(text: text, isFinal: true)
|
||
for segment in segments {
|
||
self.enqueueIncrementalSpeech(segment)
|
||
}
|
||
await self.finishIncrementalSpeech()
|
||
if !self.incrementalSpeechUsed {
|
||
await self.playAssistant(text: text)
|
||
}
|
||
}
|
||
|
||
private func streamAssistant(runId: String, gateway: GatewayNodeSession) async {
|
||
let stream = await gateway.subscribeServerEvents(bufferingNewest: 200)
|
||
for await evt in stream {
|
||
if Task.isCancelled { return }
|
||
guard evt.event == "agent", let payload = evt.payload else { continue }
|
||
guard let agentEvent = try? GatewayPayloadDecoding.decode(
|
||
payload,
|
||
as: OpenClawAgentEventPayload.self)
|
||
else {
|
||
continue
|
||
}
|
||
guard agentEvent.runId == runId, agentEvent.stream == "assistant" else { continue }
|
||
guard let text = agentEvent.data["text"]?.value as? String else { continue }
|
||
let segments = self.incrementalSpeechBuffer.ingest(text: text, isFinal: false)
|
||
if let lang = incrementalSpeechBuffer.directive?.language {
|
||
self.incrementalSpeechLanguage = ElevenLabsTTSClient.validatedLanguage(lang)
|
||
}
|
||
await self.updateIncrementalContextIfNeeded()
|
||
for segment in segments {
|
||
self.enqueueIncrementalSpeech(segment)
|
||
}
|
||
}
|
||
}
|
||
|
||
private func updateIncrementalContextIfNeeded() async {
|
||
let directive = self.incrementalSpeechBuffer.directive
|
||
if let existing = incrementalSpeechContext, directive == incrementalSpeechDirective {
|
||
if existing.language != self.incrementalSpeechLanguage {
|
||
self.incrementalSpeechContext = IncrementalSpeechContext(
|
||
apiKey: existing.apiKey,
|
||
voiceId: existing.voiceId,
|
||
modelId: existing.modelId,
|
||
outputFormat: existing.outputFormat,
|
||
language: self.incrementalSpeechLanguage,
|
||
directive: existing.directive,
|
||
canUseElevenLabs: existing.canUseElevenLabs)
|
||
}
|
||
return
|
||
}
|
||
let context = await buildIncrementalSpeechContext(directive: directive)
|
||
self.incrementalSpeechContext = context
|
||
self.incrementalSpeechDirective = directive
|
||
}
|
||
|
||
private func buildIncrementalSpeechContext(directive: TalkDirective?) async -> IncrementalSpeechContext {
|
||
let requestedVoice = directive?.voiceId?.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
let resolvedVoice = resolveVoiceAlias(requestedVoice)
|
||
if requestedVoice?.isEmpty == false, resolvedVoice == nil {
|
||
self.logger.warning("unknown voice alias \(requestedVoice ?? "?", privacy: .public)")
|
||
}
|
||
let preferredVoice = resolvedVoice ?? self.currentVoiceId ?? self.defaultVoiceId
|
||
let modelId = directive?.modelId ?? self.currentModelId ?? self.defaultModelId
|
||
let desiredOutputFormat = (directive?.outputFormat ?? self.defaultOutputFormat)?
|
||
.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
let requestedOutputFormat = (desiredOutputFormat?.isEmpty == false) ? desiredOutputFormat : nil
|
||
let outputFormat = ElevenLabsTTSClient.validatedOutputFormat(
|
||
requestedOutputFormat ?? self.effectiveDefaultOutputFormat)
|
||
if outputFormat == nil, let requestedOutputFormat {
|
||
self.logger.warning(
|
||
"talk output_format unsupported for local playback: \(requestedOutputFormat, privacy: .public)")
|
||
}
|
||
|
||
let configuredKey = self.apiKey?
|
||
.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
.isEmpty == false ? self.apiKey : nil
|
||
#if DEBUG
|
||
let resolvedKey = configuredKey ?? ProcessInfo.processInfo.environment["ELEVENLABS_API_KEY"]
|
||
#else
|
||
let resolvedKey = configuredKey
|
||
#endif
|
||
let apiKey = resolvedKey?.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
let voiceId: String? = if let apiKey, !apiKey.isEmpty {
|
||
await resolveVoiceId(preferred: preferredVoice, apiKey: apiKey)
|
||
} else {
|
||
nil
|
||
}
|
||
let canUseElevenLabs = (voiceId?.isEmpty == false) && (apiKey?.isEmpty == false)
|
||
return IncrementalSpeechContext(
|
||
apiKey: apiKey,
|
||
voiceId: voiceId,
|
||
modelId: modelId,
|
||
outputFormat: outputFormat,
|
||
language: self.incrementalSpeechLanguage,
|
||
directive: directive,
|
||
canUseElevenLabs: canUseElevenLabs)
|
||
}
|
||
|
||
private func makeIncrementalTTSRequest(
|
||
text: String,
|
||
context: IncrementalSpeechContext,
|
||
outputFormat: String?) -> ElevenLabsTTSRequest
|
||
{
|
||
ElevenLabsTTSRequest(
|
||
text: text,
|
||
modelId: context.modelId,
|
||
outputFormat: outputFormat,
|
||
speed: TalkTTSValidation.resolveSpeed(
|
||
speed: context.directive?.speed,
|
||
rateWPM: context.directive?.rateWPM),
|
||
stability: TalkTTSValidation.validatedStability(
|
||
context.directive?.stability,
|
||
modelId: context.modelId),
|
||
similarity: TalkTTSValidation.validatedUnit(context.directive?.similarity),
|
||
style: TalkTTSValidation.validatedUnit(context.directive?.style),
|
||
speakerBoost: context.directive?.speakerBoost,
|
||
seed: TalkTTSValidation.validatedSeed(context.directive?.seed),
|
||
normalize: ElevenLabsTTSClient.validatedNormalize(context.directive?.normalize),
|
||
language: context.language,
|
||
latencyTier: TalkTTSValidation.validatedLatencyTier(context.directive?.latencyTier))
|
||
}
|
||
|
||
/// Returns `mp3_44100_128` when the API has already rejected PCM, otherwise `pcm_44100`.
|
||
private var effectiveDefaultOutputFormat: String {
|
||
self.pcmFormatUnavailable ? "mp3_44100_128" : "pcm_44100"
|
||
}
|
||
|
||
private static func monitorStreamFailures(
|
||
_ stream: AsyncThrowingStream<Data, Error>,
|
||
failureBox: StreamFailureBox) -> AsyncThrowingStream<Data, Error>
|
||
{
|
||
AsyncThrowingStream { continuation in
|
||
let task = Task {
|
||
do {
|
||
for try await chunk in stream {
|
||
continuation.yield(chunk)
|
||
}
|
||
continuation.finish()
|
||
} catch {
|
||
failureBox.set(error)
|
||
continuation.finish(throwing: error)
|
||
}
|
||
}
|
||
continuation.onTermination = { _ in
|
||
task.cancel()
|
||
}
|
||
}
|
||
}
|
||
|
||
private static func isPCMFormatRejectedByAPI(_ error: Error?) -> Bool {
|
||
guard let error = error as NSError? else { return false }
|
||
guard error.domain == "ElevenLabsTTS", error.code >= 400 else { return false }
|
||
let message = (error.userInfo[NSLocalizedDescriptionKey] as? String ?? error.localizedDescription).lowercased()
|
||
return message.contains("output_format")
|
||
|| message.contains("pcm_")
|
||
|| message.contains("pcm ")
|
||
|| message.contains("subscription_required")
|
||
}
|
||
|
||
private static func makeBufferedAudioStream(chunks: [Data]) -> AsyncThrowingStream<Data, Error> {
|
||
AsyncThrowingStream { continuation in
|
||
for chunk in chunks {
|
||
continuation.yield(chunk)
|
||
}
|
||
continuation.finish()
|
||
}
|
||
}
|
||
|
||
private func speakIncrementalSegment(
|
||
_ text: String,
|
||
context preferredContext: IncrementalSpeechContext? = nil,
|
||
prefetchedAudio: IncrementalPrefetchedAudio? = nil) async
|
||
{
|
||
let context: IncrementalSpeechContext
|
||
if let preferredContext {
|
||
context = preferredContext
|
||
} else {
|
||
await self.updateIncrementalContextIfNeeded()
|
||
guard let resolvedContext = incrementalSpeechContext else {
|
||
try? await TalkSystemSpeechSynthesizer.shared.speak(
|
||
text: text,
|
||
language: self.incrementalSpeechLanguage)
|
||
return
|
||
}
|
||
context = resolvedContext
|
||
}
|
||
|
||
guard context.canUseElevenLabs, let apiKey = context.apiKey, let voiceId = context.voiceId else {
|
||
try? await TalkSystemSpeechSynthesizer.shared.speak(
|
||
text: text,
|
||
language: self.incrementalSpeechLanguage)
|
||
return
|
||
}
|
||
|
||
let client = ElevenLabsTTSClient(apiKey: apiKey)
|
||
let request = self.makeIncrementalTTSRequest(
|
||
text: text,
|
||
context: context,
|
||
outputFormat: context.outputFormat)
|
||
let rawStream: AsyncThrowingStream<Data, Error> = if let prefetchedAudio, !prefetchedAudio.chunks.isEmpty {
|
||
Self.makeBufferedAudioStream(chunks: prefetchedAudio.chunks)
|
||
} else {
|
||
client.streamSynthesize(voiceId: voiceId, request: request)
|
||
}
|
||
let playbackFormat = prefetchedAudio?.outputFormat ?? context.outputFormat
|
||
let sampleRate = TalkTTSValidation.pcmSampleRate(from: playbackFormat)
|
||
let result: StreamingPlaybackResult
|
||
if let sampleRate {
|
||
let streamFailure = StreamFailureBox()
|
||
let stream = Self.monitorStreamFailures(rawStream, failureBox: streamFailure)
|
||
self.lastPlaybackWasPCM = true
|
||
var playback = await pcmPlayer.play(stream: stream, sampleRate: sampleRate)
|
||
if !playback.finished, playback.interruptedAt == nil {
|
||
self.logger.warning("pcm playback failed; retrying mp3")
|
||
if Self.isPCMFormatRejectedByAPI(streamFailure.value) {
|
||
self.pcmFormatUnavailable = true
|
||
}
|
||
self.lastPlaybackWasPCM = false
|
||
let mp3Format = ElevenLabsTTSClient.validatedOutputFormat("mp3_44100_128")
|
||
let mp3Stream = client.streamSynthesize(
|
||
voiceId: voiceId,
|
||
request: self.makeIncrementalTTSRequest(
|
||
text: text,
|
||
context: context,
|
||
outputFormat: mp3Format))
|
||
playback = await self.mp3Player.play(stream: mp3Stream)
|
||
}
|
||
result = playback
|
||
} else {
|
||
self.lastPlaybackWasPCM = false
|
||
result = await self.mp3Player.play(stream: rawStream)
|
||
}
|
||
if !result.finished, let interruptedAt = result.interruptedAt {
|
||
self.lastInterruptedAtSeconds = interruptedAt
|
||
}
|
||
}
|
||
}
|
||
|
||
private struct IncrementalSpeechBuffer {
|
||
private static let softBoundaryMinChars = 72
|
||
|
||
private(set) var latestText: String = ""
|
||
private(set) var directive: TalkDirective?
|
||
private var spokenOffset: Int = 0
|
||
private var inCodeBlock = false
|
||
private var directiveParsed = false
|
||
|
||
mutating func ingest(text: String, isFinal: Bool) -> [String] {
|
||
let normalized = text.replacingOccurrences(of: "\r\n", with: "\n")
|
||
guard let usable = stripDirectiveIfReady(from: normalized) else { return [] }
|
||
self.updateText(usable)
|
||
return self.extractSegments(isFinal: isFinal)
|
||
}
|
||
|
||
mutating func flush() -> String? {
|
||
guard !self.latestText.isEmpty else { return nil }
|
||
let segments = self.extractSegments(isFinal: true)
|
||
return segments.first
|
||
}
|
||
|
||
private mutating func stripDirectiveIfReady(from text: String) -> String? {
|
||
guard !self.directiveParsed else { return text }
|
||
let trimmed = text.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !trimmed.isEmpty else { return nil }
|
||
if trimmed.hasPrefix("{") {
|
||
guard let newlineRange = text.range(of: "\n") else { return nil }
|
||
let firstLine = text[..<newlineRange.lowerBound]
|
||
let head = firstLine.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard head.hasSuffix("}") else { return nil }
|
||
let parsed = TalkDirectiveParser.parse(text)
|
||
if let directive = parsed.directive {
|
||
self.directive = directive
|
||
}
|
||
self.directiveParsed = true
|
||
return parsed.stripped
|
||
}
|
||
self.directiveParsed = true
|
||
return text
|
||
}
|
||
|
||
private mutating func updateText(_ newText: String) {
|
||
if newText.hasPrefix(self.latestText) {
|
||
self.latestText = newText
|
||
} else if self.latestText.hasPrefix(newText) {
|
||
// Stream reset or correction; prefer the newer prefix.
|
||
self.latestText = newText
|
||
self.spokenOffset = min(self.spokenOffset, newText.count)
|
||
} else {
|
||
// Diverged text means chunks arrived out of order or stream restarted.
|
||
let commonPrefix = Self.commonPrefixCount(self.latestText, newText)
|
||
self.latestText = newText
|
||
if self.spokenOffset > commonPrefix {
|
||
self.spokenOffset = commonPrefix
|
||
}
|
||
}
|
||
if self.spokenOffset > self.latestText.count {
|
||
self.spokenOffset = self.latestText.count
|
||
}
|
||
}
|
||
|
||
private static func commonPrefixCount(_ lhs: String, _ rhs: String) -> Int {
|
||
let left = Array(lhs)
|
||
let right = Array(rhs)
|
||
let limit = min(left.count, right.count)
|
||
var idx = 0
|
||
while idx < limit, left[idx] == right[idx] {
|
||
idx += 1
|
||
}
|
||
return idx
|
||
}
|
||
|
||
private mutating func extractSegments(isFinal: Bool) -> [String] {
|
||
let chars = Array(latestText)
|
||
guard self.spokenOffset < chars.count else { return [] }
|
||
var idx = self.spokenOffset
|
||
var lastBoundary: Int?
|
||
var inCodeBlock = self.inCodeBlock
|
||
var buffer = ""
|
||
var bufferAtBoundary = ""
|
||
var inCodeBlockAtBoundary = inCodeBlock
|
||
|
||
while idx < chars.count {
|
||
if idx + 2 < chars.count,
|
||
chars[idx] == "`",
|
||
chars[idx + 1] == "`",
|
||
chars[idx + 2] == "`"
|
||
{
|
||
inCodeBlock.toggle()
|
||
idx += 3
|
||
continue
|
||
}
|
||
|
||
if !inCodeBlock {
|
||
let currentChar = chars[idx]
|
||
buffer.append(currentChar)
|
||
if Self.isBoundary(currentChar) || Self.isSoftBoundary(currentChar, bufferedChars: buffer.count) {
|
||
lastBoundary = idx + 1
|
||
bufferAtBoundary = buffer
|
||
inCodeBlockAtBoundary = inCodeBlock
|
||
}
|
||
}
|
||
|
||
idx += 1
|
||
}
|
||
|
||
if let boundary = lastBoundary {
|
||
self.spokenOffset = boundary
|
||
self.inCodeBlock = inCodeBlockAtBoundary
|
||
let trimmed = bufferAtBoundary.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
return trimmed.isEmpty ? [] : [trimmed]
|
||
}
|
||
|
||
guard isFinal else { return [] }
|
||
self.spokenOffset = chars.count
|
||
self.inCodeBlock = inCodeBlock
|
||
let trimmed = buffer.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
return trimmed.isEmpty ? [] : [trimmed]
|
||
}
|
||
|
||
private static func isBoundary(_ ch: Character) -> Bool {
|
||
ch == "." || ch == "!" || ch == "?" || ch == "\n"
|
||
}
|
||
|
||
private static func isSoftBoundary(_ ch: Character, bufferedChars: Int) -> Bool {
|
||
bufferedChars >= self.softBoundaryMinChars && ch.isWhitespace
|
||
}
|
||
}
|
||
|
||
extension TalkModeManager {
|
||
nonisolated static func requestMicrophonePermission() async -> Bool {
|
||
switch AVAudioApplication.shared.recordPermission {
|
||
case .granted:
|
||
return true
|
||
case .denied:
|
||
return false
|
||
case .undetermined:
|
||
return await self.requestPermissionWithTimeout { completion in
|
||
AVAudioApplication.requestRecordPermission(completionHandler: { ok in
|
||
completion(ok)
|
||
})
|
||
}
|
||
@unknown default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
nonisolated static func requestSpeechPermission() async -> Bool {
|
||
let status = SFSpeechRecognizer.authorizationStatus()
|
||
switch status {
|
||
case .authorized:
|
||
return true
|
||
case .denied, .restricted:
|
||
return false
|
||
case .notDetermined:
|
||
break
|
||
@unknown default:
|
||
return false
|
||
}
|
||
|
||
return await self.requestPermissionWithTimeout { completion in
|
||
SFSpeechRecognizer.requestAuthorization { authStatus in
|
||
completion(authStatus == .authorized)
|
||
}
|
||
}
|
||
}
|
||
|
||
private nonisolated static func requestPermissionWithTimeout(
|
||
_ operation: @escaping @Sendable (@escaping @Sendable (Bool) -> Void) -> Void) async -> Bool
|
||
{
|
||
do {
|
||
return try await AsyncTimeout.withTimeout(
|
||
seconds: 8,
|
||
onTimeout: { NSError(domain: "TalkMode", code: 6, userInfo: [
|
||
NSLocalizedDescriptionKey: "permission request timed out",
|
||
]) },
|
||
operation: {
|
||
await withCheckedContinuation(isolation: nil) { cont in
|
||
Task { @MainActor in
|
||
operation { ok in
|
||
cont.resume(returning: ok)
|
||
}
|
||
}
|
||
}
|
||
})
|
||
} catch {
|
||
return false
|
||
}
|
||
}
|
||
|
||
static func permissionMessage(
|
||
kind: String,
|
||
status: AVAudioSession.RecordPermission) -> String
|
||
{
|
||
switch status {
|
||
case .denied:
|
||
return "\(kind) permission denied"
|
||
case .undetermined:
|
||
return "\(kind) permission not granted"
|
||
case .granted:
|
||
return "\(kind) permission denied"
|
||
@unknown default:
|
||
return "\(kind) permission denied"
|
||
}
|
||
}
|
||
|
||
static func permissionMessage(
|
||
kind: String,
|
||
status: SFSpeechRecognizerAuthorizationStatus) -> String
|
||
{
|
||
switch status {
|
||
case .denied:
|
||
return "\(kind) permission denied"
|
||
case .restricted:
|
||
return "\(kind) permission restricted"
|
||
case .notDetermined:
|
||
return "\(kind) permission not granted"
|
||
case .authorized:
|
||
return "\(kind) permission denied"
|
||
@unknown default:
|
||
return "\(kind) permission denied"
|
||
}
|
||
}
|
||
}
|
||
|
||
extension TalkModeManager {
|
||
func resolveVoiceAlias(_ value: String?) -> String? {
|
||
let trimmed = (value ?? "").trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !trimmed.isEmpty else { return nil }
|
||
let normalized = trimmed.lowercased()
|
||
if let mapped = voiceAliases[normalized] { return mapped }
|
||
if self.voiceAliases.values.contains(where: { $0.caseInsensitiveCompare(trimmed) == .orderedSame }) {
|
||
return trimmed
|
||
}
|
||
return Self.isLikelyVoiceId(trimmed) ? trimmed : nil
|
||
}
|
||
|
||
func resolveVoiceId(preferred: String?, apiKey: String) async -> String? {
|
||
let trimmed = preferred?.trimmingCharacters(in: .whitespacesAndNewlines) ?? ""
|
||
if !trimmed.isEmpty {
|
||
// Config / directives can provide a raw ElevenLabs voiceId (not an alias).
|
||
// Accept it directly to avoid unnecessary listVoices calls (and accidental fallback selection).
|
||
if Self.isLikelyVoiceId(trimmed) {
|
||
return trimmed
|
||
}
|
||
if let resolved = resolveVoiceAlias(trimmed) { return resolved }
|
||
self.logger.warning("unknown voice alias \(trimmed, privacy: .public)")
|
||
}
|
||
if let fallbackVoiceId { return fallbackVoiceId }
|
||
|
||
do {
|
||
let voices = try await ElevenLabsTTSClient(apiKey: apiKey).listVoices()
|
||
guard let first = voices.first else {
|
||
self.logger.warning("elevenlabs voices list empty")
|
||
return nil
|
||
}
|
||
fallbackVoiceId = first.voiceId
|
||
if self.defaultVoiceId == nil {
|
||
self.defaultVoiceId = first.voiceId
|
||
}
|
||
if !self.voiceOverrideActive {
|
||
self.currentVoiceId = first.voiceId
|
||
}
|
||
let name = first.name ?? "unknown"
|
||
self.logger
|
||
.info("default voice selected \(name, privacy: .public) (\(first.voiceId, privacy: .public))")
|
||
return first.voiceId
|
||
} catch {
|
||
self.logger.error("elevenlabs list voices failed: \(error.localizedDescription, privacy: .public)")
|
||
return nil
|
||
}
|
||
}
|
||
|
||
static func isLikelyVoiceId(_ value: String) -> Bool {
|
||
guard value.count >= 10 else { return false }
|
||
return value.allSatisfy { $0.isLetter || $0.isNumber || $0 == "-" || $0 == "_" }
|
||
}
|
||
|
||
private static func normalizedTalkApiKey(_ raw: String?) -> String? {
|
||
let trimmed = (raw ?? "").trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !trimmed.isEmpty else { return nil }
|
||
guard trimmed != Self.redactedConfigSentinel else { return nil }
|
||
// Config values may be env placeholders (for example `${ELEVENLABS_API_KEY}`).
|
||
if trimmed.hasPrefix("${"), trimmed.hasSuffix("}") { return nil }
|
||
return trimmed
|
||
}
|
||
|
||
private static func displayName(forProvider provider: String) -> String {
|
||
switch provider.trimmingCharacters(in: .whitespacesAndNewlines).lowercased() {
|
||
case "elevenlabs":
|
||
"ElevenLabs"
|
||
case "openai":
|
||
"OpenAI"
|
||
case "google":
|
||
"Google"
|
||
case "system":
|
||
"iOS System Voice"
|
||
case "realtime":
|
||
"Realtime Voice"
|
||
case let provider where !provider.isEmpty:
|
||
provider
|
||
default:
|
||
"Gateway Default"
|
||
}
|
||
}
|
||
|
||
private func applyVoiceModeDescriptor(_ descriptor: TalkVoiceModeDescriptor, persistAsConfigured: Bool = false) {
|
||
if persistAsConfigured {
|
||
self.configuredVoiceModeDescriptor = descriptor
|
||
}
|
||
self.gatewayTalkVoiceModeTitle = descriptor.title
|
||
self.gatewayTalkVoiceModeSubtitle = descriptor.subtitle
|
||
self.gatewayTalkVoiceModeAccessibilityValue = descriptor.accessibilityValue
|
||
}
|
||
|
||
private func restoreConfiguredVoiceModeDescriptor() {
|
||
self.applyVoiceModeDescriptor(self.configuredVoiceModeDescriptor)
|
||
}
|
||
|
||
private func buildConfiguredVoiceModeDescriptor(
|
||
provider: String,
|
||
providerLabel: String,
|
||
modelId: String?,
|
||
voiceId: String?,
|
||
transport: String,
|
||
isRealtime: Bool) -> TalkVoiceModeDescriptor
|
||
{
|
||
TalkVoiceModeDescriptorBuilder.build(
|
||
providerId: provider,
|
||
providerLabel: providerLabel,
|
||
modelId: modelId,
|
||
voiceId: voiceId,
|
||
transport: transport,
|
||
isRealtime: isRealtime)
|
||
}
|
||
|
||
private func ensureTalkConfigLoadedForStart() async {
|
||
if self.gatewayTalkConfigLoaded || self.gatewayTalkPermissionState.isApprovalRequestInProgress {
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline config cached permission=\(self.gatewayTalkPermissionState.statusLabel) "
|
||
+ "loadedAt=\(self.talkConfigLoadedAt?.timeIntervalSince1970 ?? 0)")
|
||
return
|
||
}
|
||
|
||
let configStartedAt = Self.nowSeconds()
|
||
await self.reloadConfig()
|
||
GatewayDiagnostics.log(
|
||
"talk.timeline config reload elapsedMs=\(Self.elapsedMs(since: configStartedAt)) "
|
||
+ "permission=\(self.gatewayTalkPermissionState.statusLabel)")
|
||
}
|
||
|
||
func reloadConfig() async {
|
||
guard let gateway else { return }
|
||
self.pcmFormatUnavailable = false
|
||
self.prefetchedRealtimeSession = nil
|
||
do {
|
||
func fetchConfig(includeSecrets: Bool) async throws -> [String: Any]? {
|
||
let paramsJSON = includeSecrets ? "{\"includeSecrets\":true}" : "{}"
|
||
let res = try await gateway.request(
|
||
method: "talk.config",
|
||
paramsJSON: paramsJSON,
|
||
timeoutSeconds: 8)
|
||
guard let json = try JSONSerialization.jsonObject(with: res) as? [String: Any] else {
|
||
return nil
|
||
}
|
||
return json["config"] as? [String: Any]
|
||
}
|
||
|
||
let config: [String: Any]
|
||
var redactedFallbackMissingScope: String?
|
||
do {
|
||
guard let fetched = try await fetchConfig(includeSecrets: true) else { return }
|
||
config = fetched
|
||
} catch {
|
||
guard let missingScope = Self.missingTalkScope(from: error),
|
||
let fetched = try await fetchConfig(includeSecrets: false)
|
||
else {
|
||
throw error
|
||
}
|
||
config = fetched
|
||
redactedFallbackMissingScope = missingScope
|
||
GatewayDiagnostics.log("talk config secrets unavailable; loaded redacted config")
|
||
}
|
||
let parsed = TalkModeGatewayConfigParser.parse(
|
||
config: config,
|
||
defaultProvider: Self.defaultTalkProvider,
|
||
defaultModelIdFallback: Self.defaultModelIdFallback,
|
||
defaultRealtimeModelIdFallback: Self.defaultRealtimeModelIdFallback,
|
||
defaultSilenceTimeoutMs: Self.defaultSilenceTimeoutMs)
|
||
if parsed.missingResolvedPayload {
|
||
GatewayDiagnostics.log(
|
||
"talk config ignored: normalized payload missing talk.resolved")
|
||
}
|
||
let providerSelection = TalkModeProviderSelection.resolved(
|
||
UserDefaults.standard.string(forKey: TalkModeProviderSelection.storageKey))
|
||
var activeProvider = parsed.activeProvider
|
||
var executionMode = parsed.executionMode
|
||
var realtimeProvider = parsed.realtimeProvider
|
||
var realtimeModelId = parsed.realtimeModelId
|
||
let realtimeVoiceOverride = TalkModeRealtimeVoiceSelection.resolvedOverride(
|
||
UserDefaults.standard.string(forKey: TalkModeRealtimeVoiceSelection.storageKey))
|
||
let realtimeVoiceId = realtimeVoiceOverride ?? parsed.realtimeVoiceId
|
||
switch providerSelection {
|
||
case .gatewayDefault:
|
||
break
|
||
case .nativeElevenLabs:
|
||
activeProvider = Self.defaultTalkProvider
|
||
executionMode = .native
|
||
case .openAIRealtime:
|
||
activeProvider = "openai"
|
||
executionMode = .native
|
||
realtimeProvider = realtimeProvider ?? "openai"
|
||
realtimeModelId = realtimeModelId ?? Self.defaultRealtimeModelIdFallback
|
||
}
|
||
let usesRealtimeConfig = activeProvider != Self.defaultTalkProvider || executionMode != .native
|
||
self.activeTalkProvider = activeProvider
|
||
self.executionMode = executionMode
|
||
self.realtimeWebRTCEnabled = usesRealtimeConfig
|
||
self.realtimeProvider = realtimeProvider
|
||
self.realtimeModelId = realtimeModelId
|
||
self.realtimeVoiceId = realtimeVoiceId
|
||
self.defaultVoiceId = parsed.defaultVoiceId
|
||
self.voiceAliases = parsed.voiceAliases
|
||
if !self.voiceOverrideActive {
|
||
self.currentVoiceId = self.defaultVoiceId
|
||
}
|
||
self.defaultModelId = parsed.defaultModelId
|
||
if !self.modelOverrideActive {
|
||
self.currentModelId = self.defaultModelId
|
||
}
|
||
self.defaultOutputFormat = parsed.defaultOutputFormat
|
||
let rawConfigApiKey = parsed.rawConfigApiKey
|
||
let configApiKey = Self.normalizedTalkApiKey(rawConfigApiKey)
|
||
let localApiKey = Self.normalizedTalkApiKey(
|
||
GatewaySettingsStore.loadTalkProviderApiKey(provider: activeProvider))
|
||
if rawConfigApiKey == Self.redactedConfigSentinel {
|
||
self.apiKey = (localApiKey?.isEmpty == false) ? localApiKey : nil
|
||
GatewayDiagnostics.log("talk config apiKey redacted; using local override if present")
|
||
} else {
|
||
self.apiKey = (localApiKey?.isEmpty == false) ? localApiKey : configApiKey
|
||
}
|
||
let gatewayOwnedVoiceProvider = usesRealtimeConfig
|
||
if gatewayOwnedVoiceProvider {
|
||
self.apiKey = nil
|
||
let credentialProvider = realtimeProvider ?? activeProvider
|
||
GatewayDiagnostics.log(
|
||
"talk realtime provider '\(credentialProvider)' uses gateway-owned credentials")
|
||
}
|
||
self.gatewayTalkDefaultVoiceId = usesRealtimeConfig ? realtimeVoiceId : self.defaultVoiceId
|
||
self.gatewayTalkDefaultModelId = usesRealtimeConfig ? realtimeModelId : self.defaultModelId
|
||
let providerLabel = providerSelection == .gatewayDefault
|
||
? Self.displayName(forProvider: activeProvider)
|
||
: providerSelection.label
|
||
let usesRealtimeRelay = executionMode == .realtimeRelay
|
||
let transport = usesRealtimeConfig ? (usesRealtimeRelay ? "gateway-relay" : "webrtc") : "native"
|
||
let transportLabel = usesRealtimeRelay ? "Gateway Relay" : (usesRealtimeConfig ? "Native WebRTC" : "Native")
|
||
self.gatewayTalkProviderLabel = providerLabel
|
||
self.gatewayTalkUsesRealtime = usesRealtimeConfig
|
||
self.gatewayTalkUsesRealtimeRelay = usesRealtimeRelay
|
||
self.gatewayTalkTransportLabel = transportLabel
|
||
self.gatewayTalkRealtimeProviderLabel = realtimeProvider.map { Self.displayName(forProvider: $0) }
|
||
self.gatewayTalkRealtimeModelId = realtimeModelId
|
||
self.gatewayTalkRealtimeVoiceId = realtimeVoiceId
|
||
let voiceModeProvider = usesRealtimeConfig ? (realtimeProvider ?? "realtime") : activeProvider
|
||
let voiceModeLabel = usesRealtimeConfig
|
||
? Self.displayName(forProvider: voiceModeProvider)
|
||
: Self.displayName(forProvider: activeProvider)
|
||
let voiceModeDescriptor = self.buildConfiguredVoiceModeDescriptor(
|
||
provider: voiceModeProvider,
|
||
providerLabel: voiceModeLabel,
|
||
modelId: usesRealtimeConfig ? realtimeModelId : self.defaultModelId,
|
||
voiceId: usesRealtimeConfig ? realtimeVoiceId : self.defaultVoiceId,
|
||
transport: transport,
|
||
isRealtime: usesRealtimeConfig)
|
||
self.applyVoiceModeDescriptor(voiceModeDescriptor, persistAsConfigured: true)
|
||
self.gatewayTalkApiKeyConfigured = gatewayOwnedVoiceProvider || (self.apiKey?.isEmpty == false)
|
||
self.gatewayTalkConfigLoaded = true
|
||
self.talkConfigLoadedAt = Date()
|
||
if let missingScope = redactedFallbackMissingScope,
|
||
gatewayOwnedVoiceProvider || self.apiKey == nil
|
||
{
|
||
self.gatewayTalkPermissionState = .missingScope(missingScope)
|
||
GatewayDiagnostics.log("talk config missing gateway scope=\(missingScope)")
|
||
} else {
|
||
self.gatewayTalkPermissionState = (self.gatewayTalkApiKeyConfigured || gatewayOwnedVoiceProvider)
|
||
? .ready
|
||
: .apiKeyMissing
|
||
}
|
||
if let interrupt = parsed.interruptOnSpeech {
|
||
self.interruptOnSpeech = interrupt
|
||
}
|
||
self.gatewaySpeechLocaleID = parsed.speechLocaleID
|
||
self.silenceWindow = TimeInterval(parsed.silenceTimeoutMs) / 1000
|
||
if parsed.normalizedPayload || parsed.defaultVoiceId != nil || parsed.rawConfigApiKey != nil {
|
||
GatewayDiagnostics.log(
|
||
"talk config provider=\(activeProvider) silenceTimeoutMs=\(parsed.silenceTimeoutMs)")
|
||
}
|
||
} catch {
|
||
self.activeTalkProvider = Self.defaultTalkProvider
|
||
self.executionMode = .native
|
||
self.realtimeWebRTCEnabled = false
|
||
self.realtimeProvider = nil
|
||
self.realtimeModelId = nil
|
||
self.realtimeVoiceId = nil
|
||
self.gatewayTalkProviderLabel = "Not loaded"
|
||
self.gatewayTalkTransportLabel = "Not loaded"
|
||
self.gatewayTalkUsesRealtime = false
|
||
self.gatewayTalkUsesRealtimeRelay = false
|
||
self.gatewayTalkRealtimeProviderLabel = nil
|
||
self.gatewayTalkRealtimeModelId = nil
|
||
self.gatewayTalkRealtimeVoiceId = nil
|
||
self.applyVoiceModeDescriptor(TalkVoiceModeDescriptor(
|
||
title: "Not loaded",
|
||
subtitle: nil,
|
||
providerId: nil,
|
||
modelId: nil,
|
||
voiceId: nil,
|
||
transport: nil,
|
||
isRealtime: false), persistAsConfigured: true)
|
||
self.defaultModelId = Self.defaultModelIdFallback
|
||
if !self.modelOverrideActive {
|
||
self.currentModelId = self.defaultModelId
|
||
}
|
||
self.gatewayTalkDefaultVoiceId = nil
|
||
self.gatewayTalkDefaultModelId = nil
|
||
self.gatewayTalkApiKeyConfigured = false
|
||
self.gatewayTalkConfigLoaded = false
|
||
self.talkConfigLoadedAt = nil
|
||
self.gatewaySpeechLocaleID = nil
|
||
self.silenceWindow = TimeInterval(Self.defaultSilenceTimeoutMs) / 1000
|
||
if let missingScope = Self.missingTalkScope(from: error) {
|
||
self.gatewayTalkPermissionState = .missingScope(missingScope)
|
||
self.statusText = "Gateway permission required"
|
||
GatewayDiagnostics.log("talk config missing gateway scope=\(missingScope)")
|
||
} else {
|
||
self.gatewayTalkPermissionState = .loadFailed(error.localizedDescription)
|
||
}
|
||
}
|
||
}
|
||
|
||
func markTalkPermissionUpgradeRequested(requestId: String?) {
|
||
self.gatewayTalkPermissionState = .upgradeRequested(requestId: requestId)
|
||
self.statusText = "Approval requested"
|
||
}
|
||
|
||
private static func missingTalkScope(from error: Error) -> String? {
|
||
let targetScope = "operator.talk.secrets"
|
||
if let gatewayError = error as? GatewayResponseError {
|
||
if Self.errorTextIndicatesMissingScope(gatewayError.message, scope: targetScope) {
|
||
return targetScope
|
||
}
|
||
if let missingScope = gatewayError.details["missingScope"]?.value as? String,
|
||
missingScope == targetScope
|
||
{
|
||
return targetScope
|
||
}
|
||
}
|
||
if Self.errorTextIndicatesMissingScope(error.localizedDescription, scope: targetScope) {
|
||
return targetScope
|
||
}
|
||
return nil
|
||
}
|
||
|
||
private static func errorTextIndicatesMissingScope(_ text: String, scope: String) -> Bool {
|
||
let lower = text.trimmingCharacters(in: .whitespacesAndNewlines).lowercased()
|
||
return lower.contains("missing scope") && lower.contains(scope.lowercased())
|
||
}
|
||
|
||
static func configureAudioSession() throws {
|
||
let session = AVAudioSession.sharedInstance()
|
||
// Prefer `.spokenAudio` for STT; it tends to preserve speech energy better than `.voiceChat`.
|
||
try session.setCategory(.playAndRecord, mode: .spokenAudio, options: [
|
||
.allowBluetoothHFP,
|
||
.defaultToSpeaker,
|
||
])
|
||
try? session.setPreferredSampleRate(48000)
|
||
try? session.setPreferredIOBufferDuration(0.02)
|
||
try session.setActive(true, options: [])
|
||
}
|
||
|
||
private static func describeAudioSession() -> String {
|
||
let session = AVAudioSession.sharedInstance()
|
||
let inputs = session.currentRoute.inputs
|
||
.map { "\($0.portType.rawValue):\($0.portName)" }
|
||
.joined(separator: ",")
|
||
let outputs = session.currentRoute.outputs
|
||
.map { "\($0.portType.rawValue):\($0.portName)" }
|
||
.joined(separator: ",")
|
||
let available = session.availableInputs?
|
||
.map { "\($0.portType.rawValue):\($0.portName)" }
|
||
.joined(separator: ",") ?? ""
|
||
return "category=\(session.category.rawValue) mode=\(session.mode.rawValue) "
|
||
+ "opts=\(session.categoryOptions.rawValue) inputAvail=\(session.isInputAvailable) "
|
||
+ "routeIn=[\(inputs)] routeOut=[\(outputs)] availIn=[\(available)]"
|
||
}
|
||
}
|
||
|
||
private final class AudioTapDiagnostics: @unchecked Sendable {
|
||
private let label: String
|
||
private let onLevel: (@Sendable (Float) -> Void)?
|
||
private let lock = NSLock()
|
||
private var bufferCount: Int = 0
|
||
private var lastLoggedAt = Date.distantPast
|
||
private var lastLevelEmitAt = Date.distantPast
|
||
private var maxRmsWindow: Float = 0
|
||
private var lastRms: Float = 0
|
||
|
||
init(label: String, onLevel: (@Sendable (Float) -> Void)? = nil) {
|
||
self.label = label
|
||
self.onLevel = onLevel
|
||
}
|
||
|
||
func onBuffer(_ buffer: AVAudioPCMBuffer) {
|
||
var shouldLog = false
|
||
var shouldEmitLevel = false
|
||
var count = 0
|
||
self.lock.lock()
|
||
self.bufferCount += 1
|
||
count = self.bufferCount
|
||
let now = Date()
|
||
if now.timeIntervalSince(self.lastLoggedAt) >= 1.0 {
|
||
self.lastLoggedAt = now
|
||
shouldLog = true
|
||
}
|
||
if now.timeIntervalSince(self.lastLevelEmitAt) >= 0.12 {
|
||
self.lastLevelEmitAt = now
|
||
shouldEmitLevel = true
|
||
}
|
||
self.lock.unlock()
|
||
|
||
let rate = buffer.format.sampleRate
|
||
let ch = buffer.format.channelCount
|
||
let frames = buffer.frameLength
|
||
|
||
var rms: Float?
|
||
if let data = buffer.floatChannelData?.pointee {
|
||
let n = Int(frames)
|
||
if n > 0 {
|
||
var sum: Float = 0
|
||
for i in 0..<n {
|
||
let v = data[i]
|
||
sum += v * v
|
||
}
|
||
rms = sqrt(sum / Float(n))
|
||
}
|
||
}
|
||
|
||
let resolvedRms = rms ?? 0
|
||
self.lock.lock()
|
||
self.lastRms = resolvedRms
|
||
if resolvedRms > self.maxRmsWindow { self.maxRmsWindow = resolvedRms }
|
||
let maxRms = self.maxRmsWindow
|
||
if shouldLog { self.maxRmsWindow = 0 }
|
||
self.lock.unlock()
|
||
|
||
if shouldEmitLevel, let onLevel {
|
||
onLevel(resolvedRms)
|
||
}
|
||
|
||
guard shouldLog else { return }
|
||
GatewayDiagnostics.log(
|
||
"\(self.label) mic: buffers=\(count) frames=\(frames) rate=\(Int(rate))Hz ch=\(ch) "
|
||
+ "rms=\(String(format: "%.4f", resolvedRms)) max=\(String(format: "%.4f", maxRms))")
|
||
}
|
||
}
|
||
|
||
extension TalkModeManager: TalkRealtimeWebRTCSessionDelegate {
|
||
func realtimeSession(_ session: TalkRealtimeWebRTCSession, didChangeStatus status: String) {
|
||
guard session === self.realtimeSession else { return }
|
||
GatewayDiagnostics.log("talk.timeline realtime status=\(status)")
|
||
self.statusText = status
|
||
self.isListening = status == "Listening"
|
||
self.isSpeaking = status == "Speaking"
|
||
if status == "Thinking" {
|
||
self.isListening = false
|
||
self.isSpeaking = false
|
||
self.isUserSpeechDetected = false
|
||
}
|
||
}
|
||
|
||
func realtimeSession(_ session: TalkRealtimeWebRTCSession, didDetectInputSpeech active: Bool) {
|
||
guard session === self.realtimeSession else { return }
|
||
self.isUserSpeechDetected = active
|
||
if active {
|
||
self.isListening = true
|
||
}
|
||
}
|
||
|
||
func realtimeSession(_ session: TalkRealtimeWebRTCSession, didReceiveUserTranscript text: String) {
|
||
guard session === self.realtimeSession else { return }
|
||
let trimmed = text.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !trimmed.isEmpty else { return }
|
||
GatewayDiagnostics.log("talk.timeline realtime user transcript chars=\(trimmed.count)")
|
||
self.lastTranscript = trimmed
|
||
self.lastHeard = Date()
|
||
}
|
||
|
||
func realtimeSession(_ session: TalkRealtimeWebRTCSession, didReceiveAssistantTranscript text: String) {
|
||
guard session === self.realtimeSession else { return }
|
||
let trimmed = text.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !trimmed.isEmpty else { return }
|
||
GatewayDiagnostics.log("talk.timeline realtime assistant transcript chars=\(trimmed.count)")
|
||
self.lastSpokenText = trimmed
|
||
}
|
||
|
||
func realtimeSessionDidFinish(_ session: TalkRealtimeWebRTCSession) {
|
||
guard session === self.realtimeSession else { return }
|
||
self.realtimeSession = nil
|
||
self.isListening = false
|
||
self.isSpeaking = false
|
||
self.isUserSpeechDetected = false
|
||
if self.isEnabled {
|
||
self.statusText = self.gatewayConnected ? "Ready" : "Offline"
|
||
}
|
||
}
|
||
}
|
||
|
||
#if DEBUG
|
||
extension TalkModeManager {
|
||
static func _test_isPCMFormatRejectedByAPI(_ error: Error?) -> Bool {
|
||
self.isPCMFormatRejectedByAPI(error)
|
||
}
|
||
|
||
func _test_seedTranscript(_ transcript: String) {
|
||
self.lastTranscript = transcript
|
||
self.lastHeard = Date()
|
||
}
|
||
|
||
func _test_handleTranscript(_ transcript: String, isFinal: Bool) async {
|
||
await self.handleTranscript(transcript: transcript, isFinal: isFinal)
|
||
}
|
||
|
||
func _test_backdateLastHeard(seconds: TimeInterval) {
|
||
self.lastHeard = Date().addingTimeInterval(-seconds)
|
||
}
|
||
|
||
func _test_runSilenceCheck() async {
|
||
await self.checkSilence()
|
||
}
|
||
|
||
func _test_incrementalReset() {
|
||
self.incrementalSpeechBuffer = IncrementalSpeechBuffer()
|
||
}
|
||
|
||
func _test_incrementalIngest(_ text: String, isFinal: Bool) -> [String] {
|
||
self.incrementalSpeechBuffer.ingest(text: text, isFinal: isFinal)
|
||
}
|
||
}
|
||
#endif
|
||
|
||
private struct IncrementalSpeechContext: Equatable {
|
||
let apiKey: String?
|
||
let voiceId: String?
|
||
let modelId: String?
|
||
let outputFormat: String?
|
||
let language: String?
|
||
let directive: TalkDirective?
|
||
let canUseElevenLabs: Bool
|
||
}
|
||
|
||
private struct IncrementalSpeechPrefetchState {
|
||
let id: UUID
|
||
let segment: String
|
||
let context: IncrementalSpeechContext
|
||
let outputFormat: String?
|
||
var chunks: [Data]?
|
||
let task: Task<Void, Never>
|
||
}
|
||
|
||
private struct IncrementalPrefetchedAudio {
|
||
let chunks: [Data]
|
||
let outputFormat: String?
|
||
}
|
||
|
||
// swiftlint:enable type_body_length file_length
|