mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
feat(grok): start the in-container SDK server + budget feed (Claude parity)
The keystone of the Grok parity work (CEO's "take Claude as baseline, create what's missing" call): the one-shot Grok container now starts the same SDK server the Claude path runs, so the per-verb circuit breaker (the flow/do MCP servers already POST /verb/attempted to it), the per-session budget/loop counters, the terminal-verb tracking, and the SessionEnd post-mortem all work on Grok instead of being silently absent. - entrypoint: launch roboco.agent_sdk.server (bare venv python, not `uv run` which would re-sync the drifted clone lock and stall), wait for /health, reset counters; run opencode WITHOUT exec so the script regains control to run the post-mortem and the silent-exit substitute after the run returns. - budget-feed.js: opencode plugin that gates on /budget/status in tool.execute.before (halt/loop deny — the only place to stop a runaway one-shot run; opencode has no PostToolUse-deny) and records the executed tool + args-hash in tool.execute.after. Fail-open; bare-verb normalization for MCP-namespaced terminal verbs. - silent-exit substitute: on a graceful exit with no terminal verb the entrypoint posts /terminal/force_substitute so the task isn't left stuck claimed/in_progress (Stop-hook parity at the boundary). - opencode_config: wire budget-feed into the plugin array; add ROBOCO_OPENCODE_EXTRA_PLUGINS so per-image role tool plugins load scoped to one role; read the per-role ROBOCO_GROK_EDIT_PERMISSION. Targeted gate green (ruff/mypy/xenon + opencode_config tests; node --check on the plugins; bash -n on the entrypoint).
This commit is contained in:
@@ -0,0 +1,108 @@
|
||||
// opencode plugin — budget / loop / terminal feed for one-shot RoboCo Grok agents.
|
||||
//
|
||||
// Ports the Claude PostToolUse budget hook (docker/scripts/post-tool-budget-hook.sh)
|
||||
// and the Stop hook's terminal-tool tracking (docker/scripts/stop-hook.sh) to
|
||||
// opencode's tool.execute.{before,after}. The in-container SDK server
|
||||
// (roboco.agent_sdk.server, :9000) is the same long-lived process the Claude
|
||||
// path runs — the grok entrypoint starts it before `opencode run`. The flow/do
|
||||
// MCP servers already POST /verb/attempted to it for the per-verb circuit
|
||||
// breaker, so starting it + this feed restores the budget/loop/terminal cluster
|
||||
// on Grok (Claude-parity, the CEO's "create what's missing" call).
|
||||
//
|
||||
// before: read /budget/status and DENY (throw) on a hard halt, or a loop with
|
||||
// loop_action=halt. opencode has no PostToolUse-deny, so the pre-exec
|
||||
// gate is the only place to stop a runaway one-shot run from burning
|
||||
// the cost cap mid-turn. The loop trips one call later than Claude
|
||||
// (record is in `after`) but still halts the burn.
|
||||
// after: record the executed tool on /terminal/tool_recorded (so a graceful
|
||||
// terminal verb is recognized) and /budget/tool_called (advances the
|
||||
// breaker/loop counters and feeds the post-exit post-mortem).
|
||||
//
|
||||
// Fail-open everywhere: a missing / slow / non-2xx SDK never blocks the agent.
|
||||
// The interactive serve images (intake / secretary) own :9000 for the human-turn
|
||||
// receiver and run NO SDK server, so these POSTs 404 there and are ignored —
|
||||
// those roles don't claim tasks or loop on verbs, so they need no budget feed.
|
||||
|
||||
const SDK_URL = process.env.ROBOCO_SDK_URL || "http://localhost:9000";
|
||||
|
||||
async function sdk(method, path, body) {
|
||||
try {
|
||||
const res = await fetch(`${SDK_URL}${path}`, {
|
||||
method,
|
||||
headers: body ? { "Content-Type": "application/json" } : undefined,
|
||||
body: body ? JSON.stringify(body) : undefined,
|
||||
signal: AbortSignal.timeout(2000),
|
||||
});
|
||||
if (!res.ok) return null;
|
||||
return await res.json();
|
||||
} catch {
|
||||
return null; // fail-open — never block the agent on SDK reachability
|
||||
}
|
||||
}
|
||||
|
||||
// Canonical, dependency-free serialization (sorted keys) so the SDK's loop
|
||||
// detector sees identical (tool, args) calls as identical. Need NOT match the
|
||||
// Claude hook's sha256 — only be stable within one session.
|
||||
function canonical(value) {
|
||||
if (value === null || typeof value !== "object") return JSON.stringify(value);
|
||||
if (Array.isArray(value)) return "[" + value.map(canonical).join(",") + "]";
|
||||
const keys = Object.keys(value).sort();
|
||||
return (
|
||||
"{" +
|
||||
keys.map((k) => JSON.stringify(k) + ":" + canonical(value[k])).join(",") +
|
||||
"}"
|
||||
);
|
||||
}
|
||||
|
||||
function argsHash(args) {
|
||||
const s = canonical(args ?? {});
|
||||
let h = 0x811c9dc5; // FNV-1a, 32-bit
|
||||
for (let i = 0; i < s.length; i++) {
|
||||
h ^= s.charCodeAt(i);
|
||||
h = Math.imul(h, 0x01000193);
|
||||
}
|
||||
return (h >>> 0).toString(16).padStart(8, "0");
|
||||
}
|
||||
|
||||
// opencode namespaces an MCP tool as "<server>_<verb>" (or "<server>.<verb>");
|
||||
// the Claude path's verbs arrive bare. Strip a known roboco-* server prefix so
|
||||
// the SDK recognizes a terminal verb (i_am_idle / i_am_done / ...) — the SDK's
|
||||
// own "__"-split is a no-op on the already-bare verb this returns.
|
||||
// UNVERIFIED-LIVE: opencode's exact MCP tool-name shape; the strip is defensive.
|
||||
function bareVerb(tool) {
|
||||
const mcp = tool.match(/^mcp__[a-z0-9-]+__(.+)$/);
|
||||
if (mcp) return mcp[1];
|
||||
const pref = tool.match(/^roboco-[a-z-]+[_.](.+)$/);
|
||||
if (pref) return pref[1];
|
||||
return tool;
|
||||
}
|
||||
|
||||
export default async () => {
|
||||
return {
|
||||
"tool.execute.before": async (input) => {
|
||||
const status = await sdk("GET", "/budget/status", null);
|
||||
if (!status) return; // fail-open
|
||||
if (status.halt) {
|
||||
throw new Error(
|
||||
`[Halt] tool budget exhausted (${status.total}/${status.halt_threshold}). ` +
|
||||
"Stop now — release the task with unclaim() or i_am_idle().",
|
||||
);
|
||||
}
|
||||
if (status.loop && status.loop_action === "halt") {
|
||||
throw new Error(
|
||||
"[Loop] same tool+args repeated in window — halting. " +
|
||||
"Release the task with unclaim() or stop with i_am_idle().",
|
||||
);
|
||||
}
|
||||
},
|
||||
"tool.execute.after": async (input) => {
|
||||
const raw = String(input?.tool || "");
|
||||
if (!raw) return;
|
||||
await sdk("POST", "/terminal/tool_recorded", { tool: bareVerb(raw) });
|
||||
await sdk("POST", "/budget/tool_called", {
|
||||
tool: bareVerb(raw),
|
||||
args_hash: argsHash(input?.args),
|
||||
});
|
||||
},
|
||||
};
|
||||
};
|
||||
Reference in New Issue
Block a user