diff --git a/package.json b/package.json index 42ad8a5..1c9a3a8 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "@ztimson/ai-utils", - "version": "0.8.4", + "version": "0.8.5", "description": "AI Utility library", "author": "Zak Timson", "license": "MIT", diff --git a/src/llm.ts b/src/llm.ts index 6a0b591..d17563d 100644 --- a/src/llm.ts +++ b/src/llm.ts @@ -151,7 +151,7 @@ class LLM { const newMem = {owner: args.owner, fact: args.fact, embeddings: [e[0][0].embedding, e[1][0].embedding]}; options.memory.splice(0, options.memory.length, ...[ ...options.memory.filter(m => { - return this.cosineSimilarity(newMem.embeddings[0], m.embeddings[0]) < 0.9 && this.cosineSimilarity(newMem.embeddings[1], m.embeddings[1]) < 0.8; + return !(this.cosineSimilarity(newMem.embeddings[0], m.embeddings[0]) >= 0.9 && this.cosineSimilarity(newMem.embeddings[1], m.embeddings[1]) >= 0.8); }), newMem ]);