diff --git a/src/hooks/ralph-loop/continuation-prompt-injector-agent-resolution.test.ts b/src/hooks/ralph-loop/continuation-prompt-injector-agent-resolution.test.ts
new file mode 100644
index 000000000..8b5af4070
--- /dev/null
+++ b/src/hooks/ralph-loop/continuation-prompt-injector-agent-resolution.test.ts
@@ -0,0 +1,47 @@
+///
+
+import { afterEach, describe, expect, test } from "bun:test"
+import type { PluginInput } from "@opencode-ai/plugin"
+
+import {
+ _resetForTesting,
+ registerAgentName,
+} from "../../features/claude-code-session-state"
+import { releaseAllPromptAsyncReservationsForTesting } from "../shared/prompt-async-gate"
+import { unsafeTestValue } from "../../../test-support/unsafe-test-value"
+import { injectContinuationPrompt } from "./continuation-prompt-injector"
+
+describe("ralph-loop continuation prompt agent resolution", () => {
+ afterEach(() => {
+ releaseAllPromptAsyncReservationsForTesting()
+ _resetForTesting()
+ })
+
+ test("#given OpenCode registered Atlas under legacy display name #when inherited agent is config key #then prompt uses registered name", async () => {
+ // given
+ registerAgentName("Atlas (Plan Executor)")
+ let capturedAgent: string | undefined
+ const ctx = unsafeTestValue({
+ client: {
+ session: {
+ messages: async () => ({ data: [{ info: { agent: "atlas" } }] }),
+ promptAsync: async (input: { readonly body: { readonly agent?: string } }) => {
+ capturedAgent = input.body.agent
+ return {}
+ },
+ },
+ },
+ })
+
+ // when
+ await injectContinuationPrompt(ctx, {
+ sessionID: "ses_ralph_registered_atlas",
+ prompt: "continue",
+ directory: "/tmp/test",
+ apiTimeoutMs: 50,
+ })
+
+ // then
+ expect(capturedAgent).toBe("Atlas (Plan Executor)")
+ })
+})
diff --git a/src/hooks/ralph-loop/continuation-prompt-injector.ts b/src/hooks/ralph-loop/continuation-prompt-injector.ts
index 7d54543cb..8d384a33a 100644
--- a/src/hooks/ralph-loop/continuation-prompt-injector.ts
+++ b/src/hooks/ralph-loop/continuation-prompt-injector.ts
@@ -10,7 +10,8 @@ import {
normalizeSDKResponse,
resolveInheritedPromptTools,
} from "../../shared"
-import { normalizeAgentForPrompt, stripAgentListSortPrefix } from "../../shared/agent-display-names"
+import { resolveRegisteredAgentName } from "../../features/claude-code-session-state"
+import { normalizeAgentForPromptKey, stripAgentListSortPrefix } from "../../shared/agent-display-names"
import { dispatchInternalPrompt } from "../shared/prompt-async-gate"
type MessageInfo = {
@@ -62,20 +63,10 @@ function createPromptAsyncError(prefix: string, error: unknown): Error {
}
function normalizeInheritedAgentForPrompt(agent: string | undefined): string | undefined {
- if (typeof agent !== "string") {
- return undefined
- }
-
- const inheritedAgent = stripAgentListSortPrefix(agent).trim()
- if (!inheritedAgent) {
- return undefined
- }
-
- if (inheritedAgent.includes(" - ")) {
- return inheritedAgent
- }
-
- return normalizeAgentForPrompt(inheritedAgent)
+ const resolvedAgent = resolveRegisteredAgentName(agent) ?? normalizeAgentForPromptKey(agent)
+ if (typeof resolvedAgent !== "string") return undefined
+ const cleanAgent = stripAgentListSortPrefix(resolvedAgent).trim()
+ return cleanAgent || undefined
}
export async function injectContinuationPrompt(
diff --git a/src/hooks/todo-continuation-enforcer/continuation-injection-agent-resolution.test.ts b/src/hooks/todo-continuation-enforcer/continuation-injection-agent-resolution.test.ts
new file mode 100644
index 000000000..3cc5df7cb
--- /dev/null
+++ b/src/hooks/todo-continuation-enforcer/continuation-injection-agent-resolution.test.ts
@@ -0,0 +1,54 @@
+///
+
+import { afterEach, describe, expect, test } from "bun:test"
+import type { PluginInput } from "@opencode-ai/plugin"
+
+import {
+ _resetForTesting,
+ registerAgentName,
+} from "../../features/claude-code-session-state"
+import { releaseAllPromptAsyncReservationsForTesting } from "../shared/prompt-async-gate"
+import { unsafeTestValue } from "../../../test-support/unsafe-test-value"
+import { injectContinuation } from "./continuation-injection"
+
+describe("todo continuation registered agent resolution", () => {
+ afterEach(() => {
+ releaseAllPromptAsyncReservationsForTesting()
+ _resetForTesting()
+ })
+
+ test("#given OpenCode registered Atlas under legacy display name #when continuation inherits config key #then prompt uses registered name", async () => {
+ // given
+ registerAgentName("Atlas (Plan Executor)")
+ let capturedAgent: string | undefined
+ const ctx = unsafeTestValue({
+ directory: "/tmp/test",
+ client: {
+ session: {
+ todo: async () => ({ data: [{ id: "1", content: "todo", status: "pending", priority: "high" }] }),
+ promptAsync: async (input: { readonly body: { readonly agent?: string } }) => {
+ capturedAgent = input.body.agent
+ return {}
+ },
+ },
+ },
+ })
+ const sessionStateStore = {
+ getExistingState: () => ({ inFlight: false, lastInjectedAt: 0, consecutiveFailures: 0 }),
+ }
+
+ // when
+ await injectContinuation({
+ ctx,
+ sessionID: "ses_todo_registered_atlas",
+ resolvedInfo: {
+ agent: "atlas",
+ model: { providerID: "openai", modelID: "gpt-5.5" },
+ },
+ sessionStateStore: unsafeTestValue(sessionStateStore),
+ })
+
+ // then
+ expect(capturedAgent).toBe("Atlas (Plan Executor)")
+ })
+})
diff --git a/src/hooks/todo-continuation-enforcer/continuation-injection.ts b/src/hooks/todo-continuation-enforcer/continuation-injection.ts
index f22da95f7..8b799595d 100644
--- a/src/hooks/todo-continuation-enforcer/continuation-injection.ts
+++ b/src/hooks/todo-continuation-enforcer/continuation-injection.ts
@@ -20,8 +20,8 @@ import { log } from "../../shared/logger"
import { isSqliteBackend } from "../../shared/opencode-storage-detection"
import {
getAgentConfigKey,
- normalizeAgentForPrompt,
normalizeAgentForPromptKey,
+ stripAgentListSortPrefix,
} from "../../shared/agent-display-names"
import { dispatchInternalPrompt, isInternalPromptDispatchAccepted } from "../shared/prompt-async-gate"
@@ -132,9 +132,8 @@ export async function injectContinuation(args: {
tools = tools ?? previousMessage?.tools
}
- const promptAgent = normalizeAgentForPromptKey(agentName)
- const resolvedAgent = resolveRegisteredAgentName(agentName)
- const launchAgent = normalizeAgentForPrompt(resolvedAgent ?? agentName)
+ const promptAgent = resolveRegisteredAgentName(agentName) ?? normalizeAgentForPromptKey(agentName)
+ const launchAgent = promptAgent ? stripAgentListSortPrefix(promptAgent).trim() || undefined : undefined
if (promptAgent && skipAgents.some(s => getAgentConfigKey(s) === getAgentConfigKey(promptAgent))) {
log(`[${HOOK_NAME}] Skipped: agent in skipAgents list`, { sessionID, agent: agentName })