hkuds--nanobot
wehub-resource-sync
43 行
1.2 KiB
Python
43 行
1.2 KiB
Python
"""Utility functions for nanobot."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from importlib import import_module
|
|
from types import ModuleType
|
|
|
|
from nanobot.utils.helpers import ensure_dir
|
|
from nanobot.utils.path import abbreviate_path
|
|
|
|
__all__ = ["ensure_dir", "abbreviate_path"]
|
|
|
|
|
|
class _LazyModuleAlias(ModuleType):
|
|
def __init__(self, name: str, target: str) -> None:
|
|
super().__init__(name)
|
|
self.__dict__["_target"] = target
|
|
|
|
def _load(self) -> ModuleType:
|
|
module = import_module(self.__dict__["_target"])
|
|
sys.modules[self.__name__] = module
|
|
return module
|
|
|
|
def __getattr__(self, name: str) -> object:
|
|
return getattr(self._load(), name)
|
|
|
|
def __dir__(self) -> list[str]:
|
|
return sorted(set(super().__dir__()) | set(dir(self._load())))
|
|
|
|
|
|
_LEGACY_MODULE_ALIASES = {
|
|
"webui_thread_disk": "nanobot.webui.thread_disk",
|
|
"webui_transcript": "nanobot.webui.transcript",
|
|
"webui_turn_helpers": "nanobot.session.webui_turns",
|
|
}
|
|
|
|
for _legacy_name, _target_name in _LEGACY_MODULE_ALIASES.items():
|
|
sys.modules.setdefault(
|
|
f"{__name__}.{_legacy_name}",
|
|
_LazyModuleAlias(f"{__name__}.{_legacy_name}", _target_name),
|
|
)
|