Implement Custom interface parse method in Connector interface

Signed-off-by: Daniel Widdis <[email protected]>
dbwiddis · May 10, 2024 · 79b2e1f · 79b2e1f
1 parent d3fdbdb
commit 79b2e1f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/common/src/main/java/org/opensearch/ml/common/connector/Connector.java b/common/src/main/java/org/opensearch/ml/common/connector/Connector.java
@@ -110,6 +110,12 @@ static Connector createConnector(XContentBuilder builder, String connectorProtoc
             throw illegalArgumentException;
         }
     }
+
+
+    @Override
+    default Custom parse(XContentParser parser) throws IOException {
+        return createConnector(parser);
+    }
 
     @SuppressWarnings("removal")
     static Connector createConnector(XContentParser parser) throws IOException {

diff --git a/common/src/main/java/org/opensearch/sdk/Custom.java b/common/src/main/java/org/opensearch/sdk/Custom.java
@@ -11,9 +11,11 @@
 import org.opensearch.core.xcontent.ToXContentObject;
 import org.opensearch.core.xcontent.XContentParser;
 
+import java.io.IOException;
+
 /**
  * Custom metadata.
  */
 public interface Custom extends ToXContentObject {
-    Custom parse(XContentParser parser);
+    Custom parse(XContentParser parser) throws IOException;
 }
diff --git a/plugin/src/main/java/org/opensearch/ml/sdkclient/XContentClient.java b/plugin/src/main/java/org/opensearch/ml/sdkclient/XContentClient.java
@@ -79,7 +79,7 @@ public CompletionStage<GetCustomResponse> getCustom(GetCustomRequest request) {
                     .complete(
                         new GetCustomResponse.Builder().id(r.getId()).custom(request.clazz().cast(new Object()).parse(parser)).build()
                     );
-            } catch (Exception e) {
+            } catch (IOException e) {
                 // Parsing error
                 future.completeExceptionally(e);
             }