From 15aeb024907c5c3b58ba35a144aa5008baa5187d Mon Sep 17 00:00:00 2001 From: conggguan Date: Mon, 8 Jul 2024 10:57:42 +0800 Subject: [PATCH] [Fix] NeuralSparseTwoPhaseProcessorIT created wrong ingest pipeline, fix it to correct API. Signed-off-by: conggguan --- .../neuralsearch/bwc/NeuralSparseTwoPhaseProcessorIT.java | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/qa/restart-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/NeuralSparseTwoPhaseProcessorIT.java b/qa/restart-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/NeuralSparseTwoPhaseProcessorIT.java index a8a6667f9..08bf166e8 100644 --- a/qa/restart-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/NeuralSparseTwoPhaseProcessorIT.java +++ b/qa/restart-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/NeuralSparseTwoPhaseProcessorIT.java @@ -10,6 +10,7 @@ import java.nio.file.Files; import java.nio.file.Path; +import java.util.List; import static org.opensearch.neuralsearch.util.TestUtils.NODES_BWC_CLUSTER; import static org.opensearch.neuralsearch.util.TestUtils.TEXT_EMBEDDING_PROCESSOR; @@ -29,13 +30,13 @@ public void testNeuralSparseQueryTwoPhaseProcessor_NeuralSearch_E2EFlow() throws String modelId = uploadSparseEncodingModel(); loadModel(modelId); neuralSparseQueryBuilder.modelId(modelId); - createPipelineProcessor(modelId, NEURAL_SPARSE_INGEST_PIPELINE_NAME); + createPipelineForSparseEncodingProcessor(modelId, NEURAL_SPARSE_INGEST_PIPELINE_NAME); createIndexWithConfiguration( getIndexNameForTest(), Files.readString(Path.of(classLoader.getResource("processor/IndexMappingMultipleShard.json").toURI())), NEURAL_SPARSE_INGEST_PIPELINE_NAME ); - addDocument(getIndexNameForTest(), "0", TEST_TEXT_FIELD, TEXT_1, null, null); + addSparseEncodingDoc(getIndexNameForTest(), "0", List.of(), List.of(), List.of(TEST_TEXT_FIELD), List.of(TEXT_1)); createNeuralSparseTwoPhaseSearchProcessor(NEURAL_SPARSE_TWO_PHASE_SEARCH_PIPELINE_NAME); updateIndexSettings( getIndexNameForTest(),