apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-25 23:35:12 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix RAG service

Browse source
This commit is contained in:
Cyber MacGeddon 2024-07-17 21:43:53 +01:00
parent 75776965f9
commit 816ee2038f
4 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
Makefile
View file

@ -1,6 +1,6 @@
# VERSION=$(shell git describe | sed 's/^v//') # VERSION=$(shell git describe | sed 's/^v//')
VERSION=0.3.1 VERSION=0.3.2
all: container all: container

2
setup.py
View file

@ -4,7 +4,7 @@ import os
with open("README.md", "r") as fh: with open("README.md", "r") as fh:
long_description = fh.read() long_description = fh.read()
version = "0.3.1" version = "0.3.2"
setuptools.setup( setuptools.setup(
name="trustgraph", name="trustgraph",

6
trustgraph/graph_rag.py
View file

@ -18,7 +18,7 @@ class GraphRag:
verbose=False, verbose=False,
entity_limit=50, entity_limit=50,
triple_limit=30, triple_limit=30,
max_sg_size=3000, max_subgraph_size=3000,
): ):
self.verbose=verbose self.verbose=verbose
@ -37,7 +37,7 @@ class GraphRag:
self.entity_limit=entity_limit self.entity_limit=entity_limit
self.query_limit=triple_limit self.query_limit=triple_limit
self.max_sg_size=max_sg_size self.max_subgraph_size=max_subgraph_size
self.label_cache = {} self.label_cache = {}
@ -149,7 +149,7 @@ class GraphRag:
subgraph = list(subgraph) subgraph = list(subgraph)
subgraph = subgraph[0:self.max_sg_size] subgraph = subgraph[0:self.max_subgraph_size]
if self.verbose: if self.verbose:
print("Subgraph:", flush=True) print("Subgraph:", flush=True)

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

@ -28,7 +28,7 @@ class Processor(ConsumerProducer):
vector_store=default_vector_store, vector_store=default_vector_store,
entity_limit=50, entity_limit=50,
triple_limit=30, triple_limit=30,
max_sg_size=3000, max_subgraph_size=3000,
): ):
super(Processor, self).__init__( super(Processor, self).__init__(
@ -48,7 +48,7 @@ class Processor(ConsumerProducer):
verbose=True, verbose=True,
entity_limit=entity_limit, entity_limit=entity_limit,
triple_limit=triple_limit, triple_limit=triple_limit,
max_sg_size=max_sg_size, max_subgraph_size=max_subgraph_size,
) )
def handle(self, msg): def handle(self, msg):