Update app.py

app.py

@@ -42,17 +42,27 @@ models = {}
 
 def get_model(model_name: str):
     logger.info(f"尝试获取模型: {model_name}")
-    model_to_load = MODEL_MAPPING.get(model_name, model_name)  # 兼容直接用开源模型名
+    model_to_load = MODEL_MAPPING.get(model_name, model_name)
+    
+    # 提前检查模型是否在已知支持列表中（可选）
+    known_models = set(MODEL_MAPPING.keys()).union(set(MODEL_MAPPING.values()))
+    if model_to_load not in known_models and not model_to_load.startswith(("BAAI/", "sentence-transformers/")):
+        raise HTTPException(status_code=400, detail=f"不支持的模型: {model_name}")
+    
     if model_name not in models:
         try:
-            models[model_name] = SentenceTransformer(model_to_load)
+            hf_token = os.getenv("HUGGINGFACE_HUB_TOKEN")
+            models[model_name] = SentenceTransformer(model_to_load, use_auth_token=hf_token)
             logger.info(f"模型 {model_name} 加载成功")
         except Exception as e:
-            error_msg = f"加载模型 {model_name} 失败: {str(e)}"
-            logger.error(error_msg)
-            raise HTTPException(status_code=500, detail=error_msg)
+            if "not a valid model identifier" in str(e):
+                raise HTTPException(status_code=400, detail=f"不支持的模型: {model_name}")
+            else:
+                raise HTTPException(status_code=500, detail=f"加载模型失败: {str(e)}")
     return models[model_name]
 
+
+
 # 验证API密钥
 def verify_api_key(authorization: Optional[str] = Header(None)):
     logger.info(f"Authorization头部内容: {authorization}")