Browse Source

Update agent_service: user MySQLUserManager

StrayWarrior 2 weeks ago
parent
commit
776e8aeddb
1 changed files with 6 additions and 2 deletions
  1. 6 2
      agent_service.py

+ 6 - 2
agent_service.py

@@ -17,7 +17,7 @@ import global_flags
 import logging_service
 import logging_service
 from chat_service import CozeChat, ChatServiceType
 from chat_service import CozeChat, ChatServiceType
 from dialogue_manager import DialogueManager, DialogueState
 from dialogue_manager import DialogueManager, DialogueState
-from user_manager import UserManager, LocalUserManager
+from user_manager import UserManager, LocalUserManager, MySQLUserManager
 from openai import OpenAI
 from openai import OpenAI
 from message_queue_backend import MessageQueueBackend, MemoryQueueBackend, AliyunRocketMQQueueBackend
 from message_queue_backend import MessageQueueBackend, MemoryQueueBackend, AliyunRocketMQQueueBackend
 from user_profile_extractor import UserProfileExtractor
 from user_profile_extractor import UserProfileExtractor
@@ -223,7 +223,11 @@ if __name__ == "__main__":
     human_queue = MemoryQueueBackend()
     human_queue = MemoryQueueBackend()
 
 
     # 初始化用户管理服务
     # 初始化用户管理服务
-    user_manager = LocalUserManager()
+    if config['debug_flags'].get('use_local_user_manager', False):
+        user_manager = LocalUserManager()
+    else:
+        db_config = config['storage']['user']
+        user_manager = MySQLUserManager(db_config['mysql'], db_config['table'])
 
 
     # 创建Agent服务
     # 创建Agent服务
     service = AgentService(
     service = AgentService(