From f77faf80237bd474e4c5f3fd33addfe2698e95fd Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Wed, 5 Feb 2025 12:36:52 +0800
Subject: [PATCH] Fix linting

---
 lightrag/api/lightrag_server.py | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py
index aadd1c09..345136aa 100644
--- a/lightrag/api/lightrag_server.py
+++ b/lightrag/api/lightrag_server.py
@@ -1793,9 +1793,9 @@ def create_app(args):
                                     error_msg = "Stream was cancelled by server"
                                 else:
                                     error_msg = f"Provider error: {error_msg}"
-                                
+
                                 logging.error(f"Stream error: {error_msg}")
-                                
+
                                 # Send error message to client
                                 error_data = {
                                     "model": ollama_server_infos.LIGHTRAG_MODEL,
@@ -1803,12 +1803,12 @@ def create_app(args):
                                     "message": {
                                         "role": "assistant",
                                         "content": f"\n\nError: {error_msg}",
-                                        "images": None
+                                        "images": None,
                                     },
-                                    "done": False
+                                    "done": False,
                                 }
                                 yield f"{json.dumps(error_data, ensure_ascii=False)}\n"
-                                
+
                                 # Send final message to close the stream
                                 final_data = {
                                     "model": ollama_server_infos.LIGHTRAG_MODEL,
@@ -1845,10 +1845,7 @@ 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
-                            },
+                            "error": {"code": "STREAM_ERROR", "message": error_msg},
                         }
                         yield f"{json.dumps(error_data, ensure_ascii=False)}\n"