Rework graph RAG implementation to work with new services

This commit is contained in:
Cyber MacGeddon 2024-08-12 21:56:34 +01:00
parent 72f36b34b2
commit 8eaaab2d16
3 changed files with 85 additions and 65 deletions

View file

@ -7,7 +7,7 @@ Connects to the trustgraph graph hosts and dumps all graph edges.
import argparse
import time
from trustgraph.trustgraph import TrustGraph
from trustgraph.direct.cassandra import TrustGraph
def show_graph(graph_hosts):

View file

@ -7,9 +7,7 @@ query = " ".join(sys.argv[1:])
gr = GraphRag(
verbose=True,
vector_store="http://localhost:19530",
pulsar_host="pulsar://localhost:6650",
graph_hosts=["localhost"],
)
if query == "":