From fc2fcb2b975b208d985dfc815943634920e7e7e2 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Wed, 17 Dec 2025 17:31:01 +0000
Subject: [PATCH] Tidy inconsistent parameters discovered while modifying the
 schema

---
 trustgraph-base/trustgraph/api/flow.py          | 8 ++++----
 trustgraph-base/trustgraph/api/socket_client.py | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/trustgraph-base/trustgraph/api/flow.py b/trustgraph-base/trustgraph/api/flow.py
index f7cda5c2..744ad2e7 100644
--- a/trustgraph-base/trustgraph/api/flow.py
+++ b/trustgraph-base/trustgraph/api/flow.py
@@ -160,14 +160,14 @@ class FlowInstance:
         )["answer"]
 
     def graph_rag(
-            self, question, user="trustgraph", collection="default",
+            self, query, user="trustgraph", collection="default",
             entity_limit=50, triple_limit=30, max_subgraph_size=150,
             max_path_length=2,
     ):
 
         # The input consists of a question
         input = {
-            "query": question,
+            "query": query,
             "user": user,
             "collection": collection,
             "entity-limit": entity_limit,
@@ -182,13 +182,13 @@ class FlowInstance:
         )["response"]
 
     def document_rag(
-            self, question, user="trustgraph", collection="default",
+            self, query, user="trustgraph", collection="default",
             doc_limit=10,
     ):
 
         # The input consists of a question
         input = {
-            "query": question,
+            "query": query,
             "user": user,
             "collection": collection,
             "doc-limit": doc_limit,
diff --git a/trustgraph-base/trustgraph/api/socket_client.py b/trustgraph-base/trustgraph/api/socket_client.py
index c0fd9cd9..b1be0195 100644
--- a/trustgraph-base/trustgraph/api/socket_client.py
+++ b/trustgraph-base/trustgraph/api/socket_client.py
@@ -284,7 +284,7 @@ class SocketFlowInstance:
 
     def graph_rag(
         self,
-        question: str,
+        query: str,
         user: str,
         collection: str,
         max_subgraph_size: int = 1000,
@@ -295,7 +295,7 @@ class SocketFlowInstance:
     ) -> Union[str, Iterator[str]]:
         """Graph RAG with optional streaming"""
         request = {
-            "question": question,
+            "query": query,
             "user": user,
             "collection": collection,
             "max-subgraph-size": max_subgraph_size,
@@ -316,7 +316,7 @@ class SocketFlowInstance:
 
     def document_rag(
         self,
-        question: str,
+        query: str,
         user: str,
         collection: str,
         doc_limit: int = 10,
@@ -325,7 +325,7 @@ class SocketFlowInstance:
     ) -> Union[str, Iterator[str]]:
         """Document RAG with optional streaming"""
         request = {
-            "question": question,
+            "query": query,
             "user": user,
             "collection": collection,
             "doc-limit": doc_limit,