fix(rag): forward owner through manager wrapper (#2991)

This commit is contained in:
Vykos
2026-06-07 12:56:57 +02:00
committed by GitHub
parent 83fca6ac62
commit 2149f0fb67
2 changed files with 50 additions and 3 deletions
+38
View File
@@ -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",
}
]