YeonGyu-Kim
70 lines
2.4 KiB
TypeScript
70 lines
2.4 KiB
TypeScript
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 }],
|
|
}
|
|
}
|