From 2d3e32a0456923c6d17b7c034c7c13073da77d7d Mon Sep 17 00:00:00 2001
From: Cizz22 <cisatraa@gmail.com>
Date: Thu, 2 Oct 2025 10:22:23 +0700
Subject: [PATCH] fix

---
 src/aeros_equipment/service.py | 8 ++++++--
 src/aeros_simulation/schema.py | 2 +-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/aeros_equipment/service.py b/src/aeros_equipment/service.py
index afbd4e8..4ed3651 100644
--- a/src/aeros_equipment/service.py
+++ b/src/aeros_equipment/service.py
@@ -373,6 +373,7 @@ async def update_equipment_for_simulation(*, db_session: DbSession, project_name
                     eq["relDisP1"] = float(custom_param["failure_rate"])
                     eq["relDisP2"] = 0
                     eq["ohDisP1"] = overhaul_duration
+                    eq["ohDisUnitCode"] = "UHour"
 
                     reqNodeInputs.append(eq)
                     results[eq["equipmentName"]] = {
@@ -380,7 +381,8 @@ async def update_equipment_for_simulation(*, db_session: DbSession, project_name
                         "distribution": eq["relDisType"],
                         "beta": eq["relDisP1"],
                         "eta": 0,
-                        "parameters": {}
+                        "parameters": {},
+                        "oh_duration": overhaul_duration
                     }
                     continue
 
@@ -389,6 +391,7 @@ async def update_equipment_for_simulation(*, db_session: DbSession, project_name
                 eq["relDisP1"] = reliabiility.get("relDisP1", 0)
                 eq["relDisP2"] = reliabiility.get("relDisP2", 0)
                 eq["ohDisP1"] = overhaul_duration
+                eq["ohDisUnitCode"] = "UHour"
 
                 reqNodeInputs.append(eq)
                 results[eq["equipmentName"]] = {
@@ -396,7 +399,8 @@ async def update_equipment_for_simulation(*, db_session: DbSession, project_name
                     "distribution": eq["relDisType"],
                     "beta": eq["relDisP1"],
                     "eta": eq["relDisP2"],
-                    "parameters": eq.get("parameters", {})
+                    "parameters": eq.get("parameters", {}),
+                    "oh_duration": overhaul_duration
                 }
             except Exception as e:
                 print(f"Error fetching data for {eq['equipmentName']}: {e}")
diff --git a/src/aeros_simulation/schema.py b/src/aeros_simulation/schema.py
index 958cd7b..0e25d0b 100644
--- a/src/aeros_simulation/schema.py
+++ b/src/aeros_simulation/schema.py
@@ -22,7 +22,7 @@ class SimulationInput(BaseModel):
     MaintenanceOutages:  Optional[int] = 0
     PlannedOutages:  Optional[int] = 0
     OverhaulInterval:  Optional[int] = Field(0)
-    OverhaulDuration:  Optional[int] = Field(50)
+    OverhaulDuration:  Optional[int] = Field(1200)
 
 class SimulationNode(BaseModel):
     id: UUID