mirror of
https://github.com/open-webui/open-webui.git
synced 2025-12-12 04:15:25 +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"])
|
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"]
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue