Fix linting

This commit is contained in:
yangdx
2025-02-03 12:39:52 +08:00
parent ede4122b63
commit 4ab02a878f

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)