Commit 9e71524c authored by alex yao's avatar alex yao

fix: 修改document_understanding_function 模型

parent b4555589
......@@ -28,7 +28,7 @@ public class DocumentUnderstandIngFunction extends AbstractLargeModelFunction {
@Resource
private LLMService llmService;
private final String MODEL = "qwen-max-longcontext";
private final String MODEL = "qwen-long";
private final String TEMPLATE = "# 工作规范:\n 1.工作流程:①对提供的文档内容进行理解,支持信息检索、摘要总结、文本分析。②根据用户提出的问题,提取或者总结文档中与问题相关的内容。2.工作限制:①要将问题与文档内容精准匹配,在理解文档时要带着问题去理解 \n\n" +
"## 文档内容\n" +
......@@ -77,7 +77,11 @@ public class DocumentUnderstandIngFunction extends AbstractLargeModelFunction {
largeModelResponse.setStream(false);
largeModelResponse.setUser("Document_Understanding");
LargeModelDemandResult largeModelDemandResult = llmService.chat(largeModelResponse);
if (largeModelDemandResult == null) {
result.setFunctionResult(StringUtils.EMPTY);
result.setPromptContent(StringUtils.EMPTY);
return result;
}
result.setFunctionResult(largeModelDemandResult.getMessage());
result.setPromptContent(largeModelDemandResult.getMessage());
......
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