mirror of
https://github.com/open-webui/open-webui
synced 2025-06-04 03:37:35 +00:00
Added proper type hints to new functions
Forgot about that...
This commit is contained in:
parent
27de981246
commit
9eccce2444
@ -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"]
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user