billion-context 0.1.140 → 0.1.142

package/dist/index.js

@@ -43467,6 +43467,10 @@ var require_lib = __commonJS({
   }
 });
 
+// src/index.ts
+import { realpathSync as realpathSync2 } from "fs";
+import { fileURLToPath as fileURLToPath8 } from "url";
+
 // node_modules/acp-kernel/dist/chunk-6RNVHY67.js
 import { createRequire } from "module";
 var require2 = createRequire(import.meta.url);
@@ -71415,24 +71419,20 @@ function withSearchContextConversationDescription(tools) {
     return copy;
   });
 }
-function handlePluginManifest(res) {
+function handlePluginManifest(res, config) {
+  const absorbOn = absorbEnabled(config);
+  const rulesOn = rulesEnabled(config);
   res.writeHead(200, { "content-type": "application/json" });
   res.end(JSON.stringify({
     ok: true,
     protocolVersion: PLUGIN_PROTOCOL_VERSION,
     proxy: "billion-context",
     version: VERSION2,
-    // Absorb and acp_rule are advertised alongside the four ACP tools.
-    // Both are host-registered opt-in (not in the kernel's
-    // ACP_TOOL_NAMES), so the manifest must list them explicitly.
-    // Per-session enablement is enforced at execution (isProxyToolFor /
-    // executeProxyTool), not here — the manifest has no request context to
-    // know which route will win.
-    toolNames: [...ACP_TOOL_NAMES, ABSORB_TOOL_NAME, RULE_TOOL_NAME],
+    toolNames: [...ACP_TOOL_NAMES, ...absorbOn ? [ABSORB_TOOL_NAME] : [], ...rulesOn ? [RULE_TOOL_NAME] : []],
     tools: {
-      anthropic: withSearchContextConversationDescription([...BILI_ACP_TOOLS_ANTHROPIC, ABSORB_TOOL, RULE_TOOL].map(withConversationIdParam)),
-      openai: withSearchContextConversationDescription([...BILI_ACP_TOOLS_OPENAI, ABSORB_TOOL_OPENAI, RULE_TOOL_OPENAI].map(withConversationIdParam)),
-      responses: withSearchContextConversationDescription([...BILI_ACP_TOOLS_RESPONSES, ABSORB_TOOL_RESPONSES, RULE_TOOL_RESPONSES].map(withConversationIdParam))
+      anthropic: withSearchContextConversationDescription([...BILI_ACP_TOOLS_ANTHROPIC, ...absorbOn ? [ABSORB_TOOL] : [], ...rulesOn ? [RULE_TOOL] : []].map(withConversationIdParam)),
+      openai: withSearchContextConversationDescription([...BILI_ACP_TOOLS_OPENAI, ...absorbOn ? [ABSORB_TOOL_OPENAI] : [], ...rulesOn ? [RULE_TOOL_OPENAI] : []].map(withConversationIdParam)),
+      responses: withSearchContextConversationDescription([...BILI_ACP_TOOLS_RESPONSES, ...absorbOn ? [ABSORB_TOOL_RESPONSES] : [], ...rulesOn ? [RULE_TOOL_RESPONSES] : []].map(withConversationIdParam))
     },
     headers: { agent: PLUGIN_AGENT_HEADER, conversation: PLUGIN_CONVERSATION_HEADER, contextWindow: PLUGIN_CONTEXT_WINDOW_HEADER, maxOutput: PLUGIN_MAX_OUTPUT_HEADER, model: PLUGIN_MODEL_HEADER, instructionsMutable: PLUGIN_INSTRUCTIONS_MUTABLE_HEADER },
     toolEndpoint: "/__bili/plugin/tool",
@@ -71460,6 +71460,13 @@ function resolveConversation(conversationId2) {
   }
   return { session, entry };
 }
+function disabledOptionalToolNote(tool, session, config) {
+  const absorbName = effectiveAbsorbConfig(session, config)?.toolName ?? ABSORB_TOOL_NAME;
+  if (tool === absorbName) return `${tool} is not enabled on this bili proxy (compress.absorb.enabled is not true) \u2014 nothing was absorbed.`;
+  if (tool === RULE_TOOL_NAME) return `${tool} is not enabled on this bili proxy (compress.rules.enabled is not true) \u2014 nothing was recorded.`;
+  if (!ccrEnabled(session) && tool === retrieveToolName(session)) return `${tool} is not enabled on this bili proxy (compress.ccr.enabled is not true) \u2014 nothing was retrieved.`;
+  return void 0;
+}
 function handlePluginStatus(conversationId2, res, deps, fallbackLatest = false) {
   const { session: resolvedSession, entry } = resolveConversation(conversationId2);
   let session = resolvedSession;
@@ -71577,6 +71584,13 @@ async function handlePluginTool(payload, res, deps) {
     return;
   }
   if (!isProxyToolFor(tool, session, deps.config)) {
+    const note = disabledOptionalToolNote(tool, session, deps.config);
+    if (note !== void 0) {
+      deps.log("info", `[${session.id}] [plugin] ${tool} called but disabled \u2014 replied with explanation`);
+      res.writeHead(200, { "content-type": "application/json" });
+      res.end(JSON.stringify({ ok: true, result: note }));
+      return;
+    }
     const allowed = [...ACP_TOOL_NAMES];
     const absorb = effectiveAbsorbConfig(session, deps.config);
     if (absorb?.enabled === true) allowed.push(absorb.toolName ?? ABSORB_TOOL_NAME);
@@ -73073,13 +73087,48 @@ function artifactSeedHit(body) {
   return tagSeed || toolSeed;
 }
 function detectAcpArtifacts(body, parsed) {
-  if (body.includes("<acp ") && ACP_TAG_RE.test(body.toString("utf8"))) return "tags";
+  if (parsed && typeof parsed === "object" && !Array.isArray(parsed)) {
+    const encoded = JSON.stringify(historyTextOf(parsed));
+    if (encoded.includes("<acp ") && ACP_TAG_RE.test(encoded)) return "tags";
+  } else if (body.includes("<acp ") && ACP_TAG_RE.test(body.toString("utf8"))) {
+    return "tags";
+  }
   if (parsed && typeof parsed === "object" && !Array.isArray(parsed)) {
     const names = historyToolCallNames(parsed);
     if (names.has("acp_status") && names.has("search_context")) return "tool-history";
   }
   return null;
 }
+function historyTextOf(parsed) {
+  const parts = [];
+  const container = Array.isArray(parsed.messages) ? parsed.messages : Array.isArray(parsed.input) ? parsed.input : Array.isArray(parsed.contents) ? parsed.contents : null;
+  if (container) {
+    for (const item of container) collectHistoryText(item, parts);
+  }
+  return parts.join("\n");
+}
+function collectHistoryText(item, out) {
+  if (!item || typeof item !== "object" || Array.isArray(item)) return;
+  const it2 = item;
+  if (it2.role === "system" || it2.role === "developer") return;
+  const content = it2.content;
+  if (typeof content === "string") out.push(content);
+  else if (Array.isArray(content)) {
+    for (const block of content) {
+      if (block && typeof block === "object" && !Array.isArray(block) && typeof block.text === "string") {
+        out.push(block.text);
+      }
+    }
+  }
+  const parts = it2.parts;
+  if (Array.isArray(parts)) {
+    for (const part of parts) {
+      if (part && typeof part === "object" && !Array.isArray(part) && typeof part.text === "string") {
+        out.push(part.text);
+      }
+    }
+  }
+}
 function historyToolCallNames(parsed) {
   const names = /* @__PURE__ */ new Set();
   const container = Array.isArray(parsed.messages) ? parsed.messages : Array.isArray(parsed.input) ? parsed.input : Array.isArray(parsed.contents) ? parsed.contents : null;
@@ -73243,6 +73292,10 @@ async function startServer(opts) {
   const log2 = (level, msg) => logMsg(opts, level, msg);
   const instanceId = randomUUID4();
   const instanceStartedAt = Date.now();
+  const parentWatchPid = Number.parseInt(process.env.BILI_PARENT_PID ?? "", 10);
+  const initialWatcherPid = Number.isInteger(parentWatchPid) && parentWatchPid > 0 && parentWatchPid !== process.pid ? parentWatchPid : null;
+  const proxyWatchers = /* @__PURE__ */ new Set();
+  if (initialWatcherPid !== null) proxyWatchers.add(initialWatcherPid);
   await initSessions();
   loadConversations();
   log2("info", `[persist] ${getStore().enabled ? "enabled" : "disabled"}`);
@@ -73263,7 +73316,7 @@ async function startServer(opts) {
   const server = http.createServer(async (req, res) => {
     armRequestWatchdog(req, res, log2);
     try {
-      await handle(req, res, opts, core, config, log2, instanceId, instanceStartedAt);
+      await handle(req, res, opts, core, config, log2, instanceId, instanceStartedAt, proxyWatchers, initialWatcherPid);
     } catch (err2) {
       const msg = String(err2);
       const e = err2;
@@ -73419,10 +73472,30 @@ Content-Length: ${Buffer.byteLength(body)}\r
   if (process.platform === "win32") {
     process.on("SIGBREAK", () => shutdown("SIGBREAK"));
   }
-  const parentPid = Number.parseInt(process.env.BILI_PARENT_PID ?? "", 10);
-  if (Number.isInteger(parentPid) && parentPid > 0 && parentPid !== process.pid) {
+  if (initialWatcherPid !== null) {
+    let idleSince = null;
+    let lastDead = [];
     const watcher = setInterval(() => {
-      if (!isPidAlive(parentPid)) shutdown(`parent-gone (pid ${parentPid})`);
+      const dead = [];
+      for (const pid of proxyWatchers) {
+        if (!isPidAlive(pid)) {
+          proxyWatchers.delete(pid);
+          dead.push(pid);
+        }
+      }
+      if (dead.length > 0) lastDead = dead;
+      if (proxyWatchers.size > 0) {
+        idleSince = null;
+        return;
+      }
+      if (idleSince === null) {
+        idleSince = Date.now();
+        return;
+      }
+      if (Date.now() - idleSince >= WATCHER_IDLE_GRACE_MS) {
+        const reason = lastDead.length === 1 && lastDead[0] === initialWatcherPid ? `parent-gone (pid ${lastDead[0]})` : `watchers-gone (pids: ${lastDead.join(", ")})`;
+        shutdown(reason);
+      }
     }, 2e3);
     watcher.unref?.();
   }
@@ -73458,7 +73531,8 @@ function adminTrustedHosts(bindHost, port) {
   return set;
 }
 var headroomFallbackLogged = /* @__PURE__ */ new Set();
-async function handle(req, res, opts, core, config, log2, instanceId, instanceStartedAt) {
+var WATCHER_IDLE_GRACE_MS = 5e3;
+async function handle(req, res, opts, core, config, log2, instanceId, instanceStartedAt, proxyWatchers, initialWatcherPid) {
   const selfProbePath = selfAdminProbePath(req.url ?? "", req.socket.localPort);
   if (selfProbePath !== void 0) req.url = selfProbePath;
   const isAdminPath = req.url === "/__bili/" || req.url?.startsWith("/__bili/") || req.url === "/__acp/" || req.url?.startsWith("/__acp/");
@@ -73575,7 +73649,7 @@ async function handle(req, res, opts, core, config, log2, instanceId, instanceSt
     }
     return;
   }
-  if (req.method === "GET" && req.url === "/__bili/plugin/manifest") return handlePluginManifest(res);
+  if (req.method === "GET" && req.url === "/__bili/plugin/manifest") return handlePluginManifest(res, config);
   if (req.method === "GET" && req.url?.startsWith("/__bili/plugin/status")) {
     const query = req.url.slice(req.url.indexOf("?") + 1);
     const params = new URLSearchParams(query);
@@ -73587,6 +73661,29 @@ async function handle(req, res, opts, core, config, log2, instanceId, instanceSt
     }
     return handlePluginStatus(conversationId2, res, { core, config, log: log2 }, params.get("fallback") === "latest");
   }
+  if (req.method === "POST" && req.url === "/__bili/watcher") {
+    try {
+      const body = await readBody(req);
+      const parsed2 = JSON.parse(body.toString("utf8"));
+      const pid = parsed2.pid;
+      if (typeof pid !== "number" || !Number.isInteger(pid) || pid <= 1 || pid === process.pid) {
+        res.writeHead(400, { "content-type": "application/json" });
+        res.end(JSON.stringify({ ok: false, error: "expected { pid: <integer> }" }));
+      } else if (initialWatcherPid === null) {
+        res.writeHead(409, { "content-type": "application/json" });
+        res.end(JSON.stringify({ ok: false, error: "watchdog not armed (no parent pid) \u2014 this proxy does not take watchers" }));
+      } else {
+        proxyWatchers.add(pid);
+        res.writeHead(200, { "content-type": "application/json" });
+        res.end(JSON.stringify({ ok: true, watchers: proxyWatchers.size }));
+      }
+      return;
+    } catch (err2) {
+      res.writeHead(err2 instanceof BodyTooLargeError ? 413 : 400, { "content-type": "application/json" });
+      res.end(JSON.stringify({ ok: false, error: String(err2) }));
+      return;
+    }
+  }
   if (req.method === "POST" && req.url === "/__bili/plugin/tool") {
     try {
       const body = await readBody(req);
@@ -73894,7 +73991,8 @@ ${bodyBuffer.toString("utf8")}`);
     const sessionId = anonAffinity ? anonAffinity.sessionId : conversation;
     if (artifactSeed) {
       const artifactKind = detectAcpArtifacts(bodyBuffer, parsed);
-      if (artifactKind !== null && !hasProcessedState(sessionId, { protocol })) {
+      const pluginAnnounced = pluginAgentHeader(req.headers) !== void 0;
+      if (artifactKind !== null && !pluginAnnounced && !hasProcessedState(sessionId, { protocol })) {
         if (!warnedChainSessions.has(sessionId)) {
           warnedChainSessions.add(sessionId);
           if (warnedChainSessions.size > WARNED_CHAIN_SESSION_CAP) {
@@ -81073,10 +81171,26 @@ async function main() {
 }
 
 // src/index.ts
-main().catch((err2) => {
-  console.error("bili: failed to start:", err2);
-  process.exit(1);
-});
+function invokedAsScript() {
+  const argv1 = process.argv[1];
+  if (!argv1) return false;
+  let self2;
+  let invoked;
+  try {
+    self2 = realpathSync2(fileURLToPath8(import.meta.url));
+    invoked = realpathSync2(argv1);
+  } catch {
+    return false;
+  }
+  if (process.platform === "win32") return self2.toLowerCase() === invoked.toLowerCase();
+  return self2 === invoked;
+}
+if (invokedAsScript()) {
+  main().catch((err2) => {
+    console.error("bili: failed to start:", err2);
+    process.exit(1);
+  });
+}
 /*! Bundled license information:
 
 undici/lib/web/fetch/body.js: