From 64cfdbddcb65b8106a5a9ae291980249e8b2ab2e Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 9 Sep 2025 15:11:16 +0100 Subject: [PATCH] Fix missing document RAG user/collection params --- trustgraph-flow/trustgraph/retrieval/document_rag/rag.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/trustgraph-flow/trustgraph/retrieval/document_rag/rag.py b/trustgraph-flow/trustgraph/retrieval/document_rag/rag.py index 0cca2cff..2e5149c9 100755 --- a/trustgraph-flow/trustgraph/retrieval/document_rag/rag.py +++ b/trustgraph-flow/trustgraph/retrieval/document_rag/rag.py @@ -92,7 +92,12 @@ class Processor(FlowProcessor): else: doc_limit = self.doc_limit - response = await self.rag.query(v.query, doc_limit=doc_limit) + response = await self.rag.query( + v.query, + user=v.user, + collection=v.collection, + doc_limit=doc_limit + ) await flow("response").send( DocumentRagResponse(