server.py 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696
  1. """
  2. TrendRadar MCP Server - FastMCP 2.0 实现
  3. 使用 FastMCP 2.0 提供生产级 MCP 工具服务器。
  4. 支持 stdio 和 HTTP 两种传输模式。
  5. """
  6. import json
  7. from typing import List, Optional, Dict
  8. from fastmcp import FastMCP
  9. from .tools.data_query import DataQueryTools
  10. from .tools.analytics import AnalyticsTools
  11. from .tools.search_tools import SearchTools
  12. from .tools.config_mgmt import ConfigManagementTools
  13. from .tools.system import SystemManagementTools
  14. # 创建 FastMCP 2.0 应用
  15. mcp = FastMCP('trendradar-news')
  16. # 全局工具实例(在第一次请求时初始化)
  17. _tools_instances = {}
  18. def _get_tools(project_root: Optional[str] = None):
  19. """获取或创建工具实例(单例模式)"""
  20. if not _tools_instances:
  21. _tools_instances['data'] = DataQueryTools(project_root)
  22. _tools_instances['analytics'] = AnalyticsTools(project_root)
  23. _tools_instances['search'] = SearchTools(project_root)
  24. _tools_instances['config'] = ConfigManagementTools(project_root)
  25. _tools_instances['system'] = SystemManagementTools(project_root)
  26. return _tools_instances
  27. # ==================== 数据查询工具 ====================
  28. @mcp.tool
  29. async def get_latest_news(
  30. platforms: Optional[List[str]] = None,
  31. limit: int = 50,
  32. include_url: bool = False
  33. ) -> str:
  34. """
  35. 获取最新一批爬取的新闻数据,快速了解当前热点
  36. Args:
  37. platforms: 平台ID列表,如 ['zhihu', 'weibo', 'douyin']
  38. - 不指定时:使用 config.yaml 中配置的所有平台
  39. - 支持的平台来自 config/config.yaml 的 platforms 配置
  40. - 每个平台都有对应的name字段(如"知乎"、"微博"),方便AI识别
  41. limit: 返回条数限制,默认50,最大1000
  42. 注意:实际返回数量可能少于请求值,取决于当前可用的新闻总数
  43. include_url: 是否包含URL链接,默认False(节省token)
  44. Returns:
  45. JSON格式的新闻列表
  46. **重要:数据展示建议**
  47. 本工具会返回完整的新闻列表(通常50条)给你。但请注意:
  48. - **工具返回**:完整的50条数据 ✅
  49. - **建议展示**:向用户展示全部数据,除非用户明确要求总结
  50. - **用户期望**:用户可能需要完整数据,请谨慎总结
  51. **何时可以总结**:
  52. - 用户明确说"给我总结一下"或"挑重点说"
  53. - 数据量超过100条时,可先展示部分并询问是否查看全部
  54. **注意**:如果用户询问"为什么只显示了部分",说明他们需要完整数据
  55. """
  56. tools = _get_tools()
  57. result = tools['data'].get_latest_news(platforms=platforms, limit=limit, include_url=include_url)
  58. return json.dumps(result, ensure_ascii=False, indent=2)
  59. @mcp.tool
  60. async def get_trending_topics(
  61. top_n: int = 10,
  62. mode: str = 'current'
  63. ) -> str:
  64. """
  65. 获取个人关注词的新闻出现频率统计(基于 config/frequency_words.txt)
  66. 注意:本工具不是自动提取新闻热点,而是统计你在 config/frequency_words.txt 中
  67. 设置的个人关注词在新闻中出现的频率。你可以自定义这个关注词列表。
  68. Args:
  69. top_n: 返回TOP N关注词,默认10
  70. mode: 模式选择
  71. - daily: 当日累计数据统计
  72. - current: 最新一批数据统计(默认)
  73. Returns:
  74. JSON格式的关注词频率统计列表
  75. """
  76. tools = _get_tools()
  77. result = tools['data'].get_trending_topics(top_n=top_n, mode=mode)
  78. return json.dumps(result, ensure_ascii=False, indent=2)
  79. @mcp.tool
  80. async def get_news_by_date(
  81. date_query: Optional[str] = None,
  82. platforms: Optional[List[str]] = None,
  83. limit: int = 50,
  84. include_url: bool = False
  85. ) -> str:
  86. """
  87. 获取指定日期的新闻数据,用于历史数据分析和对比
  88. Args:
  89. date_query: 日期查询,可选格式:
  90. - 自然语言: "今天", "昨天", "前天", "3天前"
  91. - 标准日期: "2024-01-15", "2024/01/15"
  92. - 默认值: "今天"(节省token)
  93. platforms: 平台ID列表,如 ['zhihu', 'weibo', 'douyin']
  94. - 不指定时:使用 config.yaml 中配置的所有平台
  95. - 支持的平台来自 config/config.yaml 的 platforms 配置
  96. - 每个平台都有对应的name字段(如"知乎"、"微博"),方便AI识别
  97. limit: 返回条数限制,默认50,最大1000
  98. 注意:实际返回数量可能少于请求值,取决于指定日期的新闻总数
  99. include_url: 是否包含URL链接,默认False(节省token)
  100. Returns:
  101. JSON格式的新闻列表,包含标题、平台、排名等信息
  102. **重要:数据展示建议**
  103. 本工具会返回完整的新闻列表(通常50条)给你。但请注意:
  104. - **工具返回**:完整的50条数据 ✅
  105. - **建议展示**:向用户展示全部数据,除非用户明确要求总结
  106. - **用户期望**:用户可能需要完整数据,请谨慎总结
  107. **何时可以总结**:
  108. - 用户明确说"给我总结一下"或"挑重点说"
  109. - 数据量超过100条时,可先展示部分并询问是否查看全部
  110. **注意**:如果用户询问"为什么只显示了部分",说明他们需要完整数据
  111. """
  112. tools = _get_tools()
  113. result = tools['data'].get_news_by_date(
  114. date_query=date_query,
  115. platforms=platforms,
  116. limit=limit,
  117. include_url=include_url
  118. )
  119. return json.dumps(result, ensure_ascii=False, indent=2)
  120. # ==================== 高级数据分析工具 ====================
  121. @mcp.tool
  122. async def analyze_topic_trend(
  123. topic: str,
  124. analysis_type: str = "trend",
  125. date_range: Optional[Dict[str, str]] = None,
  126. granularity: str = "day",
  127. threshold: float = 3.0,
  128. time_window: int = 24,
  129. lookahead_hours: int = 6,
  130. confidence_threshold: float = 0.7
  131. ) -> str:
  132. """
  133. 统一话题趋势分析工具 - 整合多种趋势分析模式
  134. Args:
  135. topic: 话题关键词(必需)
  136. analysis_type: 分析类型,可选值:
  137. - "trend": 热度趋势分析(追踪话题的热度变化)
  138. - "lifecycle": 生命周期分析(从出现到消失的完整周期)
  139. - "viral": 异常热度检测(识别突然爆火的话题)
  140. - "predict": 话题预测(预测未来可能的热点)
  141. date_range: 日期范围(trend和lifecycle模式),可选
  142. - **格式**: {"start": "YYYY-MM-DD", "end": "YYYY-MM-DD"}
  143. - **示例**: {"start": "2025-10-18", "end": "2025-10-25"}
  144. - **说明**: AI需要根据用户的自然语言(如"最近7天")自动计算日期范围
  145. - **默认**: 不指定时默认分析最近7天
  146. granularity: 时间粒度(trend模式),默认"day"(仅支持 day,因为底层数据按天聚合)
  147. threshold: 热度突增倍数阈值(viral模式),默认3.0
  148. time_window: 检测时间窗口小时数(viral模式),默认24
  149. lookahead_hours: 预测未来小时数(predict模式),默认6
  150. confidence_threshold: 置信度阈值(predict模式),默认0.7
  151. Returns:
  152. JSON格式的趋势分析结果
  153. **AI使用说明:**
  154. 当用户使用相对时间表达时(如"最近7天"、"过去一周"、"上个月"),
  155. AI需要自动计算对应的日期范围并传递给 date_range 参数。
  156. Examples:
  157. - analyze_topic_trend(topic="人工智能", analysis_type="trend", date_range={"start": "2025-10-18", "end": "2025-10-25"})
  158. - analyze_topic_trend(topic="特斯拉", analysis_type="lifecycle", date_range={"start": "2025-10-18", "end": "2025-10-25"})
  159. - analyze_topic_trend(topic="比特币", analysis_type="viral", threshold=3.0)
  160. - analyze_topic_trend(topic="ChatGPT", analysis_type="predict", lookahead_hours=6)
  161. """
  162. tools = _get_tools()
  163. result = tools['analytics'].analyze_topic_trend_unified(
  164. topic=topic,
  165. analysis_type=analysis_type,
  166. date_range=date_range,
  167. granularity=granularity,
  168. threshold=threshold,
  169. time_window=time_window,
  170. lookahead_hours=lookahead_hours,
  171. confidence_threshold=confidence_threshold
  172. )
  173. return json.dumps(result, ensure_ascii=False, indent=2)
  174. @mcp.tool
  175. async def analyze_data_insights(
  176. insight_type: str = "platform_compare",
  177. topic: Optional[str] = None,
  178. date_range: Optional[Dict[str, str]] = None,
  179. min_frequency: int = 3,
  180. top_n: int = 20
  181. ) -> str:
  182. """
  183. 统一数据洞察分析工具 - 整合多种数据分析模式
  184. Args:
  185. insight_type: 洞察类型,可选值:
  186. - "platform_compare": 平台对比分析(对比不同平台对话题的关注度)
  187. - "platform_activity": 平台活跃度统计(统计各平台发布频率和活跃时间)
  188. - "keyword_cooccur": 关键词共现分析(分析关键词同时出现的模式)
  189. topic: 话题关键词(可选,platform_compare模式适用)
  190. date_range: **【对象类型】** 日期范围(可选)
  191. - **格式**: {"start": "YYYY-MM-DD", "end": "YYYY-MM-DD"}
  192. - **示例**: {"start": "2025-01-01", "end": "2025-01-07"}
  193. - **重要**: 必须是对象格式,不能传递整数
  194. min_frequency: 最小共现频次(keyword_cooccur模式),默认3
  195. top_n: 返回TOP N结果(keyword_cooccur模式),默认20
  196. Returns:
  197. JSON格式的数据洞察分析结果
  198. Examples:
  199. - analyze_data_insights(insight_type="platform_compare", topic="人工智能")
  200. - analyze_data_insights(insight_type="platform_activity", date_range={"start": "2025-01-01", "end": "2025-01-07"})
  201. - analyze_data_insights(insight_type="keyword_cooccur", min_frequency=5, top_n=15)
  202. """
  203. tools = _get_tools()
  204. result = tools['analytics'].analyze_data_insights_unified(
  205. insight_type=insight_type,
  206. topic=topic,
  207. date_range=date_range,
  208. min_frequency=min_frequency,
  209. top_n=top_n
  210. )
  211. return json.dumps(result, ensure_ascii=False, indent=2)
  212. @mcp.tool
  213. async def analyze_sentiment(
  214. topic: Optional[str] = None,
  215. platforms: Optional[List[str]] = None,
  216. date_range: Optional[Dict[str, str]] = None,
  217. limit: int = 50,
  218. sort_by_weight: bool = True,
  219. include_url: bool = False
  220. ) -> str:
  221. """
  222. 分析新闻的情感倾向和热度趋势
  223. Args:
  224. topic: 话题关键词(可选)
  225. platforms: 平台ID列表,如 ['zhihu', 'weibo', 'douyin']
  226. - 不指定时:使用 config.yaml 中配置的所有平台
  227. - 支持的平台来自 config/config.yaml 的 platforms 配置
  228. - 每个平台都有对应的name字段(如"知乎"、"微博"),方便AI识别
  229. date_range: **【对象类型】** 日期范围(可选)
  230. - **格式**: {"start": "YYYY-MM-DD", "end": "YYYY-MM-DD"}
  231. - **示例**: {"start": "2025-01-01", "end": "2025-01-07"}
  232. - **重要**: 必须是对象格式,不能传递整数
  233. limit: 返回新闻数量,默认50,最大100
  234. 注意:本工具会对新闻标题进行去重(同一标题在不同平台只保留一次),
  235. 因此实际返回数量可能少于请求的 limit 值
  236. sort_by_weight: 是否按热度权重排序,默认True
  237. include_url: 是否包含URL链接,默认False(节省token)
  238. Returns:
  239. JSON格式的分析结果,包含情感分布、热度趋势和相关新闻
  240. **重要:数据展示策略**
  241. - 本工具返回完整的分析结果和新闻列表
  242. - **默认展示方式**:展示完整的分析结果(包括所有新闻)
  243. - 仅在用户明确要求"总结"或"挑重点"时才进行筛选
  244. """
  245. tools = _get_tools()
  246. result = tools['analytics'].analyze_sentiment(
  247. topic=topic,
  248. platforms=platforms,
  249. date_range=date_range,
  250. limit=limit,
  251. sort_by_weight=sort_by_weight,
  252. include_url=include_url
  253. )
  254. return json.dumps(result, ensure_ascii=False, indent=2)
  255. @mcp.tool
  256. async def find_similar_news(
  257. reference_title: str,
  258. threshold: float = 0.6,
  259. limit: int = 50,
  260. include_url: bool = False
  261. ) -> str:
  262. """
  263. 查找与指定新闻标题相似的其他新闻
  264. Args:
  265. reference_title: 新闻标题(完整或部分)
  266. threshold: 相似度阈值,0-1之间,默认0.6
  267. 注意:阈值越高匹配越严格,返回结果越少
  268. limit: 返回条数限制,默认50,最大100
  269. 注意:实际返回数量取决于相似度匹配结果,可能少于请求值
  270. include_url: 是否包含URL链接,默认False(节省token)
  271. Returns:
  272. JSON格式的相似新闻列表,包含相似度分数
  273. **重要:数据展示策略**
  274. - 本工具返回完整的相似新闻列表
  275. - **默认展示方式**:展示全部返回的新闻(包括相似度分数)
  276. - 仅在用户明确要求"总结"或"挑重点"时才进行筛选
  277. """
  278. tools = _get_tools()
  279. result = tools['analytics'].find_similar_news(
  280. reference_title=reference_title,
  281. threshold=threshold,
  282. limit=limit,
  283. include_url=include_url
  284. )
  285. return json.dumps(result, ensure_ascii=False, indent=2)
  286. @mcp.tool
  287. async def generate_summary_report(
  288. report_type: str = "daily",
  289. date_range: Optional[Dict[str, str]] = None
  290. ) -> str:
  291. """
  292. 每日/每周摘要生成器 - 自动生成热点摘要报告
  293. Args:
  294. report_type: 报告类型(daily/weekly)
  295. date_range: **【对象类型】** 自定义日期范围(可选)
  296. - **格式**: {"start": "YYYY-MM-DD", "end": "YYYY-MM-DD"}
  297. - **示例**: {"start": "2025-01-01", "end": "2025-01-07"}
  298. - **重要**: 必须是对象格式,不能传递整数
  299. Returns:
  300. JSON格式的摘要报告,包含Markdown格式内容
  301. """
  302. tools = _get_tools()
  303. result = tools['analytics'].generate_summary_report(
  304. report_type=report_type,
  305. date_range=date_range
  306. )
  307. return json.dumps(result, ensure_ascii=False, indent=2)
  308. # ==================== 智能检索工具 ====================
  309. @mcp.tool
  310. async def search_news(
  311. query: str,
  312. search_mode: str = "keyword",
  313. date_range: Optional[Dict[str, str]] = None,
  314. platforms: Optional[List[str]] = None,
  315. limit: int = 50,
  316. sort_by: str = "relevance",
  317. threshold: float = 0.6,
  318. include_url: bool = False
  319. ) -> str:
  320. """
  321. 统一搜索接口,支持多种搜索模式
  322. Args:
  323. query: 搜索关键词或内容片段
  324. search_mode: 搜索模式,可选值:
  325. - "keyword": 精确关键词匹配(默认,适合搜索特定话题)
  326. - "fuzzy": 模糊内容匹配(适合搜索内容片段,会过滤相似度低于阈值的结果)
  327. - "entity": 实体名称搜索(适合搜索人物/地点/机构)
  328. date_range: 日期范围(可选)
  329. - **格式**: {"start": "YYYY-MM-DD", "end": "YYYY-MM-DD"}
  330. - **示例**: {"start": "2025-01-01", "end": "2025-01-07"}
  331. - **说明**: AI需要根据用户的自然语言(如"最近7天")自动计算日期范围
  332. - **默认**: 不指定时默认查询今天的新闻
  333. - **注意**: start和end可以相同(表示单日查询)
  334. platforms: 平台ID列表,如 ['zhihu', 'weibo', 'douyin']
  335. - 不指定时:使用 config.yaml 中配置的所有平台
  336. - 支持的平台来自 config/config.yaml 的 platforms 配置
  337. - 每个平台都有对应的name字段(如"知乎"、"微博"),方便AI识别
  338. limit: 返回条数限制,默认50,最大1000
  339. 注意:实际返回数量取决于搜索匹配结果(特别是 fuzzy 模式下会过滤低相似度结果)
  340. sort_by: 排序方式,可选值:
  341. - "relevance": 按相关度排序(默认)
  342. - "weight": 按新闻权重排序
  343. - "date": 按日期排序
  344. threshold: 相似度阈值(仅fuzzy模式有效),0-1之间,默认0.6
  345. 注意:阈值越高匹配越严格,返回结果越少
  346. include_url: 是否包含URL链接,默认False(节省token)
  347. Returns:
  348. JSON格式的搜索结果,包含标题、平台、排名等信息
  349. **重要:数据展示策略**
  350. - 本工具返回完整的搜索结果列表
  351. - **默认展示方式**:展示全部返回的新闻,无需总结或筛选
  352. - 仅在用户明确要求"总结"或"挑重点"时才进行筛选
  353. **AI使用说明:**
  354. 当用户使用相对时间表达时(如"最近7天"、"过去一周"、"最近半个月"),
  355. AI需要自动计算对应的日期范围。计算规则:
  356. - "最近7天" → {"start": "今天-6天", "end": "今天"}
  357. - "过去一周" → {"start": "今天-6天", "end": "今天"}
  358. - "最近30天" → {"start": "今天-29天", "end": "今天"}
  359. Examples:
  360. - 今天的新闻: search_news(query="人工智能")
  361. - 最近7天: search_news(query="人工智能", date_range={"start": "2025-10-18", "end": "2025-10-25"})
  362. - 精确日期: search_news(query="人工智能", date_range={"start": "2025-01-01", "end": "2025-01-07"})
  363. - 模糊搜索: search_news(query="特斯拉降价", search_mode="fuzzy", threshold=0.4)
  364. """
  365. tools = _get_tools()
  366. result = tools['search'].search_news_unified(
  367. query=query,
  368. search_mode=search_mode,
  369. date_range=date_range,
  370. platforms=platforms,
  371. limit=limit,
  372. sort_by=sort_by,
  373. threshold=threshold,
  374. include_url=include_url
  375. )
  376. return json.dumps(result, ensure_ascii=False, indent=2)
  377. @mcp.tool
  378. async def search_related_news_history(
  379. reference_text: str,
  380. time_preset: str = "yesterday",
  381. threshold: float = 0.4,
  382. limit: int = 50,
  383. include_url: bool = False
  384. ) -> str:
  385. """
  386. 基于种子新闻,在历史数据中搜索相关新闻
  387. Args:
  388. reference_text: 参考新闻标题(完整或部分)
  389. time_preset: 时间范围预设值,可选:
  390. - "yesterday": 昨天
  391. - "last_week": 上周 (7天)
  392. - "last_month": 上个月 (30天)
  393. - "custom": 自定义日期范围(需要提供 start_date 和 end_date)
  394. threshold: 相关性阈值,0-1之间,默认0.4
  395. 注意:综合相似度计算(70%关键词重合 + 30%文本相似度)
  396. 阈值越高匹配越严格,返回结果越少
  397. limit: 返回条数限制,默认50,最大100
  398. 注意:实际返回数量取决于相关性匹配结果,可能少于请求值
  399. include_url: 是否包含URL链接,默认False(节省token)
  400. Returns:
  401. JSON格式的相关新闻列表,包含相关性分数和时间分布
  402. **重要:数据展示策略**
  403. - 本工具返回完整的相关新闻列表
  404. - **默认展示方式**:展示全部返回的新闻(包括相关性分数)
  405. - 仅在用户明确要求"总结"或"挑重点"时才进行筛选
  406. """
  407. tools = _get_tools()
  408. result = tools['search'].search_related_news_history(
  409. reference_text=reference_text,
  410. time_preset=time_preset,
  411. threshold=threshold,
  412. limit=limit,
  413. include_url=include_url
  414. )
  415. return json.dumps(result, ensure_ascii=False, indent=2)
  416. # ==================== 配置与系统管理工具 ====================
  417. @mcp.tool
  418. async def get_current_config(
  419. section: str = "all"
  420. ) -> str:
  421. """
  422. 获取当前系统配置
  423. Args:
  424. section: 配置节,可选值:
  425. - "all": 所有配置(默认)
  426. - "crawler": 爬虫配置
  427. - "push": 推送配置
  428. - "keywords": 关键词配置
  429. - "weights": 权重配置
  430. Returns:
  431. JSON格式的配置信息
  432. """
  433. tools = _get_tools()
  434. result = tools['config'].get_current_config(section=section)
  435. return json.dumps(result, ensure_ascii=False, indent=2)
  436. @mcp.tool
  437. async def get_system_status() -> str:
  438. """
  439. 获取系统运行状态和健康检查信息
  440. 返回系统版本、数据统计、缓存状态等信息
  441. Returns:
  442. JSON格式的系统状态信息
  443. """
  444. tools = _get_tools()
  445. result = tools['system'].get_system_status()
  446. return json.dumps(result, ensure_ascii=False, indent=2)
  447. @mcp.tool
  448. async def trigger_crawl(
  449. platforms: Optional[List[str]] = None,
  450. save_to_local: bool = False,
  451. include_url: bool = False
  452. ) -> str:
  453. """
  454. 手动触发一次爬取任务(可选持久化)
  455. Args:
  456. platforms: 指定平台ID列表,如 ['zhihu', 'weibo', 'douyin']
  457. - 不指定时:使用 config.yaml 中配置的所有平台
  458. - 支持的平台来自 config/config.yaml 的 platforms 配置
  459. - 每个平台都有对应的name字段(如"知乎"、"微博"),方便AI识别
  460. - 注意:失败的平台会在返回结果的 failed_platforms 字段中列出
  461. save_to_local: 是否保存到本地 output 目录,默认 False
  462. include_url: 是否包含URL链接,默认False(节省token)
  463. Returns:
  464. JSON格式的任务状态信息,包含:
  465. - platforms: 成功爬取的平台列表
  466. - failed_platforms: 失败的平台列表(如有)
  467. - total_news: 爬取的新闻总数
  468. - data: 新闻数据
  469. Examples:
  470. - 临时爬取: trigger_crawl(platforms=['zhihu'])
  471. - 爬取并保存: trigger_crawl(platforms=['weibo'], save_to_local=True)
  472. - 使用默认平台: trigger_crawl() # 爬取config.yaml中配置的所有平台
  473. """
  474. tools = _get_tools()
  475. result = tools['system'].trigger_crawl(platforms=platforms, save_to_local=save_to_local, include_url=include_url)
  476. return json.dumps(result, ensure_ascii=False, indent=2)
  477. # ==================== 启动入口 ====================
  478. def run_server(
  479. project_root: Optional[str] = None,
  480. transport: str = 'stdio',
  481. host: str = '0.0.0.0',
  482. port: int = 3333
  483. ):
  484. """
  485. 启动 MCP 服务器
  486. Args:
  487. project_root: 项目根目录路径
  488. transport: 传输模式,'stdio' 或 'http'
  489. host: HTTP模式的监听地址,默认 0.0.0.0
  490. port: HTTP模式的监听端口,默认 3333
  491. """
  492. # 初始化工具实例
  493. _get_tools(project_root)
  494. # 打印启动信息
  495. print()
  496. print("=" * 60)
  497. print(" TrendRadar MCP Server - FastMCP 2.0")
  498. print("=" * 60)
  499. print(f" 传输模式: {transport.upper()}")
  500. if transport == 'stdio':
  501. print(" 协议: MCP over stdio (标准输入输出)")
  502. print(" 说明: 通过标准输入输出与 MCP 客户端通信")
  503. elif transport == 'http':
  504. print(f" 监听地址: http://{host}:{port}")
  505. print(f" HTTP端点: http://{host}:{port}/mcp")
  506. print(" 协议: MCP over HTTP (生产环境)")
  507. if project_root:
  508. print(f" 项目目录: {project_root}")
  509. else:
  510. print(" 项目目录: 当前目录")
  511. print()
  512. print(" 已注册的工具:")
  513. print(" === 基础数据查询(P0核心)===")
  514. print(" 1. get_latest_news - 获取最新新闻")
  515. print(" 2. get_news_by_date - 按日期查询新闻(支持自然语言)")
  516. print(" 3. get_trending_topics - 获取趋势话题")
  517. print()
  518. print(" === 智能检索工具 ===")
  519. print(" 4. search_news - 统一新闻搜索(关键词/模糊/实体)")
  520. print(" 5. search_related_news_history - 历史相关新闻检索")
  521. print()
  522. print(" === 高级数据分析 ===")
  523. print(" 6. analyze_topic_trend - 统一话题趋势分析(热度/生命周期/爆火/预测)")
  524. print(" 7. analyze_data_insights - 统一数据洞察分析(平台对比/活跃度/关键词共现)")
  525. print(" 8. analyze_sentiment - 情感倾向分析")
  526. print(" 9. find_similar_news - 相似新闻查找")
  527. print(" 10. generate_summary_report - 每日/每周摘要生成")
  528. print()
  529. print(" === 配置与系统管理 ===")
  530. print(" 11. get_current_config - 获取当前系统配置")
  531. print(" 12. get_system_status - 获取系统运行状态")
  532. print(" 13. trigger_crawl - 手动触发爬取任务")
  533. print("=" * 60)
  534. print()
  535. # 根据传输模式运行服务器
  536. if transport == 'stdio':
  537. mcp.run(transport='stdio')
  538. elif transport == 'http':
  539. # HTTP 模式(生产推荐)
  540. mcp.run(
  541. transport='http',
  542. host=host,
  543. port=port,
  544. path='/mcp' # HTTP 端点路径
  545. )
  546. else:
  547. raise ValueError(f"不支持的传输模式: {transport}")
  548. if __name__ == '__main__':
  549. import sys
  550. import argparse
  551. parser = argparse.ArgumentParser(
  552. description='TrendRadar MCP Server - 新闻热点聚合 MCP 工具服务器',
  553. formatter_class=argparse.RawDescriptionHelpFormatter,
  554. epilog="""
  555. 使用示例:
  556. # STDIO 模式(用于 Cherry Studio)
  557. uv run python mcp_server/server.py
  558. # HTTP 模式(适合远程访问)
  559. uv run python mcp_server/server.py --transport http --port 3333
  560. Cherry Studio 配置示例:
  561. 设置 > MCP Servers > 添加服务器
  562. - 名称: TrendRadar
  563. - 类型: STDIO
  564. - 命令: [UV的完整路径]
  565. - 参数: --directory [项目路径] run python mcp_server/server.py
  566. 详细配置教程请查看: README-Cherry-Studio.md
  567. """
  568. )
  569. parser.add_argument(
  570. '--transport',
  571. choices=['stdio', 'http'],
  572. default='stdio',
  573. help='传输模式:stdio (默认) 或 http (生产环境)'
  574. )
  575. parser.add_argument(
  576. '--host',
  577. default='0.0.0.0',
  578. help='HTTP模式的监听地址,默认 0.0.0.0'
  579. )
  580. parser.add_argument(
  581. '--port',
  582. type=int,
  583. default=3333,
  584. help='HTTP模式的监听端口,默认 3333'
  585. )
  586. parser.add_argument(
  587. '--project-root',
  588. help='项目根目录路径'
  589. )
  590. args = parser.parse_args()
  591. run_server(
  592. project_root=args.project_root,
  593. transport=args.transport,
  594. host=args.host,
  595. port=args.port
  596. )