diff --git a/src/aeros_simulation/service.py b/src/aeros_simulation/service.py index 2b13bba..04688bf 100644 --- a/src/aeros_simulation/service.py +++ b/src/aeros_simulation/service.py @@ -25,6 +25,7 @@ from .model import ( from src.aeros_equipment.model import AerosEquipment, AerosEquipmentCustomParameterData from src.aeros_equipment.schema import EquipmentWithCustomParameters from .schema import SimulationInput, SimulationRankingParameters +from .utils import calculate_eaf client = httpx.AsyncClient(timeout=300.0) active_simulations = {} @@ -131,15 +132,10 @@ async def get_or_save_node(*, db_session: DbSession, node_data: dict, type: str return node -async def execute_simulation( - *, - db_session: DbSession, - simulation_id: Optional[UUID] = None, - sim_data: dict, - is_saved: bool = False, - eq_update: dict = {}, -): +async def execute_simulation(*, db_session: DbSession, simulation_id: Optional[UUID] = None, sim_data: dict, is_saved: bool = False, eq_update: dict = None): """Execute the actual simulation call""" + if eq_update is None: + eq_update = {} print("Executing simulation with id: %s", simulation_id, sim_data["SchematicName"]) try: @@ -177,7 +173,7 @@ async def execute_simulation( raise HTTPException( status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e) - ) + ) from e async def get_all_aeros_node(*, db_session: DbSession, schematic_name: Optional[str] = None):