feat(brain/tools) add start_cycles_check_timer in post_upgrade

arcmindai · Apr 25, 2024 · 700e113 · 700e113
1 parent 8dce2d5
commit 700e113
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/arcmindai_brain/src/lib.rs b/src/arcmindai_brain/src/lib.rs
@@ -417,6 +417,8 @@ fn post_upgrade(
  s.borrow_mut().battery_canister = battery_canister;
  s.borrow_mut().battery_api_key = battery_api_key.clone();
  });
+
+ start_cycles_check_timer(CYCLES_BALANCE_CHECK_MIN_INTERVAL_SECS);
 }
 
 // ---------------------- Candid declarations did file generator ----------------------

diff --git a/src/arcmindai_tools/src/lib.rs b/src/arcmindai_tools/src/lib.rs
@@ -378,6 +378,8 @@ fn post_upgrade(
  s.borrow_mut().battery_canister = battery_canister;
  s.borrow_mut().battery_api_key = battery_api_key.clone();
  });
+
+ start_cycles_check_timer(CYCLES_BALANCE_CHECK_MIN_INTERVAL_SECS);
 }
 
 // ---------------------- Candid declarations did file generator ----------------------