diff --git a/trustgraph-base/trustgraph/api/async_flow.py b/trustgraph-base/trustgraph/api/async_flow.py
index 68c07131..5d3cd486 100644
--- a/trustgraph-base/trustgraph/api/async_flow.py
+++ b/trustgraph-base/trustgraph/api/async_flow.py
@@ -177,11 +177,11 @@ class AsyncFlowInstance:
         result = await self.request("graph-rag", request_data)
         return result.get("response", "")
 
-    async def document_rag(self, question: str, user: str, collection: str,
+    async def document_rag(self, query: str, user: str, collection: str,
                            doc_limit: int = 10, **kwargs: Any) -> str:
         """Document RAG (non-streaming, use async_socket for streaming)"""
         request_data = {
-            "question": question,
+            "query": query,
             "user": user,
             "collection": collection,
             "doc-limit": doc_limit,
diff --git a/trustgraph-base/trustgraph/api/async_socket_client.py b/trustgraph-base/trustgraph/api/async_socket_client.py
index b0fd4fa7..cb6c8605 100644
--- a/trustgraph-base/trustgraph/api/async_socket_client.py
+++ b/trustgraph-base/trustgraph/api/async_socket_client.py
@@ -235,11 +235,11 @@ class AsyncSocketFlowInstance:
             if hasattr(chunk, 'content'):
                 yield chunk.content
 
-    async def document_rag(self, question: str, user: str, collection: str,
+    async def document_rag(self, query: str, user: str, collection: str,
                            doc_limit: int = 10, streaming: bool = False, **kwargs):
         """Document RAG with optional streaming"""
         request = {
-            "question": question,
+            "query": query,
             "user": user,
             "collection": collection,
             "doc-limit": doc_limit,
diff --git a/trustgraph-cli/trustgraph/cli/invoke_document_rag.py b/trustgraph-cli/trustgraph/cli/invoke_document_rag.py
index d30f05ac..7e88bdc4 100644
--- a/trustgraph-cli/trustgraph/cli/invoke_document_rag.py
+++ b/trustgraph-cli/trustgraph/cli/invoke_document_rag.py
@@ -24,7 +24,7 @@ def question(url, flow_id, question, user, collection, doc_limit, streaming=True
 
         try:
             response = flow.document_rag(
-                question=question,
+                query=question,
                 user=user,
                 collection=collection,
                 doc_limit=doc_limit,
@@ -42,7 +42,7 @@ def question(url, flow_id, question, user, collection, doc_limit, streaming=True
         # Use REST API for non-streaming
         flow = api.flow().id(flow_id)
         resp = flow.document_rag(
-            question=question,
+            query=question,
             user=user,
             collection=collection,
             doc_limit=doc_limit,