UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-18 08:40:58 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

feat: function exception handler

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-22 12:43:30 -07:00
parent 9205b90af6
commit cf6447eb2a
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
backend/main.py
View File

@ -913,10 +913,15 @@ async def generate_chat_completions(form_data: dict, user=Depends(get_verified_u
if form_data["stream"]:
async def stream_content():
if inspect.iscoroutinefunction(pipe):
res = await pipe(**param)
else:
res = pipe(**param)
try:
if inspect.iscoroutinefunction(pipe):
res = await pipe(**param)
else:
res = pipe(**param)
except Exception as e:
print(f"Error: {e}")
yield f"data: {json.dumps({'error': {'detail':str(e)}})}\n\n"
return
if isinstance(res, str):
message = stream_message_template(form_data["model"], res)
@ -961,6 +966,16 @@ async def generate_chat_completions(form_data: dict, user=Depends(get_verified_u
stream_content(), media_type="text/event-stream"
)
else:
try:
if inspect.iscoroutinefunction(pipe):
res = await pipe(**param)
else:
res = pipe(**param)
except Exception as e:
print(f"Error: {e}")
return {"error": {"detail":str(e)}}
if inspect.iscoroutinefunction(pipe):
res = await pipe(**param)
else: