UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-30 06:18:20 +00:00
Code Issues Actions 4 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"]: if form_data["stream"]:
async def stream_content(): async def stream_content():
if inspect.iscoroutinefunction(pipe): try:
res = await pipe(**param) if inspect.iscoroutinefunction(pipe):
else: res = await pipe(**param)
res = 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): if isinstance(res, str):
message = stream_message_template(form_data["model"], res) 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" stream_content(), media_type="text/event-stream"
) )
else: 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): if inspect.iscoroutinefunction(pipe):
res = await pipe(**param) res = await pipe(**param)
else: else: