From 0e2c00bd0026da383bd7d2e49a344a87577488be Mon Sep 17 00:00:00 2001
From: mohitdeuex <mohit.y@deuexsolutions.com>
Date: Mon, 16 Jun 2025 23:03:49 +0530
Subject: [PATCH] Add cleanSearchResponseObject()

---
 .../service/mcp/tools/SearchMetadataTool.java             | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

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<String, Object> cleanSearchResponse(Map<String, Object> searchResponse) {
+  public static Map<String, Object> cleanSearchResponse(Map<String, Object> searchResponse) {
     if (searchResponse == null) return Collections.emptyMap();
 
     Map<String, Object> topHits = safeGetMap(searchResponse.get("hits"));
@@ -142,6 +142,12 @@ public class SearchMetadataTool implements McpTool {
     return Collections.emptyMap();
   }
 
+  @SuppressWarnings("unused")
+  public static Map<String, Object> cleanSearchResponseObject(Map<String, Object> object) {
+    IGNORE_SEARCH_KEYS.forEach(object::remove);
+    return object;
+  }
+
   @SuppressWarnings("unchecked")
   private static Map<String, Object> safeGetMap(Object obj) {
     return (obj instanceof Map) ? (Map<String, Object>) obj : null;