fix: stop converting Kimi balance and switch Qwen endpoint to business.aliyuncs.com

modules/balance_client.py

@@ -1,7 +1,7 @@
 """
 Admin balance fetchers for Kimi (Moonshot), DeepSeek, and Qwen (Aliyun BSS).
 
-Credentials (read from environment):
+Credentials (优先读取仓库根目录 .env，其次环境变量):
   - MOONSHOT_API_KEY            : Bearer token for Kimi
   - DEEPSEEK_API_KEY            : Bearer token for DeepSeek
   - ALIYUN_ACCESS_KEY_ID        : AccessKey ID for Aliyun (Qwen billing)
@@ -18,11 +18,44 @@ import json
 import os
 import time
 import uuid
+from pathlib import Path
 from typing import Any, Dict, Tuple
-from urllib import parse, request, error
+from urllib import parse, request
+
+# -------- .env 读取助手 --------
+_DOTENV_CACHE: Dict[str, str] | None = None
 
 
-USD_CNY_RATE = float(os.environ.get("USD_CNY_RATE", "7.1"))
+def _read_dotenv() -> Dict[str, str]:
+    global _DOTENV_CACHE
+    if _DOTENV_CACHE is not None:
+        return _DOTENV_CACHE
+    path = Path(__file__).resolve().parents[1] / ".env"
+    result: Dict[str, str] = {}
+    if path.exists():
+        try:
+            for raw_line in path.read_text(encoding="utf-8").splitlines():
+                line = raw_line.strip()
+                if not line or line.startswith("#") or "=" not in line:
+                    continue
+                key, value = line.split("=", 1)
+                key = key.strip()
+                value = value.strip().strip('"').strip("'")
+                if key:
+                    result[key] = value
+        except Exception:
+            pass
+    _DOTENV_CACHE = result
+    return result
+
+
+def _env(key: str, default: str | None = None) -> str | None:
+    if key in os.environ:
+        return os.environ[key]
+    return _read_dotenv().get(key, default)
+
+
+USD_CNY_RATE = float(_env("USD_CNY_RATE", "7.1") or "7.1")
 
 
 def _http_get(url: str, headers: Dict[str, str] | None = None, timeout: int = 8) -> Tuple[Dict[str, Any] | None, str | None]:
@@ -38,11 +71,17 @@ def _http_get(url: str, headers: Dict[str, str] | None = None, timeout: int = 8)
 
 # -------- Kimi (Moonshot) --------
 def fetch_kimi_balance() -> Dict[str, Any]:
-    api_key = os.environ.get("MOONSHOT_API_KEY")
+    api_key = _env("MOONSHOT_API_KEY") or _env("API_KEY_KIMI") or _env("AGENT_API_KEY")
     if not api_key:
-        return {"success": False, "error": "MOONSHOT_API_KEY 未设置"}
+        return {"success": False, "error": "Kimi 密钥未设置（MOONSHOT_API_KEY / API_KEY_KIMI / AGENT_API_KEY）"}
 
-    url = "https://api.moonshot.ai/v1/users/me/balance"
+    base = (
+        _env("API_BASE_KIMI")
+        or _env("MOONSHOT_BASE_URL")
+        or _env("AGENT_API_BASE_URL")
+        or "https://api.moonshot.ai/v1"
+    ).rstrip("/")
+    url = f"{base}/users/me/balance"
     payload, err = _http_get(url, headers={"Authorization": f"Bearer {api_key}"})
     if err:
         return {"success": False, "error": err}
@@ -54,12 +93,10 @@ def fetch_kimi_balance() -> Dict[str, Any]:
         cash = float(data.get("cash_balance", 0))
         return {
             "success": True,
-            "currency": "USD",
+            "currency": payload.get("currency") or "CNY",
             "available": available,
-            "available_cny": round(available * USD_CNY_RATE, 2),
             "voucher": voucher,
             "cash": cash,
-            "rate": USD_CNY_RATE,
             "raw": payload,
         }
     except Exception as exc:  # pragma: no cover
@@ -68,9 +105,9 @@ def fetch_kimi_balance() -> Dict[str, Any]:
 
 # -------- DeepSeek --------
 def fetch_deepseek_balance() -> Dict[str, Any]:
-    api_key = os.environ.get("DEEPSEEK_API_KEY")
+    api_key = _env("DEEPSEEK_API_KEY") or _env("API_KEY_DEEPSEEK")
     if not api_key:
-        return {"success": False, "error": "DEEPSEEK_API_KEY 未设置"}
+        return {"success": False, "error": "DeepSeek 密钥未设置（DEEPSEEK_API_KEY / API_KEY_DEEPSEEK）"}
 
     url = "https://api.deepseek.com/user/balance"
     payload, err = _http_get(url, headers={"Authorization": f"Bearer {api_key}"})
@@ -110,8 +147,8 @@ def _sign(params: Dict[str, Any], secret: str) -> str:
 
 
 def fetch_qwen_balance() -> Dict[str, Any]:
-    ak = os.environ.get("ALIYUN_ACCESS_KEY_ID")
+    ak = _env("ALIYUN_ACCESS_KEY_ID")
-    sk = os.environ.get("ALIYUN_ACCESS_KEY_SECRET")
+    sk = _env("ALIYUN_ACCESS_KEY_SECRET")
     if not ak or not sk:
         return {"success": False, "error": "缺少 ALIYUN_ACCESS_KEY_ID / ALIYUN_ACCESS_KEY_SECRET"}
 
@@ -128,7 +165,10 @@ def fetch_qwen_balance() -> Dict[str, Any]:
     }
     signature = _sign(params, sk)
     params["Signature"] = signature
-    url = "https://bss.aliyuncs.com/?" + parse.urlencode(params)
+
+    # 按阿里云规范组装最终查询字符串（不可用 urlencode 的 quote_plus）
+    query = "&".join(f"{_percent_encode(k)}={_percent_encode(v)}" for k, v in params.items())
+    url = "https://business.aliyuncs.com/?" + query
 
     payload, err = _http_get(url)
     if err:
@@ -156,4 +196,3 @@ def fetch_all_balances() -> Dict[str, Any]:
         "deepseek": fetch_deepseek_balance(),
         "qwen": fetch_qwen_balance(),
     }
-