refactor(switch-agent): replace as-any with type guard, use const tuple for agent names

This commit is contained in:
ismeth
2026-03-14 16:27:56 +01:00
committed by YeonGyu-Kim
parent 9c226948d3
commit 68cd8814d8
2 changed files with 26 additions and 3 deletions
+22 -3
View File
@@ -1,7 +1,7 @@
import { tool, type ToolDefinition } from "@opencode-ai/plugin"
import { normalizeAgentForPrompt } from "../../shared/agent-display-names"
import { log } from "../../shared/logger"
import type { SwitchAgentArgs } from "./types"
import { SWITCHABLE_AGENT_NAMES, type SwitchAgentArgs } from "./types"
const DESCRIPTION =
"Switch the active session agent. After calling this tool, the session will transition to the specified agent " +
@@ -10,7 +10,15 @@ const DESCRIPTION =
"Permanent one-way handoff. Use ONLY when you're the wrong agent for the overall job, NEVER for subtasks (use task()). " +
"Targets: atlas, prometheus, sisyphus, hephaestus."
const ALLOWED_AGENTS = new Set(["atlas", "prometheus", "sisyphus", "hephaestus"])
const ALLOWED_AGENTS = new Set<string>(SWITCHABLE_AGENT_NAMES)
type TuiClient = {
post: (input: {
url: string
body: { sessionID: string }
headers?: Record<string, string>
}) => Promise<unknown>
}
type SessionClient = {
session: {
@@ -44,9 +52,20 @@ function extractSessionId(response: unknown): string | undefined {
return undefined
}
function hasTuiClient(client: SessionClient): client is SessionClient & { _client: TuiClient } {
const maybeClient = Reflect.get(client as object, "_client")
if (typeof maybeClient !== "object" || maybeClient === null) {
return false
}
return typeof Reflect.get(maybeClient, "post") === "function"
}
async function navigateTuiToSession(client: SessionClient, sessionID: string): Promise<boolean> {
if (!hasTuiClient(client)) {
return false
}
try {
await (client as any)._client.post({
await client._client.post({
url: "/tui/select-session",
body: { sessionID },
headers: { "Content-Type": "application/json" },
+4
View File
@@ -1,3 +1,7 @@
export const SWITCHABLE_AGENT_NAMES = ["atlas", "prometheus", "sisyphus", "hephaestus"] as const
export type SwitchableAgentName = (typeof SWITCHABLE_AGENT_NAMES)[number]
export interface SwitchAgentArgs {
agent: string
context: string