82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
|
"""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.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 .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",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
@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)
|