@kody-ade/kody-engine 0.4.152 → 0.4.153

package/dist/bin/kody.js

@@ -1061,7 +1061,7 @@ var init_loadPriorArt = __esm({
 // package.json
 var package_default = {
   name: "@kody-ade/kody-engine",
-  version: "0.4.152",
+  version: "0.4.153",
   description: "kody \u2014 autonomous development engine. Single-session Claude Code agent behind a generic executor + declarative executable profiles.",
   license: "MIT",
   type: "module",
@@ -1952,6 +1952,14 @@ async function runAgent(opts) {
       `[kody agent] transient connection error (attempt ${attempt + 1}/${MAX_CONNECTION_RETRIES + 1}); retrying in ${Math.round(delayMs / 1e3)}s: ${errorMessage}
 `
     );
+    if (opts.ensureBackend) {
+      try {
+        await opts.ensureBackend();
+      } catch (e) {
+        process.stderr.write(`[kody agent] backend recovery failed: ${e instanceof Error ? e.message : String(e)}
+`);
+      }
+    }
     await new Promise((r) => setTimeout(r, delayMs));
   }
   const submittedState = getSubmitted?.();
@@ -3850,65 +3858,85 @@ function generateLitellmConfigYaml(model) {
     ""
   ].join("\n");
 }
-async function startLitellmIfNeeded(model, projectDir, url = LITELLM_DEFAULT_URL) {
-  if (!needsLitellmProxy(model)) return null;
-  if (await checkLitellmHealth(url)) {
-    return { url, kill: () => {
-    } };
-  }
-  let cmd = "litellm";
+function resolveLitellmCommand() {
   try {
     execFileSync4("which", ["litellm"], { timeout: 3e3, stdio: "pipe" });
+    return "litellm";
   } catch {
     try {
       execFileSync4("python3", ["-c", "import litellm"], { timeout: 1e4, stdio: "pipe" });
-      cmd = "python3";
+      return "python3";
     } catch {
       throw new Error("litellm not installed \u2014 run: pip install 'litellm[proxy]'");
     }
   }
-  const configPath = path13.join(os2.tmpdir(), `kody-litellm-${Date.now()}.yaml`);
-  fs14.writeFileSync(configPath, generateLitellmConfigYaml(model));
+}
+async function startLitellmIfNeeded(model, projectDir, url = LITELLM_DEFAULT_URL) {
+  if (!needsLitellmProxy(model)) return null;
+  const cmd = resolveLitellmCommand();
   const portMatch = url.match(/:(\d+)/);
   const port = portMatch ? portMatch[1] : "4000";
-  const args = cmd === "litellm" ? ["--config", configPath, "--port", port] : ["-m", "litellm", "--config", configPath, "--port", port];
-  const dotenvVars = readDotenvApiKeys(projectDir);
-  const logPath = path13.join(os2.tmpdir(), `kody-litellm-${Date.now()}.log`);
-  const outFd = fs14.openSync(logPath, "w");
-  const child = spawn3(cmd, args, {
-    stdio: ["ignore", outFd, outFd],
-    detached: true,
-    env: stripBlockingEnv({ ...process.env, ...dotenvVars })
-  });
-  fs14.closeSync(outFd);
-  const timeoutMs = resolveLitellmTimeoutMs();
-  const deadline = Date.now() + timeoutMs;
-  while (Date.now() < deadline) {
-    await new Promise((r) => setTimeout(r, LITELLM_HEALTH_POLL_INTERVAL_MS));
-    if (await checkLitellmHealth(url)) {
-      return {
-        url,
-        kill: () => {
-          try {
-            child.kill();
-          } catch {
-          }
-        }
-      };
+  const childEnv = stripBlockingEnv({ ...process.env, ...readDotenvApiKeys(projectDir) });
+  let child;
+  let logPath;
+  const spawnProxy = () => {
+    const configPath = path13.join(os2.tmpdir(), `kody-litellm-${Date.now()}.yaml`);
+    fs14.writeFileSync(configPath, generateLitellmConfigYaml(model));
+    const args = cmd === "litellm" ? ["--config", configPath, "--port", port] : ["-m", "litellm", "--config", configPath, "--port", port];
+    const nextLogPath = path13.join(os2.tmpdir(), `kody-litellm-${Date.now()}.log`);
+    const outFd = fs14.openSync(nextLogPath, "w");
+    child = spawn3(cmd, args, { stdio: ["ignore", outFd, outFd], detached: true, env: childEnv });
+    fs14.closeSync(outFd);
+    logPath = nextLogPath;
+  };
+  const waitForHealth = async () => {
+    const deadline = Date.now() + resolveLitellmTimeoutMs();
+    while (Date.now() < deadline) {
+      await new Promise((r) => setTimeout(r, LITELLM_HEALTH_POLL_INTERVAL_MS));
+      if (await checkLitellmHealth(url)) return true;
     }
+    return false;
+  };
+  const readLogTail = () => {
+    if (!logPath) return "";
+    try {
+      return fs14.readFileSync(logPath, "utf-8").slice(-2e3);
+    } catch {
+      return "";
+    }
+  };
+  const killChild = () => {
+    try {
+      child?.kill();
+    } catch {
+    }
+  };
+  const ensureHealthy = async () => {
+    if (await checkLitellmHealth(url)) return true;
+    const tail = readLogTail();
+    process.stderr.write(
+      `[kody litellm] proxy unreachable mid-run; restarting.${tail ? ` Last log:
+${tail}
+` : "\n"}`
+    );
+    killChild();
+    spawnProxy();
+    return waitForHealth();
+  };
+  if (await checkLitellmHealth(url)) {
+    return { url, kill: killChild, ensureHealthy };
   }
-  let logTail = "";
-  try {
-    logTail = fs14.readFileSync(logPath, "utf-8").slice(-2e3);
-  } catch {
-  }
-  try {
-    child.kill();
-  } catch {
+  spawnProxy();
+  if (!await waitForHealth()) {
+    const tail = readLogTail();
+    killChild();
+    const seconds = Math.round(resolveLitellmTimeoutMs() / 1e3);
+    throw new Error(
+      `LiteLLM proxy failed to start within ${seconds}s (KODY_LITELLM_TIMEOUT_SEC overrides). Log tail:
+${tail}`
+    );
   }
-  const seconds = Math.round(timeoutMs / 1e3);
-  throw new Error(`LiteLLM proxy failed to start within ${seconds}s (KODY_LITELLM_TIMEOUT_SEC overrides). Log tail:
-${logTail}`);
+  return { url, kill: killChild, ensureHealthy };
 }
 function readDotenvApiKeys(projectDir) {
   const dotenvPath = path13.join(projectDir, ".env");
@@ -13106,7 +13134,10 @@ async function runExecutable(profileName, input) {
   try {
     model = parseProviderModel(modelSpec);
   } catch (err) {
-    return finishAndEnd({ exitCode: 99, reason: `agent.model invalid: ${err instanceof Error ? err.message : String(err)}` });
+    return finishAndEnd({
+      exitCode: 99,
+      reason: `agent.model invalid: ${err instanceof Error ? err.message : String(err)}`
+    });
   }
   let litellm = null;
   try {
@@ -13150,11 +13181,15 @@ async function runExecutable(profileName, input) {
     const syntheticPath = ctx.data.syntheticPluginPath;
     const pluginPaths = [...externalPlugins, ...syntheticPath ? [syntheticPath] : []];
     const agents = loadSubagents(profile);
+    const lm = litellm;
     return runAgent({
       prompt,
       model,
       cwd: input.cwd,
-      litellmUrl: litellm?.url ?? null,
+      litellmUrl: lm?.url ?? null,
+      // On a connection drop mid-run, restart the (possibly crashed) proxy
+      // before the agent retries. No-op for direct-Anthropic runs (lm null).
+      ensureBackend: lm ? () => lm.ensureHealthy().then(() => void 0) : void 0,
       verbose: input.verbose,
       quiet: input.quiet,
       ndjsonDir,
@@ -13224,7 +13259,10 @@ async function runExecutable(profileName, input) {
     } else if (!ctx.skipAgent) {
       const prompt = ctx.data.prompt;
       if (!prompt) {
-        return finishAndEnd({ exitCode: 99, reason: "composePrompt did not produce a prompt (ctx.data.prompt missing)" });
+        return finishAndEnd({
+          exitCode: 99,
+          reason: "composePrompt did not produce a prompt (ctx.data.prompt missing)"
+        });
       }
       emitEvent(input.cwd, { executable: profileName, kind: "agent_start" });
       agentResult = await invokeAgent(prompt);
@@ -13285,10 +13323,7 @@ async function runExecutable(profileName, input) {
           const runId = resolveRunId2();
           const dir = pathMod.join(input.cwd, ".kody", "runs", runId, "crashes");
           fsMod.mkdirSync(dir, { recursive: true });
-          const file = pathMod.join(
-            dir,
-            `${label.replace(/[^a-zA-Z0-9_-]/g, "_")}-${Date.now()}.json`
-          );
+          const file = pathMod.join(dir, `${label.replace(/[^a-zA-Z0-9_-]/g, "_")}-${Date.now()}.json`);
           fsMod.writeFileSync(
             file,
             JSON.stringify(
@@ -13328,10 +13363,8 @@ async function runExecutable(profileName, input) {
       try {
         const missing2 = verifyTaskArtifacts(taskArtifacts.absDir);
         if (missing2.length > 0) {
-          process.stderr.write(
-            `[task-artifacts] task ${taskArtifacts.taskId} missing: ${missing2.join(", ")}
-`
-          );
+          process.stderr.write(`[task-artifacts] task ${taskArtifacts.taskId} missing: ${missing2.join(", ")}
+`);
         }
       } catch {
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kody-ade/kody-engine",
-  "version": "0.4.152",
+  "version": "0.4.153",
   "description": "kody — autonomous development engine. Single-session Claude Code agent behind a generic executor + declarative executable profiles.",
   "license": "MIT",
   "type": "module",