feat:rag graph

pilot/graph_engine/graph_engine.py

@@ -107,7 +107,7 @@ class RAGGraphEngine:
         """Build the index from nodes."""
         index_struct = self.index_struct_cls()
         num_threads = 5
-        chunk_size = len(documents) if (len(documents) < num_threads) else len(documents) / num_threads
+        chunk_size = len(documents) if (len(documents) < num_threads) else len(documents) // num_threads
 
         import concurrent
         future_tasks = []
@@ -132,6 +132,7 @@ class RAGGraphEngine:
         #
         # return index_struct
 
+
     def search(self, query):
         from pilot.graph_engine.graph_search import RAGGraphSearch
 

pilot/scene/base_chat.py

@@ -109,7 +109,7 @@ class BaseChat(ABC):
         import inspect
         input_values = (
             await self.generate_input_values()
-            if inspect.isawaitable(self.generate_input_values)
+            if inspect.isawaitable(self.generate_input_values())
             else self.generate_input_values()
         )
         ### Chat sequence advance

pilot/scene/chat_knowledge/extract_entity/chat.py

@@ -24,7 +24,7 @@ class ExtractEntity(BaseChat):
         self.user_input = chat_param["current_user_input"]
         self.extract_mode = chat_param["select_param"]
 
-    async def generate_input_values(self):
+    def generate_input_values(self):
         input_values = {
             "text": self.user_input,
         }

pilot/scene/chat_knowledge/extract_triplet/chat.py

@@ -24,7 +24,7 @@ class ExtractTriplet(BaseChat):
         self.user_input = chat_param["current_user_input"]
         self.extract_mode = chat_param["select_param"]
 
-    async def generate_input_values(self):
+    def generate_input_values(self):
         input_values = {
             "text": self.user_input,
         }