From 1a7b225e9071a3b0e26886c40b9792c66f32bc45 Mon Sep 17 00:00:00 2001 From: yangdx Date: Mon, 21 Apr 2025 00:06:15 +0800 Subject: [PATCH] Fix stream respone error for naive query mode --- lightrag/operate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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, "")