Added proper type hints to new functions

Forgot about that...
This commit is contained in:
toriset 2025-05-27 15:44:20 +03:00 committed by GitHub
parent 27de981246
commit 9eccce2444
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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"]