Register hot_topic_toolkit

pqai_agent/toolkit/__init__.py

@@ -7,6 +7,7 @@ from pqai_agent.toolkit.image_describer import ImageDescriber
 from pqai_agent.toolkit.message_notifier import MessageNotifier
 from pqai_agent.toolkit.pq_video_searcher import PQVideoSearcher
 from pqai_agent.toolkit.search_toolkit import SearchToolkit
+from pqai_agent.toolkit.hot_topic_toolkit import HotTopicToolkit
 
 global_tool_map = ToolRegistry.tool_map
 

pqai_agent/toolkit/hot_topic_toolkit.py

@@ -4,8 +4,12 @@ import requests
 from pqai_agent.logging import logger
 from pqai_agent.toolkit.base import BaseToolkit
 from pqai_agent.toolkit.function_tool import FunctionTool
+from pqai_agent.toolkit.tool_registry import register_toolkit
 
+
+@register_toolkit
 class HotTopicToolkit(BaseToolkit):
+    @staticmethod
     def get_hot_topics(self) -> List[Dict[str, Any]]:
         """
         获取热点内容列表