feat(modelRegistry): add MinimalProvider type

This commit is contained in:
ItzCrazyKns
2025-10-15 12:53:36 +05:30
parent cc5eea17e4
commit 30fb1e312b
2 changed files with 18 additions and 6 deletions

View File

@@ -4,7 +4,7 @@ import BaseModelProvider, {
} from './providers/baseProvider';
import { getConfiguredModelProviders } from '../config/serverRegistry';
import { providers } from './providers';
import { ModelList } from './types';
import { MinimalProvider, Model } from './types';
class ModelRegistry {
activeProviders: (ConfigModelProvider & {
@@ -35,18 +35,23 @@ class ModelRegistry {
});
}
async getActiveModels() {
const models: ModelList[] = [];
async getActiveProviders() {
const providers: MinimalProvider[] = [];
await Promise.all(
this.activeProviders.map(async (p) => {
const m = await p.provider.getModelList();
models.push(m);
providers.push({
id: p.id,
name: p.name,
chatModels: m.chat,
embeddingModels: m.embedding,
});
}),
);
return models;
return providers;
}
}

View File

@@ -13,4 +13,11 @@ type ProviderMetadata = {
key: string;
};
export type { Model, ModelList, ProviderMetadata };
type MinimalProvider = {
id: string;
name: string;
chatModels: Model[];
embeddingModels: Model[];
};
export type { Model, ModelList, ProviderMetadata, MinimalProvider };