From 2e05a326a1c2d92fdc33f19bb7fb8e9575683044 Mon Sep 17 00:00:00 2001 From: Cizz22 Date: Fri, 10 Apr 2026 12:36:03 +0700 Subject: [PATCH] fix: add support for iter_bytes when iter_content is unavailable in simulation save service --- src/aeros_simulation/simulation_save_service.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) 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