yangdx
commited on
Commit
·
c63b233
1
Parent(s):
5e539c5
Fix linting
Browse files
lightrag/api/lightrag_server.py
CHANGED
|
@@ -1793,9 +1793,9 @@ def create_app(args):
|
|
| 1793 |
error_msg = "Stream was cancelled by server"
|
| 1794 |
else:
|
| 1795 |
error_msg = f"Provider error: {error_msg}"
|
| 1796 |
-
|
| 1797 |
logging.error(f"Stream error: {error_msg}")
|
| 1798 |
-
|
| 1799 |
# Send error message to client
|
| 1800 |
error_data = {
|
| 1801 |
"model": ollama_server_infos.LIGHTRAG_MODEL,
|
|
@@ -1803,12 +1803,12 @@ def create_app(args):
|
|
| 1803 |
"message": {
|
| 1804 |
"role": "assistant",
|
| 1805 |
"content": f"\n\nError: {error_msg}",
|
| 1806 |
-
"images": None
|
| 1807 |
},
|
| 1808 |
-
"done": False
|
| 1809 |
}
|
| 1810 |
yield f"{json.dumps(error_data, ensure_ascii=False)}\n"
|
| 1811 |
-
|
| 1812 |
# Send final message to close the stream
|
| 1813 |
final_data = {
|
| 1814 |
"model": ollama_server_infos.LIGHTRAG_MODEL,
|
|
@@ -1845,10 +1845,7 @@ def create_app(args):
|
|
| 1845 |
error_data = {
|
| 1846 |
"model": ollama_server_infos.LIGHTRAG_MODEL,
|
| 1847 |
"created_at": ollama_server_infos.LIGHTRAG_CREATED_AT,
|
| 1848 |
-
"error": {
|
| 1849 |
-
"code": "STREAM_ERROR",
|
| 1850 |
-
"message": error_msg
|
| 1851 |
-
},
|
| 1852 |
}
|
| 1853 |
yield f"{json.dumps(error_data, ensure_ascii=False)}\n"
|
| 1854 |
|
|
|
|
| 1793 |
error_msg = "Stream was cancelled by server"
|
| 1794 |
else:
|
| 1795 |
error_msg = f"Provider error: {error_msg}"
|
| 1796 |
+
|
| 1797 |
logging.error(f"Stream error: {error_msg}")
|
| 1798 |
+
|
| 1799 |
# Send error message to client
|
| 1800 |
error_data = {
|
| 1801 |
"model": ollama_server_infos.LIGHTRAG_MODEL,
|
|
|
|
| 1803 |
"message": {
|
| 1804 |
"role": "assistant",
|
| 1805 |
"content": f"\n\nError: {error_msg}",
|
| 1806 |
+
"images": None,
|
| 1807 |
},
|
| 1808 |
+
"done": False,
|
| 1809 |
}
|
| 1810 |
yield f"{json.dumps(error_data, ensure_ascii=False)}\n"
|
| 1811 |
+
|
| 1812 |
# Send final message to close the stream
|
| 1813 |
final_data = {
|
| 1814 |
"model": ollama_server_infos.LIGHTRAG_MODEL,
|
|
|
|
| 1845 |
error_data = {
|
| 1846 |
"model": ollama_server_infos.LIGHTRAG_MODEL,
|
| 1847 |
"created_at": ollama_server_infos.LIGHTRAG_CREATED_AT,
|
| 1848 |
+
"error": {"code": "STREAM_ERROR", "message": error_msg},
|
|
|
|
|
|
|
|
|
|
| 1849 |
}
|
| 1850 |
yield f"{json.dumps(error_data, ensure_ascii=False)}\n"
|
| 1851 |
|