diff --git a/src/main/java/com/bfd/entity/Constants.java b/src/main/java/com/bfd/entity/Constants.java index 514077b..2f1ee2b 100644 --- a/src/main/java/com/bfd/entity/Constants.java +++ b/src/main/java/com/bfd/entity/Constants.java @@ -180,6 +180,7 @@ public class Constants { public static final String DEEPSEEK_VALUE = "6"; public static final String DEEPSEEK_LOCAL_VALUE = "7"; public static final String QWEN_MAX_VALUE = "8"; + public static final String BAIDU_MAX_VALUE = "9"; public static final String MODELADDRESS = "modelAddress"; public static final String AUTHORIZATIONS = "authorizations"; public static final String AUTHORIZATION = "authorization"; diff --git a/src/main/java/com/bfd/process/ModelSelectProcess.java b/src/main/java/com/bfd/process/ModelSelectProcess.java index be204ec..263d7de 100644 --- a/src/main/java/com/bfd/process/ModelSelectProcess.java +++ b/src/main/java/com/bfd/process/ModelSelectProcess.java @@ -65,6 +65,9 @@ public class ModelSelectProcess implements Runnable{ }else if(llm.equals(Constants.QWEN_MAX_VALUE)){ log.info("QWEN 智能模型任务触发-----"); response = DownLoadUtil.doPost(address, JSONObject.toJSONString(task)); + }else if(llm.equals(Constants.BAIDU_MAX_VALUE)){ + log.info("文心一言 智能模型任务触发-----"); + response = DownLoadUtil.doPost(address, JSONObject.toJSONString(task)); }else{ log.error("未知模型,不处理"); }