diff --git a/src/main/java/com/olympus/apollo/controllers/KnowledgeTreeController.java b/src/main/java/com/olympus/apollo/controllers/KnowledgeTreeController.java
index 0d34e61..7343b8f 100644
--- a/src/main/java/com/olympus/apollo/controllers/KnowledgeTreeController.java
+++ b/src/main/java/com/olympus/apollo/controllers/KnowledgeTreeController.java
@@ -160,6 +160,7 @@ public class KnowledgeTreeController {
             ksIngestionInfo.setMinChunkSizeToEmbed(fileUploadDTO.getMinChunkSizeToEmbed());
 
             ksDocument.setIngestionInfo(ksIngestionInfo);
+            ksDocument.setApolloVersion("2");
             ksDocumentRepository.save(ksDocument);
 
             logger.info("[V2-UPLOAD] Document saved successfully with ID: {}", ksDocument.getId());
@@ -257,6 +258,8 @@ public class KnowledgeTreeController {
             ksVideoIngestionInfo.setLanguage(videoUploadDTO.getLanguage());
 
             ksVideo.setIngestionInfo(ksVideoIngestionInfo);
+            ksVideo.setCallDocumentIngestionApi(true);
+            ksVideo.setApolloVersion("2");
             ksVideoRepository.save(ksVideo);
 
             logger.info("[V2-VIDEO-UPLOAD] Video saved successfully with ID: {}", ksVideo.getId());