标题生成品类任务

title_process_task.py

@@ -1,7 +1,8 @@
 """
 @author: luojunhui
 """
-from tasks.ai_tasks.category_generation_task import CategoryGenerationTask
+from tasks.ai_tasks.category_generation_task import ArticlePoolCategoryGenerationTask
+from tasks.ai_tasks.category_generation_task import VideoPoolCategoryGenerationTask
 from tasks.title_rewrite_task import TitleRewriteTask
 
 
@@ -10,7 +11,11 @@ if __name__ == '__main__':
     title_rewrite_task = TitleRewriteTask()
     title_rewrite_task.deal()
 
-    # 2. 标题分类
-    category_generation_task = CategoryGenerationTask()
-    category_generation_task.deal()
+    # 2. 视频内容池标题分类
+    video_pool_category_generation_task = VideoPoolCategoryGenerationTask()
+    video_pool_category_generation_task.deal()
+
+    # 3. 文章内容池标题分类
+    article_pool_category_generation_task = ArticlePoolCategoryGenerationTask()
+    article_pool_category_generation_task.deal()