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

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