evoagentx/hitl/__init__.py

from .hitl import (
    HITLDecision,
    HITLInteractionType,
    HITLMode,
    HITLContext,
    HITLRequest,
    HITLResponse,
)

from .approval_manager import (
    HITLManager,
)

from .interceptor_agent import (
    HITLBaseAgent,
    HITLInterceptorAgent,
    HITLUserInputCollectorAgent,
    HITLConversationAgent,
    HITLInterceptorAction,
    HITLUserInputCollectorAction,
    HITLPostExecutionAction,
    HITLConversationAction
)

from .special_hitl_agent import (
    HITLOutsideConversationAgent,
    HITLOutsideConversationAction,
)

SPECIAL_HITL_AGENT_REGISTRY = [
    HITLOutsideConversationAgent,
]

__all__ = [
    # HITL data model
    'HITLDecision',
    'HITLInteractionType', 
    'HITLMode',
    'HITLContext',
    'HITLRequest',
    'HITLResponse',
    
    'HITLManager',
    
    # HITL Agent and Action
    'HITLBaseAgent',
    'HITLInterceptorAgent',
    'HITLUserInputCollectorAgent',
    'HITLConversationAgent',
    'HITLInterceptorAction',
    'HITLUserInputCollectorAction',
    'HITLPostExecutionAction',
    'HITLConversationAction',
    
    # Special HITL Agent
    'HITLOutsideConversationAgent',
    'HITLOutsideConversationAction',

    # Special HITL Agent Registry
    'SPECIAL_HITL_AGENT_REGISTRY',
]