Improve endpoints feedback

huridocs · Jul 1, 2024 · d4cfb41 · d4cfb41
1 parent f8749eb
commit d4cfb41
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/app.py b/app.py
@@ -46,13 +46,13 @@ async def get_xml():
 
 @app.post("/xml_to_train/{tenant}/{extractor_id}")
 async def to_train_xml_file(tenant, extractor_id, file: UploadFile = File(...)):
-    print(to_train_xml_file, tenant, extractor_id)
+    print("received file to train", tenant, extractor_id)
     return "xml_to_train saved"
 
 
 @app.post("/xml_to_predict/{tenant}/{extractor_id}")
 async def to_predict_xml_file(tenant, extractor_id, file: UploadFile = File(...)):
-    print(to_predict_xml_file, tenant, extractor_id)
+    print("received file to predict", tenant, extractor_id)
     return "xml_to_train saved"
 
 

diff --git a/worker_translations.py b/worker_translations.py
@@ -25,15 +25,15 @@ def __init__(self):
 
     def process(self, id, message, rc, ts):
         task = TranslationTaskMessage(**message)
-        print(task.dict())
+        print(task.model_dump())
 
         translations: list[Translation] = [self.get_translation(task, language) for language in task.languages_to]
         response = TranslationResponseMessage(
-            **task.dict(),
+            **task.model_dump(),
             translations=translations,
         )
 
-        self.results_queue.sendMessage(delay=5).message(response.dict()).execute()
+        self.results_queue.sendMessage(delay=5).message(response.model_dump()).execute()
         return True
 
     def subscribe_to_tasks_queue(self):