diff --git a/trustgraph/retrieval/graph_rag/rag.py b/trustgraph/retrieval/graph_rag/rag.py index 993557a0..80cbfafb 100755 --- a/trustgraph/retrieval/graph_rag/rag.py +++ b/trustgraph/retrieval/graph_rag/rag.py @@ -52,6 +52,7 @@ class Processor(ConsumerProducer): entity_limit=entity_limit, triple_limit=triple_limit, max_subgraph_size=max_subgraph_size, + module=module, ) def handle(self, msg):