@@ -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