MLE-26953 Chunks now capture model name

rjrudin · rjrudin · commit 5463b41b10eb · 2026-02-03T13:04:43.000-05:00
This is using embeddingModel.getModelName() in the LangChain4j API, and it will allow for Nuclia integration to easily add the model name found in each chunk response.
diff --git a/marklogic-spark-connector/src/main/java/com/marklogic/langchain4j/embedding/EmbeddingGenerator.java b/marklogic-spark-connector/src/main/java/com/marklogic/langchain4j/embedding/EmbeddingGenerator.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2023-2025 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
+ * Copyright (c) 2023-2026 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
  */
 package com.marklogic.langchain4j.embedding;
 
@@ -29,6 +29,7 @@ public class EmbeddingGenerator implements EmbeddingProducer {
     private final EmbeddingModel embeddingModel;
     private final int batchSize;
     private final String prompt;
+    private final String modelName;
 
     // Only used for debug logging.
     private static final AtomicLong tokenCount = new AtomicLong(0);
@@ -38,6 +39,7 @@ public EmbeddingGenerator(EmbeddingModel embeddingModel, int batchSize, String p
         this.embeddingModel = embeddingModel;
         this.batchSize = batchSize;
         this.prompt = prompt;
+        this.modelName = embeddingModel.modelName();
     }
 
     @Override
@@ -62,7 +64,7 @@ private int generateAndAddEmbeddings(List<TextSegment> segments, List<Chunk> chu
         List<Embedding> embeddings = generateEmbeddings(segments);
         for (int i = 0; i < embeddings.size(); i++) {
             Embedding embedding = embeddings.get(i);
-            chunks.get(chunkCounter).addEmbedding(embedding.vector());
+            chunks.get(chunkCounter).addEmbedding(embedding.vector(), modelName);
             chunkCounter++;
         }
         return chunkCounter;
diff --git a/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/ChunkInputs.java b/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/ChunkInputs.java
@@ -12,6 +12,7 @@ public class ChunkInputs {
     private final String text;
     private float[] embedding;
     private byte[] classification;
+    private String modelName;
 
     public ChunkInputs(String text) {
         this.text = text;
@@ -36,4 +37,12 @@ public byte[] getClassification() {
     public void setClassification(byte[] classification) {
         this.classification = classification;
     }
+
+    public String getModelName() {
+        return modelName;
+    }
+
+    public void setModelName(String modelName) {
+        this.modelName = modelName;
+    }
 }
diff --git a/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/DocumentInputs.java b/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/DocumentInputs.java
@@ -87,14 +87,15 @@ public void addChunkClassification(byte[] classification) {
         throw new IllegalStateException("Cannot add classification: all chunks already have classifications");
     }
 
-    public void addEmbedding(float[] embedding) {
+    public void addEmbedding(float[] embedding, String modelName) {
         if (chunkInputsList == null || chunkInputsList.isEmpty()) {
             throw new IllegalStateException("Cannot add embedding: no chunks exist");
         }
         // Find the next chunk without an embedding
         for (ChunkInputs chunk : chunkInputsList) {
             if (chunk.getEmbedding() == null) {
                 chunk.setEmbedding(embedding);
+                chunk.setModelName(modelName);
                 return;
             }
         }
diff --git a/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/DocumentPipeline.java b/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/DocumentPipeline.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2023-2025 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
+ * Copyright (c) 2023-2026 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
  */
 package com.marklogic.spark.core;
 
@@ -119,8 +119,8 @@ public String getEmbeddingText() {
         }
 
         @Override
-        public void addEmbedding(float[] embedding) {
-            inputs.addEmbedding(embedding);
+        public void addEmbedding(float[] embedding, String modelName) {
+            inputs.addEmbedding(embedding, modelName);
         }
     }
 
diff --git a/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/embedding/Chunk.java b/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/embedding/Chunk.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2023-2025 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
+ * Copyright (c) 2023-2026 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
  */
 package com.marklogic.spark.core.embedding;
 
@@ -17,6 +17,7 @@ public interface Chunk {
      * Add the vector data in the given embedding to the chunk.
      *
      * @param embedding
+     * @param modelName
      */
-    void addEmbedding(float[] embedding);
+    void addEmbedding(float[] embedding, String modelName);
 }
diff --git a/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/embedding/DOMChunk.java b/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/embedding/DOMChunk.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2023-2025 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
+ * Copyright (c) 2023-2026 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
  */
 package com.marklogic.spark.core.embedding;
 
@@ -56,7 +56,13 @@ public String getEmbeddingText() {
     }
 
     @Override
-    public void addEmbedding(float[] embedding) {
+    public void addEmbedding(float[] embedding, String modelName) {
+        if (modelName != null) {
+            final Element modelNameElement = document.createElementNS(xmlChunkConfig.getEmbeddingNamespace(), "model-name");
+            modelNameElement.setTextContent(modelName);
+            chunkElement.appendChild(modelNameElement);
+        }
+
         // DOM is fine with null as a value for the namespace.
         final Element embeddingElement = document.createElementNS(xmlChunkConfig.getEmbeddingNamespace(), xmlChunkConfig.getEmbeddingName());
 
diff --git a/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/embedding/JsonChunk.java b/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/embedding/JsonChunk.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2023-2025 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
+ * Copyright (c) 2023-2026 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
  */
 package com.marklogic.spark.core.embedding;
 
@@ -34,7 +34,11 @@ public String getEmbeddingText() {
     }
 
     @Override
-    public void addEmbedding(float[] embedding) {
+    public void addEmbedding(float[] embedding, String modelName) {
+        if (modelName != null) {
+            chunk.put("model-name", modelName);
+        }
+
         if (base64EncodeVectors) {
             String base64Vector = VectorUtil.base64Encode(embedding);
             chunk.put(this.embeddingArrayName, base64Vector);
diff --git a/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/splitter/JsonChunkDocumentProducer.java b/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/splitter/JsonChunkDocumentProducer.java
@@ -56,7 +56,7 @@ protected DocumentWriteOperation addChunksToSourceDocument() {
             }
             var jsonChunk = new JsonChunk(chunk, null, chunkConfig.getEmbeddingName(), chunkConfig.isBase64EncodeVectors());
             if (chunkInputs.getEmbedding() != null) {
-                jsonChunk.addEmbedding(chunkInputs.getEmbedding());
+                jsonChunk.addEmbedding(chunkInputs.getEmbedding(), chunkInputs.getModelName());
             }
             chunks.add(jsonChunk);
         }
@@ -93,7 +93,7 @@ protected DocumentWriteOperation makeChunkDocument() {
             }
             var jsonChunk = new JsonChunk(chunk, null, chunkConfig.getEmbeddingName(), chunkConfig.isBase64EncodeVectors());
             if (chunkInputs.getEmbedding() != null) {
-                jsonChunk.addEmbedding(chunkInputs.getEmbedding());
+                jsonChunk.addEmbedding(chunkInputs.getEmbedding(), chunkInputs.getModelName());
             }
             chunks.add(jsonChunk);
             listIndex++;
diff --git a/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/splitter/XmlChunkDocumentProducer.java b/marklogic-spark-connector/src/main/java/com/marklogic/spark/core/splitter/XmlChunkDocumentProducer.java
@@ -70,7 +70,7 @@ protected DocumentWriteOperation makeChunkDocument() {
             ChunkInputs chunkInputs = chunkInputsList.get(listIndex);
             Element classificationResponseNode = chunkInputs.getClassification() != null ?
                 getClassificationResponseElement(chunkInputs.getClassification()) : null;
-            addChunk(doc, chunkInputs.getText(), chunksElement, chunks, classificationResponseNode, chunkInputs.getEmbedding());
+            addChunk(doc, chunkInputs.getText(), chunksElement, chunks, classificationResponseNode, chunkInputs.getEmbedding(), chunkInputs.getModelName());
             listIndex++;
         }
 
@@ -91,7 +91,7 @@ protected DocumentWriteOperation addChunksToSourceDocument() {
         for (ChunkInputs chunkInputs : chunkInputsList) {
             Element classificationResponseNode = chunkInputs.getClassification() != null ?
                 getClassificationResponseElement(chunkInputs.getClassification()) : null;
-            addChunk(doc, chunkInputs.getText(), chunksElement, chunks, classificationResponseNode, chunkInputs.getEmbedding());
+            addChunk(doc, chunkInputs.getText(), chunksElement, chunks, classificationResponseNode, chunkInputs.getEmbedding(), chunkInputs.getModelName());
         }
 
         return new DocumentAndChunks(
@@ -110,7 +110,7 @@ private Element getClassificationResponseElement(byte[] classificationBytes) {
         }
     }
 
-    private void addChunk(Document doc, String textSegment, Element chunksElement, List<Chunk> chunks, Element classificationResponse, float[] embedding) {
+    private void addChunk(Document doc, String textSegment, Element chunksElement, List<Chunk> chunks, Element classificationResponse, float[] embedding, String modelName) {
         Element chunk = doc.createElementNS(chunkConfig.getXmlNamespace(), "chunk");
         chunksElement.appendChild(chunk);
 
@@ -129,7 +129,7 @@ private void addChunk(Document doc, String textSegment, Element chunksElement, L
 
         var domChunk = new DOMChunk(doc, chunk, this.xmlChunkConfig, this.xPathFactory);
         if (embedding != null) {
-            domChunk.addEmbedding(embedding);
+            domChunk.addEmbedding(embedding, modelName);
         }
         chunks.add(domChunk);
     }
diff --git a/marklogic-spark-connector/src/test/java/com/marklogic/spark/writer/embedding/AbstractEmbeddingTest.java b/marklogic-spark-connector/src/test/java/com/marklogic/spark/writer/embedding/AbstractEmbeddingTest.java
@@ -0,0 +1,20 @@
+/*
+ * Copyright (c) 2023-2026 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
+ */
+package com.marklogic.spark.writer.embedding;
+
+import com.marklogic.spark.AbstractIntegrationTest;
+import org.junit.jupiter.api.AfterEach;
+
+abstract class AbstractEmbeddingTest extends AbstractIntegrationTest {
+
+    static final String TEST_EMBEDDING_FUNCTION_CLASS = "com.marklogic.spark.writer.embedding.MinilmEmbeddingModelFunction";
+
+    // The minilm embedding model returns "unknown" as its model name.
+    static final String EXPECTED_MODEL_NAME = "unknown";
+
+    @AfterEach
+    void teardown() {
+        TestEmbeddingModel.reset();
+    }
+}
diff --git a/marklogic-spark-connector/src/test/java/com/marklogic/spark/writer/embedding/AddEmbeddingsFromTextTest.java b/marklogic-spark-connector/src/test/java/com/marklogic/spark/writer/embedding/AddEmbeddingsFromTextTest.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2023-2025 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
+ * Copyright (c) 2023-2026 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
  */
 package com.marklogic.spark.writer.embedding;
 
@@ -27,9 +27,7 @@
  * need to specify the location of the chunks. The connector is expected to determine the location based on whether the
  * sidecar docs are JSON or XML.
  */
-class AddEmbeddingsFromTextTest extends AbstractIntegrationTest {
-
-    private static final String TEST_EMBEDDING_FUNCTION_CLASS = "com.marklogic.spark.writer.embedding.MinilmEmbeddingModelFunction";
+class AddEmbeddingsFromTextTest extends AbstractEmbeddingTest {
 
     @Test
     void jsonSidecarDocuments() {
@@ -43,9 +41,11 @@ void jsonSidecarDocuments() {
             assertTrue(uri.endsWith(".json"));
             JsonNode doc = readJsonDocument(uri);
             assertEquals(JsonNodeType.ARRAY, doc.get("chunks").getNodeType());
-            doc.get("chunks").forEach(chunkEmbedding -> {
-                String textValue = chunkEmbedding.get("text").asText();
-                String embeddingValue = chunkEmbedding.get("_vector").toPrettyString();
+            doc.get("chunks").forEach(chunk -> {
+                assertEquals(EXPECTED_MODEL_NAME, chunk.get("model-name").asText(), "Model name should be stored as of 3.1.0");
+
+                String textValue = chunk.get("text").asText();
+                String embeddingValue = chunk.get("_vector").toPrettyString();
                 if (!chunkEmbeddings.containsKey(textValue)) {
                     chunkEmbeddings.put(textValue, embeddingValue);
                 }
@@ -76,6 +76,9 @@ void xmlSidecarDocuments() {
             if (!chunkEmbeddings.containsKey(textValue)) {
                 chunkEmbeddings.put(textValue, embeddingValue);
             }
+
+            assertEquals(EXPECTED_MODEL_NAME, doc.getElementValue("/node()/model:chunks/model:chunk/vec:model-name"),
+                "Model name should be stored as of 3.1.0");
         }
 
         // collapse embedding values in map to set. Any duplicates will be ignored. If there's a duplicate embedding
@@ -100,6 +103,4 @@ private DataFrameWriter<Row> prepareToWriteChunks() {
             .option(Options.WRITE_SPLITTER_SIDECAR_COLLECTIONS, "text-chunks")
             .option(Options.WRITE_EMBEDDER_MODEL_FUNCTION_CLASS_NAME, TEST_EMBEDDING_FUNCTION_CLASS);
     }
-
-
 }
diff --git a/marklogic-spark-connector/src/test/java/com/marklogic/spark/writer/embedding/AddEmbeddingsToJsonTest.java b/marklogic-spark-connector/src/test/java/com/marklogic/spark/writer/embedding/AddEmbeddingsToJsonTest.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2023-2025 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
+ * Copyright (c) 2023-2026 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.
  */
 package com.marklogic.spark.writer.embedding;
 
@@ -11,7 +11,6 @@
 import com.marklogic.client.row.RowRecord;
 import com.marklogic.client.row.RowSet;
 import com.marklogic.junit5.RequiresMarkLogic12;
-import com.marklogic.spark.AbstractIntegrationTest;
 import com.marklogic.spark.ConnectorException;
 import com.marklogic.spark.Options;
 import org.apache.spark.sql.DataFrameWriter;
@@ -24,14 +23,7 @@
 
 import static org.junit.jupiter.api.Assertions.*;
 
-class AddEmbeddingsToJsonTest extends AbstractIntegrationTest {
-
-    private static final String TEST_EMBEDDING_FUNCTION_CLASS = "com.marklogic.spark.writer.embedding.MinilmEmbeddingModelFunction";
-
-    @AfterEach
-    void teardown() {
-        TestEmbeddingModel.reset();
-    }
+class AddEmbeddingsToJsonTest extends AbstractEmbeddingTest {
 
     /**
      * Tests the use case where a user wants to split the text into chunks and generate embeddings for each chunk, all
@@ -223,6 +215,8 @@ void jsonSidecarWithVersion2XDefaultsForEmbeddings() {
             assertTrue(firstChunk.has("text"));
             assertTrue(firstChunk.has("embedding"), "Verifies that a user can override the default of '_vector' in the 3.0 " +
                 "release to be the default value of 'embedding' from the 2.x releases. Actual chunk doc: " + firstChunk.toPrettyString());
+
+            assertEquals(EXPECTED_MODEL_NAME, firstChunk.get("model-name").asText(), "Model name should be stored as of 3.1.0");
         }
     }
 
@@ -242,6 +236,7 @@ void chunksIsAnObjectInsteadOfAnArray() {
         JsonNode doc = readJsonDocument("/custom-path-test.json");
         assertTrue(doc.has("_vector"));
         assertEquals(JsonNodeType.ARRAY, doc.get("_vector").getNodeType());
+        assertEquals(EXPECTED_MODEL_NAME, doc.get("model-name").asText(), "Model name should be stored as of 3.1.0");
     }
 
     @Test
@@ -267,6 +262,7 @@ void testBatchSize() {
         for (int i = 0; i < chunks.size(); i++) {
             JsonNode chunk = chunks.get(i);
             assertTrue(chunk.has("_vector"), "No embedding found in chunk " + i);
+            assertEquals(EXPECTED_MODEL_NAME, chunk.get("model-name").asText(), "Model name should be stored as of 3.1.0");
         }
 
         assertEquals(3, TestEmbeddingModel.batchCounter, "Expecting 3 batches to be sent to the test " +
@@ -371,6 +367,8 @@ void arbitraryRowWithNoInitialUri() {
                 "that URI caused the construction of a DocumentWriteOperationImpl to fail. The fix - providing a " +
                 "temporary initial URI - avoids this bug, with the real URI being set after the embedding " +
                 "generation process.");
+
+        assertEquals(EXPECTED_MODEL_NAME, doc.get("model-name").asText(), "Model name should be stored as of 3.1.0");
     }
 
     @Test
@@ -463,6 +461,7 @@ private void verifyEachChunkOnDocumentHasAnEmbedding(String uri) {
             assertTrue(node.has("text"));
             assertTrue(node.has("_vector"));
             assertEquals(JsonNodeType.ARRAY, node.get("_vector").getNodeType());
+            assertEquals(EXPECTED_MODEL_NAME, node.get("model-name").asText(), "Model name should be stored as of 3.1.0");
         });
     }
 
@@ -501,6 +500,8 @@ private void verifyDocumentHasTwoChunksWithEncodedVectors(JsonNode doc) {
             assertFalse(chunk.has("lang"), "Due to MLE-22918, the 'lang' field is not set to 'zxx' since this " +
                 "will disable stemming on data outside the intended scope of the 'lang' field. A user is free to " +
                 "e.g. use a REST transform to add this if desired.");
+
+            assertEquals(EXPECTED_MODEL_NAME, chunk.get("model-name").asText(), "Model name should be stored as of 3.1.0");
         }
     }
 }
diff --git a/marklogic-spark-connector/src/test/java/com/marklogic/spark/writer/embedding/AddEmbeddingsToXmlTest.java b/marklogic-spark-connector/src/test/java/com/marklogic/spark/writer/embedding/AddEmbeddingsToXmlTest.java
diff --git a/marklogic-spark-connector/src/test/java/com/marklogic/spark/writer/embedding/TestEmbeddingModel.java b/marklogic-spark-connector/src/test/java/com/marklogic/spark/writer/embedding/TestEmbeddingModel.java

Original file line number	Diff line number	Diff line change
`@@ -87,14 +87,15 @@ public void addChunkClassification(byte[] classification) {`
`87`	`87`	`throw new IllegalStateException("Cannot add classification: all chunks already have classifications");`
`88`	`88`	`}`
`89`	`89`
`90`		`- public void addEmbedding(float[] embedding) {`
	`90`	`+ public void addEmbedding(float[] embedding, String modelName) {`
`91`	`91`	`if (chunkInputsList == null \|\| chunkInputsList.isEmpty()) {`
`92`	`92`	`throw new IllegalStateException("Cannot add embedding: no chunks exist");`
`93`	`93`	`}`
`94`	`94`	`// Find the next chunk without an embedding`
`95`	`95`	`for (ChunkInputs chunk : chunkInputsList) {`
`96`	`96`	`if (chunk.getEmbedding() == null) {`
`97`	`97`	`chunk.setEmbedding(embedding);`
	`98`	`+ chunk.setModelName(modelName);`
`98`	`99`	`return;`
`99`	`100`	`}`
`100`	`101`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,5 @@`
`1`	`1`	`/*`
`2`		`- * Copyright (c) 2023-2025 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.`
	`2`	`+ * Copyright (c) 2023-2026 Progress Software Corporation and/or its subsidiaries or affiliates. All Rights Reserved.`
`3`	`3`	`*/`
`4`	`4`	`package com.marklogic.spark.core;`
`5`	`5`
`@@ -119,8 +119,8 @@ public String getEmbeddingText() {`
`119`	`119`	`}`
`120`	`120`
`121`	`121`	`@Override`
`122`		`- public void addEmbedding(float[] embedding) {`
`123`		`- inputs.addEmbedding(embedding);`
	`122`	`+ public void addEmbedding(float[] embedding, String modelName) {`
	`123`	`+ inputs.addEmbedding(embedding, modelName);`
`124`	`124`	`}`
`125`	`125`	`}`
`126`	`126`
Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ protected DocumentWriteOperation addChunksToSourceDocument() {`
`56`	`56`	`}`
`57`	`57`	`var jsonChunk = new JsonChunk(chunk, null, chunkConfig.getEmbeddingName(), chunkConfig.isBase64EncodeVectors());`
`58`	`58`	`if (chunkInputs.getEmbedding() != null) {`
`59`		`- jsonChunk.addEmbedding(chunkInputs.getEmbedding());`
	`59`	`+ jsonChunk.addEmbedding(chunkInputs.getEmbedding(), chunkInputs.getModelName());`
`60`	`60`	`}`
`61`	`61`	`chunks.add(jsonChunk);`
`62`	`62`	`}`
`@@ -93,7 +93,7 @@ protected DocumentWriteOperation makeChunkDocument() {`
`93`	`93`	`}`
`94`	`94`	`var jsonChunk = new JsonChunk(chunk, null, chunkConfig.getEmbeddingName(), chunkConfig.isBase64EncodeVectors());`
`95`	`95`	`if (chunkInputs.getEmbedding() != null) {`
`96`		`- jsonChunk.addEmbedding(chunkInputs.getEmbedding());`
	`96`	`+ jsonChunk.addEmbedding(chunkInputs.getEmbedding(), chunkInputs.getModelName());`
`97`	`97`	`}`
`98`	`98`	`chunks.add(jsonChunk);`
`99`	`99`	`listIndex++;`
Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ protected DocumentWriteOperation makeChunkDocument() {`
`70`	`70`	`ChunkInputs chunkInputs = chunkInputsList.get(listIndex);`
`71`	`71`	`Element classificationResponseNode = chunkInputs.getClassification() != null ?`
`72`	`72`	`getClassificationResponseElement(chunkInputs.getClassification()) : null;`
`73`		`- addChunk(doc, chunkInputs.getText(), chunksElement, chunks, classificationResponseNode, chunkInputs.getEmbedding());`
	`73`	`+ addChunk(doc, chunkInputs.getText(), chunksElement, chunks, classificationResponseNode, chunkInputs.getEmbedding(), chunkInputs.getModelName());`
`74`	`74`	`listIndex++;`
`75`	`75`	`}`
`76`	`76`
`@@ -91,7 +91,7 @@ protected DocumentWriteOperation addChunksToSourceDocument() {`
`91`	`91`	`for (ChunkInputs chunkInputs : chunkInputsList) {`
`92`	`92`	`Element classificationResponseNode = chunkInputs.getClassification() != null ?`
`93`	`93`	`getClassificationResponseElement(chunkInputs.getClassification()) : null;`
`94`		`- addChunk(doc, chunkInputs.getText(), chunksElement, chunks, classificationResponseNode, chunkInputs.getEmbedding());`
	`94`	`+ addChunk(doc, chunkInputs.getText(), chunksElement, chunks, classificationResponseNode, chunkInputs.getEmbedding(), chunkInputs.getModelName());`
`95`	`95`	`}`
`96`	`96`
`97`	`97`	`return new DocumentAndChunks(`
`@@ -110,7 +110,7 @@ private Element getClassificationResponseElement(byte[] classificationBytes) {`
`110`	`110`	`}`
`111`	`111`	`}`
`112`	`112`
`113`		`- private void addChunk(Document doc, String textSegment, Element chunksElement, List<Chunk> chunks, Element classificationResponse, float[] embedding) {`
	`113`	`+ private void addChunk(Document doc, String textSegment, Element chunksElement, List<Chunk> chunks, Element classificationResponse, float[] embedding, String modelName) {`
`114`	`114`	`Element chunk = doc.createElementNS(chunkConfig.getXmlNamespace(), "chunk");`
`115`	`115`	`chunksElement.appendChild(chunk);`
`116`	`116`
`@@ -129,7 +129,7 @@ private void addChunk(Document doc, String textSegment, Element chunksElement, L`
`129`	`129`
`130`	`130`	`var domChunk = new DOMChunk(doc, chunk, this.xmlChunkConfig, this.xPathFactory);`
`131`	`131`	`if (embedding != null) {`
`132`		`- domChunk.addEmbedding(embedding);`
	`132`	`+ domChunk.addEmbedding(embedding, modelName);`
`133`	`133`	`}`
`134`	`134`	`chunks.add(domChunk);`
`135`	`135`	`}`