#!/usr/bin/env python3

from trustgraph.graph_rag import GraphRag
import sys

query = " ".join(sys.argv[1:])

gr = GraphRag(verbose=True)

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)