From c8df9a4fa2fff47ed4a1ac35bc3a0ee0acd9ad55 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Wed, 12 Mar 2025 23:46:14 +0000 Subject: [PATCH] Graph RAG and document RAG working and configurable --- trustgraph-cli/scripts/tg-invoke-document-rag | 2 +- trustgraph-flow/trustgraph/gateway/document_rag.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/trustgraph-cli/scripts/tg-invoke-document-rag b/trustgraph-cli/scripts/tg-invoke-document-rag index 6a5443dc..759d4200 100755 --- a/trustgraph-cli/scripts/tg-invoke-document-rag +++ b/trustgraph-cli/scripts/tg-invoke-document-rag @@ -44,7 +44,7 @@ def main(): # ) parser.add_argument( - '-q', '--query', + '-q', '--question', required=True, help=f'Question to answer', ) diff --git a/trustgraph-flow/trustgraph/gateway/document_rag.py b/trustgraph-flow/trustgraph/gateway/document_rag.py index 9f1428b4..94d8f788 100644 --- a/trustgraph-flow/trustgraph/gateway/document_rag.py +++ b/trustgraph-flow/trustgraph/gateway/document_rag.py @@ -23,7 +23,7 @@ class DocumentRagRequestor(ServiceRequestor): query=body["query"], user=body.get("user", "trustgraph"), collection=body.get("collection", "default"), - doc_limit=body.get("doc-limit", 20), + doc_limit=int(body.get("doc-limit", 20)), ) def from_response(self, message):