michael/merlyn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix similarity score bug in lance/chroma dbs (#2986)

Browse Source 
* 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/utils/vectorDbProviders/chroma/index.js
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) {

2
server/utils/vectorDbProviders/lance/index.js
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 () {