瀏覽代碼

Update agent_service: add history dialogue database

StrayWarrior 1 月之前
父節點
當前提交
82f786be42
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      pqai_agent/agent_service.py

+ 4 - 1
pqai_agent/agent_service.py

@@ -23,6 +23,7 @@ from pqai_agent.logging_service import logger
 from pqai_agent import chat_service
 from pqai_agent.chat_service import CozeChat, ChatServiceType
 from pqai_agent.dialogue_manager import DialogueManager, DialogueState, DialogueStateCache
+from pqai_agent.history_dialogue_service import HistoryDialogueDatabase
 from pqai_agent.push_service import PushScanThread, PushTaskWorkerPool
 from pqai_agent.rate_limiter import MessageSenderRateLimiter
 from pqai_agent.response_type_detector import ResponseTypeDetector
@@ -41,6 +42,8 @@ class AgentService:
         user_relation_manager: UserRelationManager,
         chat_service_type: ChatServiceType = ChatServiceType.OPENAI_COMPATIBLE
     ):
+        self.config = configs.get()
+
         self.receive_queue = receive_backend
         self.send_queue = send_backend
         self.human_queue = human_backend
@@ -52,8 +55,8 @@ class AgentService:
         self.user_profile_extractor = UserProfileExtractor()
         self.response_type_detector = ResponseTypeDetector()
         self.agent_registry: Dict[str, DialogueManager] = {}
+        self.history_dialogue_db = HistoryDialogueDatabase(self.config['storage']['user']['mysql'])
 
-        self.config = configs.get()
         chat_config = self.config['chat_api']['openai_compatible']
         self.text_model_name = chat_config['text_model']
         self.multimodal_model_name = chat_config['multimodal_model']