@storecraft/database-turso 1.0.11 → 1.0.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@storecraft/database-turso",
-  "version": "1.0.11",
+  "version": "1.0.13",
   "description": "`Storecraft` database driver for `Turso` (cloud sqlite)",
   "license": "MIT",
   "author": "Tomer Shalev (https://github.com/store-craft)",
@@ -35,7 +35,8 @@
   "scripts": {
     "database-turso:test": "node ./tests/runner.test.js",
     "test": "npm run database-turso:test",
-    "prepublishOnly": "npm version patch --force"
+    "prepublishOnly": "npm version patch --force",
+    "sc-publish": "npm publish --access public"
   },
   "peerDependencies": {
     "kysely": "*"

package/vector-store/index.js

@@ -92,12 +92,11 @@ export class LibSQLVectorStore {
 
   /** @type {VectorStore["onInit"]} */
   onInit = (app) => {
-    this.config.authToken ??= app.platform.env[
-      LibSQLVectorStore.EnvConfig.authToken ?? 'LIBSQL_AUTH_TOKEN'
-    ];
-    this.config.url ??= app.platform.env[
-      LibSQLVectorStore.EnvConfig.url ?? 'LIBSQL_URL'
-    ];
+    this.config.authToken ??= app.platform.env[LibSQLVectorStore.EnvConfig.authToken] 
+        ?? app.platform.env['LIBSQL_AUTH_TOKEN'];
+
+    this.config.url ??= app.platform.env[LibSQLVectorStore.EnvConfig.url] 
+        ?? app.platform.env['LIBSQL_URL'];
   }
 
   /** @type {VectorStore["embedder"]} */
@@ -193,7 +192,8 @@ export class LibSQLVectorStore {
 
   /** @type {VectorStore["similaritySearch"]} */
   similaritySearch = async (query, k, namespaces) => {
-
+    // console.log({query,k,namespaces})
+    
     const embedding_result = await this.embedder.generateEmbeddings(
       {
         content: [
@@ -218,10 +218,12 @@ export class LibSQLVectorStore {
     /** @type {InArgs} */
     let args = [];
     let sql = `
-    SELECT id, metadata, pageContent, updated_at, namespace, ${distance_fn}(embedding, vector(?)) AS score
-    FROM vector_top_k('${index_name}', vector(?), ?) as top_k_view
-    JOIN ${table} ON ${table}.rowid = top_k_view.id
+    SELECT ${table}.id, metadata, pageContent, updated_at, namespace, ${distance_fn}(embedding, vector(?)) AS score
+    FROM vector_top_k('${index_name}', vector(?), CAST(? AS INTEGER)) as top_k_view
+    JOIN ${table} ON ${table}.rowid = top_k_view.rowid
     `;
+
+    // console.log(typeof k)
     args.push(vector_sql_value, vector_sql_value, k);
 
     if(Array.isArray(namespaces) && namespaces.length) {
@@ -236,9 +238,10 @@ export class LibSQLVectorStore {
     `
     args.push(vector_sql_value);
 
-
     const result = await this.client.execute({ sql, args });
 
+    // console.log({result})
+
     return result.rows.map(
       (row) => (
         {