Update IT to work with session tokens.

Signed-off-by: Austin Lee <[email protected]>
austintlee · Sep 3, 2024 · e6b21da · e6b21da
1 parent ad211f3
commit e6b21da
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRAGSearchProcessorIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRAGSearchProcessorIT.java
@@ -199,6 +199,9 @@ public class RestMLRAGSearchProcessorIT extends RestMLRemoteInferenceIT {
         + "\",\n"
         + "    \"secret_key\": \""
         + AWS_SECRET_ACCESS_KEY
+        + "\",\n"
+        + "    \"session_token\": \""
+        + AWS_SESSION_TOKEN
         + "\"\n"
         + "  },\n"
         + "  \"actions\": [\n"
@@ -239,6 +242,9 @@ public class RestMLRAGSearchProcessorIT extends RestMLRemoteInferenceIT {
         + "\",\n"
         + "    \"secret_key\": \""
         + AWS_SECRET_ACCESS_KEY
+        + "\",\n"
+        + "    \"session_token\": \""
+        + AWS_SESSION_TOKEN
         + "\"\n"
         + "  },\n"
         + "  \"actions\": [\n"