diff --git a/main.py b/main.py index 14c8ae3..115cff4 100644 --- a/main.py +++ b/main.py @@ -600,7 +600,12 @@ async def fast_get_pump_schema(network: str) -> dict[str, dict[str, Any]]: @app.post("/addpump/") async def fastapi_add_pump(network: str, pump: str, node1: str, node2: str) -> ChangeSet: - return add_pump(network, pump, node1, node2) + props: dict[str, any] = { + 'id' : pump, + 'node1' : node1, + 'node2' : node2 + } + return add_pump(network, ChangeSet(props)) @app.post("/deletepump/") async def fastapi_delete_pump(network: str, pump: str) -> ChangeSet: