From 95ec39637c2528257de70e7bc4d04e6539801cec Mon Sep 17 00:00:00 2001
From: Ryan Bogan <rbogan@amazon.com>
Date: Wed, 3 Jan 2024 10:34:39 -0800
Subject: [PATCH] Add necessary imports

Signed-off-by: Ryan Bogan <rbogan@amazon.com>
---
 .../opensearch/knn/plugin/transport/GetModelResponseTests.java  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/test/java/org/opensearch/knn/plugin/transport/GetModelResponseTests.java b/src/test/java/org/opensearch/knn/plugin/transport/GetModelResponseTests.java
index e0ee6bf3b2..a54605d247 100644
--- a/src/test/java/org/opensearch/knn/plugin/transport/GetModelResponseTests.java
+++ b/src/test/java/org/opensearch/knn/plugin/transport/GetModelResponseTests.java
@@ -14,7 +14,9 @@
 import org.mockito.MockedStatic;
 import org.opensearch.Version;
 import org.opensearch.common.io.stream.BytesStreamOutput;
+import org.opensearch.core.xcontent.MediaTypeRegistry;
 import org.opensearch.core.xcontent.XContentBuilder;
+import org.opensearch.common.xcontent.XContentType;
 import org.opensearch.knn.KNNTestCase;
 import org.opensearch.knn.index.KNNClusterUtil;
 import org.opensearch.knn.index.SpaceType;