mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add custom embedding model name as suffix to hypaVector storage keys

Browse Source 
- Add custom embedding model name as suffix to hypaVector storage keys when using custom embedding models to prevent vector dimension mismatch errors.
This commit is contained in:
Bo26fhmC5M
2025-03-14 23:58:01 +09:00
parent fef10c1e56
commit 825bcaf63e
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/ts/process/memory/hypamemory.ts
View File

@@ -133,9 +133,11 @@ export class HypaProcesser{
} }
async addText(texts:string[]) { async addText(texts:string[]) {
const db = getDatabase()
const suffix = (this.model === 'custom' && db.hypaCustomSettings.model) ? `-${db.hypaCustomSettings.model}` : ""
for(let i=0;i<texts.length;i++){ for(let i=0;i<texts.length;i++){
const itm:memoryVector = await this.forage.getItem(texts[i] + '|' + this.model) const itm:memoryVector = await this.forage.getItem(texts[i] + '|' + this.model + suffix)
if(itm){ if(itm){
itm.alreadySaved = true itm.alreadySaved = true
this.vectors.push(itm) this.vectors.push(itm)
@@ -164,7 +166,7 @@ export class HypaProcesser{
for(let i=0;i<memoryVectors.length;i++){ for(let i=0;i<memoryVectors.length;i++){
const vec = memoryVectors[i] const vec = memoryVectors[i]
if(!vec.alreadySaved){ if(!vec.alreadySaved){
await this.forage.setItem(texts[i] + '|' + this.model, vec) await this.forage.setItem(texts[i] + '|' + this.model + suffix, vec)
} }
} }