@span-io/agent-link 0.1.2 → 0.1.4

package/dist/index.js

@@ -161,35 +161,47 @@ function handleControl(message) {
     }
 }
 function setupAgentPiping(agentId, proc) {
+    const rawFlushSize = Number.parseInt(process.env.AGENT_LINK_STREAM_FLUSH_CHARS ?? "16384", 10);
+    const flushSize = Number.isFinite(rawFlushSize) && rawFlushSize > 0 ? rawFlushSize : 16384;
     const setupStream = (stream, name) => {
         if (!stream)
             return;
         let buffer = "";
         stream.setEncoding("utf8");
+        const flushChunk = (chunk) => {
+            if (chunk.length > 0) {
+                transport.sendLog(agentId, name, chunk);
+            }
+        };
         stream.on("data", (chunk) => {
             buffer += chunk;
             let index = buffer.indexOf("\n");
             while (index >= 0) {
                 const line = buffer.slice(0, index + 1);
                 buffer = buffer.slice(index + 1);
-                transport.sendLog(agentId, name, line);
+                flushChunk(line);
                 index = buffer.indexOf("\n");
             }
+            // Prevent unbounded memory growth when tools stream without newlines.
+            while (buffer.length >= flushSize) {
+                const part = buffer.slice(0, flushSize);
+                buffer = buffer.slice(flushSize);
+                flushChunk(part);
+            }
         });
         stream.on("end", () => {
-            if (buffer.length > 0) {
-                transport.sendLog(agentId, name, buffer);
-            }
+            flushChunk(buffer);
+            buffer = "";
         });
     };
     setupStream(proc.child.stdout, "stdout");
     setupStream(proc.child.stderr, "stderr");
-    proc.child.on("exit", (code, signal) => {
+    proc.child.on("exit", (_code, _signal) => {
         transport.sendStatus(agentId, "exited");
         activeAgents.delete(agentId);
     });
     proc.child.on("error", (error) => {
-        transport.sendLog(agentId, "stderr", `Process error: ${error.message}\n`);
+        transport.sendLog(agentId, "stderr", "Process error: " + error.message + "\n");
         transport.sendStatus(agentId, "error");
         activeAgents.delete(agentId);
     });

package/dist/process-runner.js

@@ -1,3 +1,4 @@
+import { mkdirSync } from "fs";
 import { spawn } from "child_process";
 const DEFAULT_CODEX_ARGS = "exec --skip-git-repo-check";
 const DEFAULT_GEMINI_ARGS = "";
@@ -12,6 +13,50 @@ function shellQuote(value) {
     // Simple single-quote wrapping for display purposes
     return `'${value.replace(/'/g, `'"'"'`)}'`;
 }
+function collectDirectoriesFromArgs(args) {
+    const directories = [];
+    for (let i = 0; i < args.length; i += 1) {
+        const arg = args[i];
+        const next = args[i + 1];
+        if ((arg === "--cd" || arg === "-C" || arg === "--add-dir" || arg === "--include-directories") && typeof next === "string") {
+            directories.push(next);
+            i += 1;
+            continue;
+        }
+        if (arg.startsWith("--cd=")) {
+            directories.push(arg.slice("--cd=".length));
+            continue;
+        }
+        if (arg.startsWith("--add-dir=")) {
+            directories.push(arg.slice("--add-dir=".length));
+            continue;
+        }
+        if (arg.startsWith("--include-directories=")) {
+            const raw = arg.slice("--include-directories=".length);
+            raw
+                .split(",")
+                .map((entry) => entry.trim())
+                .filter(Boolean)
+                .forEach((entry) => directories.push(entry));
+            continue;
+        }
+    }
+    return directories;
+}
+function ensureDirectoriesExist(rawDirs) {
+    const deduped = new Set(rawDirs
+        .map((entry) => entry.trim())
+        .filter((entry) => entry.length > 0));
+    for (const dir of deduped) {
+        try {
+            mkdirSync(dir, { recursive: true });
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            throw new Error(`Failed to create working directory '${dir}': ${message}`);
+        }
+    }
+}
 export function buildCommand({ agent, prompt, optionsArgs = [], executablePath }, promptModeOverride) {
     // 1. Gemini Models
     if (agent.model.startsWith("gemini-")) {
@@ -34,7 +79,13 @@ export function buildCommand({ agent, prompt, optionsArgs = [], executablePath }
     // 2. Claude Models
     if (agent.model.startsWith("claude-")) {
         const command = executablePath ?? process.env.CLAUDE_BIN ?? "claude";
-        const args = ["-p", prompt, "--model", agent.model];
+        const args = ["-p", prompt];
+        if (optionsArgs.length > 0) {
+            args.push(...optionsArgs);
+        }
+        if (!args.includes("--model") && !args.includes("-m")) {
+            args.push("--model", agent.model);
+        }
         return { command, args, promptMode: "args" };
     }
     // 3. Generic Codex/Other Models
@@ -81,11 +132,11 @@ export function spawnAgentProcess(config) {
     const { agent, optionsArgs = [] } = config;
     const startedAt = new Date().toISOString();
     const isCodexModel = !agent.model.startsWith("gemini-") && !agent.model.startsWith("claude-");
-    const sanitizedOptions = agent.model.startsWith("claude-")
-        ? []
-        : optionsArgs;
+    const sanitizedOptions = optionsArgs;
     const spawnWithMode = (promptModeOverride) => {
         const { command, args, promptMode } = buildCommand({ ...config, optionsArgs: sanitizedOptions }, promptModeOverride);
+        // Ensure requested project/working directories exist before launching CLI.
+        ensureDirectoriesExist(collectDirectoriesFromArgs(args));
         const ttyMode = process.env.CODEX_TTY_MODE ?? DEFAULT_TTY_MODE;
         const hasExec = args.includes("exec");
         const useScriptWrapper = ttyMode === "script" || (ttyMode === "auto" && hasExec);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@span-io/agent-link",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "Secure bridge between Span (AI control plane) and local agent CLI tools.",
   "type": "module",
   "bin": {