gitea/edward
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改微头条prompt
All checks were successful
Gitea Actions Demo / deploy (push) Successful in 2m51s
Details

Browse Source
This commit is contained in:
konjacpotato
2026-02-27 10:35:49 +08:00
parent 5d35cd955c
commit 9361eb97bf
2 changed files with 32 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
task/hot_topic/real_estate_story.py
View File

@ -113,43 +113,7 @@ def story_edit_task():
db.commit()
# break # 目前先处理一条内容，后续再改成批量处理
llm_engine = LLMThinkingEngine(system_prompt_file="real_estate_story_system_prompt.txt")
for content in to_processed_contents:
logger.info(f"story_edit_task content id: {content.id}, title: {content.link}, platform: {content.platform}")
story = llm_engine.think(f"故事素材：{content.content}")
logger.info(f"story_edit_task content id: {content.id} story: {story}")
# llm生成的结果有时不是json结构，会在前后增加一些文本，需要提取出json部分进行解析
try:
json_start = story.find("{")
json_end = story.rfind("}") + 1
if json_start != -1 and json_end != -1:
story = story[json_start:json_end]
else:
logger.warning(f"story_edit_task content id: {content.id} llm生成的结果不是有效的json格式，无法提取故事内容")
continue
except json.JSONDecodeError:
logger.warning(f"story_edit_task content id: {content.id} llm生成的结果不是有效的json格式，无法解析故事内容")
continue
# 将生成的故事写入Article表
try:
json_story = json.loads(story)
except json.JSONDecodeError:
logger.warning(f"story_edit_task content id: {content.id} llm生成的故事内容不是有效的json格式，无法解析故事内容")
continue
title = json_story.get("title", "无标题")
paragraphs = json_story.get("body", ["无内容"])
article_content = "\n".join(paragraphs)
article = Article(
title=title,
keywords=None,
content=article_content,
used=False
)
db.add(article)
db.commit()
# break # 目前先处理一条内容，后续再改成批量处理
llm_engine = LLMThinkingEngine(system_prompt_file="real_estate_story_short_system_prompt.txt")
llm_engine = LLMThinkingEngine(system_prompt_file="wei_toutiao_system_prompt.txt")
for content in to_processed_contents:
logger.info(f"story_edit_task content id: {content.id}, title: {content.link}, platform: {content.platform}")
story = llm_engine.think(f"故事素材：{content.content}")
@ -185,5 +149,6 @@ def story_edit_task():
db.commit()
# break # 目前先处理一条内容，后续再改成批量处理
if __name__ == "__main__":
execute_task(story_edit_task)