trustgraph/tests/test-graph-rag

#!/usr/bin/env python3

import pulsar
from trustgraph.clients.graph_rag_client import GraphRagClient

rag = GraphRagClient(pulsar_host="pulsar://localhost:6650")

query="""
This knowledge graph describes the Space Shuttle disaster.
Present 20 facts which are present in the knowledge graph."""

resp = rag.request(query)

print(resp)