agents/specialized/__init__.py

"""
Specialized agents package - Domain-specific healthcare agents
"""

from .nutrition_agent import NutritionAgent
from .exercise_agent import ExerciseAgent
from .symptom_agent import SymptomAgent
from .mental_health_agent import MentalHealthAgent
from .general_health_agent import GeneralHealthAgent

# Agent registry
AGENTS = {
    "nutrition_agent": NutritionAgent,
    "exercise_agent": ExerciseAgent,
    "symptom_agent": SymptomAgent,
    "mental_health_agent": MentalHealthAgent,
    "general_health_agent": GeneralHealthAgent
}

def get_agent(agent_name):
    """Get agent instance by name"""
    agent_class = AGENTS.get(agent_name)
    if agent_class:
        return agent_class()
    return GeneralHealthAgent()  # Default fallback

__all__ = [
    'NutritionAgent',
    'ExerciseAgent',
    'SymptomAgent',
    'MentalHealthAgent',
    'GeneralHealthAgent',
    'AGENTS',
    'get_agent'
]