Sign competition GC logging requests 🔐 (#66)

Author: millioner

neurons/_validator/competitions/competition_manager.py

@@ -7,6 +7,7 @@
 import json
 import os
 import enum
+import cli_parser
 from utils.gc_logging import gc_log_competition_metrics
 
 
@@ -69,6 +70,7 @@ def __init__(self, config_dir: str):
         self.config_file = os.path.join(config_dir, "competition_config.json")
         self.current_competition: Optional[CompetitionConfig] = None
         self.state = CompetitionState()
+        self.wallet = bt.wallet(config=cli_parser.config)
 
         self._load_state()
         self._load_config()
@@ -188,7 +190,7 @@ def log_metrics(self, metrics: dict):
                 **metrics,
             }
 
-            gc_log_competition_metrics(comp_metrics)
+            gc_log_competition_metrics(comp_metrics, self.wallet.hotkey)
         except Exception as e:
             bt.logging.error(f"Error logging metrics: {e}")
 

neurons/utils/gc_logging.py

@@ -69,16 +69,27 @@ def log_responses(
         return None
 
 
-def gc_log_competition_metrics(metrics: dict) -> Optional[requests.Response]:
+def gc_log_competition_metrics(
+    metrics: dict, hotkey: bt.Keypair
+) -> Optional[requests.Response]:
     """
     Log competition metrics to the centralized logging server.
     """
     try:
+        metrics["validator_key"] = hotkey.ss58_address
         input_bytes = json.dumps(metrics).encode("utf-8")
+        # sign the inputs with your hotkey
+        signature = hotkey.sign(input_bytes)
+        # encode the inputs and signature as base64
+        signature_str = base64.b64encode(signature).decode("utf-8")
+
         return session.post(
             COMPETITION_LOGGING_URL,
             data=input_bytes,
-            headers={"Content-Type": "application/json"},
+            headers={
+                "Content-Type": "application/json",
+                "X-Request-Signature": signature_str,
+            },
             timeout=5,
         )
     except requests.exceptions.RequestException as e: