hmo/AgentsMeeting
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Kimi/SenseNova usage not auto-refreshed — auto-timer only collected OCG; extract _collect_kimi_usage/_collect_sensenova_usage, refresh every 30min in timer

Browse Source
This commit is contained in:
hmo
2026-08-15 12:45:12 +08:00
parent 1febfd0cb5
commit fd71e8ffcd
1 changed files with 70 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
gateway/scripts/dashboard.py
+45 -18
View File
@@ -1462,8 +1462,32 @@ def api_usage_refresh():
if proc.returncode != 0 and proc.stderr:
log.warning(f"usage_collector stderr:\n{proc.stderr[-1000:]}")
# 2. Kimi 远程采集（SSH 到 Windows）
# 2. Kimi 远程采集（SSH 到 Windows）+ scp 回 246
try:
_collect_kimi_usage()
except Exception as e:
log.warning(f"usage: Kimi collection failed: {e}")
# 3. SenseNova 远程采集（SSH 到 Windows，CDP 刮 console 余量）
try:
_collect_sensenova_usage()
except Exception as e:
log.warning(f"usage: SenseNova collection failed: {e}")
except Exception as e:
log.error(f"usage: collector crash: {e}")
finally:
_usage_collector_running = False
t = threading.Thread(target=_runner, name="usage_collector", daemon=True)
t.start()
return jsonify({"ok": True, "message": "collection triggered (~15s), poll /api/usage shortly"})
def _collect_kimi_usage():
"""SSH 到 Windows 跑 usage_collector_kimi.py，scp 结果回 246。
被 /api/usage/refresh 和 usage auto-timer 共用，保证 Kimi 数据新鲜。
"""
log.info("usage: starting Kimi collector via SSH")
kimi_cmd = [
"ssh", "-o", "StrictHostKeyChecking=no", "-o", "ConnectTimeout=10",
@@ -1475,7 +1499,7 @@ def api_usage_refresh():
if proc2.returncode != 0 and proc2.stderr:
log.warning(f"usage_kimi stderr: {proc2.stderr[-300:]}")
# 3. scp Kimi 结果回 246
# scp Kimi 结果回 246
scp_cmd = [
"scp", "-o", "StrictHostKeyChecking=no", "-o", "ConnectTimeout=10",
"hmo@192.168.1.16:D:/F/NewI/opencode/daily-workspace/projects/AgentsMeeting/gateway/temp/usage_stats_kimi.json",
@@ -1484,8 +1508,9 @@ def api_usage_refresh():
proc3 = subprocess.run(scp_cmd, capture_output=True, timeout=15)
log.info(f"usage: Kimi scp done (rc={proc3.returncode})")
# 4. SenseNova 远程采集（SSH 到 Windows，CDP 刮 console 余量）
try:
def _collect_sensenova_usage():
"""SSH 到 Windows 跑 usage_collector_sensenova.py，scp 结果回 246。"""
log.info("usage: starting SenseNova collector via SSH")
sn_cmd = [
"ssh", "-o", "StrictHostKeyChecking=no", "-o", "ConnectTimeout=10",
@@ -1502,18 +1527,6 @@ def api_usage_refresh():
str(TEMP_DIR / "usage_stats_sensenova.json")
], capture_output=True, timeout=15)
log.info(f"usage: SenseNova scp done (rc={scp4.returncode})")
except Exception as e:
log.warning(f"usage: SenseNova collection failed: {e}")
except Exception as e:
log.warning(f"usage: Kimi collection failed: {e}")
except Exception as e:
log.error(f"usage: collector crash: {e}")
finally:
_usage_collector_running = False
t = threading.Thread(target=_runner, name="usage_collector", daemon=True)
t.start()
return jsonify({"ok": True, "message": "collection triggered (~15s), poll /api/usage shortly"})
# ── Key Registry ──────────────────────────────────────────────
@@ -1625,9 +1638,11 @@ def api_keys():
_USAGE_AUTO_INTERVAL = 300
def _start_usage_auto_timer():
"""后台线程：每 5 分钟自动采集 + cookie 保活。246 自足，不依赖 Windows。"""
"""后台线程：OCG 每 5 分钟自动采集；Kimi/SenseNova 每 30 分钟采集（SSH 到 Windows）。"""
_KIMI_INTERVAL_MULT = 6 # 每 6 个 OCG 周期（30min）采一次 Kimi/SenseNova
def _loop():
time.sleep(15) # initial delay: let dashboard stabilize
tick = 0
while True:
try:
global _usage_collector_running
@@ -1639,6 +1654,18 @@ def _start_usage_auto_timer():
log.info(f"usage auto-timer: rc={proc.returncode}")
if proc.returncode != 0 and proc.stderr:
log.warning(f"usage auto-timer stderr:\n{proc.stderr[-800:]}")
# Kimi/SenseNova：每 30 分钟刷新一次（SSH 到 Windows + CDP，较重）
tick += 1
if tick % _KIMI_INTERVAL_MULT == 0:
try:
_collect_kimi_usage()
except Exception as e:
log.warning(f"usage auto-timer: Kimi failed: {e}")
try:
_collect_sensenova_usage()
except Exception as e:
log.warning(f"usage auto-timer: SenseNova failed: {e}")
_usage_collector_running = False
except Exception as e:
log.error(f"usage auto-timer error: {e}")
@@ -1647,7 +1674,7 @@ def _start_usage_auto_timer():
t = threading.Thread(target=_loop, name="usage_auto_timer", daemon=True)
t.start()
log.info(f"usage auto-timer started (interval={_USAGE_AUTO_INTERVAL}s)")
log.info(f"usage auto-timer started (interval={_USAGE_AUTO_INTERVAL}s, kimi every {_KIMI_INTERVAL_MULT * _USAGE_AUTO_INTERVAL}s)")
# ════════════════════════════════════════════════════════════