From 0c976daed89ccd2b970eeed3627d691c8b5d6fe1 Mon Sep 17 00:00:00 2001
From: Justin Parker <jparkerweb@gmail.com>
Date: Thu, 7 Nov 2024 19:17:31 -0800
Subject: [PATCH] fix model cache issue with Web UI

---
 CHANGELOG.md            | 4 ++++
 embeddingUtils.js       | 3 +++
 package-lock.json       | 4 ++--
 package.json            | 2 +-
 webui/package-lock.json | 4 ++--
 webui/package.json      | 2 +-
 6 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 5ef2332..a61ba52 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 All notable changes to this project will be documented in this file.
 
+## [2.2.4] - 2024-11-08
+### Fixed
+- Fixed issue with Web UI embedding cache not being cleared when a new model is initialized.
+
 ## [2.2.3] - 2024-11-07
 ### Added
 - Web UI adjustments for display of truncated JSON results on screen but still allowing download of full results.
diff --git a/embeddingUtils.js b/embeddingUtils.js
index 7b02b93..b6e088f 100644
--- a/embeddingUtils.js
+++ b/embeddingUtils.js
@@ -14,6 +14,9 @@ export async function initializeEmbeddingUtils(onnxEmbeddingModel, onnxEmbedding
         quantized: onnxEmbeddingModelQuantized,
     });
 
+    // Clear the embedding cache when initializing with a new model
+    embeddingCache.clear();
+
     return {
         modelName: onnxEmbeddingModel,
         isQuantized: onnxEmbeddingModelQuantized
diff --git a/package-lock.json b/package-lock.json
index d31cf91..bce54cf 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "semantic-chunking",
-  "version": "2.2.1",
+  "version": "2.2.4",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "semantic-chunking",
-      "version": "2.2.1",
+      "version": "2.2.4",
       "license": "ISC",
       "dependencies": {
         "@xenova/transformers": "^2.17.2",
diff --git a/package.json b/package.json
index 69ac305..b1663c1 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "semantic-chunking",
-  "version": "2.2.3",
+  "version": "2.2.4",
   "description": "Semantically create chunks from large texts. Useful for workflows involving large language models (LLMs).",
   "repository": {
     "type": "git",
diff --git a/webui/package-lock.json b/webui/package-lock.json
index 88c7b13..1e7c43b 100644
--- a/webui/package-lock.json
+++ b/webui/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "semantic-chunking-webui",
-  "version": "1.1.0",
+  "version": "1.1.4",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "semantic-chunking-webui",
-      "version": "1.1.0",
+      "version": "1.1.4",
       "license": "ISC",
       "dependencies": {
         "cors": "^2.8.5",
diff --git a/webui/package.json b/webui/package.json
index ad7e90b..7202b10 100644
--- a/webui/package.json
+++ b/webui/package.json
@@ -1,6 +1,6 @@
 {
   "name": "semantic-chunking-webui",
-  "version": "1.1.3",
+  "version": "1.1.4",
   "type": "module",
   "description": "Web UI for semantic-chunking library",
   "main": "server.js",