Update alz_companion/agent.py

alz_companion/agent.py

@@ -116,6 +116,10 @@ FACTUAL_KEYPHRASES = [
     r"\b(where is|where was) my\b",
     r"\b(how old am i)\b",
     # r"\b(when did|what did) the journal say\b"
+    # NEW below to handle what is/are movdies/videos separating from songs/music
+    r"\b(what|who|where|when|which)\b.*(is|are|was|were|am)\b.*\b(my|i|me|our)\b",
+    r"\b(do you remember|tell me about|what do you know about)\b.*\b(my|i|me|our)\b",
+    r"\b(my|our)\b.*\bfavorite\b"
 ]
 _FQ_PATTERNS = [re.compile(p, re.IGNORECASE) for p in FACTUAL_KEYPHRASES]
 
@@ -459,6 +463,14 @@ def route_query_type(query: str, severity: str = "Normal / Unspecified"):
         print(f"Query classified as: {sum_hit} (summarization pre-router)")
         return sum_hit
 
+    # --- START: ADDED FACTUAL CHECK ---
+    # Priority 3: Check for personal factual questions.
+    factual_hit = _pre_router_factual(query)
+    if factual_hit:
+        print(f"Query classified as: {factual_hit} (factual pre-router)")
+        return factual_hit
+    # --- END: ADDED FACTUAL CHECK ---
+
     # Priority 4: Check for music requests.
     # NEW Add Music Support before care_hit = _pre_router(query)
     # the general "caregiving" keyword checker (_pre_router) is called before 
@@ -479,7 +491,7 @@ def route_query_type(query: str, severity: str = "Normal / Unspecified"):
     query_type = _llm_route_with_prompt(query, temperature=0.0)
     print(f"Query classified as: {query_type} (LLM router)")
     return query_type
-
+# END route_query_type
 
 # helper: put near other small utils in agent.py
 # In agent.py, replace the _source_ids_for_eval function
@@ -665,6 +677,7 @@ def make_rag_chain(vs_general: FAISS, vs_personal: FAISS, *, for_evaluation: boo
             # This logic retrieves all documents from the personal FAISS store and then
             # filters them to include ONLY text-based sources, excluding media files.
             print("[DEBUG] Personal Memory Route Activated. Retrieving all personal text documents...")
+            
             if vs_personal and vs_personal.docstore and len(vs_personal.index_to_docstore_id) > 0:
                 # 1. Get all documents from the FAISS docstore
                 all_personal_docs = list(vs_personal.docstore._dict.values())