DONT_USE_HIVE_BINARIES, AUTH_TOKEN env parameters

clore_hosting/main.py

@@ -116,7 +116,10 @@ class CloreClient:
                     self.restart_docker = True # Restart docker when it's loaded under systemd (accual restart will happen only if no orders running to not disrupt workload)
 
         docker_interface.verify_docker_version()
-        nvml.init()
+
+        self.dont_use_hive_binaries = True if 'DONT_USE_HIVE_BINARIES' in os.environ else False
+
+        nvml.init(allow_hive_binaries=not self.dont_use_hive_binaries)
 
         self.gpu_oc_specs = nvml.get_gpu_oc_specs()
         self.last_oc_service_submit = 0
@@ -492,10 +495,10 @@ class CloreClient:
                 await monitoring.put("oc_service")
                 oc_apply_allowed = True
                 ### OC Service should also hande Hive stuff
-                if self.use_hive_flightsheet and self.is_hive:
+                if self.use_hive_flightsheet and self.is_hive and not self.dont_use_hive_binaries:
                     await set_hive_miner_status(True)
                     oc_apply_allowed = False # Don't apply any OC when running HiveOS miner
-                elif self.is_hive:
+                elif self.is_hive and not self.dont_use_hive_binaries:
                     await set_hive_miner_status(False)
                 ### Run OC tasks
                 oc_conf = WebSocketClient.get_oc()

lib/nvml.py

@@ -70,13 +70,13 @@ is_hive = False
 all_gpus_data_list=[]
 get_data_fail=False
 
-def init(gpu_specs_file=None):
+def init(gpu_specs_file=None, allow_hive_binaries=True):
     global is_hive, all_gpus_data_list, get_data_fail
     log.info("Loading GPU OC specs [ working ]")
     try:
         pynvml.nvmlInit()
         kernel = get_specs.get_kernel()
-        if "hive" in kernel:
+        if "hive" in kernel and allow_hive_binaries:
             is_hive=True
         
         specs_file_loc = gpu_specs_file if gpu_specs_file else config.gpu_specs_file

lib/utils.py

@@ -41,6 +41,8 @@ def normalize_rule(rule_dict):
 
 def get_auth():
     try:
+        if 'AUTH_TOKEN' in os.environ:
+            return os.environ['AUTH_TOKEN']
         auth_str = ''
         with open(config.auth_file, "r", encoding="utf-8") as file:
             auth_str = file.read().strip()