feat: expose Antigravity-Codex-AML engine via FastMCP SSE

api.py

@@ -1,35 +1,48 @@
-from __future__ import annotations
-
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-
-from agents.credit_risk.router import router as credit_router
-from agents.kyc_identity.router import router as kyc_router
-from agents.risk_consultant.router import router as consultant_router
-from agents.sanctions_pep.router import router as sanctions_router
-from agents.transaction_fraud.router import router as fraud_router
-
-app = FastAPI(
-    title="RiskOS Fraud Intelligence Suite",
-    version="1.0.0",
-    description="Multi-agent AI system for real-time fraud detection, credit risk assessment, KYC identity verification, and sanctions screening.",
-)
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-app.include_router(fraud_router, prefix="/api/v1/fraud")
-app.include_router(credit_router, prefix="/api/v1/credit")
-app.include_router(kyc_router, prefix="/api/v1/kyc")
-app.include_router(sanctions_router, prefix="/api/v1/sanctions")
-app.include_router(consultant_router, prefix="/api/v1/consultant")
-
-
-@app.get("/health")
-def health() -> dict[str, object]:
-    return {"status": "ok", "agents": ["fraud", "credit", "kyc", "sanctions", "consultant"]}
+from __future__ import annotations
+
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
+from agents.credit_risk.router import router as credit_router
+from agents.kyc_identity.router import router as kyc_router
+from agents.risk_consultant.router import router as consultant_router
+from agents.sanctions_pep.router import router as sanctions_router
+from agents.transaction_fraud.router import router as fraud_router
+from mcp_server import mcp
+
+app = FastAPI(
+    title="RiskOS Fraud Intelligence Suite",
+    version="1.0.0",
+    description="Multi-agent AI system for real-time fraud detection, credit risk assessment, KYC identity verification, and sanctions screening.",
+)
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+app.include_router(fraud_router, prefix="/api/v1/fraud")
+app.include_router(credit_router, prefix="/api/v1/credit")
+app.include_router(kyc_router, prefix="/api/v1/kyc")
+app.include_router(sanctions_router, prefix="/api/v1/sanctions")
+app.include_router(consultant_router, prefix="/api/v1/consultant")
+
+
+@app.get("/health")
+def health() -> dict[str, object]:
+    return {"status": "ok", "agents": ["fraud", "credit", "kyc", "sanctions", "consultant"]}
+
+
+mcp_app = mcp.sse_app(mount_path="/mcp")
+mcp_app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+app.mount("/mcp", mcp_app)

mcp_server.py

@@ -0,0 +1,15 @@
+from __future__ import annotations
+
+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)

requirements.txt

@@ -1,5 +1,5 @@
 gradio==4.44.0
-fastapi==0.111.0
+fastapi>=0.112.0
 uvicorn==0.30.1
 pydantic==2.7.1
 xgboost==2.0.3
@@ -14,3 +14,5 @@ httpx==0.27.0
 pytest==8.2.0
 python-dotenv==1.0.1
 huggingface_hub==0.25.2
+mcp>=1.1.2
+sse-starlette>=2.0.0