feat: add bedrock runtime agent for knowledge base (#2651)

* add bedrock runtime agent for knowledge base Signed-off-by: yuye-aws <[email protected]> * update unit test Signed-off-by: yuye-aws <[email protected]> --------- Signed-off-by: yuye-aws <[email protected]>
opensearch-project · Jul 18, 2024 · 226e53f · 226e53f
1 parent 6c8720e
commit 226e53f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/common/src/test/java/org/opensearch/ml/common/connector/ConnectorTest.java b/common/src/test/java/org/opensearch/ml/common/connector/ConnectorTest.java
@@ -65,14 +65,17 @@ public void validateConnectorURL_Invalid() {
         HttpConnector connector = createHttpConnector();
         connector.validateConnectorURL(Arrays.asList("^https://runtime\\.sagemaker\\..*[a-z0-9-]\\.amazonaws\\.com/.*$",
                 "^https://api\\.openai\\.com/.*$",
-                "^https://api\\.cohere\\.ai/.*$"));
+                "^https://api\\.cohere\\.ai/.*$",
+                "^https://bedrock-agent-runtime\\\\..*[a-z0-9-]\\\\.amazonaws\\\\.com/.*$"
+            ));
     }
 
     @Test
     public void validateConnectorURL() {
         HttpConnector connector = createHttpConnector();
         connector.validateConnectorURL(Arrays.asList("^https://runtime\\.sagemaker\\..*[a-z0-9-]\\.amazonaws\\.com/.*$",
                 "^https://api\\.openai\\.com/.*$",
+                "^https://bedrock-agent-runtime\\\\..*[a-z0-9-]\\\\.amazonaws\\\\.com/.*$",
                 "^" + connector.getActions().get(0).getUrl()));
     }
 }
diff --git a/plugin/src/main/java/org/opensearch/ml/settings/MLCommonsSettings.java b/plugin/src/main/java/org/opensearch/ml/settings/MLCommonsSettings.java
@@ -144,7 +144,8 @@ private MLCommonsSettings() {}
                     "^https://runtime\\.sagemaker\\..*[a-z0-9-]\\.amazonaws\\.com/.*$",
                     "^https://api\\.openai\\.com/.*$",
                     "^https://api\\.cohere\\.ai/.*$",
-                    "^https://bedrock-runtime\\..*[a-z0-9-]\\.amazonaws\\.com/.*$"
+                    "^https://bedrock-runtime\\..*[a-z0-9-]\\.amazonaws\\.com/.*$",
+                    "^https://bedrock-agent-runtime\\..*[a-z0-9-]\\.amazonaws\\.com/.*$"
                 ),
             Function.identity(),
             Setting.Property.NodeScope,