github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-20 00:54:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(core): Fix newline escape issue (#2854)

Browse Source
This commit is contained in:
Fangyin Cheng
2025-07-16 19:49:34 +08:00
committed by GitHub
parent 845432fea0
commit 06de6ea08d
100 changed files with 146 additions and 127 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
packages/dbgpt-serve/src/dbgpt_serve/flow/service/service.py
View File

@@ -19,6 +19,7 @@ from dbgpt.core.awel.flow.flow_factory import (
)
from dbgpt.core.awel.trigger.http_trigger import CommonLLMHttpTrigger
from dbgpt.core.awel.util.chat_util import (
_v1_create_completion_response,
is_chat_flow_type,
safe_chat_stream_with_dag_task,
safe_chat_with_dag_task,
@@ -448,10 +449,17 @@ class Service(BaseService[ServeEntity, ServeRequest, ServerResponse]):
if text:
text = text.replace("\n", "\\n")
if output.error_code != 0:
yield f"data:[SERVER_ERROR]{text}\n\n"
yield _v1_create_completion_response(
f"[SERVER_ERROR]{text}",
None,
model_name=request.model,
stream_id=request.conv_uid,
)
break
else:
yield f"data:{text}\n\n"
yield _v1_create_completion_response(
text, None, model_name=request.model, stream_id=request.conv_uid
)
async def chat_stream_openai(
self, flow_uid: str, request: CommonLLMHttpRequestBody