gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix stream respone error for naive query mode

Browse Source
This commit is contained in:
yangdx
2025-04-21 00:06:15 +08:00
parent 99522a088d
commit 1a7b225e90
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lightrag/operate.py
View File

@@ -2011,9 +2011,10 @@ async def naive_query(
response = await use_model_func( response = await use_model_func(
query, query,
system_prompt=sys_prompt, 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 = (
response[len(sys_prompt) :] response[len(sys_prompt) :]
.replace(sys_prompt, "") .replace(sys_prompt, "")