Changed filter for models

mlflow_client.py

@@ -37,7 +37,7 @@ def __init__(self):
 
     def models(self):
         returns = []
-        models_list = self.client.search_registered_models()
+        models_list = self.client.search_registered_models("tag.mage_model = 'true'")
 
         if not models_list:
             return None
@@ -64,6 +64,7 @@ def model_parameters(self, name: str) -> Dict[str, Any] | None:
 
     def model_versions(self, name: str) -> List[Dict[str, str]] | None:
         versions = self.client.search_model_versions(f"name='{name}'")
+
         return [{"version": version.version,
                  "type": "" if version.tags.get("model_type") is None else version.tags["model_type"]}
                 for version in versions]