@@ -162,7 +162,7 @@ export class SqliteDatabaseAdapter extends DatabaseAdapter {
162
162
let queryParams = [ params . tableName , ...params . roomIds ] ;
163
163
164
164
if ( params . agentId ) {
165
- sql += ` AND userId = ?` ;
165
+ sql += ` AND agentId = ?` ;
166
166
queryParams . push ( params . agentId ) ;
167
167
}
168
168
@@ -219,14 +219,15 @@ export class SqliteDatabaseAdapter extends DatabaseAdapter {
219
219
const createdAt = memory . createdAt ?? Date . now ( ) ;
220
220
221
221
// Insert the memory with the appropriate 'unique' value
222
- const sql = `INSERT OR REPLACE INTO memories (id, type, content, embedding, userId, roomId, \`unique\`, createdAt) VALUES (?, ?, ?, ?, ?, ?, ?, ?)` ;
222
+ const sql = `INSERT OR REPLACE INTO memories (id, type, content, embedding, userId, roomId, agentId, \`unique\`, createdAt) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)` ;
223
223
this . db . prepare ( sql ) . run (
224
224
memory . id ?? v4 ( ) ,
225
225
tableName ,
226
226
content ,
227
227
new Float32Array ( memory . embedding ?? embeddingZeroVector ) , // Store as Float32Array
228
228
memory . userId ,
229
229
memory . roomId ,
230
+ memory . agentId ,
230
231
isUnique ? 1 : 0 ,
231
232
createdAt
232
233
) ;
@@ -235,6 +236,7 @@ export class SqliteDatabaseAdapter extends DatabaseAdapter {
235
236
async searchMemories ( params : {
236
237
tableName : string ;
237
238
roomId : UUID ;
239
+ agentId ?: UUID ;
238
240
embedding : number [ ] ;
239
241
match_threshold : number ;
240
242
match_count : number ;
@@ -256,6 +258,11 @@ export class SqliteDatabaseAdapter extends DatabaseAdapter {
256
258
sql += " AND `unique` = 1" ;
257
259
}
258
260
261
+ if ( params . agentId ) {
262
+ sql += " AND agentId = ?" ;
263
+ queryParams . push ( params . agentId ) ;
264
+ }
265
+
259
266
sql += ` ORDER BY similarity ASC LIMIT ?` ; // ASC for lower distance
260
267
// Updated queryParams order matches the placeholders
261
268
@@ -297,9 +304,8 @@ export class SqliteDatabaseAdapter extends DatabaseAdapter {
297
304
if ( params . unique ) {
298
305
sql += " AND `unique` = 1" ;
299
306
}
300
- // TODO: Test this
301
307
if ( params . agentId ) {
302
- sql += " AND userId = ?" ;
308
+ sql += " AND agentId = ?" ;
303
309
queryParams . push ( params . agentId ) ;
304
310
}
305
311
@@ -413,7 +419,7 @@ export class SqliteDatabaseAdapter extends DatabaseAdapter {
413
419
}
414
420
415
421
if ( params . agentId ) {
416
- sql += " AND userId = ?" ;
422
+ sql += " AND agentId = ?" ;
417
423
queryParams . push ( params . agentId ) ;
418
424
}
419
425
0 commit comments