UserRapport now updates properly inside the DB

nusk0 · Feb 5, 2025 · a84572d · a84572d
1 parent 21ec459
commit a84572d
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 26 deletions.
diff --git a/packages/adapter-sqlite/src/index.ts b/packages/adapter-sqlite/src/index.ts
@@ -805,10 +805,24 @@ export class SqliteDatabaseAdapter
         }
     }
 
-    async setUserRapport(userId: UUID, agentId: UUID, score: number): Promise<void> {
+    async setUserRapport(userIdOrUsername: UUID | string, agentId: UUID, score: number): Promise<void> {
         try {
-            const sql = "UPDATE accounts SET userRapport = ? WHERE id = ?";
-            this.db.prepare(sql).run(score, userId);
+            let sql;
+            let params;
+
+            // Check if we're dealing with a username or userId
+            if (userIdOrUsername.includes('-')) {
+                // It's a UUID
+                sql = "UPDATE accounts SET userRapport = userRapport + ? WHERE id = ?";
+                params = [score, userIdOrUsername];
+            } else {
+                // It's a username
+                sql = "UPDATE accounts SET userRapport = userRapport + ? WHERE username = ?";
+                params = [score, userIdOrUsername.replace('@', '')];
+            }
+
+            console.log("Setting rapport for", userIdOrUsername, "score:", score);
+            this.db.prepare(sql).run(...params);
         } catch (error) {
             console.error("Error setting user rapport:", error);
         }

diff --git a/packages/adapter-sqlite/src/sqliteTables.ts b/packages/adapter-sqlite/src/sqliteTables.ts
@@ -2,6 +2,8 @@ export const sqliteTables = `
 PRAGMA foreign_keys=OFF;
 BEGIN TRANSACTION;
 
+-- Uncomment the following line to reset all user rapport values to 0
+ --UPDATE accounts SET userRapport = 0;
 
 -- Table: accounts
 CREATE TABLE IF NOT EXISTS "accounts" (
@@ -11,17 +13,9 @@ CREATE TABLE IF NOT EXISTS "accounts" (
     "username" TEXT,
     "email" TEXT NOT NULL,
     "avatarUrl" TEXT,
-    "details" TEXT DEFAULT '{}' CHECK(json_valid("details")), -- Ensuring details is a valid JSON field
-    "userRapport" REAL NOT NULL DEFAULT 0
+    "details" TEXT DEFAULT '{}' CHECK(json_valid("details")) -- Ensuring details is a valid JSON field
 );
 
--- Add userRapport column if it doesn't exist (using correct SQLite syntax)
-SELECT CASE 
-    WHEN NOT EXISTS(SELECT 1 FROM pragma_table_info('accounts') WHERE name='userRapport') 
-    THEN 'ALTER TABLE accounts ADD COLUMN "userRapport" REAL NOT NULL DEFAULT 0;'
-END
-WHERE NOT EXISTS(SELECT 1 FROM pragma_table_info('accounts') WHERE name='userRapport');
-
 -- Table: memories
 CREATE TABLE IF NOT EXISTS "memories" (
     "id" TEXT PRIMARY KEY,

diff --git a/packages/client-twitter/src/interactions.ts b/packages/client-twitter/src/interactions.ts
@@ -128,7 +128,7 @@ export class TwitterInteractionClient {
 
             elizaLogger.log("Checking active conversations");
             // Get all active conversations
-            const activeConversations = await this.runtime.databaseAdapter.getConversationsByStatus('CLOSED');
+            const activeConversations = await this.runtime.databaseAdapter.getConversationsByStatus('ACTIVE');
 
             for (const conversation of activeConversations) {
                 const messageIds = JSON.parse(conversation.messageIds);

diff --git a/packages/client-twitter/src/utils.ts b/packages/client-twitter/src/utils.ts
@@ -484,13 +484,14 @@ export async function analyzeConversation(
         template: analysisTemplate
     });
     console.log("context", context)
- 
+
     const analysis = await generateText({
         runtime,
         context,
         modelClass: ModelClass.LARGE,
     });
 
+
     elizaLogger.log("User sentiment scores:", analysis);
 
     try {
@@ -504,18 +505,11 @@ export async function analyzeConversation(
 
         // Update user rapport based on sentiment scores
         for (const [username, score] of Object.entries(sentimentScores)) {
-            const userId = messages.find(m => 
-                (m.content.username || m.userId) === username.replace('@', '')
-            )?.userId;
-
-            if (userId) {
-                await runtime.databaseAdapter.setUserRapport(
-                    userId,
-                    runtime.agentId,
-                    score as number  // Use the sentiment score directly
-                );
-                elizaLogger.log(`Updated rapport for user ${username}:`, score);
-            }
+            await runtime.databaseAdapter.setUserRapport(
+                username,
+                runtime.agentId,
+                score as number
+            );
         }
     } catch (error) {
         elizaLogger.error("Error parsing sentiment analysis:", error);