Merge remote-tracking branch 'origin/main'

# Conflicts:
#	lightrag/llm.py
#	lightrag/operate.py
This commit is contained in:
magicyuan876
2024-12-06 15:06:00 +08:00
6 changed files with 198 additions and 6 deletions

View File

@@ -536,9 +536,10 @@ async def kg_query(
response = await use_model_func(
query,
system_prompt=sys_prompt,
stream=query_param.stream,
mode=query_param.mode,
)
if len(response) > len(sys_prompt):
if isinstance(response, str) and len(response) > len(sys_prompt):
response = (
response.replace(sys_prompt, "")
.replace("user", "")