admin-valentin/AIClient-2-API
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: read managed models from provider pools for model lists

Browse source
This commit is contained in:
HenryZ-0302 2026-04-05 14:06:21 -07:00
parent 465bbaef2b
commit 110720982f
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/utils/common.js
View file

@ -100,6 +100,17 @@ function getConfiguredSupportedModels(providerType, providerConfig = {}) {
)].sort((a, b) => a.localeCompare(b));
}
function getConfiguredSupportedModelsFromPool(providerPoolManager, providerType) {
if (!providerPoolManager?.providerStatus?.[providerType]) {
return [];
}
return [...new Set(
providerPoolManager.providerStatus[providerType]
.flatMap(providerStatus => getConfiguredSupportedModels(providerType, providerStatus.config))
)].sort((a, b) => a.localeCompare(b));
}
/**
* Extracts the protocol prefix from a given model provider string.
* This is used to determine if two providers belong to the same underlying protocol (e.g., gemini, openai, claude).
@ -882,7 +893,10 @@ export async function handleModelListRequest(req, res, service, endpointType, CO
} else {
// --- 单提供商逻辑 ---
const toProvider = CONFIG.MODEL_PROVIDER;
const configuredSupportedModels = getConfiguredSupportedModels(toProvider, CONFIG);
const pooledSupportedModels = getConfiguredSupportedModelsFromPool(providerPoolManager, toProvider);
const configuredSupportedModels = pooledSupportedModels.length > 0
? pooledSupportedModels
: getConfiguredSupportedModels(toProvider, CONFIG);
if (usesManagedModelList(toProvider) && configuredSupportedModels.length > 0) {
logger.info(`[ModelList] Returning configured supported models for ${toProvider}: ${configuredSupportedModels.join(', ')}`);