agents/assessment_agent.py

from typing import Dict, List
from .base_agent import BaseWellnessAgent

class AssessmentAgent(BaseWellnessAgent):
    """Agent specialized in mental health assessments"""
    
    def __init__(self, model_config: Dict, **kwargs):
        super().__init__(
            model_config=model_config,
            agent_type="assessment",
            role="Mental Health Evaluator",
            goal="Conduct thorough mental health assessments and provide actionable insights",
            backstory="""I am an AI agent specialized in mental health assessment. 
            I use validated assessment tools and techniques to evaluate mental well-being 
            and provide personalized recommendations.""",
            verbose=True,
            allow_delegation=False,
            tools=[],  # Tools will be added as needed
            **kwargs
        )