Procházet zdrojové kódy

yinruoxi

新增:
    1.添加电影角色自由对话
tags/B.2.3.2_20231118_base
kira před 2 roky
rodič
revize
6c21113f9a
2 změnil soubory, kde provedl 12 přidání a 2 odebrání
  1. +2
    -1
      xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/nlt/domain/LlmContext.java
  2. +10
    -1
      xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/nlt/template/MovieChatTemplate.java

+ 2
- 1
xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/nlt/domain/LlmContext.java Zobrazit soubor

@@ -35,10 +35,11 @@ public class LlmContext implements Serializable {
// 将剩余的元素按照role值为"user"和"assistant"的顺序加入到context中
for (int i = containsSystem ? 1 : 0; i < contentList.size(); i++) {
LlmContent ctx = new LlmContent();
ctx.setRole(i + cur % 2 == 0 ? "user" : "assistant");
ctx.setRole((i + cur) % 2 == 0 ? "user" : "assistant");
ctx.setContent(contentList.get(i));
llmContents.add(ctx);
}
llmContext.setContentList(llmContents);
return llmContext;
}



+ 10
- 1
xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/nlt/template/MovieChatTemplate.java Zobrazit soubor

@@ -34,6 +34,12 @@ public class MovieChatTemplate implements BaseTemplate{

@Override
public JSONObject handle(String devId, String content) {
// 判断content如果小于5五个字,返回msg:对不起,能否再描述清除一些。
if(content.length() < 5){
JSONObject resultJson = new JSONObject();
resultJson.put("msg","对不起,能否再描述清除一些。");
return resultJson;
}
Long operatorId = TerminalSecurityContextHolder.getOperatorId();
String redisKey = "group:nlp:" + SecurityContextHolder.getLocalMap().get("enterprise_id") + ":" + operatorId;
// 根据content内容调用模版并返回结果
@@ -61,7 +67,10 @@ public class MovieChatTemplate implements BaseTemplate{
break;
}
}

if(!StringUtils.isEmpty(result)){
redisTemplate.opsForList().rightPush(redisKey,content);
redisTemplate.opsForList().rightPush(redisKey,result);
}

JSONObject resultJson = new JSONObject();
resultJson.put("msg",result);


Načítá se…
Zrušit
Uložit