scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-09-10 22:43:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Merge 066adb3fc9 into 1dcbfd47fb

Browse source
This commit is contained in:
EntropyYue 2026-04-10 15:22:13 +00:00 • committed by GitHub
commit 1d696b53a0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
backend/open_webui/main.py
View file

@ -9,6 +9,7 @@ import sys
import time
import random
import re
from concurrent.futures import ThreadPoolExecutor
from uuid import uuid4
@ -646,10 +647,15 @@ async def lifespan(app: FastAPI):
if app.state.redis is not None:
app.state.redis_task_command_listener = asyncio.create_task(redis_task_command_listener(app))
executor = None
if THREAD_POOL_SIZE and THREAD_POOL_SIZE > 0:
limiter = anyio.to_thread.current_default_thread_limiter()
limiter.total_tokens = THREAD_POOL_SIZE
executor = ThreadPoolExecutor(max_workers=THREAD_POOL_SIZE)
app.state.main_loop.set_default_executor(executor)
asyncio.create_task(periodic_usage_pool_cleanup())
asyncio.create_task(periodic_session_pool_cleanup())
@ -713,6 +719,9 @@ async def lifespan(app: FastAPI):
yield
if executor is not None:
executor.shutdown()
if hasattr(app.state, 'redis_task_command_listener'):
app.state.redis_task_command_listener.cancel()