diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java index a1d17e2..29187ba 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java @@ -30,6 +30,7 @@ import org.springframework.ai.chat.prompt.Prompt; import org.springframework.ai.document.Document; import org.springframework.ai.document.DocumentReader; import org.springframework.ai.ollama.OllamaChatModel; +import org.springframework.ai.ollama.api.OllamaOptions; import org.springframework.ai.reader.TextReader; import org.springframework.ai.reader.pdf.PagePdfDocumentReader; import org.springframework.ai.transformer.splitter.TokenTextSplitter; @@ -241,7 +242,16 @@ public class ChatService { UserThreadLocal.set(userLoginInfo); aiQuestionRecordMapper.insert(record); UserThreadLocal.remove(); - return ChatClient.builder(model).defaultAdvisors(messageChatMemoryAdvisor, questionAnswerAdvisor).build().prompt(queryDTO.getMsg()).stream().content(); + return ChatClient.builder(model) + .defaultAdvisors(messageChatMemoryAdvisor, questionAnswerAdvisor) + .defaultOptions(OllamaOptions + .builder() + .temperature(0.3) + .build()) + .build() + .prompt(queryDTO.getMsg()) + .stream() + .content(); } public Flux chatStreamResponse(QueryDTO queryDTO) { @@ -251,8 +261,15 @@ public class ChatService { UserThreadLocal.set(userLoginInfo); UserThreadLocal.remove(); aiQuestionRecordMapper.insert(record); - return ChatClient.builder(model).defaultAdvisors(messageChatMemoryAdvisor, questionAnswerAdvisor).build() + return ChatClient.builder(model) + .defaultAdvisors(messageChatMemoryAdvisor, questionAnswerAdvisor) + .defaultOptions(OllamaOptions + .builder() + .temperature(0.3) + .build()) + .build() .prompt(new Prompt(new UserMessage(queryDTO.getMsg()))) - .stream().chatResponse(); + .stream() + .chatResponse(); } }