apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-06 03:42:11 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Keep 2 tests

Browse source
This commit is contained in:
Cyber MacGeddon 2024-08-07 19:10:26 +01:00
parent c4a218b6f5
commit 7a607bf90d
2 changed files with 17 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/test-graph-rag
View file

@ -5,8 +5,9 @@ from trustgraph.graph_rag_client import GraphRagClient
rag = GraphRagClient(pulsar_host="pulsar://localhost:6650") 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) resp = rag.request(query)

14
tests/test-graph-rag2 Executable file
View file

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