Update app.py

app.py

@@ -22,7 +22,7 @@ logging.basicConfig(level=logging.INFO,
 
 API_ENDPOINT = "https://api.deepseek.com/user/balance"
 TEST_MODEL_ENDPOINT = "https://api.deepseek.com/v1/chat/completions"
-MODELS_ENDPOINT = "https://api.deepseek.com/v1/models"
+MODELS_ENDPOINT = "https://api.deepseek.com/models"
 
 app = Flask(__name__)
 
@@ -103,86 +103,6 @@ def get_usd_to_cny_rate():
         logging.error(f"获取美元兑人民币汇率失败，错误信息：{e}")
         return None
 
-def test_model_availability(api_key, model_name):
-    """
-    测试指定的模型是否可用。
-    """
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
-    try:
-        response = requests.post(
-            TEST_MODEL_ENDPOINT,
-            headers=headers,
-            json={
-                "model": model_name,
-                "messages": [{"role": "user", "content": "hi"}],
-                "max_tokens": 5,
-                "stream": False
-            },
-            timeout=5
-        )
-        if response.status_code == 429 or response.status_code == 200:
-            return True
-        else:
-            return False
-    except requests.exceptions.RequestException as e:
-        logging.error(
-            f"测试模型 {model_name} 可用性失败，"
-            f"API Key：{api_key}，错误信息：{e}"
-        )
-        return False
-
-def refresh_models():
-    """
-    刷新模型列表和免费模型列表。
-    """
-    global text_models, free_text_models
-
-    text_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
-    free_text_models = []
-
-    ban_models_str = os.environ.get("BAN_MODELS")
-    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 解析失败，请检查格式。"
-            )
-            ban_models = []
-    
-    text_models = [model for model in text_models if model not in ban_models]
-
-    with concurrent.futures.ThreadPoolExecutor(
-        max_workers=1000
-    ) as executor:
-        future_to_model = {
-            executor.submit(
-                test_model_availability,
-                FREE_MODEL_TEST_KEY,
-                model
-            ): model for model in text_models
-        }
-        for future in concurrent.futures.as_completed(future_to_model):
-            model = future_to_model[future]
-            try:
-                is_free = future.result()
-                if is_free:
-                    free_text_models.append(model)
-            except Exception as exc:
-                logging.error(f"模型 {model} 测试生成异常: {exc}")
-
-    logging.info(f"所有文本模型列表：{text_models}")
-    logging.info(f"免费文本模型列表：{free_text_models}")
-
 def load_keys():
     """
     从环境变量中加载 keys，进行去重，
@@ -190,12 +110,6 @@ def load_keys():
     然后记录到日志中。
     使用线程池并发处理每个 key。
     """
-    keys_str = os.environ.get("KEYS")
-    test_model = os.environ.get(
-        "TEST_MODEL",
-        "Pro/google/gemma-2-9b-it"
-    )
-
     if keys_str:
         keys = [key.strip() for key in keys_str.split(',')]
         unique_keys = list(set(keys))