From 7a607bf90d91a40aa02e114e7f0c197e269d2481 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Wed, 7 Aug 2024 19:10:26 +0100
Subject: [PATCH] Keep 2 tests

---
 tests/test-graph-rag  |  5 +++--
 tests/test-graph-rag2 | 14 ++++++++++++++
 2 files changed, 17 insertions(+), 2 deletions(-)
 create mode 100755 tests/test-graph-rag2

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)
+