from .BaseAgent import BaseAgent
from .AgentRegister import (
    AgentRegistry,
    get_agent_registry,
    register_agent,
    get_agent,
    list_available_models
)


from .CADAgent import CADAgent


__all__ = [
    'BaseAgent',
    'AgentRegistry',
    'get_agent_registry',
    'register_agent',
    'get_agent',
    'list_available_models',
]


# Register CADAgent.
register_agent("cadagent", CADAgent)