From 2149f0fb676e8cbe329446f4d9850ce48e0012c0 Mon Sep 17 00:00:00 2001
From: Vykos <illjaesterhazy@gmail.com>
Date: Sun, 7 Jun 2026 12:56:57 +0200
Subject: [PATCH] fix(rag): forward owner through manager wrapper (#2991)

---
 src/rag_manager.py                     | 15 ++++++++--
 tests/test_rag_manager_owner_compat.py | 38 ++++++++++++++++++++++++++
 2 files changed, 50 insertions(+), 3 deletions(-)
 create mode 100644 tests/test_rag_manager_owner_compat.py

diff --git a/src/rag_manager.py b/src/rag_manager.py
index 87f370472..8a7767761 100644
--- a/src/rag_manager.py
+++ b/src/rag_manager.py
@@ -5,7 +5,7 @@ A thin wrapper around VectorRAG for backward compatibility and additional featur
 """
 
 import logging
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Optional
 
 # Try to import from different possible locations
 try:
@@ -34,9 +34,18 @@ class RAGManager:
         """Search for documents - delegates to VectorRAG."""
         return self.vector_rag.search(query, k)
     
-    def index_personal_documents(self, directory: str) -> Dict[str, Any]:
+    def index_personal_documents(
+        self,
+        directory: str,
+        file_extensions: Optional[set] = None,
+        owner: Optional[str] = None,
+    ) -> Dict[str, Any]:
         """Index documents - delegates to VectorRAG."""
-        return self.vector_rag.index_personal_documents(directory)
+        return self.vector_rag.index_personal_documents(
+            directory,
+            file_extensions=file_extensions,
+            owner=owner,
+        )
     
     def retrieve(self, query: str, k: int = 5) -> List[str]:
         """Retrieve relevant chunks - delegates to VectorRAG."""
diff --git a/tests/test_rag_manager_owner_compat.py b/tests/test_rag_manager_owner_compat.py
new file mode 100644
index 000000000..8bc925371
--- /dev/null
+++ b/tests/test_rag_manager_owner_compat.py
@@ -0,0 +1,38 @@
+from src.rag_manager import RAGManager
+
+
+class _FakeVectorRAG:
+    def __init__(self):
+        self.calls = []
+
+    def index_personal_documents(self, directory, file_extensions=None, owner=None):
+        self.calls.append(
+            {
+                "directory": directory,
+                "file_extensions": file_extensions,
+                "owner": owner,
+            }
+        )
+        return {"success": True, "indexed_count": 1}
+
+
+def test_rag_manager_forwards_owner_and_file_extensions():
+    fake = _FakeVectorRAG()
+    manager = RAGManager.__new__(RAGManager)
+    manager.vector_rag = fake
+    extensions = {".md", ".txt"}
+
+    result = manager.index_personal_documents(
+        "/tmp/personal",
+        file_extensions=extensions,
+        owner="alice",
+    )
+
+    assert result == {"success": True, "indexed_count": 1}
+    assert fake.calls == [
+        {
+            "directory": "/tmp/personal",
+            "file_extensions": extensions,
+            "owner": "alice",
+        }
+    ]