diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/mcp/tools/SearchMetadataTool.java b/openmetadata-service/src/main/java/org/openmetadata/service/mcp/tools/SearchMetadataTool.java index 8df71a1cdae..e622e3111a5 100644 --- a/openmetadata-service/src/main/java/org/openmetadata/service/mcp/tools/SearchMetadataTool.java +++ b/openmetadata-service/src/main/java/org/openmetadata/service/mcp/tools/SearchMetadataTool.java @@ -119,7 +119,7 @@ public class SearchMetadataTool implements McpTool { "SearchMetadataTool does not support limits enforcement."); } - private static Map cleanSearchResponse(Map searchResponse) { + public static Map cleanSearchResponse(Map searchResponse) { if (searchResponse == null) return Collections.emptyMap(); Map topHits = safeGetMap(searchResponse.get("hits")); @@ -142,6 +142,12 @@ public class SearchMetadataTool implements McpTool { return Collections.emptyMap(); } + @SuppressWarnings("unused") + public static Map cleanSearchResponseObject(Map object) { + IGNORE_SEARCH_KEYS.forEach(object::remove); + return object; + } + @SuppressWarnings("unchecked") private static Map safeGetMap(Object obj) { return (obj instanceof Map) ? (Map) obj : null;