Fixed models errors

Author: UtkarshTheDev

locallab/__init__.py

@@ -2,7 +2,7 @@
 LocalLab - A lightweight AI inference server for running LLMs locally
 """
 
-__version__ = "0.4.22"
+__version__ = "0.4.37"
 
 # Only import what's necessary initially, lazy-load the rest
 from .logger import get_logger

locallab/routes/models.py

@@ -15,8 +15,11 @@
 # Get logger
 logger = get_logger("locallab.routes.models")
 
-# Create router
-router = APIRouter(tags=["Models"])
+# Create router with prefix
+router = APIRouter(
+    prefix="/models",
+    tags=["Models"]
+)
 
 class ModelInfo(BaseModel):
     """Model information response"""
@@ -42,7 +45,7 @@ class LoadModelRequest(BaseModel):
     """Request model for loading a model"""
     model_id: str
 
-@router.post("/models/load")
+@router.post("/load")
 async def load_model(request: LoadModelRequest) -> Dict[str, str]:
     """Load a specific model"""
     try:
@@ -54,7 +57,7 @@ async def load_model(request: LoadModelRequest) -> Dict[str, str]:
         logger.error(f"Failed to load model {request.model_id}: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))
 
-@router.get("", response_model=ModelsListResponse)
+@router.get("/list", response_model=ModelsListResponse)
 async def list_models() -> ModelsListResponse:
     """List all available models"""
     models_list = []
@@ -72,14 +75,12 @@ async def list_models() -> ModelsListResponse:
         current_model=model_manager.current_model
     )
 
-
 @router.get("/available", response_model=ModelsListResponse)
 async def available_models() -> ModelsListResponse:
     """List all available models (alternative endpoint)"""
     # This endpoint exists to provide compatibility with different API patterns
     return await list_models()
 
-
 @router.get("/current", response_model=ModelResponse)
 async def get_current_model() -> ModelResponse:
     """Get information about the currently loaded model"""
@@ -96,7 +97,6 @@ async def get_current_model() -> ModelResponse:
         loading_progress=1.0
     )
 
-
 @router.post("/load/{model_id}", response_model=Dict[str, str])
 async def load_model(model_id: str, background_tasks: BackgroundTasks) -> Dict[str, str]:
     """Load a specific model"""
@@ -115,7 +115,6 @@ async def load_model(model_id: str, background_tasks: BackgroundTasks) -> Dict[s
         logger.error(f"Failed to load model {model_id}: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))
 
-
 @router.post("/load", response_model=Dict[str, str])
 async def load_model_from_body(request: LoadModelRequest, background_tasks: BackgroundTasks) -> Dict[str, str]:
     """Load a specific model using model_id from request body"""
@@ -135,7 +134,6 @@ async def load_model_from_body(request: LoadModelRequest, background_tasks: Back
         logger.error(f"Failed to load model {model_id}: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))
 
-
 @router.post("/unload", response_model=Dict[str, str])
 async def unload_model() -> Dict[str, str]:
     """Unload the current model to free up resources"""
@@ -150,7 +148,6 @@ async def unload_model() -> Dict[str, str]:
         logger.error(f"Failed to unload model: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))
 
-
 @router.get("/status/{model_id}", response_model=ModelResponse)
 async def get_model_status(model_id: str) -> ModelResponse:
     """Get the loading status of a specific model"""

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="locallab",
-    version="0.4.36",
+    version="0.4.37",
     packages=find_packages(include=["locallab", "locallab.*"]),
     install_requires=[
         "fastapi>=0.95.0,<1.0.0",