From 056f727bfdb91a9eb7cb1af128af7ca182ed29aa Mon Sep 17 00:00:00 2001
From: Will Miao <13051207myq@gmail.com>
Date: Sat, 4 Oct 2025 17:08:02 +0800
Subject: [PATCH] feat(model-scanner): enhance page type determination for
 model types

---
 py/routes/handlers/misc_handlers.py | 1 -
 py/services/model_scanner.py        | 7 ++++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/py/routes/handlers/misc_handlers.py b/py/routes/handlers/misc_handlers.py
index caaf3c66..398013dc 100644
--- a/py/routes/handlers/misc_handlers.py
+++ b/py/routes/handlers/misc_handlers.py
@@ -22,7 +22,6 @@ from aiohttp import web
 from ...config import config
 from ...services.metadata_service import (
     get_metadata_archive_manager,
-    get_metadata_provider,
     update_metadata_providers,
 )
 from ...services.service_registry import ServiceRegistry
diff --git a/py/services/model_scanner.py b/py/services/model_scanner.py
index 61917736..4e1c0eac 100644
--- a/py/services/model_scanner.py
+++ b/py/services/model_scanner.py
@@ -211,7 +211,12 @@ class ModelScanner:
             self._is_initializing = True
             
             # Determine the page type based on model type
-            page_type = 'loras' if self.model_type == 'lora' else 'checkpoints'
+            page_type_map = {
+                'lora': 'loras',
+                'checkpoint': 'checkpoints',
+                'embedding': 'embeddings'
+            }
+            page_type = page_type_map.get(self.model_type, self.model_type)
             
             # First, try to load from cache
             await ws_manager.broadcast_init_progress({