Update alz_companion/agent.py

alz_companion/agent.py

@@ -661,14 +661,14 @@ 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.")
-            if vs_personal and vs_personal._collection.count() > 0:
-                personal_docs_data = vs_personal.get(include=["metadatas", "documents"])
-                all_retrieved_docs.extend([
-                    Document(page_content=doc_content, metadata=metadata)
-                    for doc_content, metadata in zip(personal_docs_data['documents'], personal_docs_data['metadatas'])
-                ])
+            # --- START: FAISS-COMPATIBLE FIX ---
+            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 ---
+            
         else:
             # For caregiving scenarios, use our powerful Multi-Stage Retrieval algorithm.
             print("[DEBUG] Using Multi-Stage Retrieval for caregiving scenario...")