private val context: Context,

private val scope: CoroutineScope,

private val session: GatewaySession,

private val supportsChatSubscribe: Boolean,

private val isConnected: () -> Boolean,

private val onBeforeSpeak: suspend () -> Unit = {},

private val onAfterSpeak: suspend () -> Unit = {},

private const val realtimeSampleRateHz = 24_000

private const val realtimeAudioFrameMs = 100

private const val listenWatchdogMs = 12_000L

private const val chatFinalWaitWithSubscribeMs = 45_000L

private const val chatFinalWaitWithoutSubscribeMs = 6_000L

private const val chatFinalWaitMs = 45_000L

private const val maxCachedRunCompletions = 128

private const val maxConversationEntries = 40

private const val realtimePlaybackBufferMs = 240

private val completedRunsLock = Any()

private val completedRunStates = LinkedHashMap<String, Boolean>()

private val completedRunTexts = LinkedHashMap<String, String>()

private var chatSubscribedSessionKey: String? = null

private var configLoaded = false

private var executionMode = TalkModeExecutionMode.Native

private val startGeneration = AtomicLong(0L)

}

}

suspend fun ensureChatSubscribed() {

reloadConfig()

subscribeChatIfNeeded(session = session, sessionKey = mainSessionKey.ifBlank { "main" })

}

fun setMainSessionKey(sessionKey: String?) {

val trimmed = sessionKey?.trim().orEmpty()

if (trimmed.isEmpty()) return

scope.launch {

try {

reloadConfig()

subscribeChatIfNeeded(session = session, sessionKey = mainSessionKey.ifBlank { "main" })

val startedAt = System.currentTimeMillis().toDouble() / 1000.0

val prompt = buildPrompt(command)

val runId = sendChat(prompt, session)

_statusText.value = "Off"

stopRealtimeRelay()

stopSpeaking()

chatSubscribedSessionKey = null

pendingRunId = null

pendingFinal?.cancel()

pendingFinal = null

try {

val startedAt = System.currentTimeMillis().toDouble() / 1000.0

subscribeChatIfNeeded(session = session, sessionKey = mainSessionKey)

Log.d(tag, "chat.send start sessionKey=${mainSessionKey.ifBlank { "main" }} chars=${prompt.length}")

val runId = sendChat(prompt, session)

Log.d(tag, "chat.send ok runId=$runId")

return payload

}

private suspend fun subscribeChatIfNeeded(

session: GatewaySession,

sessionKey: String,

) {

if (!supportsChatSubscribe) return

val key = sessionKey.trim()

if (key.isEmpty()) return

if (chatSubscribedSessionKey == key) return

val sent = session.sendNodeEvent("chat.subscribe", """{"sessionKey":"$key"}""")

if (sent) {

chatSubscribedSessionKey = key

Log.d(tag, "chat.subscribe ok sessionKey=$key")

} else {

Log.w(tag, "chat.subscribe failed sessionKey=$key")

}

}

private fun buildPrompt(transcript: String): String {

val lines =

mutableListOf(

consumeRunCompletion(runId)?.let { return it }

val timeoutMs = if (supportsChatSubscribe) chatFinalWaitWithSubscribeMs else chatFinalWaitWithoutSubscribeMs

val result =

try {

withTimeout(timeoutMs) { deferred.await() }

withTimeout(chatFinalWaitMs) { deferred.await() }

} catch (_: TimeoutCancellationException) {

false

}