diff --git a/src/libsql-db.cjs b/src/libsql-db.cjs index 076e838a1a144fa80687aa27ed92f488c2b532cb..fd00187134cda688c20b2929934ca5bfd4899c42 100644 --- a/src/libsql-db.cjs +++ b/src/libsql-db.cjs @@ -45,9 +45,9 @@ class LibSqlDb { } async similaritySearch(query, k) { const statement = `SELECT id, pageContent, uniqueLoaderId, source, metadata, - vector_distance_cos(vector, vector32('[${query.join(',')}]')) + vector_distance_cos(vector, vector32('[${query.join(',')}]')) as distance FROM ${this.tableName} - ORDER BY vector_distance_cos(vector, vector32('[${query.join(',')}]')) ASC + ORDER BY distance ASC LIMIT ${k};`; this.debug(`Executing statement - ${(0, embedjs_utils_1.truncateCenterString)(statement, 700)}`); const results = await this.client.execute(statement); @@ -56,7 +56,7 @@ class LibSqlDb { return { metadata, pageContent: result.pageContent.toString(), - score: 1, + score: result.distance, }; }); } diff --git a/src/libsql-db.js b/src/libsql-db.js index 58c42e4910bd0e53bc497ff9b9702b1f7a961266..3b124ff26f3b7cdca72eed5b872d7410c32eb70a 100644 --- a/src/libsql-db.js +++ b/src/libsql-db.js @@ -41,9 +41,9 @@ export class LibSqlDb { } async similaritySearch(query, k) { const statement = `SELECT id, pageContent, uniqueLoaderId, source, metadata, - vector_distance_cos(vector, vector32('[${query.join(',')}]')) + vector_distance_cos(vector, vector32('[${query.join(',')}]')) as distance FROM ${this.tableName} - ORDER BY vector_distance_cos(vector, vector32('[${query.join(',')}]')) ASC + ORDER BY distance ASC LIMIT ${k};`; this.debug(`Executing statement - ${truncateCenterString(statement, 700)}`); const results = await this.client.execute(statement); @@ -52,7 +52,7 @@ export class LibSqlDb { return { metadata, pageContent: result.pageContent.toString(), - score: 1, + score: result.distance, }; }); }