Skip to content

Commit ffa4a8e

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

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

backend/open_webui/functions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -318,7 +318,7 @@ 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):
321+
if isinstance(res, str) or isinstance(res, Generator) or isinstance(res, AsyncGenerator):
322322
finish_message = openai_chat_chunk_message_template(form_data['model'], '')
323323
finish_message['choices'][0]['finish_reason'] = 'stop'
324324
yield f'data: {json.dumps(finish_message)}\n\n'

0 commit comments

Comments
 (0)