Przeglądaj źródła

Rename message.py to mq_message.py

StrayWarrior 2 tygodni temu
rodzic
commit
65bae398d8

+ 1 - 1
pqai_agent/agent_service.py

@@ -31,7 +31,7 @@ from pqai_agent.response_type_detector import ResponseTypeDetector
 from pqai_agent.user_manager import UserManager, UserRelationManager
 from pqai_agent.message_queue_backend import MessageQueueBackend, AliyunRocketMQQueueBackend
 from pqai_agent.user_profile_extractor import UserProfileExtractor
-from pqai_agent.message import MessageType, Message, MessageChannel
+from pqai_agent.mq_message import MessageType, Message, MessageChannel
 from pqai_agent.utils.db_utils import create_sql_engine
 
 

+ 1 - 1
pqai_agent/agents/message_push_agent.py

@@ -4,7 +4,7 @@ from typing import Optional, List, Dict
 from pqai_agent.agents.simple_chat_agent import SimpleOpenAICompatibleChatAgent
 from pqai_agent.chat_service import VOLCENGINE_MODEL_DEEPSEEK_V3
 from pqai_agent.logging_service import logger
-from pqai_agent.message import MessageType
+from pqai_agent.mq_message import MessageType
 from pqai_agent.toolkit.function_tool import FunctionTool
 from pqai_agent.toolkit.image_describer import ImageDescriber
 from pqai_agent.toolkit.message_notifier import MessageNotifier

+ 1 - 1
pqai_agent/agents/message_reply_agent.py

@@ -4,7 +4,7 @@ from typing import Optional, List, Dict
 from pqai_agent.agents.simple_chat_agent import SimpleOpenAICompatibleChatAgent
 from pqai_agent.chat_service import VOLCENGINE_MODEL_DEEPSEEK_V3
 from pqai_agent.logging_service import logger
-from pqai_agent.message import MessageType
+from pqai_agent.mq_message import MessageType
 from pqai_agent.toolkit.function_tool import FunctionTool
 from pqai_agent.toolkit.image_describer import ImageDescriber
 from pqai_agent.toolkit.message_notifier import MessageNotifier

+ 1 - 1
pqai_agent/dialogue_manager.py

@@ -21,7 +21,7 @@ from pqai_agent import chat_service, prompt_templates
 from pqai_agent.history_dialogue_service import HistoryDialogueService
 
 from pqai_agent.chat_service import ChatServiceType
-from pqai_agent.message import MessageType, Message
+from pqai_agent.mq_message import MessageType, Message
 from pqai_agent.toolkit.lark_alert_for_human_intervention import LarkAlertForHumanIntervention
 from pqai_agent.toolkit.lark_sheet_record_for_human_intervention import LarkSheetRecordForHumanIntervention
 from pqai_agent.user_manager import UserManager

+ 1 - 1
pqai_agent/history_dialogue_service.py

@@ -11,7 +11,7 @@ from pqai_agent.logging_service import logger
 import time
 
 from pqai_agent import configs
-from pqai_agent.message import MessageType
+from pqai_agent.mq_message import MessageType
 
 
 class HistoryDialogueService:

+ 1 - 1
pqai_agent/message_queue_backend.py

@@ -11,7 +11,7 @@ from rocketmq import ClientConfiguration, Credentials, SimpleConsumer
 from pqai_agent import configs
 from pqai_agent import logging_service
 from pqai_agent.logging_service import logger
-from pqai_agent.message import Message, MessageType, MessageChannel
+from pqai_agent.mq_message import Message, MessageType, MessageChannel
 
 
 

+ 0 - 0
pqai_agent/message.py → pqai_agent/mq_message.py


+ 1 - 1
pqai_agent/push_service.py

@@ -16,7 +16,7 @@ from pqai_agent.agents.message_push_agent import MessagePushAgent, DummyMessageP
 from pqai_agent.configs import apollo_config
 from pqai_agent.data_models.agent_push_record import AgentPushRecord
 from pqai_agent.logging_service import logger
-from pqai_agent.message import MessageType
+from pqai_agent.mq_message import MessageType
 
 
 class TaskType(Enum):

+ 1 - 1
pqai_agent/response_type_detector.py

@@ -13,7 +13,7 @@ from pqai_agent import configs
 from pqai_agent import prompt_templates
 from pqai_agent.dialogue_manager import DialogueManager
 from pqai_agent.logging_service import logger
-from pqai_agent.message import MessageType
+from pqai_agent.mq_message import MessageType
 
 
 class ResponseTypeDetector:

+ 1 - 1
pqai_agent_server/agent_server.py

@@ -6,7 +6,7 @@ from pqai_agent import configs, logging_service
 from pqai_agent.agent_service import AgentService
 from pqai_agent.chat_service import ChatServiceType
 from pqai_agent.logging_service import logger
-from pqai_agent.message import MessageType, Message, MessageChannel
+from pqai_agent.mq_message import MessageType, Message, MessageChannel
 from pqai_agent.message_queue_backend import AliyunRocketMQQueueBackend, MemoryQueueBackend
 from pqai_agent.push_service import PushTaskWorkerPool, PushScanThread
 from pqai_agent.user_manager import LocalUserManager, LocalUserRelationManager, MySQLUserManager, \

+ 1 - 1
pqai_agent_server/utils/prompt_util.py

@@ -9,7 +9,7 @@ from pqai_agent import logging_service, chat_service
 from pqai_agent.response_type_detector import ResponseTypeDetector
 from pqai_agent.user_profile_extractor import UserProfileExtractor
 from pqai_agent.dialogue_manager import DialogueManager
-from pqai_agent.message import MessageType
+from pqai_agent.mq_message import MessageType
 from pqai_agent.utils.prompt_utils import format_agent_profile
 
 logger = logging_service.logger

+ 1 - 1
scripts/mq_sender.py

@@ -1,6 +1,6 @@
 import sys
 from pqai_agent import configs
-from pqai_agent.message import Message, MessageType, MessageChannel
+from pqai_agent.mq_message import Message, MessageType, MessageChannel
 from pqai_agent.message_queue_backend import AliyunRocketMQQueueBackend
 import time
 from argparse import ArgumentParser

+ 1 - 1
scripts/resend_lost_message.py

@@ -5,7 +5,7 @@
 from datetime import datetime
 import re
 from pqai_agent import configs
-from pqai_agent.message import MessageChannel, Message, MessageType
+from pqai_agent.mq_message import MessageChannel, Message, MessageType
 from pqai_agent.message_queue_backend import AliyunRocketMQQueueBackend
 from pqai_agent.user_manager import MySQLUserRelationManager
 

+ 1 - 1
tests/unit_test.py

@@ -6,7 +6,7 @@ import pytest
 from unittest.mock import Mock, MagicMock
 from pqai_agent.agent_service import AgentService, MemoryQueueBackend
 from pqai_agent.dialogue_manager import DialogueState, TimeContext
-from pqai_agent.message import MessageType, Message, MessageChannel
+from pqai_agent.mq_message import MessageType, Message, MessageChannel
 from pqai_agent.response_type_detector import ResponseTypeDetector
 from pqai_agent.user_manager import LocalUserManager
 import time