update model select

This commit is contained in:
duanfuxiang
2025-06-18 08:57:49 +08:00
parent c3cc81624f
commit 9b4bfe97b4
2 changed files with 21 additions and 3 deletions

View File

@@ -25,7 +25,16 @@ export class RAGEngine {
this.app = app
this.settings = settings
this.vectorManager = dbManager.getVectorManager()
this.embeddingModel = getEmbeddingModel(settings)
if (settings.embeddingModelId && settings.embeddingModelId.trim() !== '') {
try {
this.embeddingModel = getEmbeddingModel(settings)
} catch (error) {
console.warn('Failed to initialize embedding model:', error)
this.embeddingModel = null
}
} else {
this.embeddingModel = null
}
}
cleanup() {
@@ -35,7 +44,16 @@ export class RAGEngine {
setSettings(settings: InfioSettings) {
this.settings = settings
this.embeddingModel = getEmbeddingModel(settings)
if (settings.embeddingModelId && settings.embeddingModelId.trim() !== '') {
try {
this.embeddingModel = getEmbeddingModel(settings)
} catch (error) {
console.warn('Failed to initialize embedding model:', error)
this.embeddingModel = null
}
} else {
this.embeddingModel = null
}
}
async initializeDimension(): Promise<void> {