🎉 ModelCache: add get_loaded_model_ids method.

pyproject.toml

@@ -2,7 +2,7 @@
 
 [tool.poetry]
 name = "codenames-solvers"
-version = "1.7.2"
+version = "1.7.3"
 description = "Solvers implementation for Codenames board game in python."
 authors = ["Michael Kali <[email protected]>", "Asaf Kali <[email protected]>"]
 readme = "README.md"

solvers/models/cache.py

@@ -2,7 +2,7 @@
 import os
 from os.path import expanduser
 from threading import Lock
-from typing import Dict
+from typing import Dict, Set
 
 import gensim.downloader as gensim_api
 from generic_iterative_stemmer.models import StemmedKeyedVectors
@@ -25,6 +25,11 @@ def _get_model_lock(self, model_identifier: ModelIdentifier) -> Lock:
             model_lock = self._model_locks.setdefault(model_identifier, Lock())
         return model_lock
 
+    def get_loaded_model_ids(self) -> Set[ModelIdentifier]:
+        with self._main_lock:
+            keys = set(self._cache.keys())
+        return keys
+
     def is_loaded(self, model_identifier: ModelIdentifier) -> bool:
         return model_identifier in self._cache