Merge pull request #2161 from backend-developers-ltd/fix_to_headers_speed

fix Synapse base performance (more than 10x speed up)

Author: thealligatorking

bittensor/synapse.py

@@ -369,6 +369,7 @@ class Synapse(BaseModel):
     """
 
     model_config = ConfigDict(validate_assignment=True)
+    _model_json_schema: ClassVar[Dict[str, Any]]
 
     def deserialize(self) -> "Synapse":
         """
@@ -580,11 +581,27 @@ def failed_verification(self) -> bool:
         """
         return self.dendrite is not None and self.dendrite.status_code == 401
 
+    @classmethod
+    def _get_cached_model_json_schema(cls) -> dict:
+        """
+        Returns the JSON schema for the Synapse model.
+
+        This method returns a cached version of the JSON schema for the Synapse model.
+        The schema is stored in the class variable ``_model_json_schema`` and is only
+        generated once to improve performance.
+
+        Returns:
+            dict: The JSON schema for the Synapse model.
+        """
+        if "_model_json_schema" not in cls.__dict__:
+            cls._model_json_schema = cls.model_json_schema()
+        return cls._model_json_schema
+
     def get_required_fields(self):
         """
         Get the required fields from the model's JSON schema.
         """
-        schema = self.__class__.model_json_schema()
+        schema = self._get_cached_model_json_schema()
         return schema.get("required", [])
 
     def to_headers(self) -> dict:
@@ -635,16 +652,15 @@ def to_headers(self) -> dict:
         # Getting the fields of the instance
         instance_fields = self.model_dump()
 
+        required = set(self.get_required_fields())
         # Iterating over the fields of the instance
         for field, value in instance_fields.items():
             # If the object is not optional, serializing it, encoding it, and adding it to the headers
-            required = self.get_required_fields()
-
             # Skipping the field if it's already in the headers or its value is None
             if field in headers or value is None:
                 continue
 
-            elif required and field in required:
+            elif field in required:
                 try:
                     # create an empty (dummy) instance of type(value) to pass pydantic validation on the axon side
                     serialized_value = json.dumps(value.__class__.__call__())