diff --git a/Makefile b/Makefile index 9dea4f59..ab6a1347 100644 --- a/Makefile +++ b/Makefile @@ -1,6 +1,6 @@ # VERSION=$(shell git describe | sed 's/^v//') -VERSION=0.3.1 +VERSION=0.3.2 all: container diff --git a/setup.py b/setup.py index c6c32f97..eb22db9a 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,7 @@ import os with open("README.md", "r") as fh: long_description = fh.read() -version = "0.3.1" +version = "0.3.2" setuptools.setup( name="trustgraph", diff --git a/trustgraph/graph_rag.py b/trustgraph/graph_rag.py index aa24eef0..220231f6 100644 --- a/trustgraph/graph_rag.py +++ b/trustgraph/graph_rag.py @@ -18,7 +18,7 @@ class GraphRag: verbose=False, entity_limit=50, triple_limit=30, - max_sg_size=3000, + max_subgraph_size=3000, ): self.verbose=verbose @@ -37,7 +37,7 @@ class GraphRag: self.entity_limit=entity_limit self.query_limit=triple_limit - self.max_sg_size=max_sg_size + self.max_subgraph_size=max_subgraph_size self.label_cache = {} @@ -149,7 +149,7 @@ class GraphRag: subgraph = list(subgraph) - subgraph = subgraph[0:self.max_sg_size] + subgraph = subgraph[0:self.max_subgraph_size] if self.verbose: print("Subgraph:", flush=True) diff --git a/trustgraph/rag/graph/rag.py b/trustgraph/rag/graph/rag.py index 8a7a484d..1299eb19 100755 --- a/trustgraph/rag/graph/rag.py +++ b/trustgraph/rag/graph/rag.py @@ -28,7 +28,7 @@ class Processor(ConsumerProducer): vector_store=default_vector_store, entity_limit=50, triple_limit=30, - max_sg_size=3000, + max_subgraph_size=3000, ): super(Processor, self).__init__( @@ -48,7 +48,7 @@ class Processor(ConsumerProducer): verbose=True, entity_limit=entity_limit, triple_limit=triple_limit, - max_sg_size=max_sg_size, + max_subgraph_size=max_subgraph_size, ) def handle(self, msg):