apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-26 08:56:21 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tidied scripts, added 2 query scripts (#53)

Browse source
This commit is contained in:
cybermaggedon 2024-09-05 16:45:22 +01:00 • committed by GitHub
parent 65d7f6d261
commit 6e4534e35c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 167 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

21
scripts/query
View file

@ -1,21 +0,0 @@
#!/usr/bin/env python3
from trustgraph.graph_rag import GraphRag
import sys
query = " ".join(sys.argv[1:])
gr = GraphRag(
verbose=True,
pulsar_host="pulsar://localhost:6650",
pr_request_queue="non-persistent://tg/request/prompt",
pr_response_queue="non-persistent://tg/response/prompt-response",
)
if query == "":
query="""This knowledge graph describes the Space Shuttle disaster.
Present 20 facts which are present in the knowledge graph."""
resp = gr.query(query)
print(resp)