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

Fix graph-rag

Browse source
This commit is contained in:
Cyber MacGeddon 2024-07-17 22:10:25 +01:00
parent 816ee2038f
commit 0574eb6177
7 changed files with 44 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

5
trustgraph/rag/graph/rag.py
View file

@ -12,7 +12,7 @@ from ... base import ConsumerProducer
default_input_queue = 'graph-rag-query'
default_output_queue = 'graph-rag-response'
default_subscriber = 'graph-rag'
default_graph_hosts = [ 'localhost' ]
default_graph_hosts = 'localhost'
default_vector_store = 'http://localhost:19530'
class Processor(ConsumerProducer):
@ -43,7 +43,7 @@ class Processor(ConsumerProducer):
self.rag = GraphRag(
pulsar_host=pulsar_host,
graph_hosts=graph_hosts,
graph_hosts=graph_hosts.split(","),
vector_store=vector_store,
verbose=True,
entity_limit=entity_limit,
@ -114,4 +114,3 @@ def run():
Processor.start('graph-rag', __doc__)