feat(model-registry): add loading method

This commit is contained in:
ItzCrazyKns
2025-10-16 17:56:57 +05:30
parent 36fdb6491d
commit 9219593ee1

View File

@@ -53,6 +53,26 @@ class ModelRegistry {
return providers;
}
async loadChatModel(providerId: string, modelName: string) {
const provider = this.activeProviders.find((p) => p.id === providerId);
if (!provider) throw new Error('Invalid provider id');
const model = await provider.provider.loadChatModel(modelName);
return model;
}
async loadEmbeddingModel(providerId: string, modelName: string) {
const provider = this.activeProviders.find((p) => p.id === providerId);
if (!provider) throw new Error('Invalid provider id');
const model = await provider.provider.loadEmbeddingModel(modelName);
return model;
}
}
export default ModelRegistry;