cursor support

Author: acer-king

Diff for: validators/core/axon.py

@@ -34,6 +34,10 @@
     InternalServerError,
 )
 from bittensor.threadpool import PriorityThreadPoolExecutor
+import bittensor
+import bittensor as bt
+from substrateinterface import Keypair
+from bittensor.errors import SynapseDendriteNoneException
 
 
 class CortexAxon(bt.axon):
@@ -49,6 +53,26 @@ def __init__(self,
         super().__init__(wallet, config, port, ip, external_ip, external_port, max_workers)
         self.app.add_middleware(CortexAxonMiddleware, axon=self)
 
+    def default_verify(self, synapse: bittensor.Synapse):
+        if synapse.dendrite is not None:
+            keypair = Keypair(ss58_address=synapse.dendrite.hotkey)
+
+            # Build the signature messages.
+            message = f"{synapse.dendrite.nonce}.{synapse.dendrite.hotkey}.{self.wallet.hotkey.ss58_address}.{synapse.dendrite.uuid}.{synapse.computed_body_hash}"
+
+            # Build the unique endpoint key.
+            endpoint_key = f"{synapse.dendrite.hotkey}:{synapse.dendrite.uuid}"
+
+            if not keypair.verify(message, synapse.dendrite.signature):
+                raise Exception(
+                    f"Signature mismatch with {message} and {synapse.dendrite.signature}"
+                )
+
+            # Success
+            self.nonces[endpoint_key] = synapse.dendrite.nonce  # type: ignore
+        else:
+            raise SynapseDendriteNoneException()
+
 
 class CortexAxonMiddleware(BaseHTTPMiddleware):
     """

Diff for: validators/weight_setter.py

@@ -24,8 +24,8 @@
 from validators.services.cache import QueryResponseCache
 from validators.utils import error_handler, setup_max_capacity, load_entire_questions
 from validators.task_manager import TaskMgr
+from validators.core.axon import CortexAxon
 from cortext.dendrite import CortexDendrite
-from cortext.axon import CortexAxon
 from cursor.app.endpoints.text import chat
 from cursor.app.endpoints.generic import models
 from cursor.app.core.middleware import APIKeyMiddleware