Graph RAG and document RAG working and configurable

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',
     )

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):