vpp-ai-platform/skills-py/vpp_skills/app.py

"""HTTP/JSON skill service (docs/11 §3.4, docs/09 §4).

One process, one route per skill, generated pydantic contracts on both request
and response. The TS runtime's registerSkill wrapper is the only intended
caller; it snapshots request/response for lineage. Run locally with
``uvicorn vpp_skills.app:app``.
"""

from __future__ import annotations

from fastapi import FastAPI, HTTPException
from fastapi.responses import JSONResponse

from vpp_contracts.bid_optimization_request import BidOptimizationRequest
from vpp_contracts.bid_optimization_result import BidOptimizationResult
from vpp_contracts.dispatch_optimization_request import DispatchOptimizationRequest
from vpp_contracts.dispatch_optimization_result import DispatchOptimizationResult
from vpp_contracts.potential_assessment_request import PotentialAssessmentRequest
from vpp_contracts.potential_assessment_result import PotentialAssessmentResult
from vpp_contracts.forecast_bundle import ForecastBundle
from vpp_contracts.forecast_request import ForecastRequest
from vpp_contracts.report_request import ReportRequest
from vpp_contracts.skill_report import SkillReport

from . import SKILL_VERSIONS
from .bid_milp import optimize_bid
from .dispatch_opt import optimize_dispatch
from .potential import assess_potential
from .forecast import forecast
from .report import generate_report

app = FastAPI(title="vpp-skills", version="0.1.0")

ROUTES = {
    "load-forecast": "/v1/forecast/load",
    "pv-forecast": "/v1/forecast/pv",
    "price-forecast": "/v1/forecast/price",
    "bid-optimization-milp": "/v1/optimize/bid",
    "report-generator": "/v1/report",
    "potential-assessment": "/v1/assess/potential",
    "dispatch-optimization": "/v1/optimize/dispatch",
}


@app.exception_handler(ValueError)
async def _value_error(_, exc: ValueError) -> JSONResponse:
    return JSONResponse(status_code=422, content={"detail": str(exc)})


@app.get("/health")
def health() -> dict:
    return {"status": "ok"}


@app.get("/v1/skills")
def skills() -> list[dict]:
    """Registry view: what the TS tool registry pins versions against."""
    return [{"id": sid, "version": ver, "endpoint": ROUTES[sid]} for sid, ver in SKILL_VERSIONS.items()]


def _forecast_for(kind: str, req: ForecastRequest) -> ForecastBundle:
    if req.kind.value != kind:
        raise HTTPException(status_code=422, detail=f"this endpoint serves kind={kind}, got {req.kind.value}")
    return forecast(req)


@app.post(ROUTES["load-forecast"], response_model=ForecastBundle)
def forecast_load(req: ForecastRequest) -> ForecastBundle:
    return _forecast_for("LOAD", req)


@app.post(ROUTES["pv-forecast"], response_model=ForecastBundle)
def forecast_pv(req: ForecastRequest) -> ForecastBundle:
    return _forecast_for("PV", req)


@app.post(ROUTES["price-forecast"], response_model=ForecastBundle)
def forecast_price(req: ForecastRequest) -> ForecastBundle:
    return _forecast_for("PRICE", req)


@app.post(ROUTES["bid-optimization-milp"], response_model=BidOptimizationResult)
def optimize(req: BidOptimizationRequest) -> BidOptimizationResult:
    return optimize_bid(req)


@app.post(ROUTES["report-generator"], response_model=SkillReport)
def report(req: ReportRequest) -> SkillReport:
    return generate_report(req)


@app.post(ROUTES["potential-assessment"], response_model=PotentialAssessmentResult)
def potential(req: PotentialAssessmentRequest) -> PotentialAssessmentResult:
    return assess_potential(req)


@app.post(ROUTES["dispatch-optimization"], response_model=DispatchOptimizationResult)
def dispatch(req: DispatchOptimizationRequest) -> DispatchOptimizationResult:
    return optimize_dispatch(req)