Fix validation

adap · Mar 28, 2024 · 8057db1 · 8057db1
1 parent da1587e
commit 8057db1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/src/cc/flwr/src/communicator.cc b/src/cc/flwr/src/communicator.cc
@@ -26,7 +26,7 @@ bool validate_task_ins(const flwr::proto::TaskIns &task_ins,
 
 bool validate_task_res(const flwr::proto::TaskRes &task_res) {
   // Retrieve initialized fields in TaskRes
-  return (!task_res.task_id().empty());
+  return true;
 }
 
 flwr::proto::TaskRes

diff --git a/src/cc/flwr/src/message_handler.cc b/src/cc/flwr/src/message_handler.cc
@@ -73,6 +73,10 @@ handle_task(flwr_local::Client *client, const flwr::proto::TaskIns &task_ins) {
 
   flwr::proto::TaskRes task_res;
 
+  task_res.set_task_id("");
+  task_res.set_group_id(task_ins.group_id());
+  task_res.set_run_id(task_ins.run_id());
+
   std::unique_ptr<flwr::proto::Task> task =
       std::make_unique<flwr::proto::Task>();
 
@@ -82,10 +86,12 @@ handle_task(flwr_local::Client *client, const flwr::proto::TaskIns &task_ins) {
 
   task->set_allocated_recordset(proto_recordset_ptr.release());
   task->set_task_type(received_task.task_type());
-  task->set_ttl(3600);
-  task->set_created_at(std::chrono::duration_cast<std::chrono::seconds>(
-                           std::chrono::system_clock::now().time_since_epoch())
-                           .count());
+  task->set_ttl("");
+  task->set_created_at("");
+  task->set_allocated_consumer(
+      std::make_unique<flwr::proto::Node>(received_task.producer()).release());
+  task->set_allocated_producer(
+      std::make_unique<flwr::proto::Node>(received_task.consumer()).release());
 
   task_res.set_allocated_task(task.release());