From 216e42206e89ea656b71f15ee7a620286192c2a2 Mon Sep 17 00:00:00 2001 From: Cizz22 Date: Mon, 13 Apr 2026 12:24:24 +0700 Subject: [PATCH] feat: include simulation start and end timestamps in forecast response message --- src/aeros_simulation/router.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/aeros_simulation/router.py b/src/aeros_simulation/router.py index 10984bc..7c528ed 100644 --- a/src/aeros_simulation/router.py +++ b/src/aeros_simulation/router.py @@ -1,3 +1,4 @@ +from datetime import timedelta from collections import defaultdict from datetime import datetime import io @@ -529,6 +530,9 @@ async def get_forecast_eaf( # Using the helper from utils.py offset_hours = hours_between(oh_date, now) + #start date is oh + offset + start_date = oh_date + timedelta(hours=offset_hours) + # 4. Define Forecast Window duration = 168 if time_range == "weekly" else 720 @@ -578,7 +582,7 @@ async def get_forecast_eaf( "sim_duration_hours": duration, "reference_oh_date": oh_date_str }, - message="Forecast generated successfully" + message=f"Forecast generated successfully, start date: {start_date.strftime('%Y-%m-%d %H:%M:%S')}, end_date: {(start_date + timedelta(hours=duration)).strftime('%Y-%m-%d %H:%M:%S')}" ) except Exception as e: import logging