task id is now string

Author: erhant

Cargo.lock

@@ -7,7 +7,7 @@ default-members = ["compute"]
 
 [workspace.package]
 edition = "2021"
-version = "0.5.6"
+version = "0.5.7"
 license = "Apache-2.0"
 readme = "README.md"
 

Cargo.toml

@@ -32,9 +32,8 @@ impl TaskResponder {
             Err(err) => {
                 let err_string = format!("{:#}", err);
                 log::error!(
-                    "Task {}/{} failed due to parsing error: {}",
-                    task.file_id,
-                    task.task_id,
+                    "Task {} failed due to parsing error: {}",
+                    task.row_id,
                     err_string
                 );
 
@@ -103,10 +102,9 @@ impl TaskResponder {
             Ok(result) => {
                 // prepare signed and encrypted payload
                 log::info!(
-                    "Publishing {} result for {}/{}",
+                    "Publishing {} result for {}",
                     "task".yellow(),
-                    task_metadata.file_id,
-                    task_metadata.task_id
+                    task_output.row_id
                 );
 
                 // TODO: will get better token count from `TaskWorkerOutput`
@@ -131,12 +129,7 @@ impl TaskResponder {
             Err(err) => {
                 // use pretty display string for error logging with causes
                 let err_string = format!("{:#}", err);
-                log::error!(
-                    "Task {}/{} failed: {}",
-                    task_metadata.file_id,
-                    task_metadata.task_id,
-                    err_string
-                );
+                log::error!("Task {} failed: {}", task_output.row_id, err_string);
 
                 // prepare error payload
                 let error_payload = TaskResponsePayload {

compute/src/reqres/task.rs

@@ -10,7 +10,7 @@ use uuid::Uuid;
 /// This is put into a map before execution, and then removed after the task is done.
 pub struct TaskWorkerMetadata {
     pub model_name: String,
-    pub task_id: Uuid,
+    pub task_id: String,
     pub file_id: Uuid,
     /// If for any reason this object is dropped before `channel` is responded to,
     /// the task will be lost and the channel will be abruptly closed, causing an error on

compute/src/workers/task.rs

@@ -12,15 +12,18 @@ pub const TASK_RESULT_TOPIC: &str = "results";
 /// `result` and `error` are mutually-exclusive, only one of them can be `Some`:
 /// - if `result` is `Some`, then it contains the result.
 /// - if `error` is `Some`, then it contains the error message.
+///
+/// Each task belongs to a file (uniquely identified by `file_id`), and has a unique identifier (`row_id`).
+/// THe `task_id` is a custom identifier given by a user.
 #[derive(Debug, Clone, Serialize, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub struct TaskResponsePayload {
-    /// The unique identifier of the task.
-    pub row_id: Uuid,
-    /// The custom identifier of the task.
-    pub task_id: Uuid,
     /// The file that this task is associated with.
     pub file_id: Uuid,
+    /// The unique identifier of the task.
+    pub row_id: Uuid,
+    /// The custom identifier of the task, not necessarily unique.
+    pub task_id: String,
     /// Name of the model used for this task.
     pub model: String,
     /// Stats about the task execution.
@@ -38,15 +41,18 @@ pub struct TaskResponsePayload {
 }
 
 /// A generic task request, given by Dria.
+///
+/// Each task belongs to a file (uniquely identified by `file_id`), and has a unique identifier (`row_id`).
+/// THe `task_id` is a custom identifier given by a user.
 #[derive(Debug, Clone, Serialize, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub struct TaskRequestPayload<T> {
-    /// The unique identifier of the task.
-    pub row_id: Uuid,
-    /// The unique identifier of the task.
-    pub task_id: Uuid,
     /// The file that this task is associated with.
     pub file_id: Uuid,
+    /// The unique identifier of the task.
+    pub row_id: Uuid,
+    /// The custom identifier of the task, not necessarily unique.
+    pub task_id: String,
     /// The input to the compute function.
     pub input: T,
 }