mirror of
https://github.com/open-webui/open-webui.git
synced 2025-12-11 20:05:19 +00:00
Added proper type hints to new functions
Forgot about that...
This commit is contained in:
parent
27de981246
commit
9eccce2444
1 changed files with 2 additions and 2 deletions
|
|
@ -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 a new issue