Batch import wasn't updated

app/services/impl/training/training.py

@@ -1,4 +1,5 @@
 import re
+import uuid
 from datetime import datetime
 from functools import reduce
 from logging import getLogger
@@ -23,9 +24,9 @@ class TrainingService(ITrainingService):
     ]
     # strategy word_link ct_focus reading_skill word_partners writing_skill language_for_writing
 
-    def __init__(self, llm: ILLMService, firestore: IDocumentStore, training_kb: IKnowledgeBase):
+    def __init__(self, llm: ILLMService, document_store: IDocumentStore, training_kb: IKnowledgeBase):
         self._llm = llm
-        self._db = firestore
+        self._db = document_store
         self._kb = training_kb
         self._logger = getLogger(__name__)
 
@@ -96,16 +97,15 @@ class TrainingService(ITrainingService):
         for area in training_content.weak_areas:
             weak_areas["weak_areas"].append(area.dict())
 
-        new_id = str(uuid.uuid4())
         training_doc = {
-            'id': new_id,
             'created_at': int(datetime.now().timestamp() * 1000),
             **exam_map,
             **usefull_tips.dict(),
             **weak_areas,
             "user": user
         }
-        doc_id = await self._db.save_to_db('training', training_doc)
+        new_id = await self._db.save_to_db('training', training_doc)
+
         return {
             "id": new_id
         }