work on role cache

elizaOS · Feb 14, 2025 · d0d606f · d0d606f
1 parent 538d421
commit d0d606f
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 43 deletions.
diff --git a/packages/agent/src/swarm/communityManager/actions/timeout.ts b/packages/agent/src/swarm/communityManager/actions/timeout.ts
@@ -12,23 +12,14 @@ import {
     logger,
 } from "@elizaos/core";
 import { type Message } from "discord.js";
+import { ServerRoleState } from "../../shared/role/types";
 
 interface TimeoutSettings {
     enabled: boolean;
     duration: number; // timeout duration in minutes
     lastUpdated: number;
 }
 
-interface ServerRoleCache {
-    roles: {
-        [userId: string]: {
-            role: string;
-            platformId: string;
-        };
-    };
-    lastUpdated: number;
-}
-
 export const timeoutAction: Action = {
     name: "TIMEOUT_USER",
     similes: ["MUTE_USER", "SILENCE_USER", "RESTRICT_USER"],
@@ -66,7 +57,7 @@ export const timeoutAction: Action = {
             }
 
             // Get user roles
-            const roles = await runtime.cacheManager.get<ServerRoleCache>(
+            const roles = await runtime.cacheManager.get<ServerRoleState>(
                 `server_${serverId}_user_roles`
             );
 
@@ -148,7 +139,7 @@ export const timeoutAction: Action = {
             }
 
             // Get roles to check if we can timeout this user
-            const roles = await runtime.cacheManager.get<ServerRoleCache>(
+            const roles = await runtime.cacheManager.get<ServerRoleState>(
                 `server_${serverId}_user_roles`
             );
 

diff --git a/packages/agent/src/swarm/shared/role/action.ts b/packages/agent/src/swarm/shared/role/action.ts
@@ -22,6 +22,7 @@ import {
   logger,
 } from "@elizaos/core";
 import { type Message } from "discord.js";
+import { ServerRoleState } from "./types";
 
 export enum RoleName {
   ADMIN = "ADMIN",
@@ -31,22 +32,6 @@ export enum RoleName {
   IGNORE = "IGNORE",
 }
 
-interface UserRole {
-  userId: string;
-  platformId: string;
-  serverId: string;
-  role: RoleName;
-  assignedBy: string;
-  assignedAt: number;
-}
-
-interface ServerRoleCache {
-  roles: {
-    [userId: string]: UserRole;
-  };
-  lastUpdated: number;
-}
-
 const canModifyRole = (
   currentRole: RoleName,
   targetRole: RoleName,
@@ -92,7 +77,7 @@ const updateOrgRoleAction: Action = {
 
     try {
       // Get roles cache
-      const roleCache = await runtime.cacheManager.get<ServerRoleCache>(
+      const roleCache = await runtime.cacheManager.get<ServerRoleState>(
         `server_${serverId}_user_roles`
       );
 
@@ -159,7 +144,7 @@ const updateOrgRoleAction: Action = {
 
     try {
       // Get roles cache
-      let roleCache = await runtime.cacheManager.get<ServerRoleCache>(
+      let roleCache = await runtime.cacheManager.get<ServerRoleState>(
         `server_${serverId}_user_roles`
       );
 
@@ -222,7 +207,6 @@ const updateOrgRoleAction: Action = {
       // Update role
       roleCache.roles[mentionedUser.id] = {
         userId: mentionedUser.id,
-        platformId: mentionedUser.id,
         serverId,
         role: newRole,
         assignedBy: requesterId,

diff --git a/packages/agent/src/swarm/shared/role/types.ts b/packages/agent/src/swarm/shared/role/types.ts
@@ -8,6 +8,14 @@ export enum RoleName {
     IGNORE = "IGNORE"
 }
 
+export interface UserRole {
+    userId: string;
+    serverId: string;
+    role: RoleName;
+    assignedBy: string;
+    assignedAt: number;
+}
+
 export interface ServerRoleState {
     roles: {
         [userId: string]: UserRole;
@@ -35,15 +43,6 @@ export function canModifyRole(modifierRole: RoleName, targetRole: RoleName, newR
     return false; // Other roles can't modify roles
 }
 
-export interface UserRole {
-    userId: string;
-    platformId: string; // Discord ID, Telegram ID, etc
-    serverId: string;
-    role: RoleName;
-    assignedBy: string;
-    assignedAt: number;
-}
-
 // Role access helpers
 export async function getUserServerRole(
     runtime: IAgentRuntime,

diff --git a/packages/plugin-bootstrap/src/providers/anxiety.ts b/packages/plugin-bootstrap/src/providers/anxiety.ts
@@ -19,11 +19,10 @@ interface AnxietyScore {
     lastDecayTimestamp: number;
 }
 
-interface ServerRoleCache {
+interface ServerRoleState {
     roles: {
         [userId: string]: {
             role: string;
-            platformId: string;
         };
     };
     lastUpdated: number;
@@ -66,7 +65,7 @@ const socialAwarenessProvider: Provider = {
             }
 
             // Check roles for message sender
-            const roles = await runtime.cacheManager.get<ServerRoleCache>(
+            const roles = await runtime.cacheManager.get<ServerRoleState>(
                 `server_${serverId}_user_roles`
             );