diff --git a/lightrag/operate.py b/lightrag/operate.py
index 466eca50..73d559e6 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -2011,9 +2011,10 @@ async def naive_query(
     response = await use_model_func(
         query,
         system_prompt=sys_prompt,
+        stream=query_param.stream,
     )
 
-    if len(response) > len(sys_prompt):
+    if isinstance(response, str) and len(response) > len(sys_prompt):
         response = (
             response[len(sys_prompt) :]
             .replace(sys_prompt, "")