From 5fb88f78bc72d6b6d8970d7949f38abd4e955598 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Sun, 29 Sep 2024 20:37:06 +0100 Subject: [PATCH] Fix queue name invocation, use correct names, not defaults --- trustgraph/document_rag.py | 4 ++-- trustgraph/graph_rag.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/trustgraph/document_rag.py b/trustgraph/document_rag.py index 61da645b..f3c8b158 100644 --- a/trustgraph/document_rag.py +++ b/trustgraph/document_rag.py @@ -73,8 +73,8 @@ class DocumentRag: self.lang = PromptClient( pulsar_host=pulsar_host, - input_queue=prompt_request_queue, - output_queue=prompt_response_queue, + input_queue=pr_request_queue, + output_queue=pr_response_queue, subscriber=module + "-de-prompt", ) diff --git a/trustgraph/graph_rag.py b/trustgraph/graph_rag.py index c44066b4..15acb609 100644 --- a/trustgraph/graph_rag.py +++ b/trustgraph/graph_rag.py @@ -96,8 +96,8 @@ class GraphRag: self.lang = PromptClient( pulsar_host=pulsar_host, - input_queue=prompt_request_queue, - output_queue=prompt_response_queue, + input_queue=pr_request_queue, + output_queue=pr_response_queue, subscriber=module + "-prompt", )