From b7c7cb221dd6f463d23570c3fb59d6915308cc6d Mon Sep 17 00:00:00 2001 From: YeonGyu-Kim Date: Fri, 8 May 2026 17:44:37 +0900 Subject: [PATCH] feat(delegate-task): add model string parser for variants --- .../delegate-task/model-string-parser.ts | 63 +++++++++++++++++++ 1 file changed, 63 insertions(+) create mode 100644 src/tools/delegate-task/model-string-parser.ts diff --git a/src/tools/delegate-task/model-string-parser.ts b/src/tools/delegate-task/model-string-parser.ts new file mode 100644 index 000000000..820bb3cc3 --- /dev/null +++ b/src/tools/delegate-task/model-string-parser.ts @@ -0,0 +1,63 @@ +const KNOWN_VARIANTS = new Set([ + "low", + "medium", + "high", + "xhigh", + "max", + "minimal", + "none", + "auto", + "thinking", +]) + +export function parseVariantFromModelID(rawModelID: string): { modelID: string; variant?: string } { + const trimmedModelID = rawModelID.trim() + if (!trimmedModelID) { + return { modelID: "" } + } + + const parenthesizedVariant = trimmedModelID.match(/^(.*)\(([^()]+)\)\s*$/) + if (parenthesizedVariant) { + const modelID = parenthesizedVariant[1]?.trim() ?? "" + const variant = parenthesizedVariant[2]?.trim() + return variant ? { modelID, variant } : { modelID } + } + + const spaceVariant = trimmedModelID.match(/^(.*\S)\s+([a-z][a-z0-9_-]*)$/i) + if (spaceVariant) { + const modelID = spaceVariant[1]?.trim() ?? "" + const variant = spaceVariant[2]?.trim().toLowerCase() + if (variant && KNOWN_VARIANTS.has(variant)) { + return { modelID, variant } + } + } + + return { modelID: trimmedModelID } +} + +export function parseModelString( + model: string, +): { providerID: string; modelID: string; variant?: string } | undefined { + const trimmedModel = model.trim() + if (!trimmedModel) return undefined + + const parts = trimmedModel.split("/") + if (parts.length < 2) { + return undefined + } + + const providerID = parts[0]?.trim() + const rawModelID = parts.slice(1).join("/").trim() + if (!providerID || !rawModelID) { + return undefined + } + + const parsedModel = parseVariantFromModelID(rawModelID) + if (!parsedModel.modelID) { + return undefined + } + + return parsedModel.variant + ? { providerID, modelID: parsedModel.modelID, variant: parsedModel.variant } + : { providerID, modelID: parsedModel.modelID } +}