Update alz_companion/agent.py

alz_companion/agent.py

@@ -705,34 +705,36 @@ def make_rag_chain(vs_general: FAISS, vs_personal: FAISS, *, for_evaluation: boo
             # Implements the simple fallback logic for personal routes.
             # the logic of it always returns a personal doc unless it's not loaded with personal memory
             if filtered_personal_docs:
+                print("[DEBUG] Factual/Sum/Multi: Prioritizing personal docs.")
                 all_retrieved_docs = filtered_personal_docs
             else:
+                print("[DEBUG] Factual/Sum/Multi: Prioritizing general docs.")
                 all_retrieved_docs = filtered_general_docs
             # --- END OF MODIFICATION ---
         else: # caregiving_scenario
             if disease_stage in ["Moderate Stage", "Advanced Stage"]:
                 # --- STAGE-AWARE LOGIC FOR CAREGIVING SCENARIOS ---
                 if filtered_general_docs:
-                    print("[DEBUG] Stage-Aware Logic: Prioritizing general documents.")
+                    print("[DEBUG] Moderate / Advanced: Prioritizing general documents.")
                     all_retrieved_docs = filtered_general_docs
                 elif filtered_personal_docs:
-                    print("[DEBUG] Stage-Aware Logic: Falling back to personal documents.")
+                    print("[DEBUG] Moderate / Advanced: Falling back to personal documents.")
                     all_retrieved_docs = filtered_personal_docs
                 else:
-                    print("[DEBUG] Stage-Aware Logic: No relevant documents found.")
+                    print("[DEBUG] Moderate / Advanced: No relevant documents found.")
                     all_retrieved_docs = []
                 # --- END STAGE-AWARE BLOCK ---
             else:
                 # --- NORMAL ROUTING LOGIC ---
                 # Conditional Blending logic for caregiving remains.
                 if abs(best_personal_score - best_general_score) <= SCORE_MARGIN:
-                    print("[DEBUG] Normal Logic: Blending personal and general docs (scores are close).")
+                    print("[DEBUG] Caregiving Case: Blending personal and general docs (scores are close).")
                     all_retrieved_docs = list({doc.page_content: doc for doc in filtered_personal_docs + filtered_general_docs}.values())[:4]
                 elif best_personal_score < best_general_score:
-                    print("[DEBUG] Normal Logic: Prioritizing personal docs (better score).")
+                    print("[DEBUG] Caregiving Case: Prioritizing personal docs (better score).")
                     all_retrieved_docs = filtered_personal_docs
                 else:
-                    print("[DEBUG] Normal Logic: Prioritizing general docs (better score).")
+                    print("[DEBUG] Caregiving Case: Prioritizing general docs (better score).")
                     all_retrieved_docs = filtered_general_docs
         
         # --- Prompt Generation and LLM Call ---