fix schema response post simulation

main
MrWaradana 4 weeks ago
parent e81f3508d8
commit ef265379e6

@ -7,6 +7,7 @@ from src.database.core import DbSession
from src.database.service import CommonParameters
from src.models import StandardResponse
from src.simulations.schema import (
SimulationBase,
SimulationCreate,
SimulationPagination,
SimulationRead,
@ -47,7 +48,7 @@ async def get_simulation(db_session: DbSession, simulation_id: str):
return StandardResponse(data=simulation, message="Data retrieved successfully")
@router.post("", response_model=StandardResponse[SimulationRead])
@router.post("", response_model=StandardResponse[SimulationBase])
async def create_simulation(
db_session: DbSession, simulation_in: SimulationCreate, current_user: CurrentUser
):
@ -56,7 +57,7 @@ async def create_simulation(
return StandardResponse(data=simulation, message="Data created successfully")
@router.post("/run", response_model=StandardResponse[SimulationRead])
@router.post("/run", response_model=StandardResponse[SimulationBase])
async def run_simulation_endpoint(
db_session: DbSession,
payload: SimulationRunPayload,
@ -80,7 +81,7 @@ async def run_simulation_endpoint(
)
@router.put("/{simulation_id}", response_model=StandardResponse[SimulationRead])
@router.put("/{simulation_id}", response_model=StandardResponse[SimulationBase])
async def update_simulation(
db_session: DbSession,
simulation_id: str,
@ -103,7 +104,7 @@ async def update_simulation(
return StandardResponse(data=updated_simulation, message="Data updated successfully")
@router.delete("/{simulation_id}", response_model=StandardResponse[SimulationRead])
@router.delete("/{simulation_id}", response_model=StandardResponse[SimulationBase])
async def delete_simulation(db_session: DbSession, simulation_id: str):
simulation = await get(db_session=db_session, simulation_id=simulation_id)

Loading…
Cancel
Save