hansjone/oclaw
1
0
Fork 0
mirror of https://github.com/hansjone/oclaw.git synced 2026-10-09 04:40:45 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

Trigger async runtime prewarm immediately after Admin MCP tools sync.

Browse source 
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
oliver 2026-08-12 22:59:17 +08:00
parent e095901f61
commit 91c5a1a754
3 changed files with 31 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

21
interfaces/admin/routes.py
View file

@ -129,6 +129,23 @@ def _bailian_webparser_virtual_tools() -> list[dict[str, Any]]:
return bailian_webparser_virtual_tools()
def _spawn_async_runtime_prewarm(*, reason: str) -> dict[str, Any]:
"""Fire-and-forget prewarm after Admin MCP catalog changes (does not block the HTTP response)."""
import threading
reason_s = str(reason or "mcp_admin_sync").strip() or "mcp_admin_sync"
def _run() -> None:
try:
run_runtime_prewarm(reason=reason_s, store=get_assistant_store())
except Exception:
pass
th = threading.Thread(target=_run, name="oclaw-mcp-sync-prewarm", daemon=True)
th.start()
return {"accepted": True, "mode": "async", "reason": reason_s}
def _http_get_json(url: str, *, timeout: float = 8.0) -> dict[str, Any]:
req = urllib_request.Request(
url,
@ -3884,12 +3901,14 @@ def build_admin_router() -> APIRouter:
invalidate_tool_wire_cache(reason=f"mcp_admin_sync:{server_id}")
except Exception:
pass
prewarm = _spawn_async_runtime_prewarm(reason=f"mcp_admin_sync:{server_id}")
return {
"ok": True,
"server_id": server_id,
"synced_tools": len(norm),
"tool_names": [str(t.get("tool_name") or "") for t in norm if str(t.get("tool_name") or "").strip()],
"compat_mode": "bailian_webparser",
"prewarm": prewarm,
}
rt = mcp_runtime_for_row(row, store=store)
try:
@ -3919,11 +3938,13 @@ def build_admin_router() -> APIRouter:
invalidate_tool_wire_cache(reason=f"mcp_admin_sync:{server_id}")
except Exception:
pass
prewarm = _spawn_async_runtime_prewarm(reason=f"mcp_admin_sync:{server_id}")
return {
"ok": True,
"server_id": server_id,
"synced_tools": len(norm),
"tool_names": [str(t.get("tool_name") or "") for t in norm if str(t.get("tool_name") or "").strip()],
"prewarm": prewarm,
}
finally:
rt.stop()