diff --git a/examples/graph_visual_with_html.py b/examples/graph_visual_with_html.py
index 1a3ff144..56642185 100644
--- a/examples/graph_visual_with_html.py
+++ b/examples/graph_visual_with_html.py
@@ -11,6 +11,7 @@ net = Network(height="100vh", notebook=True)
 # Convert NetworkX graph to Pyvis network
 net.from_nx(G)
 
+
 # Add colors and title to nodes
 for node in net.nodes:
     node["color"] = "#{:06x}".format(random.randint(0, 0xFFFFFF))
diff --git a/lightrag/llm.py b/lightrag/llm.py
index 72af880e..53626b76 100644
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@@ -616,7 +616,6 @@ async def jina_embedding(
     data_list = await fetch_data(url, headers, data)
     return np.array([dp["embedding"] for dp in data_list])
 
-
 @wrap_embedding_func_with_attrs(embedding_dim=2048, max_token_size=512)
 @retry(
     stop=stop_after_attempt(3),
diff --git a/lightrag/operate.py b/lightrag/operate.py
index 45c9ef16..72734867 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -222,7 +222,7 @@ async def _merge_edges_then_upsert(
                 },
             )
     description = await _handle_entity_relation_summary(
-        (src_id, tgt_id), description, global_config
+        f"({src_id}, {tgt_id})", description, global_config
     )
     await knowledge_graph_inst.upsert_edge(
         src_id,
@@ -572,7 +572,6 @@ async def kg_query(
             mode=query_param.mode,
         ),
     )
-
     return response
 
 
@@ -1104,6 +1103,7 @@ async def naive_query(
     response = await use_model_func(
         query,
         system_prompt=sys_prompt,
+        mode=query_param.mode,
     )
 
     if len(response) > len(sys_prompt):
diff --git a/lightrag/utils.py b/lightrag/utils.py
index 32d5c87f..49a7b498 100644
--- a/lightrag/utils.py
+++ b/lightrag/utils.py
@@ -383,7 +383,7 @@ async def get_best_cached_response(
             except Exception as e:  # Catch all possible exceptions
                 logger.warning(f"LLM similarity check failed: {e}")
                 return None  # Return None directly when LLM check fails
-
+              
         prompt_display = (
             best_prompt[:50] + "..." if len(best_prompt) > 50 else best_prompt
         )