added support for tags to categorise the models

2025-01-10 11:08:10 +01:00 · 2025-01-10 11:08:10 +01:00 · c6d779f591
commit c6d779f591
parent b89fafdc96
3 changed files with 9 additions and 1 deletions
--- a/source/model/base/BaseModel.py
+++ b/source/model/base/BaseModel.py
@ -18,6 +18,8 @@ class BaseModel(abc.ABC):
        self.manager = manager
        # the mimetype of the model responses
        self.output_type: str = configuration.get("output_type", "application/json")
+        # get the tags of the model
+        self.tags = configuration.get("tags", [])

        self._loaded = False