Browse Source

Update user_manager: fix abstract method signature

StrayWarrior 2 months ago
parent
commit
787019d8f1
1 changed files with 2 additions and 2 deletions
  1. 2 2
      pqai_agent/user_manager.py

+ 2 - 2
pqai_agent/user_manager.py

@@ -77,7 +77,7 @@ class UserRelationManager(abc.ABC):
         pass
 
     @abc.abstractmethod
-    def list_staff_users(self) -> List[Dict]:
+    def list_staff_users(self, staff_id: str = None, tag_id: int = None) -> List[Dict]:
         pass
 
     @abc.abstractmethod
@@ -227,7 +227,7 @@ class LocalUserRelationManager(UserRelationManager):
     def list_users(self, staff_id: str, page: int = 1, page_size: int = 100):
         return []
 
-    def list_staff_users(self):
+    def list_staff_users(self, staff_id: str = None, tag_id: int = None):
         return [
             {"staff_id": "1688854492669990", "user_id": "7881299670930896"}
         ]