add class variable as a cache

fetchai · Jul 6, 2023 · f552b91 · f552b91
1 parent 069d70c
commit f552b91
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/uagents/models.py b/src/uagents/models.py
@@ -5,6 +5,8 @@
 
 
 class Model(BaseModel):
+    _schema_no_descr = None
+
     @staticmethod
     def remove_descriptions(schema: Dict[str, Dict[str, str]]):
         fields_with_descr = []
@@ -23,9 +25,11 @@ def remove_descriptions(schema: Dict[str, Dict[str, str]]):
 
     @classmethod
     def schema_no_descr(cls) -> Dict[str, Any]:
-        orig_schema = json.loads(cls.schema_json(indent=None, sort_keys=True))
-        Model.remove_descriptions(orig_schema)
-        return orig_schema
+        if cls._schema_no_descr is None:
+            schema = json.loads(cls.schema_json(indent=None, sort_keys=True))
+            Model.remove_descriptions(schema)
+            cls._schema_no_descr = schema
+        return cls._schema_no_descr
 
     @classmethod
     def schema_json_no_descr(cls) -> str: