Merge pull request #147 from web-genie-ai/feat/ddos

Feat/ddos

Author: sangar-1028

.gitignore

@@ -193,3 +193,4 @@ developer_doc.md
 
 debug_images/
 submit_results.py
+*.pem

neurons/validators/validator.py

@@ -87,13 +87,11 @@ def __init__(self, config=None):
 
     def resync_metagraph(self):
         # Copies state of metagraph before syncing.
-        previous_metagraph = copy.deepcopy(self.metagraph)
-
+        previous_axons = copy.deepcopy(self.metagraph.axons)
         # Sync the metagraph.
         self.metagraph.sync(subtensor=self.subtensor)
-
         # Check if the metagraph axon info has changed.
-        if previous_metagraph.axons == self.metagraph.axons:
+        if previous_axons == self.metagraph.axons:
             return
 
         bt.logging.info(

pyproject.toml

@@ -1,17 +1,17 @@
 [project]
 name = "web-genie-ai"
-version = "1.1.20"
+version = "1.3.0"
 description = "The first bittensor subnet for web generation"
 readme = "README.md"
 requires-python = ">=3.12.4"
 dependencies = [
     "ansible-vault==2.1.0",
-    "async-substrate-interface==1.0.0",
+    "async-substrate-interface==1.0.3",
     "beautifulsoup4==4.12.3",
     "bert-score==0.3.13",
     "bt-decode==0.5.0a2",
-    "bittensor==9.0.0",
-    "bittensor-cli==9.0.0",
+    "bittensor==9.0.3",
+    "bittensor-cli==9.0.3",
     "clip",
     "datasets==3.2.0",
     "ddt==1.6.0",
@@ -26,7 +26,7 @@ dependencies = [
     "peft",
     "pip-chill==1.0.3",
     "playwright==1.49.1",
-    "pydantic==2.6",
+    "pydantic==2.10.6",
     "python-dotenv==1.0.1",
     "scikit-learn==1.6.0",
     "shtab==1.6.5",
@@ -40,6 +40,7 @@ dependencies = [
     "tensorflow>=2.18.0",
     "tf-keras",
     "uvicorn",
+    "bt-ddos-shield-client>=0.1.3",
 ]
 
 [project.urls]

uv.lock

@@ -21,6 +21,7 @@
 import bittensor as bt
 
 from abc import ABC, abstractmethod
+from bt_ddos_shield import ShieldMetagraph
 
 # Sync calls set weights and also resyncs the metagraph.
 from webgenie.constants import NEURON_EPOCH_LENGTH, SPEC_VERSION
@@ -90,7 +91,7 @@ def __init__(self, config=None):
         else:
             self.wallet = bt.wallet(config=self.config)
             self.subtensor = bt.subtensor(config=self.config)
-            self.metagraph = self.subtensor.metagraph(self.config.netuid)
+            self.metagraph = ShieldMetagraph(self.wallet, self.config.netuid, subtensor=self.subtensor)
 
         bt.logging.info(f"Wallet: {self.wallet}")
         bt.logging.info(f"Subtensor: {self.subtensor}")

webgenie/base/neuron.py

@@ -3,7 +3,7 @@
 import psutil
 # Change this value when updating your code base.
 # Define the version of the webgenie.
-__VERSION__ = "1.2.2" # version
+__VERSION__ = "1.3.0" # version
 
 SPEC_VERSION = (
     (1000 * int(__VERSION__.split(".")[0]))

webgenie/constants.py

@@ -48,5 +48,5 @@ def get_html(self, session:int, task_number:int)->str:
 
         if response.status_code != 200:
             raise Exception(f"Failed to get HTML: {response.status_code} {response.text}")
-        bt.logging.info(f"HTML: {response.json()}")
+        
         return response.json()["html"]