mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2026-07-07 15:12:18 +08:00
bd87e9a6cb
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Signed-off-by: suyao <sy20010504@gmail.com>
39 lines
1.9 KiB
TypeScript
39 lines
1.9 KiB
TypeScript
/**
|
|
* Build-time canonicalization — the id-folding the catalog generator applies to every model id.
|
|
* Shares the runtime resolver's `normalizeModelId` helpers but is INTENTIONALLY different in one way:
|
|
* it does NOT strip parameter size, so the catalog keeps `qwen3-235b` ≠ `qwen3-30b`.
|
|
*
|
|
* Lives in its own module (not inside `generate-catalog.ts`, whose import runs the generation IIFE) so
|
|
* tests can import `canonOf` without triggering a live upstream fetch.
|
|
*/
|
|
import {
|
|
expandKnownPrefixes,
|
|
normalizeVersionSeparators,
|
|
stripAggregatorPrefixes,
|
|
stripBedrockRevision,
|
|
stripBedrockVendorPrefix,
|
|
stripVariantQuantDateSuffixes
|
|
} from '../src/utils/normalize'
|
|
|
|
// strip the same org/host routing prefixes the runtime resolver does (zai-org-, databricks-, …),
|
|
// so a host that flattens `zai-org/glm-5` → `zai-org-glm-5` folds into the real `glm-5`; then peel the
|
|
// bedrock cross-vendor `[region.]vendor.` / `vendor-` prefix (shared with the runtime).
|
|
const base = (id: string) => stripBedrockVendorPrefix(stripAggregatorPrefixes(id.toLowerCase().split('/').pop()!))
|
|
|
|
// Minus param-size stripping — the catalog keeps `qwen3-235b` ≠ `qwen3-30b`.
|
|
export const canonOf = (id: string): string => {
|
|
let s = base(id) // split('/').pop, lowercase, strip aggregator + bedrock-vendor prefix
|
|
s = stripBedrockRevision(s) // bedrock arn revision: claude-…-v1:0 / …:0 (keeps whisper-v3)
|
|
s = expandKnownPrefixes(s) // mm- → minimax-
|
|
s = stripVariantQuantDateSuffixes(s) // variant/quant/date suffixes, iterated to a fixpoint
|
|
s = normalizeVersionSeparators(s) // 4.6 → 4-6
|
|
return s
|
|
.replace(/[^a-z0-9-]/g, '-')
|
|
.replace(/-+/g, '-')
|
|
.replace(/^-|-$/g, '')
|
|
}
|
|
|
|
// a `-` OR a digit ends the prefix word, so `qwen` claims both `qwen-max` and `qwen3-30b-a3b`
|
|
export const prefixHit = (id: string, p: string): boolean =>
|
|
id === p || id.startsWith(`${p}-`) || (id.startsWith(p) && /\d/.test(id[p.length] ?? ''))
|