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-05 09:47:39 +08:00
parent 69f200faf2
commit ff40e61fad
1 changed files with 8 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
lightrag/api/lightrag_server.py
View file

@ -1721,7 +1721,7 @@ def create_app(args):
first_chunk_time = None
last_chunk_time = None
total_response = ""
try:
# Ensure response is an async generator
if isinstance(response, str):
@ -1786,9 +1786,9 @@ def create_app(args):
"created_at": ollama_server_infos.LIGHTRAG_CREATED_AT,
"error": {
"code": "STREAM_CANCELLED",
"message": "Stream was cancelled by server"
"message": "Stream was cancelled by server",
},
"done": False
"done": False,
}
yield f"{json.dumps(error_data, ensure_ascii=False)}\n"
raise
@ -1815,24 +1815,21 @@ def create_app(args):
except Exception as e:
error_msg = f"Error in stream_generator: {str(e)}"
logging.error(error_msg)
# 发送错误消息给客户端
error_data = {
"model": ollama_server_infos.LIGHTRAG_MODEL,
"created_at": ollama_server_infos.LIGHTRAG_CREATED_AT,
"error": {
"code": "STREAM_ERROR",
"message": error_msg
},
"done": False
"error": {"code": "STREAM_ERROR", "message": error_msg},
"done": False,
}
yield f"{json.dumps(error_data, ensure_ascii=False)}\n"
# 确保发送结束标记
final_data = {
"model": ollama_server_infos.LIGHTRAG_MODEL,
"created_at": ollama_server_infos.LIGHTRAG_CREATED_AT,
"done": True
"done": True,
}
yield f"{json.dumps(final_data, ensure_ascii=False)}\n"
raise