Files
devclaw-gitea/lib/tiers.ts

98 lines
3.0 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 = ["qa-engineer", "manual-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",
"qa-engineer": "anthropic/claude-sonnet-4-5",
"manual-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: {
"qa-engineer": "anthropic/claude-sonnet-4-5",
"manual-tester": "anthropic/claude-haiku-4-5",
},
};
/** Emoji used in announcements per tier. */
export const TIER_EMOJI: Record<Tier, string> = {
junior: "⚡",
medior: "🔧",
senior: "🧠",
"qa-engineer": "🔍",
"manual-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;
}
/**
* Migration map from old model-alias session keys to tier names.
* Used by migrateWorkerState() in projects.ts.
*/
export const TIER_MIGRATION: Record<string, string> = {
haiku: "junior",
sonnet: "medior",
opus: "senior",
grok: "qa-engineer",
qa: "qa-engineer",
};