kifaru 1.0.194 → 1.0.195

package/kifaru-gateway.js

@@ -54207,7 +54207,7 @@ var init_lib2 = __esm(() => {
 });
 
 // src/index.ts
-import fs14 from "fs";
+import fs15 from "fs";
 
 // src/config.ts
 init_zod();
@@ -74295,13 +74295,66 @@ function workbenchSocketHandlers(service) {
   };
 }
 
+// src/local-run-supervisor.ts
+import { spawn } from "child_process";
+import fs14 from "fs";
+import path14 from "path";
+function localRunConcurrency(value = process.env.KIFARU_LOCAL_RUN_CONCURRENCY) {
+  const parsed = Number(value ?? "3");
+  return Number.isInteger(parsed) && parsed > 0 ? Math.min(parsed, 8) : 3;
+}
+function localRunWorkerEntry(fromDir = import.meta.dir) {
+  const bundled = path14.join(fromDir, "kifaru-local-run-worker.js");
+  if (fs14.existsSync(bundled))
+    return bundled;
+  const source = path14.join(fromDir, "local-run-worker.ts");
+  return fs14.existsSync(source) ? source : null;
+}
+function startLocalRunWorkers(options = {}) {
+  const env = options.env ?? process.env;
+  if (env.KIFARU_LOCAL_RUNS !== "1")
+    return () => {};
+  const entry = localRunWorkerEntry(options.fromDir);
+  if (!entry)
+    throw new Error("Local headless worker entry is missing from this Kifaru installation");
+  const children = new Set;
+  let stopping = false;
+  const count = localRunConcurrency(env.KIFARU_LOCAL_RUN_CONCURRENCY);
+  const launch = (slot) => {
+    if (stopping)
+      return;
+    const child = spawn(process.execPath, [entry], { env, stdio: "inherit" });
+    children.add(child);
+    child.on("exit", (code) => {
+      children.delete(child);
+      if (!stopping) {
+        if (code !== 0)
+          console.error(`[local-run] worker ${slot} exited (${code ?? "signal"}); restarting`);
+        setTimeout(() => launch(slot), code === 0 ? 100 : 1000);
+      }
+    });
+    child.on("error", (error92) => console.error(`[local-run] worker ${slot} failed:`, error92));
+  };
+  for (let slot = 1;slot <= count; slot++)
+    launch(slot);
+  console.log(`[local-run] ${count} worker slots ready`);
+  return () => {
+    stopping = true;
+    for (const child of children)
+      child.kill("SIGTERM");
+  };
+}
+
 // src/index.ts
 var config3 = loadConfig();
+if (process.env.KIFARU_LOCAL_RUNS === undefined && ["127.0.0.1", "::1", "localhost"].includes(config3.host)) {
+  process.env.KIFARU_LOCAL_RUNS = "1";
+}
 var catalog = loadModelCatalog(config3);
 var storage = createSqliteStorage(config3.sessionDbPath);
 var workbench = config3.workbenchCwd ? await createWorkbench() : undefined;
 async function createWorkbench() {
-  fs14.mkdirSync(config3.workbenchCwd, { recursive: true });
+  fs15.mkdirSync(config3.workbenchCwd, { recursive: true });
   try {
     const service = await WorkbenchAgentService.create(config3, catalog, createPlatformRepository(storage.db));
     console.log(`[session-gateway] live workspace enabled at ${config3.workbenchCwd}`);
@@ -74354,6 +74407,7 @@ var server = Bun.serve({
     return new Response("Internal server error", { status: 500 });
   }
 });
+var stopLocalRunWorkers = startLocalRunWorkers();
 console.log(`[session-gateway] listening on http://${server.hostname}:${server.port} with ${Object.keys(catalog.models).length} model(s)`);
 if (config3.hasOAuthProviders) {
   console.log(`[session-gateway] OAuth providers enabled: ${config3.oauthProviderIds.join(", ")}`);
@@ -74365,6 +74419,7 @@ if (config3.hasBootstrapToken) {
 }
 for (const signal of ["SIGINT", "SIGTERM"]) {
   process.on(signal, () => {
+    stopLocalRunWorkers();
     storage.close();
     server.stop(true);
     process.exit(0);