diff --git a/src/main/java/com/olympus/apollo/services/KSIngestor.java b/src/main/java/com/olympus/apollo/services/KSIngestor.java index e45dbe6..0db711a 100644 --- a/src/main/java/com/olympus/apollo/services/KSIngestor.java +++ b/src/main/java/com/olympus/apollo/services/KSIngestor.java @@ -225,6 +225,7 @@ public class KSIngestor { docs.forEach(doc -> { List splitDocs = splitter.split(doc); + int docIndex = 0; logger.info("Number of documents: " + splitDocs.size()); HashMap meta=(HashMap) ksTexts.getIngestionInfo().get("metadata"); @@ -234,10 +235,15 @@ public class KSIngestor { meta2.putAll(meta1); meta2.put("KsInternalMainEntityId",ksTexts.getId()); meta2.put("KsExternalDocUniqueID",KsExternalDocUniqueID); + + for (Document splitDoc : splitDocs) { + meta2.put("KsDocumentUIndex",docIndex); splitDoc.getMetadata().putAll(meta2); + docIndex++; } embedtexts(splitDocs); + }); //ksTexts.setIngestionStatus("INGESTED"); ksTexts.setIngestionDate(new Date());