slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-05-12 16:53:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(LocalRunnerClient): support initializing in-process dynamic action MCP Server

Browse Source
This commit is contained in:
slhafzjw
2025-12-24 21:36:39 +08:00
parent 592e2604d9
commit d614ac0b15
2 changed files with 34 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Partner-Main/pom.xml
View File

@@ -56,6 +56,12 @@
<version>0.5.0</version> <version>0.5.0</version>
<scope>test</scope> <scope>test</scope>
</dependency> </dependency>
<dependency>
<groupId>work.slhaf</groupId>
<artifactId>Partner-Common</artifactId>
<version>0.5.0</version>
<scope>compile</scope>
</dependency>
</dependencies> </dependencies>
<properties> <properties>

29
Partner-Main/src/main/java/work/slhaf/partner/core/action/runner/LocalRunnerClient.java
View File

@@ -10,6 +10,8 @@ import io.modelcontextprotocol.client.transport.ServerParameters;
import io.modelcontextprotocol.client.transport.StdioClientTransport; import io.modelcontextprotocol.client.transport.StdioClientTransport;
import io.modelcontextprotocol.client.transport.customizer.McpSyncHttpClientRequestCustomizer; import io.modelcontextprotocol.client.transport.customizer.McpSyncHttpClientRequestCustomizer;
import io.modelcontextprotocol.json.McpJsonMapper; import io.modelcontextprotocol.json.McpJsonMapper;
import io.modelcontextprotocol.server.McpServer;
import io.modelcontextprotocol.server.McpStatelessAsyncServer;
import io.modelcontextprotocol.spec.McpClientTransport; import io.modelcontextprotocol.spec.McpClientTransport;
import io.modelcontextprotocol.spec.McpSchema; import io.modelcontextprotocol.spec.McpSchema;
import lombok.Data; import lombok.Data;
@@ -17,6 +19,7 @@ import lombok.extern.slf4j.Slf4j;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable; import org.jetbrains.annotations.Nullable;
import org.jetbrains.annotations.UnknownNullability; import org.jetbrains.annotations.UnknownNullability;
import work.slhaf.partner.common.mcp.InProcessMcpTransport;
import work.slhaf.partner.core.action.entity.McpData; import work.slhaf.partner.core.action.entity.McpData;
import work.slhaf.partner.core.action.entity.MetaAction; import work.slhaf.partner.core.action.entity.MetaAction;
import work.slhaf.partner.core.action.entity.MetaActionInfo; import work.slhaf.partner.core.action.entity.MetaActionInfo;
@@ -29,7 +32,6 @@ import java.io.File;
import java.io.IOException; import java.io.IOException;
import java.io.InputStreamReader; import java.io.InputStreamReader;
import java.nio.file.*; import java.nio.file.*;
import java.nio.file.attribute.BasicFileAttributes;
import java.time.Duration; import java.time.Duration;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.HashMap; import java.util.HashMap;
@@ -46,14 +48,32 @@ import static work.slhaf.partner.common.Constant.Path.TMP_ACTION_DIR_LOCAL;
public class LocalRunnerClient extends RunnerClient { public class LocalRunnerClient extends RunnerClient {
private final Map<String, McpSyncClient> mcpClients = new HashMap<>(); private final Map<String, McpSyncClient> mcpClients = new HashMap<>();
/**
* 动态生成的行动程序都将挂载至该 McpServer
*/
private McpStatelessAsyncServer dynamicActionMcpServer;
public LocalRunnerClient(ConcurrentHashMap<String, MetaActionInfo> existedMetaActions, ExecutorService executor, @Nullable String actionWatchPath) { public LocalRunnerClient(ConcurrentHashMap<String, MetaActionInfo> existedMetaActions, ExecutorService executor, @Nullable String actionWatchPath) {
super(existedMetaActions, executor); super(existedMetaActions, executor);
ActionWatchService watchService = new ActionWatchService(actionWatchPath); ActionWatchService watchService = new ActionWatchService(actionWatchPath);
watchService.launch(); watchService.launch();
registerDynamicActionMcp();
setupShutdownHook(); setupShutdownHook();
} }
private void registerDynamicActionMcp() {
InProcessMcpTransport.Pair pair = InProcessMcpTransport.pair();
McpSchema.ServerCapabilities serverCapabilities = McpSchema.ServerCapabilities.builder()
.tools(true)
.resources(true, true)
.build();
dynamicActionMcpServer = McpServer.async(pair.serverSide())
.capabilities(serverCapabilities)
.jsonMapper(McpJsonMapper.getDefault())
.build();
registerMcpClient("dynamic-action", pair.clientSide(), 10);
}
@Override @Override
protected RunnerResponse doRun(MetaAction metaAction) { protected RunnerResponse doRun(MetaAction metaAction) {
RunnerResponse response; RunnerResponse response;
@@ -224,8 +244,13 @@ public class LocalRunnerClient extends RunnerClient {
*/ */
private void registerMcpClient(String id, McpClientTransportParams mcpClientTransportParams) { private void registerMcpClient(String id, McpClientTransportParams mcpClientTransportParams) {
McpClientTransport clientTransport = createTransport(mcpClientTransportParams); McpClientTransport clientTransport = createTransport(mcpClientTransportParams);
int timeout = mcpClientTransportParams.timeout;
registerMcpClient(id, clientTransport, timeout);
}
private void registerMcpClient(String id, McpClientTransport clientTransport, int timeout) {
McpSyncClient client = McpClient.sync(clientTransport) McpSyncClient client = McpClient.sync(clientTransport)
.requestTimeout(Duration.ofSeconds(mcpClientTransportParams.timeout)) .requestTimeout(Duration.ofSeconds(timeout))
.clientInfo(new McpSchema.Implementation(id, "PARTNER")) .clientInfo(new McpSchema.Implementation(id, "PARTNER"))
// 行动程序(现 MCP Tool)的描述文本将直接由resources返回 // 行动程序(现 MCP Tool)的描述文本将直接由resources返回
// 原因: ToolChange 发送的内容侧重调用，缺少可承担描述文本的字段 // 原因: ToolChange 发送的内容侧重调用，缺少可承担描述文本的字段