add batch predict job actiontype in connector

Zhangxunmt · Zhangxunmt · commit 95d39886da9b · 2024-06-05T14:36:34.000-07:00
Signed-off-by: Xun Zhang &lt;xunzh@amazon.com&gt;
diff --git a/common/src/main/java/org/opensearch/ml/common/connector/ConnectorAction.java b/common/src/main/java/org/opensearch/ml/common/connector/ConnectorAction.java
@@ -182,6 +182,7 @@ public static ConnectorAction parse(XContentParser parser) throws IOException {
     }
 
     public enum ActionType {
-        PREDICT
+        PREDICT,
+        BATCH_PREDICT_JOB
     }
 }
diff --git a/plugin/src/main/java/org/opensearch/ml/settings/MLCommonsSettings.java b/plugin/src/main/java/org/opensearch/ml/settings/MLCommonsSettings.java
@@ -132,6 +132,7 @@ private MLCommonsSettings() {}
             ImmutableList
                 .of(
                     "^https://runtime\\.sagemaker\\..*[a-z0-9-]\\.amazonaws\\.com/.*$",
+                    "^https://api\\.sagemaker\\..*[a-z0-9-]\\.amazonaws\\.com/.*$",
                     "^https://api\\.openai\\.com/.*$",
                     "^https://api\\.cohere\\.ai/.*$",
                     "^https://bedrock-runtime\\..*[a-z0-9-]\\.amazonaws\\.com/.*$"

Original file line number	Diff line number	Diff line change
`@@ -182,6 +182,7 @@ public static ConnectorAction parse(XContentParser parser) throws IOException {`
`182`	`182`	`}`
`183`	`183`
`184`	`184`	`public enum ActionType {`
`185`		`- PREDICT`
	`185`	`+ PREDICT,`
	`186`	`+ BATCH_PREDICT_JOB`
`186`	`187`	`}`
`187`	`188`	`}`