CU-8699vq0he: Improve addon access from CAT (#55)

* CU-8699vq0he: Add methods for easier addon access

* CU-8699vq0he: Add some docstrings for addon-related methods

Author: mart-r

medcat-v2/medcat/cat.py

@@ -1,5 +1,5 @@
 from typing import Optional, Union, Any, overload, Literal, Iterable, Iterator
-from typing import cast
+from typing import cast, Type, TypeVar
 import os
 import json
 from datetime import date
@@ -39,6 +39,9 @@
 logger = logging.getLogger(__name__)
 
 
+AddonType = TypeVar("AddonType", bound="AddonComponent")
+
+
 class CAT(AbstractSerialisable):
     """This is a collection of serialisable model parts.
     """
@@ -839,9 +842,36 @@ def __eq__(self, other: Any) -> bool:
     # addon (e.g MetaCAT) related stuff
 
     def add_addon(self, addon: AddonComponent) -> None:
+        """Add the addon to the model pack an pipe.
+
+        Args:
+            addon (AddonComponent): The addon to add.
+        """
         self.config.components.addons.append(addon.config)
         self._pipeline.add_addon(addon)
 
+    def get_addons(self) -> list[AddonComponent]:
+        """Get the list of all addons in this model pack.
+
+        Returns:
+            list[AddonComponent]: The list of addons present.
+        """
+        return list(self._pipeline.iter_addons())
+
+    def get_addons_of_type(self, addon_type: Type[AddonType]) -> list[AddonType]:
+        """Get a list of addons of a specific type.
+
+        Args:
+            addon_type (Type[AddonType]): The type of addons to look for.
+
+        Returns:
+            list[AddonType]: The list of addons of this specific type.
+        """
+        return [
+            addon for addon in self.get_addons()
+            if isinstance(addon, addon_type)
+        ]
+
 
 class OutOfDataException(ValueError):
     pass