Update main.py

main.py

@@ -205,6 +205,11 @@ def create_response(content: str, model: str, finish_reason: Optional[str] = Non
 
 @app.post("/niansuhai/v1/chat/completions")
 async def chat_completions(request: ChatRequest):
+    # Validate the model
+    valid_models = Blackbox.models + list(Blackbox.userSelectedModel.keys()) + list(Blackbox.model_aliases.keys())
+    if request.model not in valid_models:
+        raise HTTPException(status_code=400, detail=f"Invalid model name: {request.model}. Valid models are: {valid_models}")
+
     messages = [{"role": msg.role, "content": msg.content} for msg in request.messages]
 
     async_generator = Blackbox.create_async_generator(
@@ -253,6 +258,7 @@ async def chat_completions(request: ChatRequest):
             "usage": None,
         }
 
+
 @app.get("/niansuhai/v1/models")
 async def get_models():
     return {"models": Blackbox.models}