ensemble-core/main.py

from typing import Any, Dict

from fastapi import FastAPI
from pydantic import BaseModel

app = FastAPI(title="Ensemble Core Service")


class EnsembleRequest(BaseModel):
    media_type: str
    model_results: Dict[str, Any]
    method: str = "stacking"


@app.post("/calculate")
async def calculate_ensemble(request: EnsembleRequest):
    # TODO: Migrate proprietary ensemble logic from gateway
    return {"verdict": "fake", "confidence": 0.95, "method_used": request.method}


@app.get("/health")
async def health():
    return {"status": "healthy"}