Update app.py

app.py

@@ -47,6 +47,8 @@ def get_credit_summary(api_key):
         logging.error(f"total_balance 无法转换为浮点数，API Key：{api_key}，错误信息：{e}")
         return None
 
+FREE_MODEL_TEST_KEY = "sk-bmjbjzleaqfgtqfzmcnsbagxrlohriadnxqrzfocbizaxukw"  # 硬编码免费模型测试 KEY
+
 def test_model_availability(api_key, model_name):
     """
     测试指定的模型是否可用。
@@ -65,18 +67,31 @@ def test_model_availability(api_key, model_name):
                                      "stream": False
                                  },
                                  timeout=10)
-        # 检查是否是429错误
-        if response.status_code == 429:
+        # 正常返回或返回 429 都被认为是免费模型
+        if response.status_code == 429 or response.status_code < 500 :
             return True
-      
-        response.json()  # 尝试解析 JSON 响应
-        return True
+        else:
+            return False
     except requests.exceptions.RequestException as e:
         logging.error(f"测试模型 {model_name} 可用性失败，API Key：{api_key}，错误信息：{e}")
         return False
-    except ValueError:
-        logging.error(f"测试模型 {model_name} 可用性失败，API Key：{api_key}，响应不是有效的 JSON 格式")
-        return False
+
+def refresh_models():
+    """
+    刷新模型列表和免费模型列表。
+    """
+    global all_models, free_models
+
+    # 使用 FREE_MODEL_TEST_KEY 获取所有模型列表
+    all_models = get_all_models(FREE_MODEL_TEST_KEY)
+
+    free_models = []
+    for model in all_models:
+        if test_model_availability(FREE_MODEL_TEST_KEY, model):
+            free_models.append(model)
+
+    logging.info(f"所有模型列表：{all_models}")
+    logging.info(f"免费模型列表：{free_models}")
 
 def load_keys():
     """
@@ -148,30 +163,6 @@ def get_all_models(api_key):
         logging.error(f"解析模型列表失败，API Key：{api_key}，错误信息：{e}")
         return []
 
-def refresh_models():
-    """
-    刷新模型列表和免费模型列表。
-    """
-    global all_models, free_models
-  
-    # 使用 valid_keys_global 中的第一个 key 获取完整模型列表
-    if valid_keys_global:
-        all_models = get_all_models(valid_keys_global[0])
-    else:
-        logging.warning("没有有效的key，无法获取完整模型列表。")
-        all_models = []
-
-    # 从环境变量读取免费模型列表
-    free_models_str = os.environ.get("FREE_MODELS")
-    if free_models_str:
-        free_models = [model.strip() for model in free_models_str.split(',')]
-    else:
-        logging.warning("环境变量 FREE_MODELS 未设置或为空，免费模型列表将为空。")
-        free_models = []
-
-    logging.info(f"所有模型列表：{all_models}")
-    logging.info(f"免费模型列表：{free_models}")
-
 def determine_request_type(model_name):
     """
     根据用户请求的模型判断请求类型。