Spaces:
Runtime error
Runtime error
| """ | |
| Model listing routes. | |
| """ | |
| from fastapi import APIRouter | |
| from app.core.logging import get_logger | |
| from app.schemas.models import ModelsListResponse, ModelInfo | |
| from app.services.model_registry import get_model_registry | |
| logger = get_logger(__name__) | |
| router = APIRouter(tags=["models"]) | |
| async def list_models() -> ModelsListResponse: | |
| """ | |
| List all loaded models. | |
| Returns information about the fusion model and all submodels, | |
| including their Hugging Face repository IDs and configurations. | |
| """ | |
| registry = get_model_registry() | |
| models = registry.list_models() | |
| fusion_info = None | |
| submodels_info = [] | |
| for model in models: | |
| model_info = ModelInfo( | |
| repo_id=model["repo_id"], | |
| name=model["name"], | |
| model_type=model["model_type"], | |
| config=model.get("config") | |
| ) | |
| if model["model_type"] == "fusion": | |
| fusion_info = model_info | |
| else: | |
| submodels_info.append(model_info) | |
| return ModelsListResponse( | |
| fusion=fusion_info, | |
| submodels=submodels_info, | |
| total_count=len(models) | |
| ) | |