Skip to content

Commit 9d683e5

Browse files
committed
fix: emit [DONE] for AsyncGenerator pipe returns
1 parent e7d8c9f commit 9d683e5

1 file changed

Lines changed: 4 additions & 5 deletions

File tree

backend/open_webui/functions.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -318,11 +318,10 @@ async def stream_content():
318318
async for line in res:
319319
yield process_line(form_data, line)
320320

321-
if isinstance(res, str) or isinstance(res, Generator):
322-
finish_message = openai_chat_chunk_message_template(form_data['model'], '')
323-
finish_message['choices'][0]['finish_reason'] = 'stop'
324-
yield f'data: {json.dumps(finish_message)}\n\n'
325-
yield 'data: [DONE]'
321+
finish_message = openai_chat_chunk_message_template(form_data['model'], '')
322+
finish_message['choices'][0]['finish_reason'] = 'stop'
323+
yield f'data: {json.dumps(finish_message)}\n\n'
324+
yield 'data: [DONE]'
326325

327326
return StreamingResponse(stream_content(), media_type='text/event-stream')
328327
else:

0 commit comments

Comments
 (0)