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

Change document-rag and graph-rag processing so that the user can

Browse source 
specify parameters.  Changes in Pulsar services, Pulsar message
schemas, gateway and command-line tools.  User-visible changes in
new parameters on command-line tools.
This commit is contained in:
Cyber MacGeddon 2025-03-12 17:51:47 +00:00
parent f1559c5944
commit 8830e99e57
12 changed files with 154 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

33
trustgraph-cli/scripts/tg-invoke-graph-rag
View file

@ -12,12 +12,18 @@ default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
default_user = 'trustgraph'
default_collection = 'default'
def question(url, question, user, collection):
def question(
url, question, user, collection, entity_limit, triple_limit,
subgraph_limit
):
rag = Api(url)
# user=user, collection=collection,
resp = rag.graph_rag(question=question)
resp = rag.graph_rag(
question=question, user=user, collection=collection,
entity_limit=entity_limit, triple_limit=triple_limit,
subgraph_limit=subgraph_limit
)
print(resp)
@ -52,6 +58,24 @@ def main():
help=f'Collection ID (default: {default_collection})'
)
parser.add_argument(
'-e', '--entity-limit',
default=default_entity_limit,
help=f'Entity limit (default: {default_entity_limit})'
)
parser.add_argument(
'-t', '--triple-limit',
default=default_triple_limit,
help=f'Triple limit (default: {default_triple_limit})'
)
parser.add_argument(
'-s', '--subgraph-limit',
default=default_subgraph_limit,
help=f'Subgraph limit (default: {subgraph_limit})'
)
args = parser.parse_args()
try:
@ -61,6 +85,9 @@ def main():
question=args.question,
user=args.user,
collection=args.collection,
entity_limit=args.entity_limit,
triple_limit=args.triple_limit,
subgraph_limit=args.subgraph_limit,
)
except Exception as e: