githubmirror/ChatDev

runtime/bootstrap/schema.py

28 строк772 B

NA-Wen

initial commit of chatdev 2.0

07 янв 2026, 11:24

07 янв 2026, 11:24f0db945

"""Ensure schema registry is populated with runtime-provided registrations."""

from importlib import import_module
from typing import Iterable

_BOOTSTRAPPED = False


def _modules_to_import() -> Iterable[str]:
    return (
        "runtime.node.builtin_nodes",
        "runtime.node.agent.memory.builtin_stores",
        "runtime.node.agent.thinking.builtin_thinking",
        "runtime.edge.conditions.builtin_types",
        "runtime.node.agent.providers.builtin_providers",
    )


def ensure_schema_registry_populated() -> None:
    """Import built-in runtime registration modules exactly once."""
    global _BOOTSTRAPPED
    if _BOOTSTRAPPED:
        return

    for module_name in _modules_to_import():
        import_module(module_name)

    _BOOTSTRAPPED = True