add agentId to filtering

Author: lalalune

core/src/adapters/sqlite.ts

@@ -152,32 +152,29 @@ export class SqliteDatabaseAdapter extends DatabaseAdapter {
         tableName: string;
         agentId?: UUID;
     }): Promise<Memory[]> {
+        console.log("getMemoriesByRoomIds", params);
         if (!params.tableName) {
             // default to messages
             params.tableName = "messages";
         }
         const placeholders = params.roomIds.map(() => "?").join(", ");
         let sql = `SELECT * FROM memories WHERE type = ? AND roomId IN (${placeholders})`;
-        const stmt = this.db.prepare(sql);
-        const queryParams = [params.tableName, ...params.roomIds];
+        let queryParams = [params.tableName, ...params.roomIds];
 
         if (params.agentId) {
             sql += ` AND userId = ?`;
             queryParams.push(params.agentId);
         }
 
-        const memories: Memory[] = [];
+        const stmt = this.db.prepare(sql);
         const rows = stmt.all(...queryParams) as (Memory & {
             content: string;
         })[];
-        rows.forEach((row) => {
-            memories.push({
-                ...row,
-                content: JSON.parse(row.content),
-            });
-        });
-
-        return memories;
+
+        return rows.map(row => ({
+            ...row,
+            content: JSON.parse(row.content)
+        }));
     }
 
     async getMemoryById(memoryId: UUID): Promise<Memory | null> {

core/src/adapters/sqlite/sqliteTables.ts

@@ -22,9 +22,11 @@ CREATE TABLE IF NOT EXISTS "memories" (
     "embedding" BLOB NOT NULL, -- TODO: EMBEDDING ARRAY, CONVERT TO BEST FORMAT FOR SQLITE-VSS (JSON?)
     "userId" TEXT,
     "roomId" TEXT,
+    "agentId" TEXT,
     "unique" INTEGER DEFAULT 1 NOT NULL,
     FOREIGN KEY ("userId") REFERENCES "accounts"("id"),
-    FOREIGN KEY ("roomId") REFERENCES "rooms"("id")
+    FOREIGN KEY ("roomId") REFERENCES "rooms"("id"),
+    FOREIGN KEY ("agentId") REFERENCES "accounts"("id")
 );
 
 -- Table: goals