84: Change field name of numerical back to progress for numerical progress in TaskUpdateProgress messages and send START instead of 0.0 progress on worker start.

Author: lfse-slafleur

src/omotes_sdk/internal/orchestrator_worker_events/messages/task_pb2.py

@@ -80,13 +80,13 @@ class TaskProgressUpdate(google.protobuf.message.Message):
     CELERY_TASK_ID_FIELD_NUMBER: builtins.int
     CELERY_TASK_TYPE_FIELD_NUMBER: builtins.int
     STATUS_FIELD_NUMBER: builtins.int
-    NUMERICAL_FIELD_NUMBER: builtins.int
+    PROGRESS_FIELD_NUMBER: builtins.int
     MESSAGE_FIELD_NUMBER: builtins.int
     job_id: builtins.str
     celery_task_id: builtins.str
     celery_task_type: builtins.str
     status: global___TaskProgressUpdate.ProgressStatus.ValueType
-    numerical: builtins.float
+    progress: builtins.float
     message: builtins.str
     def __init__(
         self,
@@ -95,11 +95,11 @@ class TaskProgressUpdate(google.protobuf.message.Message):
         celery_task_id: builtins.str = ...,
         celery_task_type: builtins.str = ...,
         status: global___TaskProgressUpdate.ProgressStatus.ValueType = ...,
-        numerical: builtins.float = ...,
+        progress: builtins.float = ...,
         message: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["numerical", b"numerical", "progress", b"progress", "status", b"status"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["celery_task_id", b"celery_task_id", "celery_task_type", b"celery_task_type", "job_id", b"job_id", "message", b"message", "numerical", b"numerical", "progress", b"progress", "status", b"status"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["progress", b"progress"]) -> typing_extensions.Literal["status", "numerical"] | None: ...
+    def HasField(self, field_name: typing_extensions.Literal["progress", b"progress", "progress_type", b"progress_type", "status", b"status"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["celery_task_id", b"celery_task_id", "celery_task_type", b"celery_task_type", "job_id", b"job_id", "message", b"message", "progress", b"progress", "progress_type", b"progress_type", "status", b"status"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["progress_type", b"progress_type"]) -> typing_extensions.Literal["status", "progress"] | None: ...
 
 global___TaskProgressUpdate = TaskProgressUpdate

src/omotes_sdk/internal/orchestrator_worker_events/messages/task_pb2.pyi

@@ -45,6 +45,25 @@ def __init__(self, job_id: UUID, task: CeleryTask, broker_if: BrokerInterface):
         self.task = task
         self.broker_if = broker_if
 
+    def send_start(self) -> None:
+        """Send a START progress update to the orchestrator."""
+        logger.debug(
+            "Sending START progress update for job %s (celery id %s)",
+            self.job_id,
+            self.task.request.id,
+        )
+        self.broker_if.send_message_to(
+            None,
+            WORKER.config.task_progress_queue_name,
+            TaskProgressUpdate(
+                job_id=str(self.job_id),
+                celery_task_id=self.task.request.id,
+                celery_task_type=WORKER_TASK_TYPE,
+                status=TaskProgressUpdate.START,
+                message="Started job at worker.",
+            ).SerializeToString(),
+        )
+
     def update_progress(self, fraction: float, message: str) -> None:
         """Send a progress update to the orchestrator.
 
@@ -243,7 +262,7 @@ def wrapped_worker_task(
     """
     logger.info("Worker started new task %s with reference %s", job_id, job_reference)
     task_util = TaskUtil(job_id, task, task.broker_if)
-    task_util.update_progress(0, "Job calculation started")
+    task_util.send_start()
     output_esdl = WORKER_TASK_FUNCTION(input_esdl, params_dict, task_util.update_progress)
 
     input_esh = pyesdl_from_string(input_esdl)

src/omotes_sdk/internal/worker/worker.py

@@ -18,9 +18,9 @@ message TaskProgressUpdate {
   string job_id = 1;
   string celery_task_id = 2;
   string celery_task_type = 5;
-  oneof progress {
+  oneof progress_type {
     ProgressStatus status = 6;
-    double numerical = 7;
+    double progress = 7;
   }
   string message = 4;