Browse Source

yinruoxi

feature:
    1.修改:【三国&朝阳大模型Demo】Lagi大模型切换逻辑。
tags/B.2.7.4_20240220_release
kira 1 year ago
parent
commit
87d272fda8
2 changed files with 5 additions and 2 deletions
  1. +1
    -0
      xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/nlt/config/LagiConfig.java
  2. +4
    -2
      xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/nlt/controller/DmIntentController.java

+ 1
- 0
xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/nlt/config/LagiConfig.java View File

@@ -12,5 +12,6 @@ public class LagiConfig {
tenantLlmMap.put("三国演义","sanguo");
tenantLlmMap.put("朝阳政务","chaoyang");
tenantLlmMap.put("朝阳科技园","chaoyang");
tenantLlmMap.put("北京朝阳大数据公司","chaoyang");
}
}

+ 4
- 2
xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/nlt/controller/DmIntentController.java View File

@@ -403,6 +403,7 @@ public class DmIntentController extends BaseController<DmIntentQuery, DmIntentDt
if (StringUtils.isEmpty(category)) {
category = "airport";
}
String llmRoute = category;
JSONObject joResult = new JSONObject();
joResult.put("msg","");
joResult.put("target",0);
@@ -410,14 +411,14 @@ public class DmIntentController extends BaseController<DmIntentQuery, DmIntentDt
vo.setCategory(category);
vo.addDmLlm("user", intent.getContent());
JSONObject testJ = new JSONObject();
testJ.put("category","airport");
testJ.put("category",category);
JSONArray ja1 = new JSONArray();
JSONObject jo = new JSONObject();
jo.put("role","user");
jo.put("content",intent.getContent());
ja1.add(jo);
testJ.put("messages",ja1);
JSONObject resultJson = remoteLandingLlmService.query(category,testJ);
JSONObject resultJson = remoteLandingLlmService.query(llmRoute,testJ);
log.info("北方大返回:{}",resultJson.toString());
if (resultJson.get("status").equals("success")) {
JSONArray ja = resultJson.getJSONArray("data");
@@ -510,6 +511,7 @@ public class DmIntentController extends BaseController<DmIntentQuery, DmIntentDt
testJ.put("messages",ja1);
String gatewayheader = LagiConfig.tenantLlmMap.get(enterpriseDtoR.getData().getName());
gatewayheader = StringUtils.isEmpty(gatewayheader) ? "": gatewayheader;
log.info("北方大请求:{}",gatewayheader);
JSONObject resultJson = remoteLandingLlmService.query(gatewayheader,testJ);
log.info("北方大返回:{}",resultJson.toString());
if (resultJson.get("status").equals("success")) {


Loading…
Cancel
Save