From 99afe0e21aa978216cde3fd9b721c81756a1ca89 Mon Sep 17 00:00:00 2001 From: musistudio Date: Sun, 3 Aug 2025 07:47:11 +0800 Subject: [PATCH] compatible model name case problem --- src/utils/router.ts | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/utils/router.ts b/src/utils/router.ts index 2743d458..22129f6b 100644 --- a/src/utils/router.ts +++ b/src/utils/router.ts @@ -64,6 +64,16 @@ const calculateTokenCount = ( const getUseModel = async (req: any, tokenCount: number, config: any) => { if (req.body.model.includes(",")) { + const [provider, model] = req.body.model.split(","); + const finalProvider = config.Providers.find( + (p: any) => p.name.toLowerCase() === provider + ); + const finalModel = finalProvider?.models?.find( + (m: any) => m.name.toLowerCase() === model + ); + if (finalProvider && finalModel) { + return `${finalProvider.name},${finalModel}`; + } return req.body.model; } // if tokenCount is greater than the configured threshold, use the long context model