51 lines
2.2 KiB
Diff
51 lines
2.2 KiB
Diff
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,
|
|
};
|
|
});
|
|
}
|