1
0
Эх сурвалжийг харах

Added proper type hints to new functions

Forgot about that...
toriset 4 сар өмнө
parent
commit
9eccce2444

+ 2 - 2
backend/open_webui/utils/models.py

@@ -34,7 +34,7 @@ log = logging.getLogger(__name__)
 log.setLevel(SRC_LOG_LEVELS["MAIN"])
 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)
     raw_ollama_models = await ollama.get_all_models(request, user=user)
     return [
     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)
     openai_response = await openai.get_all_models(request, user=user)
     return openai_response["data"]
     return openai_response["data"]