|
@@ -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(
|