fix(ai-gateway): 修复 BUDGET_DAILY_TOKEN_LIMIT 默认 0 导致全站 429

Author: Aparencia

server/ai-gateway/.env.example

@@ -0,0 +1,10 @@
+# ===========================================
+# 熵减 AI 网关 — 环境变量模板
+# ===========================================
+# 复制此文件为 .env 并填入实际值
+# 切勿将含有真实密钥的 .env 文件提交到 Git
+
+# --- 预算控制（每日 token/费用上限） ---
+# 0 = 不限制；默认 free 档 200K tokens / ¥0.5
+BUDGET_DAILY_TOKEN_LIMIT=200000
+BUDGET_DAILY_COST_LIMIT=0.5

server/ai-gateway/cost/budget.py

@@ -77,7 +77,7 @@ async def dispatch(self, request: Request, call_next):
         tracker = get_cost_tracker()
         usage = await tracker.get_user_daily_usage(user_id)
 
-        if usage["tokens"] >= DAILY_TOKEN_LIMIT:
+        if DAILY_TOKEN_LIMIT > 0 and usage["tokens"] >= DAILY_TOKEN_LIMIT:
             logger.warning(
                 "预算拦截(token): user=%s, tokens=%d/%d",
                 user_id, usage["tokens"], DAILY_TOKEN_LIMIT,
@@ -94,7 +94,7 @@ async def dispatch(self, request: Request, call_next):
                 },
             )
 
-        if usage["yuan"] >= daily_cost_limit:
+        if daily_cost_limit > 0 and usage["yuan"] >= daily_cost_limit:
             logger.warning(
                 "预算拦截(cost): user=%s, yuan=%.2f/%.2f",
                 user_id, usage["yuan"], daily_cost_limit,

server/docker-compose.prod.yml

@@ -121,9 +121,9 @@ services:
       SUPABASE_SERVICE_KEY: ${SUPABASE_SERVICE_KEY:-}
       # 开发者白名单（邮箱或 user_id，逗号分隔）：完全豁免配额/费用限制并赋予 lifetime 身份
       DEV_USER_IDS: ${DEV_USER_IDS:-}
-      # 预算控制（每日 token/费用上限，0=不限制）
-      BUDGET_DAILY_TOKEN_LIMIT: ${BUDGET_DAILY_TOKEN_LIMIT:-0}
-      BUDGET_DAILY_COST_LIMIT: ${BUDGET_DAILY_COST_LIMIT:-0}
+      # 预算控制（0 = 不限制；默认 free 档 200K tokens / ¥0.5）
+      BUDGET_DAILY_TOKEN_LIMIT: ${BUDGET_DAILY_TOKEN_LIMIT:-200000}
+      BUDGET_DAILY_COST_LIMIT: ${BUDGET_DAILY_COST_LIMIT:-0.5}
       # 付费系统（面包多，可选）
       PAYMENT_PROVIDER: ${PAYMENT_PROVIDER:-}
       PAYMENT_API_KEY: ${PAYMENT_API_KEY:-}