Healthcare

"""
Healthcare Tools
================
Tools for nurse triage: patient verification, symptom assessment, clinical KB search.
"""

from typing import Any, Dict

# ═══════════════════════════════════════════════════════════════════════════════
# TOOL SCHEMAS
# ═══════════════════════════════════════════════════════════════════════════════

verify_patient_identity_schema = {
    "name": "verify_patient_identity",
    "description": "Verify patient identity using name and DOB (HIPAA requirement)",
    "parameters": {
        "type": "object",
        "properties": {
            "full_name": {"type": "string", "description": "Patient's full name"},
            "date_of_birth": {"type": "string", "description": "DOB in YYYY-MM-DD format"},
        },
        "required": ["full_name", "date_of_birth"],
    },
}

search_clinical_knowledge_base_schema = {
    "name": "search_clinical_knowledge_base",
    "description": "Search clinical protocols and guidelines for symptom triage",
    "parameters": {
        "type": "object",
        "properties": {
            "query": {"type": "string", "description": "Symptom or condition to search"},
            "category": {
                "type": "string",
                "enum": ["symptoms", "protocols", "medications", "emergency"],
            },
        },
        "required": ["query"],
    },
}

assess_symptom_urgency_schema = {
    "name": "assess_symptom_urgency",
    "description": "Evaluate urgency: EMERGENCY, URGENT, or ROUTINE",
    "parameters": {
        "type": "object",
        "properties": {
            "symptoms": {"type": "array", "items": {"type": "string"}},
            "severity": {"type": "integer", "minimum": 1, "maximum": 10},
            "duration_hours": {"type": "number"},
        },
        "required": ["symptoms"],
    },
}

transfer_to_emergency_schema = {
    "name": "transfer_to_emergency",
    "description": "EMERGENCY: Transfer to 911 for life-threatening situations",
    "parameters": {
        "type": "object",
        "properties": {
            "emergency_type": {"type": "string"},
            "symptoms": {"type": "array", "items": {"type": "string"}},
        },
        "required": ["emergency_type"],
    },
}


# ═══════════════════════════════════════════════════════════════════════════════
# RED FLAG DETECTION
# ═══════════════════════════════════════════════════════════════════════════════

RED_FLAG_KEYWORDS = [
    "chest pain", "can't breathe", "difficulty breathing",
    "stroke", "face drooping", "slurred speech",
    "severe bleeding", "unconscious", "suicide", "overdose",
    "allergic reaction", "swelling throat", "anaphylaxis",
]


async def assess_symptom_urgency(args: Dict[str, Any]) -> Dict[str, Any]:
    """Assess urgency level of reported symptoms."""
    symptoms = args.get("symptoms", [])
    severity = args.get("severity", 5)
    symptoms_text = " ".join(s.lower() for s in symptoms)

    # Check for emergency keywords
    for keyword in RED_FLAG_KEYWORDS:
        if keyword in symptoms_text:
            return {
                "urgency": "EMERGENCY",
                "action": "transfer_to_emergency",
                "message": "Red flag detected. Transfer to emergency immediately.",
            }

    if severity >= 8:
        return {
            "urgency": "URGENT",
            "action": "handoff_specialist_nurse",
            "message": "High severity. Recommend specialist nurse.",
        }

    return {
        "urgency": "ROUTINE",
        "action": "schedule_appointment",
        "message": "Symptoms appear routine. Schedule appointment.",
    }

# Connect to FHIR API for patient data
from fhirclient import client

async def get_patient_profile(args: Dict[str, Any]) -> Dict[str, Any]:
    """Retrieve patient from FHIR-compliant EHR."""
    settings = {
        'app_id': 'nurse_triage',
        'api_base': os.getenv('FHIR_ENDPOINT')
    }
    smart = client.FHIRClient(settings=settings)
    patient = smart.patient.read(args['patient_id'])

    return {
        "success": True,
        "profile": {
            "full_name": patient.name[0].text,
            "allergies": [a.code.text for a in patient.allergyIntolerance],
            "medications": [m.code.text for m in patient.medicationStatement],
        },
    }

from registries.scenariostore.loader import load_scenario, build_handoff_map_from_scenario

# Load healthcare scenario
scenario = load_scenario("healthcare")
print(f"Start agent: {scenario['start_agent']}")
# → Start agent: NurseTriage

# Build handoff map
handoff_map = build_handoff_map_from_scenario("healthcare")
print(handoff_map)
# → {"handoff_specialist_nurse": "SpecialistNurse", "handoff_appointment": "AppointmentAgent"}

import pytest
from registries.scenariostore.loader import load_scenario, get_handoff_config

def test_healthcare_scenario_loads():
    scenario = load_scenario("healthcare")
    assert scenario["start_agent"] == "NurseTriage"
    assert "NurseTriage" in scenario["agents"]

def test_specialist_handoff_is_announced():
    config = get_handoff_config("healthcare", "NurseTriage", "handoff_specialist_nurse")
    assert config.type == "announced"
    assert config.share_context is True

def test_appointment_handoff_is_discrete():
    config = get_handoff_config("healthcare", "NurseTriage", "handoff_appointment")
    assert config.type == "discrete"