Update private_gpt/launcher.py

private_gpt/launcher.py

@@ -7,6 +7,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.openapi.utils import get_openapi
 from injector import Injector
 from fastapi import APIRouter
+from fastapi import JSONResponse
 
 from private_gpt.paths import docs_path
 from private_gpt.server.chat.chat_router import chat_router
@@ -103,6 +104,29 @@ def create_app(root_injector: Injector) -> FastAPI:
 
             return {"message": f"Model switched to {new_model}"}
 
+        # Define a new APIRouter for the model_list
+        model_list_router = APIRouter(prefix="/v1/models_list", dependencies=[Depends(get_current_user)])    
+
+        @model_list_router.get("/", response_model=list[dict])
+        async def model_list(current_user: dict = Depends(get_current_user)):
+            """
+            Get a list of models with their details.
+            """
+            # In this example, hardcoding some sample model data
+            models_data = [
+                {"id": 1, "name": "gpt-3.5-turbo", "access": ["user", "admin"]},
+                {"id": 2, "name": "gpt-4", "access": ["admin"]},
+                # Add more models as needed
+            ]
+
+            # If the current user is an admin, return the full list
+            if "admin" in current_user.get("role", []):
+                return models_data
+
+            # If the current user is a regular user, filter models based on their access
+            accessible_models = [model for model in models_data if "user" in model["access"]]
+            return accessible_models
+
         # @model_router.post("/switch_model")
         # async def switch_model(new_model: str):
         #     # Implement logic to switch the LLM model based on the user's request
@@ -170,6 +194,7 @@ def create_app(root_injector: Injector) -> FastAPI:
         app.include_router(embeddings_router)
         app.include_router(health_router)
         app.include_router(model_router)
+        app.include_router(model_list_router)
 
         settings = root_injector.get(Settings)
         if settings.server.cors.enabled: