diff --git a/ai-hub/app/core/services/preference.py b/ai-hub/app/core/services/preference.py index 4ccdffb..268f8b0 100644 --- a/ai-hub/app/core/services/preference.py +++ b/ai-hub/app/core/services/preference.py @@ -56,6 +56,11 @@ merged_statuses = copy.deepcopy(system_statuses) merged_statuses.update(user_statuses) + # M6: Propagate personal statuses to their suffixed IDs (e.g. llm_gemini -> llm_gemini_personal) + # This ensures the UI shows green/red dots for personal accounts correctly. + for k, v in user_statuses.items(): + merged_statuses[f"{k}_personal"] = v + def is_provider_healthy(section: str, provider_id: str, p_data: dict = None) -> bool: # M6: Check health using either the full ID or the base ID (in case of personal suffix) status_key = f"{section}_{provider_id}"