Fix bugs in setup.py

This commit is contained in:
Cyber MacGeddon 2024-07-15 21:48:51 +01:00
parent 178bd6943e
commit b63ffbd95e
2 changed files with 10 additions and 5 deletions

View file

@ -5,7 +5,12 @@ import sys
query = " ".join(sys.argv[1:])
gr = GraphRag(verbose=True)
gr = GraphRag(
verbose=True,
vector_store="http://localhost:19530",
pulsar_host="pulsar://localhost:6650",
graph_hosts=["localhost"],
)
if query == "":
query="""This knowledge graph describes the Space Shuttle disaster.