mentalwellness / agents /crisis_agent.py
invincible-jha's picture
Implement Mistral LLM support for all agents
3e274d5
raw
history blame contribute delete
835 Bytes
from typing import Dict
from .base_agent import BaseWellnessAgent
class CrisisAgent(BaseWellnessAgent):
"""Agent specialized in crisis intervention"""
def __init__(self, model_config: Dict, **kwargs):
super().__init__(
model_config=model_config,
agent_type="crisis",
role="Crisis Intervention Specialist",
goal="Provide immediate support and intervention in crisis situations",
backstory="""I am an AI agent specialized in crisis intervention.
I prioritize user safety and provide immediate support while ensuring
appropriate escalation to emergency services when necessary.""",
verbose=True,
allow_delegation=False,
tools=[], # Tools will be added as needed
**kwargs
)