fix(chat): restore pending approval guard, tighten encapsulation, drop dormant tracker
This commit is contained in:
1 parent
0af4782da1
commit
7b198fa6d6
6 files changed
+40
-216
No files matched your search
@@ -81,8 +81,12 @@ class ChatApprovalsDelegate(
|
||||
@Suppress("UNCHECKED_CAST")
|
||||
val approvals = (result as? Map<*, *>)?.get("approvals") as? List<*>
|
||||
val first = approvals?.filterIsInstance<Map<*, *>>()?.firstOrNull() ?: return
|
||||
val requestId = first["request_id"] as? String ?: return
|
||||
val sessionId = runtimeSessionId() ?: uiState.value.currentSessionId
|
||||
maybeSurfacePendingApproval(first, sessionId)
|
||||
val alreadyShown =
|
||||
uiState.value.messages.any { it.approvalInfo?.requestId == requestId }
|
||||
if (alreadyShown) return
|
||||
handleApprovalRequest(parseApprovalMap(first, sessionId))
|
||||
}
|
||||
|
||||
fun handleApprovalRespondResult(result: Any?) {
|
||||
|
||||
@@ -19,7 +19,7 @@ import java.util.concurrent.ConcurrentHashMap
|
||||
* Tracks an in-flight `config.set key=model` RPC and the model it would replace
|
||||
* so confirmation dialog dismiss can roll back optimistically.
|
||||
*/
|
||||
data class ActiveModelSwitch(
|
||||
private data class ActiveModelSwitch(
|
||||
val spec: String,
|
||||
val previousModel: String?,
|
||||
)
|
||||
@@ -47,10 +47,10 @@ class ChatModelSwitchDelegate(
|
||||
private val getPinnedModels: () -> List<PinnedModel> = { AuthManager.getPinnedModels() },
|
||||
private val savePinnedModels: (List<PinnedModel>) -> Unit = { AuthManager.savePinnedModels(it) },
|
||||
) {
|
||||
val pendingModelSwitchRequests = ConcurrentHashMap<String, ActiveModelSwitch>()
|
||||
var activeModelSwitchConfirmation: ActiveModelSwitch? = null
|
||||
var optimisticPreviousModel: String? = null
|
||||
var cachedModelOptions: List<ModelProvider> = emptyList()
|
||||
private val pendingModelSwitchRequests = ConcurrentHashMap<String, ActiveModelSwitch>()
|
||||
private var activeModelSwitchConfirmation: ActiveModelSwitch? = null
|
||||
private var optimisticPreviousModel: String? = null
|
||||
private var cachedModelOptions: List<ModelProvider> = emptyList()
|
||||
|
||||
fun isModelPickerCommand(command: String): Boolean {
|
||||
val trimmed = command.trim()
|
||||
|
||||
@@ -1,111 +0,0 @@
|
||||
package com.m57.hermescontrol.ui.chat
|
||||
|
||||
import java.util.concurrent.ConcurrentHashMap
|
||||
|
||||
/**
|
||||
* Tracks an in-flight RPC request associated with a specific session generation
|
||||
* and resume sequence to guard against stale results/errors from previous sessions.
|
||||
*/
|
||||
data class TrackedSessionRequest(
|
||||
val generation: Long,
|
||||
val resumeSequence: Long = 0L,
|
||||
val sessionId: String? = null,
|
||||
)
|
||||
|
||||
/**
|
||||
* Owns RPC id-to-method mapping, session request generations, hydration sequences,
|
||||
* and staleness detection.
|
||||
*
|
||||
* Extracted behavior-preservingly from [ChatViewModel] to ensure deterministic
|
||||
* RPC tracking across session creation, switches, resumes, and reconnects.
|
||||
*/
|
||||
class ChatRpcRequestTracker(
|
||||
private val getCurrentSessionId: () -> String?,
|
||||
) {
|
||||
/** Maps an in-flight RPC id to its method for UI error labeling. */
|
||||
private val idToMethod = ConcurrentHashMap<String, String>()
|
||||
private val sessionRequestById = ConcurrentHashMap<String, TrackedSessionRequest>()
|
||||
|
||||
var sessionGeneration: Long = 0L
|
||||
private set
|
||||
|
||||
var resumeRequestSequence: Long = 0L
|
||||
private set
|
||||
var activeResumeRequestSequence: Long = 0L
|
||||
private set
|
||||
|
||||
var hydrationRequestSequence: Long = 0L
|
||||
private set
|
||||
var activeHydrationRequestSequence: Long = 0L
|
||||
private set
|
||||
|
||||
var resumedGeneration: Long = -1L
|
||||
var hydratedGeneration: Long = -1L
|
||||
|
||||
fun nextSessionGeneration(): Long {
|
||||
sessionGeneration++
|
||||
return sessionGeneration
|
||||
}
|
||||
|
||||
fun nextResumeSequence(): Long {
|
||||
resumeRequestSequence++
|
||||
activeResumeRequestSequence = resumeRequestSequence
|
||||
return resumeRequestSequence
|
||||
}
|
||||
|
||||
fun nextHydrationSequence(): Long {
|
||||
hydrationRequestSequence++
|
||||
activeHydrationRequestSequence = hydrationRequestSequence
|
||||
return hydrationRequestSequence
|
||||
}
|
||||
|
||||
fun trackRequest(
|
||||
id: String,
|
||||
method: String,
|
||||
) {
|
||||
idToMethod[id] = method
|
||||
}
|
||||
|
||||
fun trackSessionRequest(
|
||||
id: String,
|
||||
method: String,
|
||||
generation: Long,
|
||||
resumeSequence: Long = 0L,
|
||||
sessionId: String? = null,
|
||||
) {
|
||||
sessionRequestById[id] = TrackedSessionRequest(generation, resumeSequence, sessionId)
|
||||
trackRequest(id, method)
|
||||
}
|
||||
|
||||
fun isCurrentSessionRequest(
|
||||
sessionId: String,
|
||||
generation: Long,
|
||||
): Boolean = generation == sessionGeneration && sessionId == getCurrentSessionId()
|
||||
|
||||
fun isCurrentHydration(
|
||||
sessionId: String,
|
||||
generation: Long,
|
||||
requestSequence: Long,
|
||||
): Boolean = requestSequence == activeHydrationRequestSequence && isCurrentSessionRequest(sessionId, generation)
|
||||
|
||||
fun isStaleSessionRequest(id: String): Boolean = sessionRequestById[id]?.let(::isStaleSessionRequest) == true
|
||||
|
||||
fun isStaleSessionRequest(request: TrackedSessionRequest): Boolean =
|
||||
request.generation != sessionGeneration ||
|
||||
(request.sessionId != null && request.sessionId != getCurrentSessionId()) ||
|
||||
(request.resumeSequence != 0L && request.resumeSequence != activeResumeRequestSequence)
|
||||
|
||||
fun removeMethod(id: String): String? = idToMethod.remove(id)
|
||||
|
||||
fun removeSessionRequest(id: String): TrackedSessionRequest? = sessionRequestById.remove(id)
|
||||
|
||||
fun forgetRequest(id: String) {
|
||||
idToMethod.remove(id)
|
||||
sessionRequestById.remove(id)
|
||||
}
|
||||
|
||||
fun resetSessionStateTracking() {
|
||||
resumedGeneration = -1L
|
||||
hydratedGeneration = -1L
|
||||
}
|
||||
}
|
||||
@@ -510,11 +510,6 @@ class ChatViewModel(
|
||||
private val sessionRequestById = ConcurrentHashMap<String, SessionRequest>()
|
||||
private var sessionGeneration = 0L
|
||||
|
||||
private data class ActiveModelSwitch(
|
||||
val spec: String,
|
||||
val previousModel: String?,
|
||||
)
|
||||
|
||||
private var resumeRequestSequence = 0L
|
||||
private var activeResumeRequestSequence = 0L
|
||||
private var hydrationRequestSequence = 0L
|
||||
|
||||
Reference in new issue
Block a user