From 825bcaf63e84060a4ae7e40c7f828b1e7f81c81f Mon Sep 17 00:00:00 2001
From: Bo26fhmC5M <88071760+Bo26fhmC5M@users.noreply.github.com>
Date: Fri, 14 Mar 2025 23:58:01 +0900
Subject: [PATCH] fix: add custom embedding model name as suffix to hypaVector
 storage keys - Add custom embedding model name as suffix to hypaVector
 storage keys when using custom embedding models to prevent vector dimension
 mismatch errors.

---
 src/ts/process/memory/hypamemory.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/ts/process/memory/hypamemory.ts b/src/ts/process/memory/hypamemory.ts
index 5dd8efc9..07b808fb 100644
--- a/src/ts/process/memory/hypamemory.ts
+++ b/src/ts/process/memory/hypamemory.ts
@@ -133,9 +133,11 @@ export class HypaProcesser{
     }
     
     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++){
-            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){
                 itm.alreadySaved = true
                 this.vectors.push(itm)
@@ -164,7 +166,7 @@ export class HypaProcesser{
         for(let i=0;i<memoryVectors.length;i++){
             const vec = memoryVectors[i]
             if(!vec.alreadySaved){
-                await this.forage.setItem(texts[i] + '|' + this.model, vec)
+                await this.forage.setItem(texts[i] + '|' + this.model + suffix, vec)
             }
         }