elias/oh-my-opencode
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(athena): enable retry and bound growth for agent-switch fallback markers

Browse Source 
Delete marker from processedFallbackMessages on failure so message can be retried. Add MAX_PROCESSED_FALLBACK_MARKERS=500 with eviction to prevent unbounded Set growth.

Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-opencode)

Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
ismeth 2026-02-18 20:56:22 +01:00 committed by YeonGyu-Kim
parent e8042fa445
commit 5dfe0a34fc
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/hooks/agent-switch/hook.ts
View File

@ -12,6 +12,7 @@ import {
} from "./fallback-handoff"
const processedFallbackMessages = new Set<string>()
const MAX_PROCESSED_FALLBACK_MARKERS = 500
function getSessionIDFromStatusEvent(input: { event: { properties?: Record<string, unknown> } }): string | undefined {
const props = input.event.properties as Record<string, unknown> | undefined
@ -97,6 +98,15 @@ export function createAgentSwitchHook(ctx: PluginInput) {
}
processedFallbackMessages.add(marker)
// Prevent unbounded growth of the Set
if (processedFallbackMessages.size > MAX_PROCESSED_FALLBACK_MARKERS) {
const iterator = processedFallbackMessages.values()
const oldest = iterator.next().value
if (oldest) {
processedFallbackMessages.delete(oldest)
}
}
// If switch_agent already queued a handoff, do not synthesize fallback behavior.
if (getPendingSwitch(sessionID)) {
return
@ -125,6 +135,7 @@ export function createAgentSwitchHook(ctx: PluginInput) {
source: "athena-message-fallback",
})
} catch (error) {
processedFallbackMessages.delete(marker)
log("[agent-switch] Failed to recover fallback handoff from Athena message", {
sessionID,
messageID,