rebase main

Signed-off-by: zane-neo <[email protected]>
br3no · Oct 21, 2024 · 179ce62 · 179ce62
1 parent 86d193f
commit 179ce62
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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);