From c695c3b4048ac0e17e0e0c6a6dd979ebb0f24f79 Mon Sep 17 00:00:00 2001 From: chenxudong Date: Tue, 8 Apr 2025 11:16:19 +0800 Subject: [PATCH] =?UTF-8?q?=E8=AE=BE=E7=BD=AEAi=E4=B8=AD=E7=9A=84temperatu?= =?UTF-8?q?re?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../service/serviceimpl/ChatService.java | 23 ++++++++++++++++--- 1 file changed, 20 insertions(+), 3 deletions(-) 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(); } }