mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 16:36:21 +02:00
22 lines
546 B
Python
Executable file
22 lines
546 B
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import pulsar
|
|
from trustgraph.clients.graph_rag_client import GraphRagClient
|
|
|
|
rag = GraphRagClient(
|
|
pulsar_host="pulsar://localhost:6650",
|
|
subscriber="test1",
|
|
input_queue = "non-persistent://tg/request/graph-rag:default",
|
|
output_queue = "non-persistent://tg/response/graph-rag:default",
|
|
)
|
|
|
|
#query="""
|
|
#This knowledge graph describes the Space Shuttle disaster.
|
|
#Present 20 facts which are present in the knowledge graph."""
|
|
|
|
query = "How many cats does Mark have?"
|
|
|
|
resp = rag.request(query)
|
|
|
|
print(resp)
|
|
|