xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-08 09:17:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: mcp 1.0.0

Browse Source
This commit is contained in:
ageer
2025-04-16 21:56:05 +08:00
parent d1006f50ad
commit 761d954ef1
1 changed files with 27 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OpenAIServiceImpl.java
View File

@@ -6,16 +6,15 @@ import io.modelcontextprotocol.client.transport.WebFluxSseClientTransport;
import io.modelcontextprotocol.spec.McpSchema; import io.modelcontextprotocol.spec.McpSchema;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.ruoyi.chat.listener.SSEEventSourceListener;
import org.ruoyi.chat.service.chat.IChatService; import org.ruoyi.chat.service.chat.IChatService;
import org.ruoyi.common.chat.entity.chat.ChatChoice; import org.ruoyi.common.chat.entity.chat.*;
import org.ruoyi.common.chat.entity.chat.ChatCompletion;
import org.ruoyi.common.chat.entity.chat.ChatCompletionResponse;
import org.ruoyi.common.chat.entity.chat.Parameters;
import org.ruoyi.common.chat.entity.chat.tool.ToolCallFunction; import org.ruoyi.common.chat.entity.chat.tool.ToolCallFunction;
import org.ruoyi.common.chat.entity.chat.tool.Tools; import org.ruoyi.common.chat.entity.chat.tool.Tools;
import org.ruoyi.common.chat.entity.chat.tool.ToolsFunction; import org.ruoyi.common.chat.entity.chat.tool.ToolsFunction;
import org.ruoyi.common.chat.openai.OpenAiStreamClient; import org.ruoyi.common.chat.openai.OpenAiStreamClient;
import org.ruoyi.common.chat.request.ChatRequest; import org.ruoyi.common.chat.request.ChatRequest;
import org.ruoyi.common.core.exception.ServiceException;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.web.reactive.function.client.WebClient; import org.springframework.web.reactive.function.client.WebClient;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
@@ -33,9 +32,27 @@ public class OpenAIServiceImpl implements IChatService {
private final OpenAiStreamClient openAiStreamClient; private final OpenAiStreamClient openAiStreamClient;
@Override @Override
public SseEmitter chat(ChatRequest chatRequest,SseEmitter emitter) { public SseEmitter chat(ChatRequest chatRequest,SseEmitter emitter) {
String toolString = mcpChat(chatRequest);
Message userMessage = Message.builder().content("工具返回信息："+toolString).role(Message.Role.ASSISTANT).build();
List<Message> messages = chatRequest.getMessages();
messages.add(userMessage);
SSEEventSourceListener openAIEventSourceListener = new SSEEventSourceListener(emitter);
ChatCompletion completion = ChatCompletion
.builder()
.messages(messages)
.model(chatRequest.getModel())
.stream(true)
.build();
openAiStreamClient.streamChatCompletion(completion, openAIEventSourceListener);
return emitter;
}
public String mcpChat(ChatRequest chatRequest) {
WebFluxSseClientTransport webFluxSseClientTransport = new WebFluxSseClientTransport(WebClient.builder().baseUrl("http://localhost:8080")); WebFluxSseClientTransport webFluxSseClientTransport = new WebFluxSseClientTransport(WebClient.builder().baseUrl("http://localhost:8080"));
ChatCompletion completion = ChatCompletion ChatCompletion completion = ChatCompletion
.builder() .builder()
@@ -44,6 +61,7 @@ public class OpenAIServiceImpl implements IChatService {
.stream(false) .stream(false)
.build(); .build();
List<Tools> tools = new ArrayList<>(); List<Tools> tools = new ArrayList<>();
McpSchema.Content content = null;
try (var client = McpClient.sync(webFluxSseClientTransport).build()) { try (var client = McpClient.sync(webFluxSseClientTransport).build()) {
client.initialize(); client.initialize();
McpSchema.ListToolsResult toolsList = client.listTools(); McpSchema.ListToolsResult toolsList = client.listTools();
@@ -63,26 +81,17 @@ public class OpenAIServiceImpl implements IChatService {
.build(); .build();
tools.add(tool); tools.add(tool);
} }
completion.setTools(tools); completion.setTools(tools);
ChatCompletionResponse chatCompletionResponse = openAiStreamClient.chatCompletion(completion); ChatCompletionResponse chatCompletionResponse = openAiStreamClient.chatCompletion(completion);
String arguments = chatCompletionResponse.getChoices().get(0).getMessage().getToolCalls().get(0).getFunction().getArguments(); String arguments = chatCompletionResponse.getChoices().get(0).getMessage().getToolCalls().get(0).getFunction().getArguments();
String name = chatCompletionResponse.getChoices().get(0).getMessage().getToolCalls().get(0).getFunction().getName(); String name = chatCompletionResponse.getChoices().get(0).getMessage().getToolCalls().get(0).getFunction().getName();
Map<String, Object> map = JSONUtil.toBean(arguments, Map.class); Map<String, Object> map = JSONUtil.toBean(arguments, Map.class);
McpSchema.CallToolResult sumResult = client.callTool(new McpSchema.CallToolRequest(name, map)); McpSchema.CallToolResult sumResult = client.callTool(new McpSchema.CallToolRequest(name, map));
System.out.println("add a+ b = " + sumResult.content().get(0)); content= sumResult.content().get(0);
} catch (Exception e) {
throw new ServiceException("请求失败"+e);
McpSchema.Content content = sumResult.content().get(0);
emitter.send(sumResult.content().get(0));
} catch (IOException e) {
emitter.completeWithError(e);
} }
emitter.complete(); return content.toString();
return emitter;
} }
} }