|
|
@@ -14,7 +14,6 @@ def generate_prompt(ori_title):
|
|
|
生成prompt
|
|
|
"""
|
|
|
prompt = f"""
|
|
|
- 生成prompt
|
|
|
请将以下标题改写成适合公众号中小程序点击和传播的文章标题，文章标题的写作规范如下，请学习后进行文章标题的编写。直接输出最终的文章标题，文章标题撰写规范如下：
|
|
|
1. 标题结构：要点前置，信息明确
|
|
|
核心信息前置：标题开头直接点出文章的核心内容或亮点，吸引读者注意。例如：
|
|
|
@@ -101,7 +100,7 @@ class TitleRewriteTask:
|
|
|
self.db = DatabaseConnector(db_config=long_articles_config)
|
|
|
self.db.connect()
|
|
|
|
|
|
- def get_articles_batch(self, batch_size=20000):
|
|
|
+ def get_articles_batch(self, batch_size=1000):
|
|
|
"""
|
|
|
从数据库中获取文章
|
|
|
"""
|
|
|
@@ -114,6 +113,17 @@ class TitleRewriteTask:
|
|
|
res = self.db.fetch(query=sql, cursor_type=DictCursor)
|
|
|
return res
|
|
|
|
|
|
+ def update_title_rewrite_status(self, content_trace_id, ori_status, new_status):
|
|
|
+ """
|
|
|
+ 更新标题重写状态
|
|
|
+ """
|
|
|
+ sql = f"""
|
|
|
+ update publish_single_video_source
|
|
|
+ set title_rewrite_status = %s
|
|
|
+ where content_trace_id = %s and ori_status = %s;
|
|
|
+ """
|
|
|
+ self.db.save(query=sql, params=(new_status, content_trace_id, ori_status))
|
|
|
+
|
|
|
def change_each_article(self, article):
|
|
|
"""
|
|
|
对每个标题进行修改
|
