mirror of
https://github.com/ItzCrazyKns/Perplexica.git
synced 2025-10-19 05:48:15 +00:00
feat(model-registry): add loading method
This commit is contained in:
@@ -53,6 +53,26 @@ class ModelRegistry {
|
|||||||
|
|
||||||
return providers;
|
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;
|
export default ModelRegistry;
|
||||||
|
Reference in New Issue
Block a user