fix: multi-ollama mounted display issue

This commit is contained in:
Timothy Jaeryang Baek 2025-08-21 00:03:38 +04:00
parent faf01bdeac
commit 1f91e5d3c6

View file

@ -415,15 +415,15 @@ async def get_all_models(request: Request, user: UserModel = None):
try: try:
loaded_models = await get_ollama_loaded_models(request, user=user) loaded_models = await get_ollama_loaded_models(request, user=user)
expires_map = { expires_map = {
m["name"]: m["expires_at"] m["model"]: m["expires_at"]
for m in loaded_models["models"] for m in loaded_models["models"]
if "expires_at" in m if "expires_at" in m
} }
for m in models["models"]: for m in models["models"]:
if m["name"] in expires_map: if m["model"] 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["model"]])
m["expires_at"] = int(dt.timestamp()) m["expires_at"] = int(dt.timestamp())
except Exception as e: except Exception as e:
log.debug(f"Failed to get loaded models: {e}") log.debug(f"Failed to get loaded models: {e}")