from mcp.server.fastmcp import FastMCP
from agents.investigative_agent.orchestrator import run_investigation

# Initialize the MCP Server
mcp = FastMCP("Antigravity-Codex-AML")

@mcp.tool()
def investigate_alert(alert_id: str) -> dict:
    """
    Run a full AML investigation on a given alert ID (e.g., 'Alert-101', 'Alert-102', 'Alert-103').
    Returns a complex JSON object containing the profile, model_scores, and a generated SAR draft.
    """
    return run_investigation(alert_id)