From 179ce62d0ee9ff0afa0bfa63fc0ecfa72662d70b Mon Sep 17 00:00:00 2001 From: zane-neo Date: Mon, 21 Oct 2024 11:04:30 +0800 Subject: [PATCH] rebase main Signed-off-by: zane-neo --- .../neuralsearch/processor/TextEmbeddingProcessorIT.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/test/java/org/opensearch/neuralsearch/processor/TextEmbeddingProcessorIT.java b/src/test/java/org/opensearch/neuralsearch/processor/TextEmbeddingProcessorIT.java index 25f41e5dd..0e1cb7556 100644 --- a/src/test/java/org/opensearch/neuralsearch/processor/TextEmbeddingProcessorIT.java +++ b/src/test/java/org/opensearch/neuralsearch/processor/TextEmbeddingProcessorIT.java @@ -245,6 +245,11 @@ private String uploadTextEmbeddingModel(String requestBody) throws Exception { return registerModelGroupAndUploadModel(requestBody); } + private String uploadTextEmbeddingModel() throws Exception { + String requestBody = Files.readString(Path.of(classLoader.getResource("processor/UploadModelRequestBody.json").toURI())); + return registerModelGroupAndUploadModel(requestBody); + } + private void createTextEmbeddingIndex() throws Exception { createIndexWithConfiguration( INDEX_NAME, @@ -262,7 +267,7 @@ public void testAsymmetricTextEmbeddingProcessor() throws Exception { loadModel(modelId); createPipelineProcessor(modelId, PIPELINE_NAME, ProcessorType.TEXT_EMBEDDING); createTextEmbeddingIndex(); - ingestDocument(); + ingestDocument(INGEST_DOC1, null); assertEquals(1, getDocCount(INDEX_NAME)); } finally { wipeOfTestResources(INDEX_NAME, PIPELINE_NAME, modelId, null);