Skip to content

Commit 3313260

Browse files
committed
feat: add logging for missing module keys in ModuleManager
1 parent c3e4a53 commit 3313260

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

adf_core_python/core/agent/module/module_manager.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
MessageCoordinator,
1212
)
1313
from adf_core_python.core.component.module.abstract_module import AbstractModule
14+
from adf_core_python.core.logger.logger import get_logger
1415

1516
if TYPE_CHECKING:
1617
from adf_core_python.core.agent.config.module_config import ModuleConfig
@@ -43,9 +44,12 @@ def __init__(
4344
self._message_coordinators: dict[str, Any] = {}
4445

4546
def get_module(self, module_name: str, default_module_name: str) -> AbstractModule:
46-
class_name = self._module_config.get_value_or_default(
47-
module_name, default_module_name
48-
)
47+
class_name = self._module_config.get_value(module_name)
48+
if class_name is None:
49+
get_logger("ModuleManager").warning(
50+
f"Module key {module_name} not found in config, using default module {default_module_name}"
51+
)
52+
class_name = default_module_name
4953

5054
try:
5155
module_class: type = self._load_module(class_name)

0 commit comments

Comments
 (0)