Skip to content

Commit 38963a9

Browse files
Add back RAG IT tests that use Converse API.
Signed-off-by: Austin Lee <austin@aryn.ai>
1 parent 3c118e5 commit 38963a9

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

plugin/src/test/java/org/opensearch/ml/rest/RestMLRAGSearchProcessorIT.java

+1-6
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
import org.apache.hc.core5.http.io.entity.EntityUtils;
3535
import org.apache.hc.core5.http.message.BasicHeader;
3636
import org.junit.Before;
37-
import org.junit.Ignore;
37+
// import org.junit.Ignore;
3838
import org.opensearch.client.Response;
3939
import org.opensearch.core.rest.RestStatus;
4040
import org.opensearch.ml.common.MLTaskState;
@@ -582,7 +582,6 @@ public void testBM25WithOpenAI() throws Exception {
582582
assertNotNull(answer);
583583
}
584584

585-
@Ignore
586585
public void testBM25WithOpenAIWithImage() throws Exception {
587586
// Skip test if key is null
588587
if (OPENAI_KEY == null) {
@@ -722,7 +721,6 @@ public void testBM25WithBedrock() throws Exception {
722721
assertNotNull(answer);
723722
}
724723

725-
@Ignore
726724
public void testBM25WithBedrockConverse() throws Exception {
727725
// Skip test if key is null
728726
if (AWS_ACCESS_KEY_ID == null) {
@@ -775,7 +773,6 @@ public void testBM25WithBedrockConverse() throws Exception {
775773
assertNotNull(answer);
776774
}
777775

778-
@Ignore
779776
public void testBM25WithBedrockConverseUsingLlmMessages() throws Exception {
780777
// Skip test if key is null
781778
if (AWS_ACCESS_KEY_ID == null) {
@@ -836,7 +833,6 @@ public void testBM25WithBedrockConverseUsingLlmMessages() throws Exception {
836833
assertNotNull(answer);
837834
}
838835

839-
@Ignore
840836
public void testBM25WithBedrockConverseUsingLlmMessagesForDocumentChat() throws Exception {
841837
// Skip test if key is null
842838
if (AWS_ACCESS_KEY_ID == null) {
@@ -953,7 +949,6 @@ public void testBM25WithOpenAIWithConversation() throws Exception {
953949
assertNotNull(interactionId);
954950
}
955951

956-
@Ignore
957952
public void testBM25WithOpenAIWithConversationAndImage() throws Exception {
958953
// Skip test if key is null
959954
if (OPENAI_KEY == null) {

0 commit comments

Comments
 (0)