fix

src/overhaul/router.py

@@ -3,7 +3,7 @@ from typing import List
 from fastapi import APIRouter, HTTPException, status
 
 from src.auth.service import Token
-from src.database.core import DbSession
+from src.database.core import CollectorDbSession, DbSession
 from src.models import StandardResponse
 from src.overhaul.service import (get_overhaul_critical_parts,
                                   get_overhaul_overview,
@@ -18,11 +18,11 @@ router = APIRouter()
 
 
 @router.get("", response_model=StandardResponse[OverhaulRead])
-async def get_overhaul(db_session: DbSession, token:Token):
+async def get_overhaul(db_session: DbSession, token:Token, collector_db_session:CollectorDbSession):
     """Get all scope pagination."""
     overview = await get_overhaul_overview(db_session=db_session)
     schedules = await get_overhaul_schedules(db_session=db_session)
-    criticalParts = await get_overhaul_critical_parts(db_session=db_session, session_id=overview["overhaul"]["id"], token=token)
+    criticalParts = await get_overhaul_critical_parts(db_session=db_session, session_id=overview["overhaul"]["id"], token=token, collector_db_session=collector_db_session)
     systemComponents = get_overhaul_system_components()
 
     return StandardResponse(

src/overhaul/service.py

@@ -55,11 +55,12 @@ async def get_simulation_results(*, simulation_id: str, token: str):
             "plant_result": plant_data
         }
 
-async def get_overhaul_critical_parts(db_session, session_id, token):
+async def get_overhaul_critical_parts(db_session, session_id, token, collector_db_session):
     """Get all overhaul critical parts."""
-    equipments, _ = await get_by_oh_session_id(
+    equipments = await get_standard_scope_by_session_id(
         db_session=db_session,
-        oh_session_id=session_id,
+        overhaul_session_id=session_id,
+        collector_db=collector_db_session
     )