Преглед на файлове

文章品类prompt改为配置

wangyunpeng преди 7 месеца
родител
ревизия
75be40c7ad

+ 5 - 36
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/service/recommend/ArticleService.java

@@ -42,6 +42,7 @@ import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.collections4.CollectionUtils;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.util.StringUtils;
 
@@ -88,6 +89,8 @@ public class ArticleService {
             "\"20240805154433785506170\", \"20240805154359027876170\", \"20241024100016206421084\", " +
             "\"20241030070010871546586\"]}")
     private static List<String> producePlanIds;
+    @Value("${kimiCategoryPrompt:}")
+    private String kimiCategoryPrompt;
 
     private final static ExecutorService pool = new CommonThreadPoolExecutor(
             32,
@@ -453,42 +456,8 @@ public class ArticleService {
     }
 
     private String buildKimiPrompt(List<String> titleList) {
-        StringBuilder prompt = new StringBuilder(
-                "请帮我完成以下任务:输入为文章的标题,根据标题判断其内容所属的类目,输出为文章标题及其对应的类目。\n" +
-                "类目需从以下范围内选择:\n" +
-                "奇闻趣事\n" +
-                "历史人物\n" +
-                "家长里短\n" +
-                "温情故事\n" +
-                "健康养生\n" +
-                "生活知识\n" +
-                "名人八卦\n" +
-                "政治新闻\n" +
-                "军事新闻\n" +
-                "影视解读\n" +
-                "为了更好地完成任务,可参考下列对文章标题的分类:\n" +
-                "{" +
-                "\"大舅病了,我取了三万元送过去,病房门口听到舅妈的话我改了主意\": \"家长里短\",\n" +
-                "\"能活到90岁的老人,基本上在70岁的时候,就不再做这些事了!\": \"健康养生\",\n" +
-                "\"去医院看望病人时,切忌带这4样东西,再亲近也不行,这是做人的根本\": \"生活知识\",\n" +
-                "\"上海一女子去饭店吃生煎包,戳了个洞想凉一凉,往里一看,瞬间惊呆了\": \"奇闻趣事\",\n" +
-                "\"卫生间放一把食盐,一年能省下好几百,涨新知识\": \"生活知识\",\n" +
-                "\"中国有一古寺,庙不大,却有武警24小时站岗,到底有何“过人”之处\": \"奇闻趣事\",\n" +
-                "\"1974年,苏联外长故意拿邓小平身高“取笑”,邓小平一句话轻松反击\": \"历史人物\",\n" +
-                "\"中国最美的女将军:上世纪曾家喻户晓,如今仍然健在\": \"历史人物\",\n" +
-                "\"北大才女蒙曼48岁仍未婚,被问最想嫁给谁,一个名字让全场笑喷\": \"名人八卦\",\n" +
-                "\"广东一老人去世,家人把老人的旧床垫扔了,环卫工人看到后,竟发现里面藏了15万元现金!家人傻眼了\": \"奇闻趣事\",\n" +
-                "\"《孤舟》结局:要不是海沫2年不生娃,周知非至死不知,顾易中要3000元的报销费的真实目的!\": \"影视解读\"\n" +
-                "}" +
-                "最后输出结果请用JSON格式输出,key为title,value为类目,仅输出JSON,不要markdown格式,不要任何其他内容," +
-                "并且内容可以被 fastJSON 的JSONObject.parseObject转换为JSON对象\n" +
-                "当标题的开头或结尾为以下字符时“”“,则在标题的开头或结尾增加\" " +
-                "输出结果格式如下:\n" +
-                "{" +
-                "\"浙江老人用“假钱”吃霸王餐9年,离世后,老板却崩溃大哭:“每天都在等他!”\": \"奇闻趣事\"," +
-                "\"“最美婴儿”迅速走红,像在娘胎里整过容,网友:看到第一眼就想抱回家\": \"奇闻趣事\"" +
-                "}" +
-                "以下是需要分析的文章标题列表,每一行是一个标题:\n");
+        StringBuilder prompt = new StringBuilder(kimiCategoryPrompt);
+        prompt.append("\n");
         for (String title : titleList) {
             prompt.append(title).append("\n");
         }