yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-08-19 06:28:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add cleanSearchResponseObject()

Browse Source
This commit is contained in:
mohitdeuex 2025-06-16 23:03:49 +05:30
parent f940f27988
commit 0e2c00bd00
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
openmetadata-service/src/main/java/org/openmetadata/service/mcp/tools/SearchMetadataTool.java
View File

@ -119,7 +119,7 @@ public class SearchMetadataTool implements McpTool {
"SearchMetadataTool does not support limits enforcement."); "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(); if (searchResponse == null) return Collections.emptyMap();
Map<String, Object> topHits = safeGetMap(searchResponse.get("hits")); Map<String, Object> topHits = safeGetMap(searchResponse.get("hits"));
@ -142,6 +142,12 @@ public class SearchMetadataTool implements McpTool {
return Collections.emptyMap(); 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") @SuppressWarnings("unchecked")
private static Map<String, Object> safeGetMap(Object obj) { private static Map<String, Object> safeGetMap(Object obj) {
return (obj instanceof Map) ? (Map<String, Object>) obj : null; return (obj instanceof Map) ? (Map<String, Object>) obj : null;