This commit is contained in:
Timothy Jaeryang Baek 2025-05-27 01:16:11 +04:00
parent aaff204e7b
commit 940a437631

View File

@ -391,18 +391,21 @@ async def get_all_models(request: Request, user: UserModel = None):
) )
} }
loaded_models = await get_ollama_loaded_models(request, user=user) try:
expires_map = { loaded_models = await get_ollama_loaded_models(request, user=user)
m["name"]: m["expires_at"] expires_map = {
for m in loaded_models["models"] m["name"]: m["expires_at"]
if "expires_at" in m for m in loaded_models["models"]
} if "expires_at" in m
}
for m in models["models"]: for m in models["models"]:
if m["name"] in expires_map: if m["name"] in expires_map:
# Parse ISO8601 datetime with offset, get unix timestamp as int # Parse ISO8601 datetime with offset, get unix timestamp as int
dt = datetime.fromisoformat(expires_map[m["name"]]) dt = datetime.fromisoformat(expires_map[m["name"]])
m["expires_at"] = int(dt.timestamp()) m["expires_at"] = int(dt.timestamp())
except Exception as e:
log.debug(f"Failed to get loaded models: {e}")
else: else:
models = {"models": []} models = {"models": []}