Add an MCP Prompt Template

from mcp.server.fastmcp.prompts import base

# ---------- Prompts: reusable templates ----------

@mcp.prompt()
def draft_sar_narrative(
    case_id: str,
    suspicious_activity_type: str,
    time_window: str = "the last 30 days",
) -> list[base.Message]:
    """Draft a Suspicious Activity Report (SAR) narrative for a case.

    The narrative will follow the firm's standard SAR template (WHO/WHAT/WHERE/WHEN/WHY),
    cite specific transactions from the case file, and reference the policy URIs used.
    """
    return [
        base.UserMessage(
            f"You are a senior compliance analyst drafting a SAR narrative. "
            f"Use ONLY information from the case file and the policy documents you have "
            f"been provided. Cite specific transactions by date and amount. "
            f"Cite policy URIs for any procedural claim. Refuse to speculate.\n\n"
            f"Case: {case_id}\n"
            f"Suspicious activity type: {suspicious_activity_type}\n"
            f"Time window: {time_window}\n\n"
            f"Step 1: Fetch the case file at compliance://cases/{case_id}.\n"
            f"Step 2: Fetch the SAR template at compliance://policies/sar-narrative-template.\n"
            f"Step 3: Draft a 5-15 sentence narrative following the template structure.\n"
            f"Step 4: List the specific URIs you cited.\n\n"
            f"Do not file the SAR. Produce a draft for human review only."
        ),
    ]


@mcp.prompt()
def explain_alert(
    alert_id: str,
    counterparty_name: str,
    counterparty_country: str,
    amount_usd: float,
) -> str:
    """Explain a transaction alert in plain language for a non-technical investigator.

    Returns a single rendered string the model uses as its user-turn input.
    """
    return (
        f"I'm a new investigator. Please explain alert {alert_id} in plain language. "
        f"The alert involves a {amount_usd:,.2f} USD transaction with counterparty "
        f"'{counterparty_name}' in {counterparty_country}.\n\n"
        f"Walk me through:\n"
        f"  1. What screening tools would you run on this?\n"
        f"  2. What the result of each tool tells you in plain English.\n"
        f"  3. What the recommended action is, and why.\n"
        f"  4. What I should ask the customer if I need more information.\n\n"
        f"Use the compliance-toolkit tools to run the screening for me as you explain."
    )

mcp dev server.py

{
  "case_id": "CASE-2026-00012",
  "suspicious_activity_type": "structuring (sub-CTR-threshold splitting)",
  "time_window": "April 18-21, 2026"
}

@mcp.prompt()
def review_sar_draft(draft_text: str) -> str:
    """Critique a draft SAR narrative against the firm's quality rubric.

    Returns a structured review pointing out missing sections, speculative language,
    uncited claims, and tone issues. Does NOT rewrite — the human decides what to fix.
    """
    return (
        f"You are a compliance lead reviewing a junior analyst's draft SAR narrative. "
        f"Apply the firm's rubric strictly. Do NOT rewrite the draft — produce a "
        f"structured critique that the analyst can act on.\n\n"
        f"--- DRAFT ---\n{draft_text}\n--- END DRAFT ---\n\n"
        f"Score each of the five criteria on a 1-5 scale and explain:\n"
        f"  1. WHO/WHAT/WHERE/WHEN/WHY structure — are all five present?\n"
        f"  2. Factuality — every claim backed by case data, no speculation?\n"
        f"  3. Citation — specific transactions cited by date and amount?\n"
        f"  4. Tone — factual, non-conclusory, regulator-appropriate?\n"
        f"  5. Length — 5-15 sentences, specific not comprehensive?\n\n"
        f"Then produce a 'top 3 things to fix' list, ordered by importance.\n"
        f"Conclude with a single line: APPROVED | REVISE | REJECT."
    )