Update alz_companion/agent.py

alz_companion/agent.py

@@ -712,7 +712,8 @@ def make_rag_chain(vs_general: FAISS, vs_personal: FAISS, *, for_evaluation: boo
                 # all_personal_docs = list(vs_personal.docstore._dict.values())
 
                 # 2. Filter this list to keep only text-based files
-                text_based_docs = []
+                # ORIG: text_based_docs = []
+                text_based_results = []
                 text_extensions = ('.txt', '.jsonl')  # Define what counts as a text source
                 # ORIG: for doc in all_personal_docs:
                 for doc, score in initial_results:
@@ -737,6 +738,7 @@ def make_rag_chain(vs_general: FAISS, vs_personal: FAISS, *, for_evaluation: boo
                 # Select the final 5 (parameter tuning) documents for the context.
                 final_personal_docs = [doc for doc, score in text_based_results[:5]]
                 all_retrieved_docs.extend(final_personal_docs)
+                
                 # ORIG code
                 # all_retrieved_docs.extend(text_based_docs)
             # --- END OF MODIFICATION ---