diff --git a/lightrag/base.py b/lightrag/base.py
index e8994bba..c6035f70 100644
--- a/lightrag/base.py
+++ b/lightrag/base.py
@@ -93,7 +93,7 @@ class QueryParam:
     This allows using different models for different query modes.
     """
 
-    user_prompt: str | None= None
+    user_prompt: str | None = None
     """User-provided prompt for the query.
     If proivded, this will be use instead of the default vaulue from prompt template.
     """
diff --git a/lightrag/operate.py b/lightrag/operate.py
index 9ffd6111..3ffca35a 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -926,7 +926,11 @@ async def kg_query(
         )
 
     # Build system prompt
-    user_prompt = query_param.user_prompt if query_param.user_prompt else PROMPTS["DEFAULT_USER_PROMPT"]
+    user_prompt = (
+        query_param.user_prompt
+        if query_param.user_prompt
+        else PROMPTS["DEFAULT_USER_PROMPT"]
+    )
     sys_prompt_temp = system_prompt if system_prompt else PROMPTS["rag_response"]
     sys_prompt = sys_prompt_temp.format(
         context_data=context,
@@ -1911,7 +1915,11 @@ async def naive_query(
         )
 
     # Build system prompt
-    user_prompt = query_param.user_prompt if query_param.user_prompt else PROMPTS["DEFAULT_USER_PROMPT"]
+    user_prompt = (
+        query_param.user_prompt
+        if query_param.user_prompt
+        else PROMPTS["DEFAULT_USER_PROMPT"]
+    )
     sys_prompt_temp = system_prompt if system_prompt else PROMPTS["naive_rag_response"]
     sys_prompt = sys_prompt_temp.format(
         content_data=text_units_str,