minimal-agent 0.6.2 → 0.6.4

package/src/tools/bash/bash.js

@@ -50,6 +50,12 @@ import { scanDestructiveCommand } from './warnings.js';
 // ---------------- 0. 常量 ----------------
 const DEFAULT_TIMEOUT_MS = 120_000; // 2 min
 const MAX_TIMEOUT_MS = 600_000; // 10 min
+/**
+ * G1：stdout / stderr 即时缓冲上限（各自）。命令狂输出（如 cat 大文件 / yes）时，
+ * 在最终统一截断之前先把内存封顶，防 OOM。取最终展示截断阈值的 2 倍 —— 正常输出不受影响，
+ * 超大输出也只占有限内存（最终仍按 DEFAULT_MAX_RESULT_SIZE_CHARS 截断展示）。
+ */
+const MAX_STREAM_BUFFER_CHARS = DEFAULT_MAX_RESULT_SIZE_CHARS * 2;
 /**
  * 灾难性命令黑名单。
  * 命中任意一条就直接拒绝执行 —— 这层是"防猪队友"而不是真沙盒。
@@ -142,6 +148,27 @@ export function checkForbidden(command) {
     }
     return null;
 }
+/**
+ * 子进程环境变量：从 process.env 复制后**剥离 agent 自己的密钥**
+ * （`MINIMAL_AGENT_*` / `TAVILY_*`）。
+ *
+ * 动机（安全）：main.tsx 启动时会把 saved config 里的 key 注入 process.env。
+ * 若原样把 process.env 透传给 Bash 子进程，模型只要跑 `echo $MINIMAL_AGENT_API_KEY`
+ * 就能把 key（绿色版里是卖家预置的）读进 tool result、回传 LLM、甚至打到 -p stdout。
+ * 这些 env 是 agent 自身配置，子进程命令并不需要 —— 直接删掉，最小化泄露面。
+ * 其余变量（PATH / HOME / 用户自己 export 的）照常透传。
+ *
+ * 导出供测试断言。
+ */
+export function sanitizedChildEnv(base = process.env) {
+    const env = { ...base };
+    for (const key of Object.keys(env)) {
+        if (key.startsWith('MINIMAL_AGENT_') || key.startsWith('TAVILY_')) {
+            delete env[key];
+        }
+    }
+    return env;
+}
 // ---------------- 1. Zod 输入 schema ----------------
 const inputSchema = z.object({
     command: z
@@ -256,17 +283,20 @@ async function call(input, signal) {
                 shell: true,
                 cwd: getWorkingDir(),
                 signal: ac.signal,
-                env: process.env,
+                env: sanitizedChildEnv(),
                 windowsHide: true,
             });
             child.stdout?.setEncoding('utf8');
             child.stderr?.setEncoding('utf8');
             child.stdout?.on('data', (chunk) => {
-                stdout += chunk;
-                // 读到一半就超额时不强行 kill；最后再统一截断,避免日志噪声
+                // G1：即时封顶，防止狂输出命令在最终截断前撑爆内存（OOM）。不强行 kill，
+                // 让命令自己跑完或被 timeout 收走；超出上限的部分直接丢弃，最终仍按上限截断展示。
+                if (stdout.length < MAX_STREAM_BUFFER_CHARS)
+                    stdout += chunk;
             });
             child.stderr?.on('data', (chunk) => {
-                stderr += chunk;
+                if (stderr.length < MAX_STREAM_BUFFER_CHARS)
+                    stderr += chunk;
             });
             child.on('error', (e) => {
                 // signal abort 会先触发 error('AbortError')；把这种 case 转成正常分支

package/src/ui/hooks/useTokenUsage.js

@@ -15,11 +15,12 @@
  *  再额外传入版本号 prop 配合依赖即可。
  * ============================================================
  */
-import { AUTOCOMPACT_BUFFER_TOKENS } from '../../context/compact.js';
+import { getCompactThreshold } from '../../context/compact.js';
 import { countMessagesTokens } from '../../context/tokenCounter.js';
 export function useTokenUsage(messages, provider) {
     const tokens = countMessagesTokens(messages);
-    const threshold = Math.max(1000, provider.contextWindow - AUTOCOMPACT_BUFFER_TOKENS);
+    // 与 loop 实际触发同源（getCompactThreshold：按 contextWindow × compactRatio 比例 + 8K 下限）
+    const threshold = getCompactThreshold(provider);
     return {
         tokens,
         threshold,