Patient portals and clinical workflows for hospitals and clinics.

# Structured LLM triage — safety-first parsing with pydantic
# The LLM classifies, but the rules engine has final say on urgency.

from typing import Literal
from pydantic import BaseModel, Field
import anthropic

class TriageDecision(BaseModel):
    urgency: Literal["critical", "urgent", "routine", "informational"]
    specialty: Literal["general", "cardiology", "orthopedics", "pediatrics",
                       "gynecology", "dermatology", "psychiatry", "unknown"]
    red_flags: list[str] = Field(default_factory=list)
    summary: str = Field(max_length=240)
    confidence: float = Field(ge=0.0, le=1.0)

RED_FLAG_TERMS = {
    "chest pain", "breathing difficulty", "unresponsive", "seizure",
    "heavy bleeding", "sudden weakness", "slurred speech", "suicidal",
}

client = anthropic.Anthropic()

def classify(patient_message: str) -> TriageDecision:
    # Deterministic red-flag check runs FIRST — never leave critical
    # cases to the LLM's discretion.
    hits = [term for term in RED_FLAG_TERMS if term in patient_message.lower()]
    if hits:
        return TriageDecision(
            urgency="critical",
            specialty="general",
            red_flags=hits,
            summary=f"Deterministic red-flag detected: {', '.join(hits)}",
            confidence=1.0,
        )

    resp = client.messages.create(
        model="claude-sonnet-5",
        max_tokens=512,
        system=(
            "You classify incoming patient messages into urgency and specialty. "
            "Never diagnose. Never suggest treatment. Never mention medication "
            "names. If uncertain, return unknown with low confidence."
        ),
        messages=[{"role": "user", "content": patient_message}],
        # In production, use tool_use with a JSON schema derived from
        # TriageDecision so the model returns validated structured output.
    )
    # ... parse response into TriageDecision, validate, return.
    # Any parse failure escalates to human review — never silently drops.
    raise NotImplementedError("wire up response parsing to TriageDecision")

def route(decision: TriageDecision) -> str:
    # Critical always goes to a human immediately.
    if decision.urgency == "critical" or decision.red_flags:
        return "on_call_doctor_queue"
    if decision.urgency == "urgent":
        return "same_day_slot_queue"
    return f"{decision.specialty}_next_available"