luojunhui 2 veckor sedan
förälder
incheckning
6f332dd546

+ 2 - 2
applications/config/__init__.py

@@ -15,7 +15,7 @@ from .deepseek_config import deep_seek_official_api_key
 from .elastic_search_mappings import es_index, es_mappings, es_settings
 
 # cold start config
-from .cold_start_config import category_config, input_source_map
+from .cold_start_config import cold_start_category_map, input_source_map
 
 # name config
 from .task_chinese_name import name_map
@@ -34,7 +34,7 @@ __all__ = [
     "es_index",
     "es_mappings",
     "es_settings",
-    "category_config",
+    "cold_start_category_map",
     "input_source_map",
     "name_map",
     "CATEGORY_FEATURES",

+ 1 - 1
applications/config/cold_start_config.py

@@ -1,4 +1,4 @@
-category_config = {
+cold_start_category_map = {
     "知识科普": "20250813032110801233225",
     "国家大事": "20250813032845706844854",
     "历史人物": "20250813033415138644740",

+ 3 - 3
applications/tasks/cold_start_tasks/article_pool_cold_start.py

@@ -12,7 +12,7 @@ from tqdm.asyncio import tqdm
 from applications.api import task_apollo, feishu_robot
 from applications.api import auto_create_crawler_task
 from applications.api import auto_bind_crawler_task_to_generate_task
-from applications.config import category_config, input_source_map
+from applications.config import cold_start_category_map, input_source_map
 from applications.utils import get_titles_from_produce_plan
 from applications.tasks.cold_start_tasks.article_pool import (
     ArticlePoolColdStartStrategy,
@@ -283,11 +283,11 @@ class ArticlePoolColdStart(ArticlePoolColdStartStrategy, ArticlePoolFilterStrate
 
             case "strategy_v2":
                 if not category_list:
-                    category_list = list(category_config.keys())
+                    category_list = list(cold_start_category_map.keys())
 
                 for category in tqdm(category_list):
                     try:
-                        plan_id = category_config[category]
+                        plan_id = cold_start_category_map[category]
                         affected_rows = await self.filter_published_titles(plan_id)
                         await self.log_client.log(
                             contents={