From d23768dfa14125f074bfc0b5ed569b4c3e514f4f Mon Sep 17 00:00:00 2001
From: jade <jade.kimhs@gmail.com>
Date: Mon, 13 Jul 2026 22:21:24 +0900
Subject: [PATCH] feat: Add categoryKey filtering support to /tools/list API

---
 .../mcp/gateway/presentation/McpRouterController.java    | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/axhub-gateway/src/main/java/io/shinhanlife/axhub/biz/mcp/gateway/presentation/McpRouterController.java b/axhub-gateway/src/main/java/io/shinhanlife/axhub/biz/mcp/gateway/presentation/McpRouterController.java
index 66b0834..7a2b52a 100644
--- a/axhub-gateway/src/main/java/io/shinhanlife/axhub/biz/mcp/gateway/presentation/McpRouterController.java
+++ b/axhub-gateway/src/main/java/io/shinhanlife/axhub/biz/mcp/gateway/presentation/McpRouterController.java
@@ -96,7 +96,8 @@ public class McpRouterController {
     }
 
     @GetMapping("/tools/list")
-    public ResponseEntity<JsonRpcResponse> listTools() {
+    public ResponseEntity<JsonRpcResponse> listTools(
+            @RequestParam(value = "categoryKey", required = false) String categoryKey) {
         List<ToolMetadata> activeTools = redisRegistryService.getAllTools()
                 .stream()
                 .filter(ToolMetadata::getVisible)
@@ -131,6 +132,12 @@ public class McpRouterController {
             }
         }
         
+        if (categoryKey != null && !categoryKey.trim().isEmpty()) {
+            activeTools = activeTools.stream()
+                    .filter(t -> categoryKey.equals(t.getCategoryKey()))
+                    .collect(Collectors.toList());
+        }
+
         JsonRpcResponse response = new JsonRpcResponse();
         response.setId(UUID.randomUUID().toString());
         response.setResult(Map.of("tools", activeTools));