github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-25 21:14:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:return conv_uid and update wechat

Browse Source
This commit is contained in:
aries_ckt 2024-03-21 15:59:46 +08:00
parent 0cf08f37b0
commit ab9d8a370e
3 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
assets/wechat.jpg
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 190 KiB

After

Width:  |  Height:  |  Size: 118 KiB

4
dbgpt/app/openapi/api_v1/api_v1.py
View File

@ -439,9 +439,7 @@ async def stream_generator(chat, incremental: bool, model_name: str):
_type_: streaming responses _type_: streaming responses
""" """
span = root_tracer.start_span("stream_generator") span = root_tracer.start_span("stream_generator")
msg = "[LLM_ERROR]: llm server has no output, maybe your prompt template is wrong."
stream_id = f"chatcmpl-{str(uuid.uuid1())}"
previous_response = "" previous_response = ""
async for chunk in chat.stream_call(): async for chunk in chat.stream_call():
if chunk: if chunk:
@ -453,7 +451,7 @@ async def stream_generator(chat, incremental: bool, model_name: str):
delta=DeltaMessage(role="assistant", content=incremental_output), delta=DeltaMessage(role="assistant", content=incremental_output),
) )
chunk = ChatCompletionStreamResponse( chunk = ChatCompletionStreamResponse(
id=stream_id, choices=[choice_data], model=model_name id=chat.chat_session_id, choices=[choice_data], model=model_name
) )
yield f"data: {chunk.json(exclude_unset=True, ensure_ascii=False)}\n\n" yield f"data: {chunk.json(exclude_unset=True, ensure_ascii=False)}\n\n"
else: else:

2
dbgpt/client/client.py
View File

@ -247,7 +247,7 @@ class Client:
if response.status_code == 200: if response.status_code == 200:
async for line in response.aiter_lines(): async for line in response.aiter_lines():
try: try:
if line == "data: [DONE]\n": if line.strip() == "data: [DONE]":
break break
if line.startswith("data:"): if line.startswith("data:"):
json_data = json.loads(line[len("data: ") :]) json_data = json.loads(line[len("data: ") :])