From 9a8e036ff92748aa9df456d5a876c2bdd2063357 Mon Sep 17 00:00:00 2001 From: Cohee Date: Wed, 5 Jul 2023 01:25:43 +0300 Subject: [PATCH] Use floor rounding on chroma auto-calc --- public/scripts/extensions/infinity-context/index.js | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/public/scripts/extensions/infinity-context/index.js b/public/scripts/extensions/infinity-context/index.js index bd0ce5899..2cef52152 100644 --- a/public/scripts/extensions/infinity-context/index.js +++ b/public/scripts/extensions/infinity-context/index.js @@ -506,7 +506,7 @@ function doAutoAdjust(chat, maxContext) { const contextMessagesRounded = Math.ceil(contextMessages / 10) * 10; console.debug('CHROMADB: Estimated context messages (rounded): %o', contextMessagesRounded); // Messages to keep (proportional, rounded to nearest 5, minimum 5, maximum 500) - const messagesToKeep = Math.min(defaultSettings.keep_context_max, Math.max(5, Math.ceil(contextMessagesRounded * extension_settings.chromadb.keep_context_proportion / 5) * 5)); + const messagesToKeep = Math.min(defaultSettings.keep_context_max, Math.max(5, Math.floor(contextMessagesRounded * extension_settings.chromadb.keep_context_proportion / 5) * 5)); console.debug('CHROMADB: Estimated messages to keep: %o', messagesToKeep); // Messages to query (rounded, maximum 500) const messagesToQuery = Math.min(defaultSettings.n_results_max, contextMessagesRounded - messagesToKeep);