From 731352fd04a6e8b483a8490fe1a833439302591c Mon Sep 17 00:00:00 2001
From: ageer <ageerle@163.com>
Date: 星期一, 14 四月 2025 00:22:21 +0800
Subject: [PATCH] feat: mcp测试版
---
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OllamaServiceImpl.java | 21 ++++++++++++---------
1 files changed, 12 insertions(+), 9 deletions(-)
diff --git a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OllamaServiceImpl.java b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OllamaServiceImpl.java
index 8b337b2..aace823 100644
--- a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OllamaServiceImpl.java
+++ b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OllamaServiceImpl.java
@@ -39,13 +39,21 @@
@Autowired
private IChatModelService chatModelService;
- @Autowired
- private ChatClient chatClient;
- @Autowired
- private ToolCallbackProvider tools;
+
+ private final ChatClient chatClient;
private final ChatMemory chatMemory = new InMemoryChatMemory();
+ public OllamaServiceImpl(ChatClient.Builder chatClientBuilder,ToolCallbackProvider tools) {
+ this.chatClient = chatClientBuilder
+ .defaultTools(tools)
+ .defaultOptions(
+ OllamaOptions.builder()
+ .model(OllamaModel.QWEN_2_5_7B)
+ .temperature(0.4)
+ .build())
+ .build();
+ }
@Override
public SseEmitter chat(ChatRequest chatRequest,SseEmitter emitter) {
@@ -104,11 +112,6 @@
this.chatClient.prompt(chatRequest.getPrompt())
.advisors(messageChatMemoryAdvisor)
- .tools(tools)
- .options(OllamaOptions.builder()
- .model(OllamaModel.QWEN_2_5_7B)
- .temperature(0.4)
- .build())
.stream()
.chatResponse()
.subscribe(
--
Gitblit v1.9.3