|
@@ -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']
|