Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 03:47:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

refac: close subprocess gracefully

Browse Source
This commit is contained in:
Timothy J. Baek
2024-04-21 01:46:09 -05:00
parent 8651bec915
commit 3c382d4c6c
2 changed files with 52 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
backend/main.py
View File

@@ -20,7 +20,11 @@ from starlette.middleware.base import BaseHTTPMiddleware
from apps.ollama.main import app as ollama_app
from apps.openai.main import app as openai_app
from apps.litellm.main import app as litellm_app, start_litellm_background
from apps.litellm.main import (
app as litellm_app,
start_litellm_background,
shutdown_litellm_background,
)
from apps.audio.main import app as audio_app
from apps.images.main import app as images_app
from apps.rag.main import app as rag_app
@@ -316,3 +320,8 @@ app.mount(
SPAStaticFiles(directory=FRONTEND_BUILD_DIR, html=True),
name="spa-static-files",
)
@app.on_event("shutdown")
async def shutdown_event():
await shutdown_litellm_background()