mirror of
https://github.com/open-webui/open-webui
synced 2025-06-08 15:37:22 +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"])
|
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 New Issue
Block a user