From 9eccce244444ffd7c6d07923022d0afb4ed7ad16 Mon Sep 17 00:00:00 2001 From: toriset Date: Tue, 27 May 2025 15:44:20 +0300 Subject: [PATCH] Added proper type hints to new functions Forgot about that... --- backend/open_webui/utils/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/utils/models.py b/backend/open_webui/utils/models.py index 464680124..6d55d7876 100644 --- a/backend/open_webui/utils/models.py +++ b/backend/open_webui/utils/models.py @@ -34,7 +34,7 @@ log = logging.getLogger(__name__) log.setLevel(SRC_LOG_LEVELS["MAIN"]) -async def fetch_ollama_models(request, user): +async def fetch_ollama_models(request: Request, user: UserModel = None): raw_ollama_models = await ollama.get_all_models(request, user=user) return [ { @@ -51,7 +51,7 @@ async def fetch_ollama_models(request, user): ] -async def fetch_openai_models(request, user): +async def fetch_openai_models(request: Request, user: UserModel = None): openai_response = await openai.get_all_models(request, user=user) return openai_response["data"]