apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 12:41:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Tidy inconsistent parameters discovered while modifying the schema

Browse source
This commit is contained in:
Cyber MacGeddon 2025-12-17 17:31:01 +00:00
parent f19363c640
commit fc2fcb2b97
2 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

8
trustgraph-base/trustgraph/api/flow.py
View file

@ -160,14 +160,14 @@ class FlowInstance:
)["answer"] )["answer"]
def graph_rag( 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, entity_limit=50, triple_limit=30, max_subgraph_size=150,
max_path_length=2, max_path_length=2,
): ):
# The input consists of a question # The input consists of a question
input = { input = {
"query": question, "query": query,
"user": user, "user": user,
"collection": collection, "collection": collection,
"entity-limit": entity_limit, "entity-limit": entity_limit,
@ -182,13 +182,13 @@ class FlowInstance:
)["response"] )["response"]
def document_rag( def document_rag(
self, question, user="trustgraph", collection="default", self, query, user="trustgraph", collection="default",
doc_limit=10, doc_limit=10,
): ):
# The input consists of a question # The input consists of a question
input = { input = {
"query": question, "query": query,
"user": user, "user": user,
"collection": collection, "collection": collection,
"doc-limit": doc_limit, "doc-limit": doc_limit,

8
trustgraph-base/trustgraph/api/socket_client.py
View file

@ -284,7 +284,7 @@ class SocketFlowInstance:
def graph_rag( def graph_rag(
self, self,
question: str, query: str,
user: str, user: str,
collection: str, collection: str,
max_subgraph_size: int = 1000, max_subgraph_size: int = 1000,
@ -295,7 +295,7 @@ class SocketFlowInstance:
) -> Union[str, Iterator[str]]: ) -> Union[str, Iterator[str]]:
"""Graph RAG with optional streaming""" """Graph RAG with optional streaming"""
request = { request = {
"question": question, "query": query,
"user": user, "user": user,
"collection": collection, "collection": collection,
"max-subgraph-size": max_subgraph_size, "max-subgraph-size": max_subgraph_size,
@ -316,7 +316,7 @@ class SocketFlowInstance:
def document_rag( def document_rag(
self, self,
question: str, query: str,
user: str, user: str,
collection: str, collection: str,
doc_limit: int = 10, doc_limit: int = 10,
@ -325,7 +325,7 @@ class SocketFlowInstance:
) -> Union[str, Iterator[str]]: ) -> Union[str, Iterator[str]]:
"""Document RAG with optional streaming""" """Document RAG with optional streaming"""
request = { request = {
"question": question, "query": query,
"user": user, "user": user,
"collection": collection, "collection": collection,
"doc-limit": doc_limit, "doc-limit": doc_limit,