Update app.py

app.py

@@ -17,7 +17,7 @@ logging.basicConfig(level=logging.INFO,
 
 API_ENDPOINT = "https://api-st.siliconflow.cn/v1/user/info"
 TEST_MODEL_ENDPOINT = "https://api-st.siliconflow.cn/v1/chat/completions"
-MODELS_ENDPOINT = "https://api-st.siliconflow.cn/v1/models"
+MODELS_ENDPOINT = "https://api.openai.com/v1/models"
 EMBEDDINGS_ENDPOINT = "https://api-st.siliconflow.cn/v1/embeddings"
 IMAGE_ENDPOINT = "https://api-st.siliconflow.cn/v1/images/generations"
 
@@ -239,60 +239,12 @@ def get_siliconflow_data(model_name, data):
 def refresh_models():
     global models
     
-    models["text"] = get_all_models(FREE_MODEL_TEST_KEY, "chat")
-    models["embedding"] = get_all_models(FREE_MODEL_TEST_KEY, "embedding")
-    models["image"] = get_all_models(FREE_MODEL_TEST_KEY, "text-to-image")
-    
-    models["free_text"] = []
-    models["free_embedding"] = []
-    models["free_image"] = []
-
-    ban_models = []
-    ban_models_str = os.environ.get("BAN_MODELS")
-    if ban_models_str:
-        try:
-            ban_models = json.loads(ban_models_str)
-            if not isinstance(ban_models, list):
-                logging.warning("环境变量 BAN_MODELS 格式不正确，应为 JSON 数组。")
-                ban_models = []
-        except json.JSONDecodeError:
-            logging.warning("环境变量 BAN_MODELS JSON 解析失败，请检查格式。")
-    
-    models["text"] = [model for model in models["text"] if model not in ban_models]
-    models["embedding"] = [model for model in models["embedding"] if model not in ban_models]
-    models["image"] = [model for model in models["image"] if model not in ban_models]
-
-    model_types = [
-        ("text", "chat"),
-        ("embedding", "embedding"),
-        ("image", "image")
-    ]
-    
-    for model_type, test_type in model_types:
-        with concurrent.futures.ThreadPoolExecutor(max_workers=10000) as executor:
-            future_to_model = {
-                executor.submit(
-                    test_model_availability,
-                    FREE_MODEL_TEST_KEY,
-                    model,
-                    test_type
-                ): model for model in models[model_type]
-            }
-            
-            for future in concurrent.futures.as_completed(future_to_model):
-                model = future_to_model[future]
-                try:
-                    is_free = future.result()
-                    if is_free:
-                        models[f"free_{model_type}"].append(model)
-                except Exception as exc:
-                    logging.error(f"{model_type}模型 {model} 测试生成异常: {exc}")
+    models["text"] = get_all_models(FREE_MODEL_TEST_KEY)
 
-    for model_type in ["text", "embedding", "image"]:
+    for model_type in ["text"]:
         logging.info(f"所有{model_type}模型列表：{models[model_type]}")
         logging.info(f"免费{model_type}模型列表：{models[f'free_{model_type}']}")
 
-
 def load_keys():
     global key_status
     for status in key_status:
@@ -319,7 +271,7 @@ def process_key(key, test_model):
             else:
                 return "unverified"
 
-def get_all_models(api_key, sub_type):
+def get_all_models(api_key):
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json"
@@ -327,8 +279,7 @@ def get_all_models(api_key, sub_type):
     try:
         response = session.get(
             MODELS_ENDPOINT,
-            headers=headers,
-            params={"sub_type": sub_type}
+            headers=headers
         )
         response.raise_for_status()
         data = response.json()