trustgraph/tests/test-graph-rag

16 lines
335 B
Text
Raw Normal View History

2024-07-10 23:20:06 +01:00
#!/usr/bin/env python3
import pulsar
from trustgraph.clients.graph_rag_client import GraphRagClient
2024-07-10 23:20:06 +01:00
rag = GraphRagClient(pulsar_host="pulsar://localhost:6650")
2024-07-15 18:58:03 +01:00
query="""
This knowledge graph describes the Space Shuttle disaster.
Present 20 facts which are present in the knowledge graph."""
2024-07-10 23:20:06 +01:00
resp = rag.request(query)
print(resp)