Update alz_companion/agent.py

alz_companion/agent.py

@@ -661,13 +661,25 @@ def make_rag_chain(vs_general: FAISS, vs_personal: FAISS, *, for_evaluation: boo
 
         # --- NEW: DEDICATED LOGIC PATHS FOR RETRIEVAL ---
         if is_personal_route:
-            
-            # For personal queries, semantic search is unreliable. We retrieve ALL personal documents.
-            print("[DEBUG] Personal Memory Route Activated. Retrieving all personal documents.")
-            # --- START: FAISS-COMPATIBLE FIX ---
+            # --- START OF MODIFICATION ---
+            # 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:
-                all_retrieved_docs.extend(list(vs_personal.docstore._dict.values()))
-            # --- END: FAISS-COMPATIBLE FIX ---
+                # 1. Get all documents from the FAISS docstore
+                all_personal_docs = list(vs_personal.docstore._dict.values())
+
+                # 2. Filter this list to keep only text-based files
+                text_based_docs = []
+                text_extensions = ('.txt', '.jsonl')  # Define what counts as a text source
+                for doc in all_personal_docs:
+                    source = doc.metadata.get("source", "").lower()
+                    if source.endswith(text_extensions):
+                        text_based_docs.append(doc)
+                
+                # 3. Extend the final list with only the filtered, text-based documents
+                all_retrieved_docs.extend(text_based_docs)
+            # --- END OF MODIFICATION ---
             
         else:
             # For caregiving scenarios, use our powerful Multi-Stage Retrieval algorithm.