mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-08 23:33:16 +08:00
Trigger async runtime prewarm immediately after Admin MCP tools sync.
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
parent
e095901f61
commit
91c5a1a754
3 changed files with 31 additions and 5 deletions
|
|
@ -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()
|
||||
|
|
|
|||
|
|
@ -11,6 +11,8 @@ function summarizeApiStatus(prefix, r) {
|
|||
const err = String(r.error_code || r.error || r.detail || "error").slice(0, 120);
|
||||
return `${p} fail: ${err}`;
|
||||
}
|
||||
const prewarmTail =
|
||||
r.prewarm && typeof r.prewarm === "object" && r.prewarm.accepted ? " · prewarm=async" : "";
|
||||
if (Array.isArray(r.tools)) {
|
||||
const names = r.tools
|
||||
.map((x) => String((x && (x.tool_name || x.name)) || "").trim())
|
||||
|
|
@ -18,7 +20,7 @@ function summarizeApiStatus(prefix, r) {
|
|||
const shown = names.slice(0, 10).join(", ");
|
||||
const more = names.length > 10 ? ` (+${names.length - 10})` : "";
|
||||
const sid = r.server_id ? ` server=${r.server_id}` : "";
|
||||
return `${p} ok${sid} · ${names.length} tools${shown ? `: ${shown}${more}` : ""}`;
|
||||
return `${p} ok${sid} · ${names.length} tools${shown ? `: ${shown}${more}` : ""}${prewarmTail}`;
|
||||
}
|
||||
if (Array.isArray(r.tool_names)) {
|
||||
const names = r.tool_names.map((x) => String(x || "").trim()).filter(Boolean);
|
||||
|
|
@ -26,18 +28,18 @@ function summarizeApiStatus(prefix, r) {
|
|||
const more = names.length > 10 ? ` (+${names.length - 10})` : "";
|
||||
const n = Number(r.synced_tools || names.length || 0);
|
||||
const sid = r.server_id ? ` server=${r.server_id}` : "";
|
||||
return `${p} ok${sid} · ${n} tools${shown ? `: ${shown}${more}` : ""}`;
|
||||
return `${p} ok${sid} · ${n} tools${shown ? `: ${shown}${more}` : ""}${prewarmTail}`;
|
||||
}
|
||||
if (Array.isArray(r.items)) {
|
||||
return `${p} ok · ${r.items.length} item(s)`;
|
||||
return `${p} ok · ${r.items.length} item(s)${prewarmTail}`;
|
||||
}
|
||||
if (Array.isArray(r.results)) {
|
||||
const okN = r.results.filter((x) => x && x.ok !== false).length;
|
||||
return `${p} ok · ${okN}/${r.results.length} result(s)`;
|
||||
return `${p} ok · ${okN}/${r.results.length} result(s)${prewarmTail}`;
|
||||
}
|
||||
if (r.status != null) {
|
||||
const sid = r.server_id ? ` server=${r.server_id}` : "";
|
||||
return `${p} ${String(r.status)}${sid}`;
|
||||
return `${p} ${String(r.status)}${sid}${prewarmTail}`;
|
||||
}
|
||||
const bits = [];
|
||||
for (const k of ["ok", "server_id", "enabled", "deleted", "synced_tools", "compat_mode"]) {
|
||||
|
|
@ -45,6 +47,7 @@ function summarizeApiStatus(prefix, r) {
|
|||
bits.push(`${k}=${String(r[k]).slice(0, 40)}`);
|
||||
if (bits.length >= 5) break;
|
||||
}
|
||||
if (prewarmTail) bits.push("prewarm=async");
|
||||
return bits.length ? `${p} ${bits.join(" · ")}` : `${p} ok`;
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -196,6 +196,7 @@ class McpAdminApiTests(unittest.TestCase):
|
|||
names = body.get("tool_names") or []
|
||||
self.assertIn("ping", names)
|
||||
self.assertEqual(int(body.get("synced_tools") or 0), len(names))
|
||||
self.assertTrue((body.get("prewarm") or {}).get("accepted"))
|
||||
stored = store.list_mcp_server_tools(server_id="dummy")
|
||||
self.assertTrue(any(str(t.get("tool_name") or "") == "ping" for t in stored))
|
||||
|
||||
|
|
@ -227,6 +228,7 @@ class McpAdminApiTests(unittest.TestCase):
|
|||
self.assertNotIn("tools", body)
|
||||
names = body.get("tool_names") or []
|
||||
self.assertIn("bailian_webparser_parse", names)
|
||||
self.assertTrue((body.get("prewarm") or {}).get("accepted"))
|
||||
stored = store.list_mcp_server_tools(server_id="webparser-compat")
|
||||
self.assertTrue(any(str(t.get("tool_name") or "") == "bailian_webparser_parse" for t in stored))
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue