feat(team-mode): add member session resolution and routing logic
This commit is contained in:
@@ -0,0 +1,63 @@
|
|||||||
|
import type { TeamModeConfig } from "../../config/schema/team-mode"
|
||||||
|
import { log } from "../../shared/logger"
|
||||||
|
import { lookupTeamSession } from "./team-session-registry"
|
||||||
|
import { listActiveTeams, loadRuntimeState } from "./team-state-store/store"
|
||||||
|
|
||||||
|
export type ResolvedMemberSession = {
|
||||||
|
teamRunId: string
|
||||||
|
memberName: string
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function findResolvedMemberSession(
|
||||||
|
sessionID: string,
|
||||||
|
config: TeamModeConfig,
|
||||||
|
logContext: string,
|
||||||
|
): Promise<ResolvedMemberSession | null> {
|
||||||
|
const registryEntry = lookupTeamSession(sessionID)
|
||||||
|
if (registryEntry?.role === "member") {
|
||||||
|
try {
|
||||||
|
const runtimeState = await loadRuntimeState(registryEntry.teamRunId, config)
|
||||||
|
const memberEntry = runtimeState.members.find(
|
||||||
|
(member) => member.name === registryEntry.memberName
|
||||||
|
&& (member.sessionId === undefined || member.sessionId === sessionID),
|
||||||
|
)
|
||||||
|
|
||||||
|
if (memberEntry !== undefined) {
|
||||||
|
return {
|
||||||
|
teamRunId: runtimeState.teamRunId,
|
||||||
|
memberName: memberEntry.name,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
log(`${logContext} registry lookup failed`, {
|
||||||
|
event: `${logContext}-registry-error`,
|
||||||
|
teamRunId: registryEntry.teamRunId,
|
||||||
|
sessionID,
|
||||||
|
error: error instanceof Error ? error.message : String(error),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const activeTeams = await listActiveTeams(config)
|
||||||
|
for (const activeTeam of activeTeams) {
|
||||||
|
try {
|
||||||
|
const runtimeState = await loadRuntimeState(activeTeam.teamRunId, config)
|
||||||
|
const memberEntry = runtimeState.members.find((member) => member.sessionId === sessionID)
|
||||||
|
if (memberEntry !== undefined) {
|
||||||
|
return {
|
||||||
|
teamRunId: runtimeState.teamRunId,
|
||||||
|
memberName: memberEntry.name,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
log(`${logContext} skipped runtime`, {
|
||||||
|
event: `${logContext}-runtime-error`,
|
||||||
|
teamRunId: activeTeam.teamRunId,
|
||||||
|
sessionID,
|
||||||
|
error: error instanceof Error ? error.message : String(error),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return null
|
||||||
|
}
|
||||||
@@ -0,0 +1,69 @@
|
|||||||
|
import { stripAgentListSortPrefix } from "../../shared/agent-display-names"
|
||||||
|
import { resolveRegisteredAgentName } from "../claude-code-session-state"
|
||||||
|
import { applySessionPromptParams } from "../../shared/session-prompt-params-helpers"
|
||||||
|
import { SessionCategoryRegistry } from "../../shared/session-category-registry"
|
||||||
|
import type { RuntimeStateMember } from "./types"
|
||||||
|
|
||||||
|
type PromptGenerationModel = {
|
||||||
|
reasoningEffort?: string
|
||||||
|
temperature?: number
|
||||||
|
top_p?: number
|
||||||
|
maxTokens?: number
|
||||||
|
thinking?: { type: "enabled" | "disabled"; budgetTokens?: number }
|
||||||
|
}
|
||||||
|
|
||||||
|
export type TeamMemberPromptBody = {
|
||||||
|
parts: Array<{ type: "text"; text: string }>
|
||||||
|
agent?: string
|
||||||
|
model?: { providerID: string; modelID: string }
|
||||||
|
variant?: string
|
||||||
|
temperature?: number
|
||||||
|
topP?: number
|
||||||
|
maxOutputTokens?: number
|
||||||
|
options?: Record<string, unknown>
|
||||||
|
}
|
||||||
|
|
||||||
|
function buildPromptGenerationParams(model: PromptGenerationModel | undefined): Omit<TeamMemberPromptBody, "parts" | "agent" | "model" | "variant"> {
|
||||||
|
if (!model) {
|
||||||
|
return {}
|
||||||
|
}
|
||||||
|
|
||||||
|
const promptOptions: Record<string, unknown> = {
|
||||||
|
...(model.reasoningEffort ? { reasoningEffort: model.reasoningEffort } : {}),
|
||||||
|
...(model.thinking ? { thinking: model.thinking } : {}),
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
...(model.temperature !== undefined ? { temperature: model.temperature } : {}),
|
||||||
|
...(model.top_p !== undefined ? { topP: model.top_p } : {}),
|
||||||
|
...(model.maxTokens !== undefined ? { maxOutputTokens: model.maxTokens } : {}),
|
||||||
|
...(Object.keys(promptOptions).length > 0 ? { options: promptOptions } : {}),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function applyMemberSessionRouting(sessionID: string, member: RuntimeStateMember): void {
|
||||||
|
if (member.category) {
|
||||||
|
SessionCategoryRegistry.register(sessionID, member.category)
|
||||||
|
}
|
||||||
|
|
||||||
|
applySessionPromptParams(sessionID, member.model)
|
||||||
|
}
|
||||||
|
|
||||||
|
export function buildMemberPromptBody(member: RuntimeStateMember, text: string): TeamMemberPromptBody {
|
||||||
|
const normalizedAgent = member.subagent_type ? stripAgentListSortPrefix(member.subagent_type) : undefined
|
||||||
|
const launchAgent = resolveRegisteredAgentName(normalizedAgent) ?? normalizedAgent
|
||||||
|
const model = member.model
|
||||||
|
? {
|
||||||
|
providerID: member.model.providerID,
|
||||||
|
modelID: member.model.modelID,
|
||||||
|
}
|
||||||
|
: undefined
|
||||||
|
|
||||||
|
return {
|
||||||
|
...(launchAgent ? { agent: launchAgent } : {}),
|
||||||
|
...(model ? { model } : {}),
|
||||||
|
...(member.model?.variant ? { variant: member.model.variant } : {}),
|
||||||
|
...buildPromptGenerationParams(member.model),
|
||||||
|
parts: [{ type: "text", text }],
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user