40 lines
1.2 KiB
TypeScript
40 lines
1.2 KiB
TypeScript
|
import { readProviderModelsCache } from "./connected-providers-cache"
|
||
|
|
import { log } from "./logger"
|
||
|
|
|
||
|
|
export function addModelsFromProviderModelsCache(
|
||
|
|
connectedProviders: Set<string>,
|
||
|
|
modelSet: Set<string>,
|
||
|
|
): boolean {
|
||
|
|
const providerModelsCache = readProviderModelsCache()
|
||
|
|
if (!providerModelsCache) {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
const providerCount = Object.keys(providerModelsCache.models).length
|
||
|
|
if (providerCount === 0) {
|
||
|
|
log("[fetchAvailableModels] provider-models cache empty, falling back to models.json")
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
log("[fetchAvailableModels] using provider-models cache (whitelist-filtered)")
|
||
|
|
const previousSize = modelSet.size
|
||
|
|
|
||
|
|
for (const [providerId, modelIds] of Object.entries(providerModelsCache.models)) {
|
||
|
|
if (!connectedProviders.has(providerId)) continue
|
||
|
|
for (const modelItem of modelIds) {
|
||
|
|
if (!modelItem) continue
|
||
|
|
const modelId = typeof modelItem === "string" ? modelItem : modelItem.id
|
||
|
|
if (modelId) {
|
||
|
|
modelSet.add(`${providerId}/${modelId}`)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
log("[fetchAvailableModels] parsed from provider-models cache", {
|
||
|
|
count: modelSet.size,
|
||
|
|
connectedProviders: Array.from(connectedProviders).slice(0, 5),
|
||
|
|
})
|
||
|
|
|
||
|
|
return modelSet.size > previousSize
|
||
|
|
}
|