diff --git a/src/aeros_simulation/simulation_save_service.py b/src/aeros_simulation/simulation_save_service.py index 2275d50..cf15c79 100644 --- a/src/aeros_simulation/simulation_save_service.py +++ b/src/aeros_simulation/simulation_save_service.py @@ -40,8 +40,12 @@ async def execute_simulation( ) response.raise_for_status() with open(tmpfile, "wb") as f: - for chunk in response.iter_content(chunk_size=8192): - f.write(chunk) + if hasattr(response, "iter_content"): + for chunk in response.iter_content(chunk_size=8192): + f.write(chunk) + else: + for chunk in response.iter_bytes(chunk_size=8192): + f.write(chunk) if not is_saved: # If not saving to DB, just return parsed JSON