c9be2e1696
- Create available-models.ts for model availability checking - Create model-selection.ts for category-to-model resolution logic - Update category-resolver, subagent-resolver, and sync modules to import from new focused modules instead of monolithic sources
72 lines
2.6 KiB
TypeScript
72 lines
2.6 KiB
TypeScript
import type { CategoryConfig, CategoriesConfig } from "../../config/schema"
|
|
import { DEFAULT_CATEGORIES, CATEGORY_PROMPT_APPENDS } from "./constants"
|
|
import { resolveModel } from "../../shared/model-resolver"
|
|
import { isModelAvailable } from "../../shared/model-availability"
|
|
import { CATEGORY_MODEL_REQUIREMENTS } from "../../shared/model-requirements"
|
|
import { log } from "../../shared/logger"
|
|
|
|
export interface ResolveCategoryConfigOptions {
|
|
userCategories?: CategoriesConfig
|
|
inheritedModel?: string
|
|
systemDefaultModel?: string
|
|
availableModels?: Set<string>
|
|
}
|
|
|
|
export interface ResolveCategoryConfigResult {
|
|
config: CategoryConfig
|
|
promptAppend: string
|
|
model: string | undefined
|
|
}
|
|
|
|
/**
|
|
* Resolve the configuration for a given category name.
|
|
* Merges default and user configurations, handles model resolution.
|
|
*/
|
|
export function resolveCategoryConfig(
|
|
categoryName: string,
|
|
options: ResolveCategoryConfigOptions
|
|
): ResolveCategoryConfigResult | null {
|
|
const { userCategories, inheritedModel, systemDefaultModel, availableModels } = options
|
|
|
|
const defaultConfig = DEFAULT_CATEGORIES[categoryName]
|
|
const userConfig = userCategories?.[categoryName]
|
|
const hasExplicitUserConfig = userConfig !== undefined
|
|
|
|
// Check if category requires a specific model - bypass if user explicitly provides config
|
|
const categoryReq = CATEGORY_MODEL_REQUIREMENTS[categoryName]
|
|
if (categoryReq?.requiresModel && availableModels && !hasExplicitUserConfig) {
|
|
if (!isModelAvailable(categoryReq.requiresModel, availableModels)) {
|
|
log(`[resolveCategoryConfig] Category ${categoryName} requires ${categoryReq.requiresModel} but not available`)
|
|
return null
|
|
}
|
|
}
|
|
const defaultPromptAppend = CATEGORY_PROMPT_APPENDS[categoryName] ?? ""
|
|
|
|
if (!defaultConfig && !userConfig) {
|
|
return null
|
|
}
|
|
|
|
// Model priority for categories: user override > category default > system default
|
|
// Categories have explicit models - no inheritance from parent session
|
|
const model = resolveModel({
|
|
userModel: userConfig?.model,
|
|
inheritedModel: defaultConfig?.model, // Category's built-in model takes precedence over system default
|
|
systemDefault: systemDefaultModel,
|
|
})
|
|
const config: CategoryConfig = {
|
|
...defaultConfig,
|
|
...userConfig,
|
|
model,
|
|
variant: userConfig?.variant ?? defaultConfig?.variant,
|
|
}
|
|
|
|
let promptAppend = defaultPromptAppend
|
|
if (userConfig?.prompt_append) {
|
|
promptAppend = defaultPromptAppend
|
|
? defaultPromptAppend + "\n\n" + userConfig.prompt_append
|
|
: userConfig.prompt_append
|
|
}
|
|
|
|
return { config, promptAppend, model }
|
|
}
|