86 lines
2.6 KiB
TypeScript
86 lines
2.6 KiB
TypeScript
/**
|
|
* Developer tier definitions and model resolution.
|
|
*
|
|
* Tasks are assigned to developer tiers (junior, medior, senior, qa)
|
|
* instead of raw model names. Each tier maps to a configurable LLM model.
|
|
*/
|
|
|
|
export const DEV_TIERS = ["junior", "medior", "senior"] as const;
|
|
export const QA_TIERS = ["reviewer", "tester"] as const;
|
|
export const ALL_TIERS = [...DEV_TIERS, ...QA_TIERS] as const;
|
|
|
|
export type DevTier = (typeof DEV_TIERS)[number];
|
|
export type QaTier = (typeof QA_TIERS)[number];
|
|
export type Tier = (typeof ALL_TIERS)[number];
|
|
|
|
export const DEFAULT_MODELS: Record<Tier, string> = {
|
|
junior: "anthropic/claude-haiku-4-5",
|
|
medior: "anthropic/claude-sonnet-4-5",
|
|
senior: "anthropic/claude-opus-4-5",
|
|
reviewer: "anthropic/claude-sonnet-4-5",
|
|
tester: "anthropic/claude-haiku-4-5",
|
|
};
|
|
|
|
/** Default models by role-tier structure. */
|
|
export const DEFAULT_MODELS_BY_ROLE = {
|
|
dev: {
|
|
junior: "anthropic/claude-haiku-4-5",
|
|
medior: "anthropic/claude-sonnet-4-5",
|
|
senior: "anthropic/claude-opus-4-5",
|
|
},
|
|
qa: {
|
|
reviewer: "anthropic/claude-sonnet-4-5",
|
|
tester: "anthropic/claude-haiku-4-5",
|
|
},
|
|
};
|
|
|
|
/** Emoji used in announcements per tier. */
|
|
export const TIER_EMOJI: Record<Tier, string> = {
|
|
junior: "⚡",
|
|
medior: "🔧",
|
|
senior: "🧠",
|
|
reviewer: "🔍",
|
|
tester: "👀",
|
|
};
|
|
|
|
/** Check if a string is a valid tier name. */
|
|
export function isTier(value: string): value is Tier {
|
|
return (ALL_TIERS as readonly string[]).includes(value);
|
|
}
|
|
|
|
/** Check if a string is a valid dev tier name. */
|
|
export function isDevTier(value: string): value is DevTier {
|
|
return (DEV_TIERS as readonly string[]).includes(value);
|
|
}
|
|
|
|
/**
|
|
* Resolve a tier name to a full model ID.
|
|
*
|
|
* Resolution order:
|
|
* 1. Plugin config `models.<role>.<tier>` nested structure (user overrides)
|
|
* 2. Plugin config `models.<tier>` flat structure (backward compatibility)
|
|
* 3. DEFAULT_MODELS (hardcoded defaults)
|
|
* 4. Treat input as raw model ID (passthrough for non-tier values)
|
|
*/
|
|
export function resolveTierToModel(
|
|
tier: string,
|
|
pluginConfig?: Record<string, unknown>,
|
|
role?: "dev" | "qa",
|
|
): string {
|
|
const models = (pluginConfig as { models?: Record<string, unknown> })?.models;
|
|
|
|
// Try nested role-tier structure first
|
|
if (role && models && typeof models === "object") {
|
|
const roleModels = models[role] as Record<string, string> | undefined;
|
|
if (roleModels?.[tier]) return roleModels[tier];
|
|
}
|
|
|
|
// Fall back to flat structure for backward compatibility
|
|
if (models && typeof models === "object") {
|
|
const flatModel = (models as Record<string, string>)[tier];
|
|
if (flatModel) return flatModel;
|
|
}
|
|
|
|
return DEFAULT_MODELS[tier as Tier] ?? tier;
|
|
}
|