From f77f6cd1e24b3fc1e6ba99c1d16bbdeb330e1cb3 Mon Sep 17 00:00:00 2001 From: Juan Calderon-Perez Date: Fri, 18 Apr 2025 09:30:34 -0400 Subject: [PATCH] Update imports to match anyio docs --- backend/open_webui/main.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py index 8c16bb837..87fa68d02 100644 --- a/backend/open_webui/main.py +++ b/backend/open_webui/main.py @@ -11,13 +11,13 @@ import random from contextlib import asynccontextmanager from urllib.parse import urlencode, parse_qs, urlparse -from anyio import to_thread from pydantic import BaseModel from sqlalchemy import text from typing import Optional from aiocache import cached import aiohttp +import anyio.to_thread import requests @@ -439,7 +439,8 @@ async def lifespan(app: FastAPI): pool_size = THREAD_POOL_SIZE if pool_size and pool_size > 0: - to_thread.current_default_thread_limiter().total_tokens = pool_size + limiter = anyio.to_thread.current_default_thread_limiter() + limiter.total_tokens = pool_size asyncio.create_task(periodic_usage_pool_cleanup()) yield