浏览代码

Update agent_service: user MySQLUserManager

StrayWarrior 2 周之前
父节点
当前提交
776e8aeddb
共有 1 个文件被更改,包括 6 次插入2 次删除
  1. 6 2
      agent_service.py

+ 6 - 2
agent_service.py

@@ -17,7 +17,7 @@ import global_flags
 import logging_service
 from chat_service import CozeChat, ChatServiceType
 from dialogue_manager import DialogueManager, DialogueState
-from user_manager import UserManager, LocalUserManager
+from user_manager import UserManager, LocalUserManager, MySQLUserManager
 from openai import OpenAI
 from message_queue_backend import MessageQueueBackend, MemoryQueueBackend, AliyunRocketMQQueueBackend
 from user_profile_extractor import UserProfileExtractor
@@ -223,7 +223,11 @@ if __name__ == "__main__":
     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服务
     service = AgentService(