Skip to content

Commit 3bb49fb

Browse files
committed
Revert "Add MCP proxy server support for prompts"
This reverts commit ebd7310.
1 parent 333eff0 commit 3bb49fb

File tree

1 file changed

+5
-16
lines changed
  • mcp/mcp-server/src/main/java/software/amazon/smithy/java/mcp/server

1 file changed

+5
-16
lines changed

mcp/mcp-server/src/main/java/software/amazon/smithy/java/mcp/server/McpService.java

Lines changed: 5 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -220,19 +220,8 @@ private JsonRpcResponse handleInitialize(JsonRpcRequest req) {
220220
}
221221

222222
private JsonRpcResponse handlePromptsList(JsonRpcRequest req) {
223-
var allPrompts = new ArrayList<>(prompts.values().stream().map(Prompt::promptInfo).toList());
224-
225-
// Add prompts from proxy servers
226-
for (McpServerProxy proxy : proxies.values()) {
227-
var response = proxy.rpc(req).join();
228-
if (response.getError() == null) {
229-
var proxyPrompts = response.getResult().asShape(ListPromptsResult.builder()).getPrompts();
230-
allPrompts.addAll(proxyPrompts);
231-
}
232-
}
233-
234223
var result = ListPromptsResult.builder()
235-
.prompts(allPrompts)
224+
.prompts(prompts.values().stream().map(Prompt::promptInfo).toList())
236225
.build();
237226
return createSuccessResponse(req.getId(), result);
238227
}
@@ -243,12 +232,12 @@ private JsonRpcResponse handlePromptsGet(JsonRpcRequest req) {
243232

244233
var prompt = prompts.get(normalize(promptName));
245234

246-
if (prompt != null) {
247-
var result = promptProcessor.buildPromptResult(prompt, promptArguments);
248-
return createSuccessResponse(req.getId(), result);
235+
if (prompt == null) {
236+
throw new RuntimeException("Prompt not found: " + promptName);
249237
}
250238

251-
throw new RuntimeException("Prompt not found: " + promptName);
239+
var result = promptProcessor.buildPromptResult(prompt, promptArguments);
240+
return createSuccessResponse(req.getId(), result);
252241
}
253242

254243
private JsonRpcResponse handleToolsList(JsonRpcRequest req, ProtocolVersion protocolVersion) {

0 commit comments

Comments
 (0)