Merge pull request #64 from JoinTheAlliance/sqlite-fixes

Sqlite adapter fixes: Handle unique and order for memories
JoinTheAlliance · Mar 21, 2024 · 30db76c · 30db76c
2 parents 5e57e16 + 877e5fb
commit 30db76c
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "bgent",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "private": false,
   "description": "bgent. because agent was taken.",
   "type": "module",

diff --git a/src/lib/adapters/sqlite.ts b/src/lib/adapters/sqlite.ts
@@ -275,9 +275,12 @@ AND room_id = ?`;
 
     const queryParams = [params.tableName, params.room_id];
 
-    // if (params.unique) {
-    //   sql += " AND `unique` = 1";
-    // }
+    if (params.unique) {
+      sql += " AND `unique` = 1";
+    }
+
+    // get the most recent memories
+    sql += " ORDER BY created_at DESC";
 
     if (params.count) {
       sql += " LIMIT ?";

diff --git a/src/lib/adapters/sqljs.ts b/src/lib/adapters/sqljs.ts
@@ -300,6 +300,12 @@ export class SqlJsDatabaseAdapter extends DatabaseAdapter {
     }
     let sql = `SELECT * FROM memories WHERE type = ? AND room_id = ?`;
 
+    if (params.unique) {
+      sql += " AND `unique` = 1";
+    }
+
+    sql += " ORDER BY created_at DESC";
+
     if (params.count) {
       sql += " LIMIT ?";
     }