Added efficient saving of document chunks

knmlprz · May 6, 2024 · 6d3c086 · 6d3c086
1 parent 8b86878
commit 6d3c086
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/api/documents/controllers.py b/api/documents/controllers.py
@@ -16,11 +16,9 @@ def create_document_controller(payload: DocumentIn) -> tuple[HTTPStatus, Documen
         document.full_clean()
         document.save()
 
-        chunks = split_document_into_chunks(document, 100)
-        for chunk_data in chunks:
-            chunk = Chunk(**chunk_data)
-            chunk.full_clean()
-            chunk.save()
+        chunks = split_document_into_chunks(document, 100) 
+        chunk_instances = [Chunk(**chunk_data) for chunk_data in chunks]
+        Chunk.objects.bulk_create(chunk_instances)
 
     return HTTPStatus.CREATED, document