fixup

Author: armanddidierjean

app/app.py

@@ -298,6 +298,7 @@ def initialize_module_visibility(
             coredata_core.ModuleVisibilityAwareness,
             db,
         )
+        known_roots = module_awareness.roots
 
         new_modules = [
             module
@@ -310,6 +311,7 @@ def initialize_module_visibility(
                 f"Startup: Some modules visibility settings are empty, initializing them ({[module.root for module in new_modules]})",
             )
             for module in new_modules:
+                known_roots.append(module.root)
                 if module.default_allowed_groups_ids is not None:
                     for group_id in module.default_allowed_groups_ids:
                         module_group_visibility = models_core.ModuleGroupVisibility(
@@ -343,9 +345,7 @@ def initialize_module_visibility(
                                 f"Startup: Could not add module visibility {module.root} in the database: {error}",
                             )
             initialization.set_core_data_sync(
-                coredata_core.ModuleVisibilityAwareness(
-                    roots=[module.root for module in get_module_list()],
-                ),
+                coredata_core.ModuleVisibilityAwareness(roots=known_roots),
                 db,
             )
             hyperion_error_logger.info(

app/module.py

@@ -3,6 +3,7 @@
 from pathlib import Path
 
 from app.core.utils.config import Settings
+from app.types.exceptions import InvalidModuleRootInDotenvError
 from app.types.module import CoreModule, Module
 
 hyperion_error_logger = logging.getLogger("hyperion.error")
@@ -12,6 +13,9 @@
 
 
 def init_module_list(settings: Settings):
+    _module_list.clear()
+    _core_module_list.clear()
+
     module_list = []
     for endpoints_file in Path().glob("app/modules/*/endpoints_*.py"):
         endpoint_module = importlib.import_module(
@@ -29,7 +33,7 @@ def init_module_list(settings: Settings):
         existing_module_roots = [module.root for module in module_list]
         for root in settings.RESTRICT_TO_MODULES:
             if root not in existing_module_roots:
-                raise ValueError()
+                raise InvalidModuleRootInDotenvError(root)
     for module in module_list:
         if (
             settings.RESTRICT_TO_MODULES

app/types/exceptions.py

@@ -32,6 +32,11 @@ def __init__(self):
         super().__init__("Google API is not configured in dotenv")
 
 
+class InvalidModuleRootInDotenvError(Exception):
+    def __init__(self, root: str):
+        super().__init__(f"Module root {root} does not exist")
+
+
 class ContentHTTPException(HTTPException):
     """
     A custom HTTPException allowing to return custom content.