diff --git a/tests/test-graph-rag b/tests/test-graph-rag index 4837d3bf..c2f629b1 100755 --- a/tests/test-graph-rag +++ b/tests/test-graph-rag @@ -5,8 +5,9 @@ from trustgraph.graph_rag_client import GraphRagClient rag = GraphRagClient(pulsar_host="pulsar://localhost:6650") -query="""List 20 key points to describe the research that led to the discovery of Leo VI. -""" +query=""" +Identify any facts which provide an explanation of the explosion of the +space shuttle rocket boosters""" resp = rag.request(query) diff --git a/tests/test-graph-rag2 b/tests/test-graph-rag2 new file mode 100755 index 00000000..4837d3bf --- /dev/null +++ b/tests/test-graph-rag2 @@ -0,0 +1,14 @@ +#!/usr/bin/env python3 + +import pulsar +from trustgraph.graph_rag_client import GraphRagClient + +rag = GraphRagClient(pulsar_host="pulsar://localhost:6650") + +query="""List 20 key points to describe the research that led to the discovery of Leo VI. +""" + +resp = rag.request(query) + +print(resp) +