2024-07-10 23:20:06 +01:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
|
|
|
|
|
from trustgraph.graph_rag import GraphRag
|
|
|
|
|
import sys
|
|
|
|
|
|
|
|
|
|
query = " ".join(sys.argv[1:])
|
|
|
|
|
|
2024-07-15 21:48:51 +01:00
|
|
|
gr = GraphRag(
|
|
|
|
|
verbose=True,
|
|
|
|
|
pulsar_host="pulsar://localhost:6650",
|
2024-08-13 09:13:17 +01:00
|
|
|
completion_request_queue="non-persistent://tg/request/text-completion-rag",
|
|
|
|
|
completion_response_queue="non-persistent://tg/response/text-completion-rag-response",
|
2024-07-15 21:48:51 +01:00
|
|
|
)
|
2024-07-10 23:20:06 +01:00
|
|
|
|
|
|
|
|
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)
|
|
|
|
|
|
