agent-office-cli 0.1.2 → 0.1.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-office-cli",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "Run and manage AI agent sessions locally, with optional relay to agentoffice.top",
   "license": "MIT",
   "engines": {

package/src/core/providers/codex.js

@@ -1,6 +1,31 @@
 const { GenericProvider } = require("./generic");
 const { findManagedCodexSessionFile, summarizeCodexSession } = require("./codex-transcript");
 
+const APPROVAL_PATTERNS = [
+  "approval requested:",
+  "approval requested by ",
+  "tool call needs your approval",
+  "requires approval by policy",
+  "requires approval:"
+];
+
+const IDLE_PATTERNS = [
+  "conversation interrupted - tell the model what to do differently",
+  "something went wrong? hit `/feedback` to",
+  "something went wrong? hit /feedback to"
+];
+
+const ATTENTION_PATTERNS = [
+  "stream disconnected before completion",
+  "error sending request for url",
+  "network error",
+  "connection timeout",
+  "timed out",
+  "failed to send request",
+  "failed to submit",
+  "panic"
+];
+
 function activeOverlayPatch(session, nextLifecycleState) {
   if (!session || !["approval", "attention"].includes(session.displayState)) {
     return null;
@@ -48,19 +73,19 @@ class CodexProvider extends GenericProvider {
 
   classifyOutput(chunk) {
     const text = String(chunk).toLowerCase();
-    if (text.includes("approval") || text.includes("press enter") || text.includes("confirm")) {
+
+    if (IDLE_PATTERNS.some((pattern) => text.includes(pattern))) {
+      return "idle";
+    }
+
+    if (APPROVAL_PATTERNS.some((pattern) => text.includes(pattern))) {
       return "approval";
     }
-    if (
-      text.includes("network error") ||
-      text.includes("connection timeout") ||
-      text.includes("timed out") ||
-      text.includes("error") ||
-      text.includes("failed") ||
-      text.includes("panic")
-    ) {
+
+    if (ATTENTION_PATTERNS.some((pattern) => text.includes(pattern))) {
       return "attention";
     }
+
     return null;
   }
 

package/src/core/providers/codex.test.js

@@ -72,3 +72,39 @@ test("classifyOutput can raise attention for transcript-backed sessions", () =>
 
   assert.equal(nextState, "attention");
 });
+
+test("classifyOutput treats user interrupted Codex screens as idle", () => {
+  const provider = new CodexProvider();
+  const nextState = provider.classifyOutput(
+    "Conversation interrupted - tell the model what to do differently. Something went wrong? Hit `/feedback` to report the issue."
+  );
+
+  assert.equal(nextState, "idle");
+});
+
+test("classifyOutput treats stream disconnects as attention", () => {
+  const provider = new CodexProvider();
+  const nextState = provider.classifyOutput(
+    "stream disconnected before completion: error sending request for url (http://54.255.64.152:3000/openai/responses)"
+  );
+
+  assert.equal(nextState, "attention");
+});
+
+test("classifyOutput does not treat plain explanatory approval text as a real approval prompt", () => {
+  const provider = new CodexProvider();
+  const nextState = provider.classifyOutput(
+    "只有真实审批提示才归到 approval"
+  );
+
+  assert.equal(nextState, null);
+});
+
+test("classifyOutput recognizes real Codex approval prompts", () => {
+  const provider = new CodexProvider();
+  const nextState = provider.classifyOutput(
+    "Approval requested: Codex wants to edit files"
+  );
+
+  assert.equal(nextState, "approval");
+});

package/src/tunnel.js

@@ -3,6 +3,42 @@ const { toSessionSummary } = require("./core");
 
 const RECONNECT_BASE_MS = 1000;
 const RECONNECT_MAX_MS = 30000;
+const LOCAL_PROXY_STRIP_HEADERS = new Set([
+  "accept-encoding",
+  "connection",
+  "content-length",
+  "cookie",
+  "origin",
+  "referer",
+  "te",
+  "trailer",
+  "transfer-encoding",
+  "upgrade"
+]);
+
+function shouldStripLocalProxyHeader(name) {
+  return (
+    LOCAL_PROXY_STRIP_HEADERS.has(name) ||
+    name.startsWith("proxy-") ||
+    name.startsWith("sec-") ||
+    name.startsWith("x-forwarded-")
+  );
+}
+
+function buildLocalRequestHeaders(headers, localServerUrl) {
+  const nextHeaders = {};
+
+  for (const [name, rawValue] of Object.entries(headers || {})) {
+    const key = String(name).toLowerCase();
+    if (shouldStripLocalProxyHeader(key) || rawValue == null) {
+      continue;
+    }
+    nextHeaders[key] = Array.isArray(rawValue) ? rawValue.join(", ") : String(rawValue);
+  }
+
+  nextHeaders.host = new URL(localServerUrl).host;
+  return nextHeaders;
+}
 
 function createTunnelClient({ key, relayUrl, localServerUrl }) {
   let ws = null;
@@ -116,7 +152,7 @@ function createTunnelClient({ key, relayUrl, localServerUrl }) {
       const fetchUrl = `${localServerUrl}${msg.path}`;
       const fetchOptions = {
         method: msg.method || "GET",
-        headers: { ...msg.headers, host: new URL(localServerUrl).host }
+        headers: buildLocalRequestHeaders(msg.headers, localServerUrl)
       };
       if (msg.body && msg.method !== "GET" && msg.method !== "HEAD") {
         fetchOptions.body = msg.body;
@@ -220,5 +256,6 @@ function createTunnelClient({ key, relayUrl, localServerUrl }) {
 }
 
 module.exports = {
+  buildLocalRequestHeaders,
   createTunnelClient
 };

package/src/tunnel.test.js

@@ -0,0 +1,31 @@
+const test = require("node:test");
+const assert = require("node:assert/strict");
+
+const { buildLocalRequestHeaders } = require("./tunnel");
+
+test("buildLocalRequestHeaders strips browser-only proxy headers and rewrites host", () => {
+  const next = buildLocalRequestHeaders(
+    {
+      authorization: "Bearer token",
+      accept: "*/*",
+      "content-type": "application/json",
+      host: "agentoffice.top",
+      connection: "keep-alive",
+      "accept-encoding": "gzip, deflate, br, zstd",
+      "content-length": "123",
+      origin: "https://agentoffice.top",
+      referer: "https://agentoffice.top/office",
+      "sec-ch-ua": "\"Chromium\";v=\"146\"",
+      "sec-fetch-mode": "cors",
+      "x-forwarded-for": "203.0.113.10"
+    },
+    "http://127.0.0.1:8765"
+  );
+
+  assert.deepEqual(next, {
+    authorization: "Bearer token",
+    accept: "*/*",
+    "content-type": "application/json",
+    host: "127.0.0.1:8765"
+  });
+});