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

update error response format in streaming API to a normal message. So user can get what's going on.

Browse Source
This commit is contained in:
yangdx
2025-02-05 11:07:31 +08:00
parent 24effb127d
commit f1ea7f7415
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/api/lightrag_server.py
View File

@@ -1793,10 +1793,12 @@ def create_app(args):
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": { "message": {
"code": "STREAM_ERROR", "role": "assistant",
"message": error_msg "content": f"\n\nError: {error_msg}",
"images": None
}, },
"done": False
} }
yield f"{json.dumps(error_data, ensure_ascii=False)}\n" yield f"{json.dumps(error_data, ensure_ascii=False)}\n"