refac: optimisation
This commit is contained in:
@@ -9,6 +9,8 @@ from typing import Optional, Union
|
||||
from urllib.parse import urlparse
|
||||
|
||||
import aiohttp
|
||||
from aiocache import cached
|
||||
|
||||
import requests
|
||||
from open_webui.apps.webui.models.models import Models
|
||||
from open_webui.config import (
|
||||
@@ -256,6 +258,7 @@ def merge_models_lists(model_lists):
|
||||
return list(merged_models.values())
|
||||
|
||||
|
||||
@cached(ttl=3)
|
||||
async def get_all_models():
|
||||
log.info("get_all_models()")
|
||||
if app.state.config.ENABLE_OLLAMA_API:
|
||||
@@ -295,8 +298,6 @@ async def get_all_models():
|
||||
for model in response.get("models", []):
|
||||
model["model"] = f"{prefix_id}.{model['model']}"
|
||||
|
||||
print(responses)
|
||||
|
||||
models = {
|
||||
"models": merge_models_lists(
|
||||
map(
|
||||
|
||||
Reference in New Issue
Block a user