LongArticleTaskServer/test_prompt.py

from applications.tasks.llm_tasks.prompts import extract_article_features
from applications.api import fetch_deepseek_completion


title_list = [
    "他从副总理一下被撸成普通工人，事后反倒安慰妻子：我心里面有数",
    "​莫言曾说：“谁都靠不住，除非你有用，在这个烟换烟，酒换酒的时代，生活万般苦，唯有自己渡。”",
    "1948年国军排长带队投诚，谁知当他来到登记处时，文书却是大惊：张定元你不是在2年前就牺牲了吗",
    "21年前抛弃侯耀文，转身嫁给师侄戴志诚的袁茵，如今活成了这样",
    "国外一对夫妻住饭店晚上开窗通风，第二天醒来后，想不到竟有256只挤满房间，当场直接吓坏了",
    "董卿说“娶谁都需要房子，娶谁都需要努力挣钱，但你永远都不要说，为了我而努力这种话，难道没有我，你就不需要给别人吗？“",
    "原来九三阅兵的解说是他俩！豪迈大气庄重激昂，难怪全国人民喜爱",
    "一颗白菜卖60多块钱！拯救韩国人今年的餐桌，全靠山东这个小城了！",
    "63岁冯远征带71岁妻子回老家，两人牵手逛景点，梁丹妮短发好利落",
    "我国最大的一块“飞地”，比7个上海市还要大，却不足2000人",
]

prompt = extract_article_features(title_list)
print(prompt)
res = fetch_deepseek_completion(model="default", prompt=prompt, output_type="json")

import json

print(json.dumps(res, ensure_ascii=False, indent=4))