diff --git a/trustgraph-base/trustgraph/api/async_flow.py b/trustgraph-base/trustgraph/api/async_flow.py index ba8b9598..68c07131 100644 --- a/trustgraph-base/trustgraph/api/async_flow.py +++ b/trustgraph-base/trustgraph/api/async_flow.py @@ -159,12 +159,12 @@ class AsyncFlowInstance: result = await self.request("text-completion", request_data) return result.get("response", "") - async def graph_rag(self, question: str, user: str, collection: str, + async def graph_rag(self, query: str, user: str, collection: str, max_subgraph_size: int = 1000, max_subgraph_count: int = 5, max_entity_distance: int = 3, **kwargs: Any) -> str: """Graph RAG (non-streaming, use async_socket for streaming)""" request_data = { - "question": question, + "query": query, "user": user, "collection": collection, "max-subgraph-size": max_subgraph_size, diff --git a/trustgraph-base/trustgraph/api/async_socket_client.py b/trustgraph-base/trustgraph/api/async_socket_client.py index 7c2a5aab..b0fd4fa7 100644 --- a/trustgraph-base/trustgraph/api/async_socket_client.py +++ b/trustgraph-base/trustgraph/api/async_socket_client.py @@ -208,12 +208,12 @@ class AsyncSocketFlowInstance: if hasattr(chunk, 'content'): yield chunk.content - async def graph_rag(self, question: str, user: str, collection: str, + async def graph_rag(self, query: str, user: str, collection: str, max_subgraph_size: int = 1000, max_subgraph_count: int = 5, max_entity_distance: int = 3, streaming: bool = False, **kwargs): """Graph RAG with optional streaming""" request = { - "question": question, + "query": query, "user": user, "collection": collection, "max-subgraph-size": max_subgraph_size, diff --git a/trustgraph-cli/trustgraph/cli/invoke_graph_rag.py b/trustgraph-cli/trustgraph/cli/invoke_graph_rag.py index 56c27585..5fa359ab 100644 --- a/trustgraph-cli/trustgraph/cli/invoke_graph_rag.py +++ b/trustgraph-cli/trustgraph/cli/invoke_graph_rag.py @@ -30,7 +30,7 @@ def question( try: response = flow.graph_rag( - question=question, + query=question, user=user, collection=collection, entity_limit=entity_limit, @@ -51,7 +51,7 @@ def question( # Use REST API for non-streaming flow = api.flow().id(flow_id) resp = flow.graph_rag( - question=question, + query=question, user=user, collection=collection, entity_limit=entity_limit,