Fix RAG service

Makefile

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

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",

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)

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):