pjandro/OmniRoute
1
0
Fork 0
mirror of https://github.com/diegosouzapw/OmniRoute.git synced 2026-08-19 21:52:21 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat: use getModelInfo for proper custom provider resolution in availability checks

Browse Source
This commit is contained in:
nyatoru
2026-02-24 18:47:38 +07:00
parent 86c566669c
commit 619c99ce4c
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/sse/handlers/chat.ts
View File

@@ -156,13 +156,14 @@ export async function handleChat(request: any, clientRawRequest: any = null) {
// Pre-check function: skip models where all accounts are in cooldown
// Uses modelAvailability module for TTL-based cooldowns
const checkModelAvailable = async (modelString: string) => {
const parsed = parseModel(modelString);
const provider = parsed.provider;
// Use getModelInfo to properly resolve custom prefixes
const modelInfo = await getModelInfo(modelString);
const provider = modelInfo.provider;
if (!provider) return true; // can't determine provider, let it try
// Check domain-level availability (cooldown)
if (!isModelAvailable(provider, parsed.model || modelString)) {
log.debug("AVAILABILITY", `${provider}/${parsed.model} in cooldown, skipping`);
if (!isModelAvailable(provider, modelInfo.model || modelString)) {
log.debug("AVAILABILITY", `${provider}/${modelInfo.model} in cooldown, skipping`);
return false;
}