src/agents/__init__.py

"""
DungeonMaster AI - Agents Package

LlamaIndex-based agents for game mastering, rules, and narration.
"""

from __future__ import annotations

# Models and enums
from src.agents.models import (
    DegradationLevel,
    DMResponse,
    GameContext,
    GameMode,
    LLMProviderHealth,
    LLMResponse,
    PacingStyle,
    RulesResponse,
    SpecialMoment,
    SpecialMomentType,
    ToolCallInfo,
    TurnResult,
    VoiceSegment,
)

# Exceptions
from src.agents.exceptions import (
    AgentError,
    DMAgentError,
    LLMAllProvidersFailedError,
    LLMProviderError,
    LLMRateLimitError,
    LLMTimeoutError,
    OrchestratorError,
    RulesAgentError,
    VoiceNarratorError,
    get_graceful_message,
)

# LLM Provider
from src.agents.llm_provider import (
    LLMFallbackChain,
    ProviderCircuitBreaker,
)

# Agents
from src.agents.dungeon_master import (
    DungeonMasterAgent,
    parse_voice_cues,
)
from src.agents.rules_arbiter import (
    RulesArbiterAgent,
    RulesCache,
)
from src.agents.voice_narrator import (
    PhraseCache,
    VoiceNarratorAgent,
    create_voice_narrator,
)

# Controllers
from src.agents.special_moments import (
    SpecialMomentHandler,
    UI_EFFECTS,
)
from src.agents.pacing_controller import (
    PACING_INSTRUCTIONS,
    PacingController,
    create_pacing_controller,
)

# Orchestrator
from src.agents.orchestrator import (
    AgentOrchestrator,
    create_orchestrator,
)


__all__ = [
    # Models
    "DegradationLevel",
    "DMResponse",
    "GameContext",
    "GameMode",
    "LLMProviderHealth",
    "LLMResponse",
    "PacingStyle",
    "RulesResponse",
    "SpecialMoment",
    "SpecialMomentType",
    "ToolCallInfo",
    "TurnResult",
    "VoiceSegment",
    # Exceptions
    "AgentError",
    "DMAgentError",
    "LLMAllProvidersFailedError",
    "LLMProviderError",
    "LLMRateLimitError",
    "LLMTimeoutError",
    "OrchestratorError",
    "RulesAgentError",
    "VoiceNarratorError",
    "get_graceful_message",
    # LLM Provider
    "LLMFallbackChain",
    "ProviderCircuitBreaker",
    # Agents
    "DungeonMasterAgent",
    "parse_voice_cues",
    "RulesArbiterAgent",
    "RulesCache",
    "VoiceNarratorAgent",
    "PhraseCache",
    "create_voice_narrator",
    # Controllers
    "SpecialMomentHandler",
    "UI_EFFECTS",
    "PACING_INSTRUCTIONS",
    "PacingController",
    "create_pacing_controller",
    # Orchestrator
    "AgentOrchestrator",
    "create_orchestrator",
]