Agent/cyber_agent/application/__init__.py

"""Experimental application framework API (no compatibility promise before M6)."""

from importlib import import_module

from cyber_agent.application.models import (
    AgentApplication,
    AgentRole,
    ApplicationRef,
    ProviderRef,
    RoleRunLimits,
    RunLimits,
    SkillSpec,
    ToolSet,
    ToolSpec,
)
_LAZY_EXPORTS = {
    "ApplicationBinding": ("cyber_agent.application.runtime", "ApplicationBinding"),
    "ApplicationRegistry": ("cyber_agent.application.runtime", "ApplicationRegistry"),
    "ApplicationRuntime": ("cyber_agent.application.runtime", "ApplicationRuntime"),
    "ApplicationServices": ("cyber_agent.application.runtime", "ApplicationServices"),
    "CandidateAdoptionReceipt": ("cyber_agent.application.candidate", "CandidateAdoptionReceipt"),
    "CandidateAdoptionRequest": ("cyber_agent.application.candidate", "CandidateAdoptionRequest"),
    "CandidateLedger": ("cyber_agent.application.candidate", "CandidateLedger"),
    "CandidatePointer": ("cyber_agent.application.candidate", "CandidatePointer"),
    "CandidateRef": ("cyber_agent.application.candidate", "CandidateRef"),
    "CandidateRepository": ("cyber_agent.application.candidate", "CandidateRepository"),
    "CandidateVersionRequest": ("cyber_agent.application.candidate", "CandidateVersionRequest"),
    "ContextMaterial": ("cyber_agent.application.ports", "ContextMaterial"),
    "ContextRequest": ("cyber_agent.application.ports", "ContextRequest"),
    "ContextResourceDescriptor": ("cyber_agent.application.ports", "ContextResourceDescriptor"),
    "TaskContextProvider": ("cyber_agent.application.ports", "TaskContextProvider"),
}


def __getattr__(name):
    target = _LAZY_EXPORTS.get(name)
    if target is None:
        raise AttributeError(name)
    module_name, attribute = target
    value = getattr(import_module(module_name), attribute)
    globals()[name] = value
    return value


__all__ = [
    "AgentApplication",
    "AgentRole",
    "ApplicationRef",
    "ApplicationBinding",
    "ApplicationRegistry",
    "ApplicationRuntime",
    "ApplicationServices",
    "CandidateAdoptionReceipt",
    "CandidateAdoptionRequest",
    "CandidateLedger",
    "CandidatePointer",
    "CandidateRef",
    "CandidateRepository",
    "CandidateVersionRequest",
    "ContextMaterial",
    "ContextRequest",
    "ContextResourceDescriptor",
    "ProviderRef",
    "RoleRunLimits",
    "RunLimits",
    "SkillSpec",
    "TaskContextProvider",
    "ToolSet",
    "ToolSpec",
]