Fix similarity score bug in lance/chroma dbs ()

* fix similarity score bug in lance/chroma dbs

* batch lower bound case

---------

Co-authored-by: timothycarambat <rambat1010@gmail.com>
This commit is contained in:
Sean Hatfield 2025-01-18 10:27:54 +08:00 committed by GitHub
parent 3d872bc639
commit f8c72786df
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
server/utils/vectorDbProviders
chroma
lance

View file

@ -100,7 +100,7 @@ const Chroma = {
distanceToSimilarity: function (distance = null) {
if (distance === null || typeof distance !== "number") return 0.0;
if (distance >= 1.0) return 1;
if (distance <= 0) return 0;
if (distance < 0) return 1 - Math.abs(distance);
return 1 - distance;
},
namespaceCount: async function (_namespace = null) {

View file

@ -29,7 +29,7 @@ const LanceDb = {
distanceToSimilarity: function (distance = null) {
if (distance === null || typeof distance !== "number") return 0.0;
if (distance >= 1.0) return 1;
if (distance <= 0) return 0;
if (distance < 0) return 1 - Math.abs(distance);
return 1 - distance;
},
heartbeat: async function () {