fix getMemoriesByRoomIds which fixes our conversation context

thegeneth · Sep 4, 2024 · 0b6780b · 0b6780b
1 parent 639b107
commit 0b6780b
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/src/adapters/sqlite.ts b/src/adapters/sqlite.ts
@@ -144,6 +144,10 @@ export class SqliteDatabaseAdapter extends DatabaseAdapter {
     roomIds: UUID[];
     tableName: string;
   }): Promise<Memory[]> {
+    if(!params.tableName) {
+      // default to messages
+      params.tableName = "messages";
+    }
     const placeholders = params.roomIds.map(() => "?").join(", ");
     const sql = `SELECT * FROM memories WHERE type = ? AND roomId IN (${placeholders})`;
     const stmt = this.db.prepare(sql);

diff --git a/src/core/runtime.ts b/src/core/runtime.ts
@@ -1052,16 +1052,22 @@ Text: ${attachment.text}
         userB,
       ]);
 
+      console.log("*** getRecentInteractions: Rooms", rooms);
+
       // Check the existing memories in the database
       const existingMemories = await this.messageManager.getMemoriesByRoomIds({
         roomIds: rooms,
       });
 
+      console.log("*** getRecentInteractions: Existing Memories", existingMemories);
+
       // Sort messages by timestamp in descending order
       existingMemories.sort(
         (a, b) => b.createdAt.getTime() - a.createdAt.getTime(),
       );
 
+      console.log("*** getRecentInteractions: Sorted Memories", existingMemories);
+
       // Take the most recent messages
       const recentInteractionsData = existingMemories.slice(0, 20);
       return recentInteractionsData;

diff --git a/supabase/migrations/20240318103238_remote_schema.sql b/supabase/migrations/20240318103238_remote_schema.sql
@@ -607,10 +607,10 @@ ALTER TABLE ONLY "public"."memories"
     ADD CONSTRAINT "memories_userId_fkey" FOREIGN KEY ("userId") REFERENCES "public"."accounts"("id");
 
 ALTER TABLE ONLY "public"."relationships"
-    ADD CONSTRAINT "relationships_user_a_fkey" FOREIGN KEY ("userA") REFERENCES "public"."accounts"("id");
+    ADD CONSTRAINT "relationships_userA_fkey" FOREIGN KEY ("userA") REFERENCES "public"."accounts"("id");
 
 ALTER TABLE ONLY "public"."relationships"
-    ADD CONSTRAINT "relationships_user_b_fkey" FOREIGN KEY ("userB") REFERENCES "public"."accounts"("id");
+    ADD CONSTRAINT "relationships_userB_fkey" FOREIGN KEY ("userB") REFERENCES "public"."accounts"("id");
 
 ALTER TABLE ONLY "public"."relationships"
     ADD CONSTRAINT "relationships_userId_fkey" FOREIGN KEY ("userId") REFERENCES "public"."accounts"("id");