Update src/rag_engine.py

src/rag_engine.py

@@ -129,10 +129,21 @@ def load_documents_from_directory(
     
     return all_docs
 
-# Quick test block
-if __name__ == "__main__":
-    # Example usage
-    print("--- Testing Rag Engine ---")
-    # You can point this to a dummy file to test
-    # docs = process_file("test_data/navy_manual.pdf", chunking_strategy="paragraph")
-    # print(f"Loaded {len(docs)} chunks.")
+def list_documents(username: str = "default") -> List[str]:
+    """
+    Lists all supported documents for a specific user.
+    Adjust 'source_documents' if your folder is named differently.
+    """
+    # Define your source directory (Update this path if you use a different one!)
+    base_dir = "source_documents" 
+    user_dir = os.path.join(base_dir, username)
+    
+    if not os.path.exists(user_dir):
+        return []
+        
+    files = []
+    for f in os.listdir(user_dir):
+        if f.lower().endswith(('.pdf', '.txt', '.md')):
+            files.append(f)
+            
+    return files