diff --git a/src/routes/proxy/shared.ts b/src/routes/proxy/shared.ts index 4daa694..75c3e78 100644 --- a/src/routes/proxy/shared.ts +++ b/src/routes/proxy/shared.ts @@ -99,8 +99,16 @@ export const apiHeaders = () => ({ ...openRouterHeaders, }); -export const resolveModel = (model: string, pool: string[]) => - pool.includes(model) ? model : pool[0]; +export const resolveModel = (model: string, pool: string[]) => { + if (pool.includes(model)) { + return model; + } + const matchedModel = pool.find((m) => m.split("/")[1] === model); + if (matchedModel) { + return matchedModel; + } + return pool[0]; +}; export const logRequest = ( c: Ctx,