ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(proxy_server.py): return [DONE] end of stream chunk

Browse Source
This commit is contained in:
Krrish Dholakia 2023-12-28 11:30:30 +05:30
parent 7822a8d421
commit bf4c4426e5
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View File

@ -913,6 +913,10 @@ async def async_data_generator(response, user_api_key_dict):
yield f"data: {json.dumps(chunk.dict())}\n\n"
except Exception as e:
yield f"data: {str(e)}\n\n"
# Streaming is done, yield the [DONE] chunk
done_message = "[DONE]"
yield f"data: {done_message}\n\n"
except Exception as e:
yield f"data: {str(e)}\n\n"