fix: relax database Ref-id checks

src/tools/navigation_toolkit.py

@@ -3,7 +3,10 @@ import json
 from agno.tools import Toolkit
 from src.services.googlemap_api_service import GoogleMapAPIService
 from src.infra.poi_repository import poi_repo
+from src.infra.context import get_session_id
+from src.infra.logger import get_logger
 
+logger = get_logger(__name__)
 
 class NavigationToolkit(Toolkit):
     def __init__(self, google_maps_api_key: str = None):
@@ -18,6 +21,16 @@ class NavigationToolkit(Toolkit):
         """
         print(f"🚗 Navigator: Loading Ref {optimization_ref_id}...")
         data = poi_repo.load(optimization_ref_id)
+
+        if not data:
+            logger.warning(f"⚠️ Warning: Ref ID '{optimization_ref_id}' not found.")
+            session_id = get_session_id()
+            if session_id:
+                latest_id = poi_repo.get_last_id_by_session(session_id)
+                if latest_id and latest_id != optimization_ref_id and latest_id.startswith("optimization"):
+                    logger.warning(f"🔄 Auto-Correcting: Switching to latest Session ID: {latest_id}")
+                    data = poi_repo.load(latest_id)
+
         if not data:
             return "❌ Error: Data not found."
 

src/tools/optimizer_toolkit.py

@@ -19,7 +19,8 @@ class OptimizationToolkit(Toolkit):
         """
         print(f"🧮 Optimizer: Fetching data for {ref_id}...")
         data = poi_repo.load(ref_id)
-        if not data: return "❌ Error: Data not found."
+        if not data:
+            return "❌ Error: Data not found."
 
         tasks = data.get("tasks", [])
 

src/tools/weather_toolkit.py

@@ -3,6 +3,7 @@ import json
 from agno.tools import Toolkit
 from src.services.openweather_service import OpenWeatherMapService
 from src.infra.poi_repository import poi_repo
+from src.infra.context import get_session_id
 from src.infra.logger import get_logger
 
 logger = get_logger(__name__)
@@ -17,7 +18,17 @@ class WeatherToolkit(Toolkit):
     def check_weather_for_timeline(self, nav_ref_id: str) -> str:
         logger.debug(f"🌤️ Weatherman: Loading Ref {nav_ref_id}...")
         data = poi_repo.load(nav_ref_id)
-        if not data: return "❌ Error: Data not found."
+        if not data:
+            logger.warning(f"⚠️ Warning: Ref ID '{nav_ref_id}' not found.")
+            session_id = get_session_id()
+            if session_id:
+                latest_id = poi_repo.get_last_id_by_session(session_id)
+                if latest_id and latest_id != nav_ref_id and latest_id.startswith("navigation"):
+                    logger.warning(f"🔄 Auto-Correcting: Switching to latest Session ID: {latest_id}")
+                    data = poi_repo.load(latest_id)
+
+        if not data:
+            return "❌ Error: Data not found."
 
         # [CRITICAL] 複製原始數據，確保 global_info, tasks 等不會遺失
         final_data = data.copy()