国内AI时讯

daily/report_data.py

@@ -8,7 +8,7 @@ from typing import Any
 
 from daily.config import OUTPUT_DIR, env_int
 from daily.delta import build_movement_baseline, build_movement_context, compare_depth
-from daily.news.fetch import prepare_wecom_news_items
+from daily.news.fetch import prepare_wecom_cn_news_items, prepare_wecom_news_items
 from daily.skills_group import group_skills_by_source
 
 
@@ -54,9 +54,14 @@ def _slim_github(item: dict[str, Any]) -> dict[str, Any]:
     }
 
 
-def _slim_news_items(ai_news: dict[str, Any], limit: int) -> list[dict[str, Any]]:
+def _slim_news_items(
+    ai_news: dict[str, Any],
+    limit: int,
+    *,
+    prepare=prepare_wecom_news_items,
+) -> list[dict[str, Any]]:
     items: list[dict[str, Any]] = []
-    for item in prepare_wecom_news_items(ai_news):
+    for item in prepare(ai_news):
         items.append(
             {
                 "link": item.get("link", ""),
@@ -98,10 +103,12 @@ def build_llm_input(
     github_topic: list[dict[str, Any]],
     topic_name: str,
     ai_news: dict[str, Any],
+    cn_ai_news: dict[str, Any],
     wecom_limits: dict[str, int],
 ) -> dict[str, Any]:
     """供 Cursor 编辑的精简 JSON（不含完整 markdown）。"""
     news_limit = wecom_limits.get("ai_news", 10)
+    cn_news_limit = wecom_limits.get("cn_ai_news", 8)
     depth = compare_depth()
     trend_cmp = trending[:depth]
     hot_cmp = hot[:depth]
@@ -153,6 +160,11 @@ def build_llm_input(
             "repos": [_slim_github(x) for x in topic_slice],
         },
         "ai_news": _slim_news_items(ai_news, news_limit) if ai_news.get("enabled") else [],
+        "cn_ai_news": _slim_news_items(
+            cn_ai_news, cn_news_limit, prepare=prepare_wecom_cn_news_items
+        )
+        if cn_ai_news.get("enabled")
+        else [],
         "movement": movement,
         "movement_baseline": movement_baseline,
     }