Browse Source

Simplify import

Juan Calderon-Perez 3 months ago
parent
commit
e19554e9a8
1 changed files with 2 additions and 3 deletions
  1. 2 3
      backend/open_webui/main.py

+ 2 - 3
backend/open_webui/main.py

@@ -11,8 +11,7 @@ import random
 
 from contextlib import asynccontextmanager
 from urllib.parse import urlencode, parse_qs, urlparse
-import anyio
-import anyio.to_thread
+from anyio import to_thread
 from pydantic import BaseModel
 from sqlalchemy import text
 
@@ -440,7 +439,7 @@ async def lifespan(app: FastAPI):
 
     pool_size = THREAD_POOL_SIZE
     if pool_size and pool_size > 0:
-        anyio.to_thread.current_default_thread_limiter().total_tokens = pool_size
+        to_thread.current_default_thread_limiter().total_tokens = pool_size
 
     asyncio.create_task(periodic_usage_pool_cleanup())
     yield