gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix linting

Browse source
This commit is contained in:
yangdx 2025-02-03 12:39:52 +08:00
parent ede4122b63
commit 4ab02a878f
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lightrag/api/lightrag_server.py
View file

@ -1707,10 +1707,10 @@ def create_app(args):
if request.system: if request.system:
rag.llm_model_kwargs["system_prompt"] = request.system rag.llm_model_kwargs["system_prompt"] = request.system
response = await rag.llm_model_func( response = await rag.llm_model_func(
cleaned_query, cleaned_query,
stream=True, stream=True,
history_messages=conversation_history, history_messages=conversation_history,
**rag.llm_model_kwargs **rag.llm_model_kwargs,
) )
else: else:
response = await rag.aquery( # Need await to get async generator response = await rag.aquery( # Need await to get async generator
@ -1826,10 +1826,10 @@ def create_app(args):
rag.llm_model_kwargs["system_prompt"] = request.system rag.llm_model_kwargs["system_prompt"] = request.system
response_text = await rag.llm_model_func( response_text = await rag.llm_model_func(
cleaned_query, cleaned_query,
stream=False, stream=False,
history_messages=conversation_history, history_messages=conversation_history,
**rag.llm_model_kwargs **rag.llm_model_kwargs,
) )
else: else:
response_text = await rag.aquery(cleaned_query, param=query_param) response_text = await rag.aquery(cleaned_query, param=query_param)