agents/__init__.py

from agents.naive_agent import NaiveAgent
from agents.rl_agent import ReinforceAgent

try:
    from agents.research_llm_agent import ResearchLLMAgent
except ImportError:
    ResearchLLMAgent = None  # type: ignore[misc, assignment]

try:
    from agents.research_generate_agent import ResearchGenerateAgent
except ImportError:
    ResearchGenerateAgent = None  # type: ignore[misc, assignment]

__all__ = [
    "NaiveAgent",
    "ReinforceAgent",
    "ResearchLLMAgent",
    "ResearchGenerateAgent",
]