Commit 846cc766 authored by alex yao's avatar alex yao

feat:长文档对话模型配置

parent 96ad2bbe
......@@ -8,6 +8,7 @@ import cn.com.poc.agent_application.service.BizAgentApplicationGcConfigService;
import cn.com.poc.ai_dialogues.entity.BizAiDialoguesEntity;
import cn.com.poc.ai_dialogues.service.BizAiDialoguesService;
import cn.com.poc.common.constant.CommonConstant;
import cn.com.poc.common.constant.FmxParamConfigConstant;
import cn.com.poc.common.model.BizFileUploadRecordModel;
import cn.com.poc.common.service.BizFileUploadRecordService;
import cn.com.poc.common.utils.DocumentLoad;
......@@ -35,6 +36,7 @@ import cn.com.poc.thirdparty.resource.demand.ai.entity.largemodel.LargeModelDema
import cn.com.poc.thirdparty.resource.demand.ai.entity.largemodel.LargeModelResponse;
import cn.com.poc.thirdparty.service.LLMService;
import cn.com.yict.framemax.core.exception.BusinessException;
import cn.com.yict.framemax.frame.service.FmxParamConfigService;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.crypto.digest.MD5;
import org.apache.commons.collections4.CollectionUtils;
......@@ -92,6 +94,9 @@ public class LongTextDialoguesServiceImpl implements LongTextDialoguesService {
@Resource
private BizLongTextSummaryCacheService bizLongTextSummaryCacheService;
@Resource
private FmxParamConfigService fmxParamConfigService;
@Override
public void call(String dialoguesId, String fileUrl, String input, Integer[] knowledgeIds, Long userId) throws Exception {
......@@ -160,11 +165,11 @@ public class LongTextDialoguesServiceImpl implements LongTextDialoguesService {
sseUtil.send(JsonUtils.serialize(result));
}
String longTextDialoguesModel = fmxParamConfigService.getParam("longtext.dialogues_model");
//组装请求参数
List<Message> messages = buildMessages(dialoguesId, userId, input, fileUrl, knowledgeResult);
LargeModelResponse largeModelResponse = new LargeModelResponse();
largeModelResponse.setModel("qwen-long");
largeModelResponse.setModel(longTextDialoguesModel);
largeModelResponse.setMessages(messages.toArray(new Message[0]));
largeModelResponse.setStream(true);
BufferedReader bufferedReader = llmService.chatChunk(largeModelResponse);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment