YeonGyu-Kim
41 lines
1.2 KiB
TypeScript
41 lines
1.2 KiB
TypeScript
import { getAgentDisplayName } from "../shared/agent-display-names";
|
|
|
|
const CORE_AGENT_ORDER: ReadonlyArray<{ displayName: string; order: number }> = [
|
|
{ displayName: getAgentDisplayName("sisyphus"), order: 1 },
|
|
{ displayName: getAgentDisplayName("hephaestus"), order: 2 },
|
|
{ displayName: getAgentDisplayName("prometheus"), order: 3 },
|
|
{ displayName: getAgentDisplayName("atlas"), order: 4 },
|
|
];
|
|
|
|
function injectOrderField(
|
|
agentConfig: unknown,
|
|
order: number,
|
|
): unknown {
|
|
if (typeof agentConfig === "object" && agentConfig !== null) {
|
|
return { ...agentConfig, order };
|
|
}
|
|
return agentConfig;
|
|
}
|
|
|
|
export function reorderAgentsByPriority(
|
|
agents: Record<string, unknown>,
|
|
): Record<string, unknown> {
|
|
const ordered: Record<string, unknown> = {};
|
|
const seen = new Set<string>();
|
|
|
|
for (const { displayName, order } of CORE_AGENT_ORDER) {
|
|
if (Object.prototype.hasOwnProperty.call(agents, displayName)) {
|
|
ordered[displayName] = injectOrderField(agents[displayName], order);
|
|
seen.add(displayName);
|
|
}
|
|
}
|
|
|
|
for (const [key, value] of Object.entries(agents)) {
|
|
if (!seen.has(key)) {
|
|
ordered[key] = value;
|
|
}
|
|
}
|
|
|
|
return ordered;
|
|
}
|