diff --git a/pilot/graph_engine/graph_engine.py b/pilot/graph_engine/graph_engine.py
index e34baba79..80cbab066 100644
--- a/pilot/graph_engine/graph_engine.py
+++ b/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
 
diff --git a/pilot/scene/base_chat.py b/pilot/scene/base_chat.py
index 58a0becc9..a1d6d9f08 100644
--- a/pilot/scene/base_chat.py
+++ b/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
diff --git a/pilot/scene/chat_knowledge/extract_entity/chat.py b/pilot/scene/chat_knowledge/extract_entity/chat.py
index 373bb4e5d..bb52961b5 100644
--- a/pilot/scene/chat_knowledge/extract_entity/chat.py
+++ b/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,
         }
diff --git a/pilot/scene/chat_knowledge/extract_triplet/chat.py b/pilot/scene/chat_knowledge/extract_triplet/chat.py
index 28152b92e..11fe871ab 100644
--- a/pilot/scene/chat_knowledge/extract_triplet/chat.py
+++ b/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,
         }