From f1ea7f7415ca53adf2576dab03d50ed137d5d61d Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Wed, 5 Feb 2025 11:07:31 +0800
Subject: [PATCH] update error response format in streaming API to a normal
 message. So user can get what's going on.

---
 lightrag/api/lightrag_server.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py
index d8412a13..c9144d0e 100644
--- a/lightrag/api/lightrag_server.py
+++ b/lightrag/api/lightrag_server.py
@@ -1793,10 +1793,12 @@ def create_app(args):
                                 error_data = {
                                     "model": ollama_server_infos.LIGHTRAG_MODEL,
                                     "created_at": ollama_server_infos.LIGHTRAG_CREATED_AT,
-                                    "error": {
-                                        "code": "STREAM_ERROR",
-                                        "message": error_msg
+                                    "message": {
+                                        "role": "assistant",
+                                        "content": f"\n\nError: {error_msg}",
+                                        "images": None
                                     },
+                                    "done": False
                                 }
                                 yield f"{json.dumps(error_data, ensure_ascii=False)}\n"