diff --git a/apps/api/src/services/chat_manager.py b/apps/api/src/services/chat_manager.py
index 42240609..78d2c2b2 100644
--- a/apps/api/src/services/chat_manager.py
+++ b/apps/api/src/services/chat_manager.py
@@ -90,6 +90,21 @@ class ChatManager:
logger.warning("openclaw_chat_failed", error="GEMINI_API_KEY not configured")
return None
+ # 月費用上限檢查 ($10 USD)
+ MONTHLY_LIMIT_USD = 10.0
+ from src.core.redis_client import get_redis
+ from src.utils.timezone import now_taipei
+ redis = get_redis()
+ month_key = f"gemini_cost:{now_taipei().strftime('%Y-%m')}"
+ try:
+ current_cost = float(await redis.get(month_key) or 0)
+ except Exception:
+ current_cost = 0.0
+
+ if current_cost >= MONTHLY_LIMIT_USD:
+ logger.warning("openclaw_gemini_monthly_limit_reached", current_usd=current_cost, limit_usd=MONTHLY_LIMIT_USD)
+ return f"🔴 OpenClaw 本月 Gemini 用量已達上限 ${MONTHLY_LIMIT_USD} USD(已用 ${current_cost:.4f})"
+
# Gemini 1.5 Flash: 快速、便宜
model = "gemini-1.5-flash"
try:
@@ -106,14 +121,22 @@ class ChatManager:
data = resp.json()
text = data["candidates"][0]["content"]["parts"][0]["text"].strip()
- # Token/費用統計
+ # Token/費用統計 + 累計到 Redis
usage = data.get("usageMetadata", {})
in_tok = usage.get("promptTokenCount", 0)
out_tok = usage.get("candidatesTokenCount", 0)
cost = (in_tok * 0.000000075) + (out_tok * 0.0000003)
- logger.info("openclaw_gemini_usage", in_tokens=in_tok, out_tokens=out_tok, cost_usd=round(cost, 6))
+ new_total = current_cost + cost
- return f"{text}\n\n📊 {in_tok+out_tok} tokens | ${cost:.4f}"
+ try:
+ await redis.set(month_key, str(round(new_total, 6)), ex=40 * 24 * 3600) # 40天 TTL
+ except Exception:
+ pass
+
+ logger.info("openclaw_gemini_usage", in_tokens=in_tok, out_tokens=out_tok,
+ cost_usd=round(cost, 6), monthly_total_usd=round(new_total, 4))
+
+ return f"{text}\n\n📊 {in_tok+out_tok} tokens | ${cost:.4f} | 本月累計 ${new_total:.4f}"
except Exception as e:
logger.warning("openclaw_chat_failed", error=str(e))
return None