Change url regex

Signed-off-by: zane-neo <[email protected]>
zane-neo · Feb 20, 2024 · 0641bc4 · 0641bc4
1 parent 689c2cb
commit 0641bc4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/ConnectorUtils.java b/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/ConnectorUtils.java
@@ -268,12 +268,12 @@ public static SdkHttpFullRequest buildSdkRequest(
         SdkHttpMethod method
     ) throws UnknownHostException {
         String endpoint = connector.getPredictEndpoint(parameters);
-        Pattern pattern = Pattern.compile("(?:(\\w+)://)?((\\w+\\.)*\\w+)(?::(\\w+))?");
+        Pattern pattern = Pattern.compile("(?:(\\w+)://)?([-a-zA-Z0-9+&@#%?=~_|!,.;]*)(?::(\\w+))?");
         Matcher matcher = pattern.matcher(endpoint);
         if (matcher.find()) {
             String protocol = matcher.group(1);
             String host = matcher.group(2);
-            String port = matcher.group(4);
+            String port = matcher.group(3);
             MLHttpClientFactory.validate(protocol, host, port);
         } else {
             throw new IllegalArgumentException("Invalid endpoint: " + endpoint);