4a82ff40fb
* refactor(shared): unify binary downloader and session path storage - Create binary-downloader.ts for common download/extract logic - Create session-injected-paths.ts for unified path tracking - Refactor comment-checker, ast-grep, grep downloaders to use shared util - Consolidate directory injector types into shared module * feat(shared): implement unified model resolution pipeline - Create ModelResolutionPipeline for centralized model selection - Refactor model-resolver to use pipeline - Update delegate-task and config-handler to use unified logic - Ensure consistent model resolution across all agent types * refactor(agents): simplify agent utils and metadata management - Extract helper functions for config merging and env context - Register prompt metadata for all agents - Simplify agent variant detection logic * cleanup: inline utilities and remove unused exports - Remove case-insensitive.ts (inline with native JS) - Simplify opencode-version helpers - Remove unused getModelLimit, createCompactionContextInjector exports - Inline transcript entry creation in claude-code-hooks - Update tests accordingly --------- Co-authored-by: justsisyphus <justsisyphus@users.noreply.github.com>
85 lines
2.4 KiB
TypeScript
85 lines
2.4 KiB
TypeScript
import type { OhMyOpenCodeConfig } from "../config"
|
|
import { AGENT_MODEL_REQUIREMENTS, CATEGORY_MODEL_REQUIREMENTS } from "./model-requirements"
|
|
|
|
export function resolveAgentVariant(
|
|
config: OhMyOpenCodeConfig,
|
|
agentName?: string
|
|
): string | undefined {
|
|
if (!agentName) {
|
|
return undefined
|
|
}
|
|
|
|
const agentOverrides = config.agents as
|
|
| Record<string, { variant?: string; category?: string }>
|
|
| undefined
|
|
const agentOverride = agentOverrides
|
|
? agentOverrides[agentName]
|
|
?? Object.entries(agentOverrides).find(([key]) => key.toLowerCase() === agentName.toLowerCase())?.[1]
|
|
: undefined
|
|
if (!agentOverride) {
|
|
return undefined
|
|
}
|
|
|
|
if (agentOverride.variant) {
|
|
return agentOverride.variant
|
|
}
|
|
|
|
const categoryName = agentOverride.category
|
|
if (!categoryName) {
|
|
return undefined
|
|
}
|
|
|
|
return config.categories?.[categoryName]?.variant
|
|
}
|
|
|
|
export function resolveVariantForModel(
|
|
config: OhMyOpenCodeConfig,
|
|
agentName: string,
|
|
currentModel: { providerID: string; modelID: string },
|
|
): string | undefined {
|
|
const agentRequirement = AGENT_MODEL_REQUIREMENTS[agentName]
|
|
if (agentRequirement) {
|
|
return findVariantInChain(agentRequirement.fallbackChain, currentModel.providerID)
|
|
}
|
|
|
|
const agentOverrides = config.agents as
|
|
| Record<string, { category?: string }>
|
|
| undefined
|
|
const agentOverride = agentOverrides
|
|
? agentOverrides[agentName]
|
|
?? Object.entries(agentOverrides).find(([key]) => key.toLowerCase() === agentName.toLowerCase())?.[1]
|
|
: undefined
|
|
const categoryName = agentOverride?.category
|
|
if (categoryName) {
|
|
const categoryRequirement = CATEGORY_MODEL_REQUIREMENTS[categoryName]
|
|
if (categoryRequirement) {
|
|
return findVariantInChain(categoryRequirement.fallbackChain, currentModel.providerID)
|
|
}
|
|
}
|
|
|
|
return undefined
|
|
}
|
|
|
|
function findVariantInChain(
|
|
fallbackChain: { providers: string[]; model: string; variant?: string }[],
|
|
providerID: string,
|
|
): string | undefined {
|
|
for (const entry of fallbackChain) {
|
|
if (entry.providers.includes(providerID)) {
|
|
return entry.variant
|
|
}
|
|
}
|
|
return undefined
|
|
}
|
|
|
|
export function applyAgentVariant(
|
|
config: OhMyOpenCodeConfig,
|
|
agentName: string | undefined,
|
|
message: { variant?: string }
|
|
): void {
|
|
const variant = resolveAgentVariant(config, agentName)
|
|
if (variant !== undefined && message.variant === undefined) {
|
|
message.variant = variant
|
|
}
|
|
}
|