Update app.py

app.py

@@ -20,10 +20,10 @@ time.tzset()
 logging.basicConfig(level=logging.INFO,
                     format='%(asctime)s - %(levelname)s - %(message)s')
 
-API_ENDPOINT = "https://api.siliconflow.cn/v1/user/info"
-TEST_MODEL_ENDPOINT = "https://api.siliconflow.cn/v1/chat/completions"
-MODELS_ENDPOINT = "https://api.siliconflow.cn/v1/models"
-EMBEDDINGS_ENDPOINT = "https://api.siliconflow.cn/v1/embeddings"
+API_ENDPOINT = "https://api.deepseek.cn/user/balance"
+TEST_MODEL_ENDPOINT = "https://api.deepseek.cn/v1/chat/completions"
+MODELS_ENDPOINT = "https://api.deepseek.cn/v1/models"
+EMBEDDINGS_ENDPOINT = "https://api.deepseek.cn/v1/embeddings"
 
 app = Flask(__name__)
 
@@ -48,7 +48,7 @@ data_lock = threading.Lock()
 
 def get_credit_summary(api_key):
     """
-    使用 API 密钥获取额度信息。
+    使用 API 密钥获取额度信息，并将美元余额转换为人民币。
     """
     headers = {
         "Authorization": f"Bearer {api_key}",
@@ -57,12 +57,55 @@ def get_credit_summary(api_key):
     try:
         response = requests.get(API_ENDPOINT, headers=headers)
         response.raise_for_status()
-        data = response.json().get("data", {})
-        total_balance = data.get("totalBalance", 0)
-        return {"total_balance": float(total_balance)}
+        data = response.json()
+        if not data.get("is_available", False):
+            logging.warning(f"API Key: {api_key} is not available.")
+            return None
+        
+        balance_infos = data.get("balance_infos", [])
+        total_balance_cny = 0.0
+        usd_balance = 0.0
+        for balance_info in balance_infos:
+            currency = balance_info.get("currency")
+            total_balance = float(balance_info.get("total_balance", 0))
+
+            if currency == "CNY":
+                total_balance_cny += total_balance
+            elif currency == "USD":
+                usd_balance = total_balance
+
+        try:
+            exchange_rate = get_usd_to_cny_rate()
+            if exchange_rate is not None:
+                total_balance_cny += usd_balance * exchange_rate
+            else:
+                 logging.warning(f"获取美元兑人民币汇率失败，无法转换美元余额，API Key：{api_key}")
+                 total_balance_cny += usd_balance * 7.2
+        except Exception as e:
+             logging.error(f"获取美元兑人民币汇率失败，API Key：{api_key}，错误信息：{e}")
+             total_balance_cny += usd_balance * 7.2 
+
+        return {"total_balance_cny": round(total_balance_cny, 2)}
     except requests.exceptions.RequestException as e:
         logging.error(f"获取额度信息失败，API Key：{api_key}，错误信息：{e}")
         return None
+    except Exception as e:
+         logging.error(f"处理额度信息失败，API Key：{api_key}，错误信息：{e}")
+         return None
+
+def get_usd_to_cny_rate():
+    """
+    获取美元兑人民币的汇率。
+    这里使用一个公共的汇率 API，你可以替换成你自己的。
+    """
+    try:
+        response = requests.get("https://api.exchangerate-api.com/v4/latest/USD")
+        response.raise_for_status()
+        data = response.json()
+        return data.get("rates", {}).get("CNY")
+    except requests.exceptions.RequestException as e:
+        logging.error(f"获取美元兑人民币汇率失败，错误信息：{e}")
+        return None
 
 FREE_MODEL_TEST_KEY = (
     "sk-bmjbjzleaqfgtqfzmcnsbagxrlohriadnxqrzfocbizaxukw"
@@ -552,38 +595,10 @@ def check_tokens():
 def list_models():
     if not check_authorization(request):
         return jsonify({"error": "Unauthorized"}), 401
-
-    detailed_models = []
     
-    for model in text_models:
-        detailed_models.append({
-            "id": model,
-            "object": "model",
-            "created": 1678888888,
-            "owned_by": "openai",
-            "permission": [
-                {
-                    "id": f"modelperm-{uuid.uuid4().hex}",
-                    "object": "model_permission",
-                    "created": 1678888888,
-                    "allow_create_engine": False,
-                    "allow_sampling": True,
-                    "allow_logprobs": True,
-                    "allow_search_indices": False,
-                    "allow_view": True,
-                    "allow_fine_tuning": False,
-                    "organization": "*",
-                    "group": None,
-                    "is_blocking": False
-                }
-            ],
-            "root": model,
-            "parent": None
-        })
-
-    for model in embedding_models:
-        detailed_models.append({
-            "id": model,
+    detailed_models = [
+        {
+            "id": "deepseek-chat",
             "object": "model",
             "created": 1678888888,
             "owned_by": "openai",
@@ -603,13 +618,11 @@ def list_models():
                     "is_blocking": False
                 }
             ],
-            "root": model,
+            "root": "deepseek-chat",
             "parent": None
-        })
-    
-    for model in image_models:
-        detailed_models.append({
-            "id": model,
+        },
+        {
+            "id": "deepseek-code",
             "object": "model",
             "created": 1678888888,
             "owned_by": "openai",
@@ -629,9 +642,10 @@ def list_models():
                     "is_blocking": False
                 }
             ],
-            "root": model,
+            "root": "deepseek-code",
             "parent": None
-        })
+        }
+    ]
 
     return jsonify({
         "success": True,