Update agent_service and push_service: call get_tools

pqai_agent/agent_service.py

@@ -32,6 +32,7 @@ from pqai_agent.push_service import PushScanThread, PushTaskWorkerPool
 from pqai_agent.rate_limiter import MessageSenderRateLimiter
 from pqai_agent.response_type_detector import ResponseTypeDetector
 from pqai_agent.service_module_manager import ServiceModuleManager
+from pqai_agent.toolkit import get_tools
 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
@@ -458,7 +459,7 @@ class AgentService:
         if agent_config:
             chat_agent = MessageReplyAgent(model=agent_config.execution_model,
                                            system_prompt=agent_config.system_prompt,
-                                           tools=None)
+                                           tools=get_tools(agent_config.tools))
         else:
             chat_agent = MessageReplyAgent()
         chat_responses = chat_agent.generate_message(

pqai_agent/push_service.py

@@ -17,6 +17,7 @@ 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.mq_message import MessageType
+from pqai_agent.toolkit import get_tools
 from pqai_agent.utils.agent_abtest_utils import get_agent_abtest_config
 
 
@@ -199,7 +200,7 @@ class PushTaskWorkerPool:
             if agent_config:
                 push_agent = MessagePushAgent(model=agent_config.execution_model,
                                               system_prompt=agent_config.system_prompt,
-                                              tools=None)
+                                              tools=get_tools(agent_config.tools))
                 query_prompt_template = agent_config.task_prompt
             else:
                 push_agent = MessagePushAgent()