@csdwd/ai-teams-agent 0.3.4 → 0.3.6

package/dist/index.js

@@ -247,6 +247,28 @@ var CLAUDE_HOOKS_ENABLED = process.env.CLAUDE_HOOKS_ENABLED !== "false";
 var WORKSPACE_CLAUDE_MD = path3.join(DEFAULT_WORKSPACE, "CLAUDE.md");
 var CLAUDE_MD_SECTION_START = "<!-- AI_TEAMS_AGENT_RULES_START -->";
 var CLAUDE_MD_SECTION_END = "<!-- AI_TEAMS_AGENT_RULES_END -->";
+function reinitializeConfig() {
+  fileConfig = loadConfigFile();
+  SERVER_PORT = process.env.AI_TEAMS_SERVER_PORT || "3789";
+  SERVER_URL = process.env.SERVER_URL || fileConfig?.serverUrl || `ws://localhost:${SERVER_PORT}`;
+  AUTH_TOKEN = process.env.AI_TEAMS_AUTH_TOKEN || fileConfig?.authToken || "";
+  EMPLOYEE_ID = process.env.EMPLOYEE_ID || fileConfig?.employeeId || "emp_local";
+  EMPLOYEE_NAME = process.env.EMPLOYEE_NAME || fileConfig?.employeeName || "Local Agent";
+  EMPLOYEE_LABELS = process.env.EMPLOYEE_LABELS?.split(",").map((item) => item.trim()).filter(Boolean) ?? [];
+  RECONNECT_MS = Number(process.env.RECONNECT_MS) || 5e3;
+  RUNNER_MODE = process.env.RUNNER_MODE || fileConfig?.runnerMode || "claude";
+  DEFAULT_WORKSPACE = process.env.DEFAULT_WORKSPACE || fileConfig?.workspace || process.cwd();
+  MAX_BUFFERED_MESSAGES = Number(process.env.AGENT_BUFFER_LIMIT) || 400;
+  AGENT_RECORDS_DIR = process.env.AGENT_RECORDS_DIR || path3.join(DEFAULT_WORKSPACE, ".ai-teams", "agents", EMPLOYEE_ID);
+  STATE_FILE = process.env.AGENT_STATE_FILE || path3.join(AGENT_RECORDS_DIR, "session-state.json");
+  LEGACY_STATE_FILE = path3.join(process.cwd(), `.agent-state.${EMPLOYEE_ID}.json`);
+  DAILY_RECORDS_DIR = path3.join(AGENT_RECORDS_DIR, "daily");
+  HOOKS_DIR = path3.join(AGENT_RECORDS_DIR, "hooks");
+  CLAUDE_HOOK_SCRIPT = path3.join(HOOKS_DIR, "claude-session-recorder.cjs");
+  CLAUDE_HOOK_SETTINGS = path3.join(HOOKS_DIR, "claude-hooks.settings.json");
+  CLAUDE_HOOKS_ENABLED = process.env.CLAUDE_HOOKS_ENABLED !== "false";
+  WORKSPACE_CLAUDE_MD = path3.join(DEFAULT_WORKSPACE, "CLAUDE.md");
+}
 
 // src/state.ts
 import fs3 from "node:fs";
@@ -1035,7 +1057,12 @@ function connect(state, getMainTask, getQueueTask, onMessage) {
       console.error(`[agent:${EMPLOYEE_ID}] invalid server message`, error);
     }
   });
-  state.socket.on("close", () => {
+  state.socket.on("close", (code, reason) => {
+    if (code === 1008) {
+      console.error(`[agent:${EMPLOYEE_ID}] \u8BA4\u8BC1\u5931\u8D25\uFF1AToken \u65E0\u6548\u6216\u670D\u52A1\u5668\u62D2\u7EDD\u8FDE\u63A5\u3002${reason ? ` (${reason})` : ""}`);
+      console.error(`[agent:${EMPLOYEE_ID}] \u8BF7\u68C0\u67E5 AI_TEAMS_AUTH_TOKEN \u914D\u7F6E\u540E\u91CD\u65B0\u542F\u52A8\u3002`);
+      process.exit(1);
+    }
     console.log(`[agent:${EMPLOYEE_ID}] disconnected, reconnecting in ${RECONNECT_MS}ms...`);
     scheduleReconnect(state, () => connect(state, getMainTask, getQueueTask, onMessage));
   });
@@ -1201,15 +1228,19 @@ function gracefulShutdown(signal) {
   if (shuttingDown) return;
   shuttingDown = true;
   console.log(`[agent:${EMPLOYEE_ID}] received ${signal}, shutting down...`);
+  const children = [];
   for (const task of [mainTask, queueTask]) {
     if (!task) continue;
     task.cancelRequested = true;
+    if (task.child) children.push(task.child);
     task.child?.kill("SIGTERM");
     send2({ type: "task.cancelled", taskId: task.taskId });
   }
+  mainTask = null;
+  queueTask = null;
   setTimeout(() => {
-    for (const task of [mainTask, queueTask]) {
-      task?.child?.kill("SIGKILL");
+    for (const child of children) {
+      child.kill("SIGKILL");
     }
   }, 2e3);
   if (connState.reconnectTimer) clearTimeout(connState.reconnectTimer);
@@ -1217,7 +1248,7 @@ function gracefulShutdown(signal) {
   if (connState.socket && connState.socket.readyState === WebSocket2.OPEN) {
     connState.socket.close(1e3, "agent shutting down");
   }
-  setTimeout(() => process.exit(0), 500);
+  setTimeout(() => process.exit(0), 3e3);
 }
 process.on("SIGTERM", () => gracefulShutdown("SIGTERM"));
 process.on("SIGINT", () => gracefulShutdown("SIGINT"));
@@ -1254,7 +1285,7 @@ if (isCli) {
   getArgValue2 = getArgValue, resolveWorkspace3 = resolveWorkspace2, resolveAgentDir2 = resolveAgentDir, resolvePidFile2 = resolvePidFile, resolveLogDir2 = resolveLogDir, applyCliArgsToEnv2 = applyCliArgsToEnv;
   const args = process.argv.slice(2);
   if (args.includes("--version") || args.includes("-v")) {
-    console.log("0.3.4");
+    console.log("0.3.6");
     process.exit(0);
   }
   if (args.includes("--help") || args.includes("-h")) {
@@ -1291,20 +1322,22 @@ if (isCli) {
   } else {
     const subcommand = args[0];
     if (subcommand === "start" || subcommand === "restart") {
-      if (subcommand === "restart") {
-        const pidFile = resolvePidFile();
-        const status = getDaemonStatus(pidFile);
-        if (status.running) {
-          void stopDaemon(pidFile);
-        }
-      } else if (!process.env.__AI_TEAMS_DAEMON_WATCHDOG && !process.env.__AI_TEAMS_DAEMON_WORKER) {
-        const status = getDaemonStatus(resolvePidFile());
-        if (status.running) {
-          console.log(`Already running (PID ${status.pid}).`);
-          process.exit(0);
+      applyCliArgsToEnv();
+      if (!process.env.__AI_TEAMS_DAEMON_WATCHDOG && !process.env.__AI_TEAMS_DAEMON_WORKER) {
+        if (subcommand === "restart") {
+          const pidFile = resolvePidFile();
+          const status = getDaemonStatus(pidFile);
+          if (status.running) {
+            void stopDaemon(pidFile);
+          }
+        } else {
+          const status = getDaemonStatus(resolvePidFile());
+          if (status.running) {
+            console.log(`Already running (PID ${status.pid}).`);
+            process.exit(0);
+          }
         }
       }
-      applyCliArgsToEnv();
       const fileConfig2 = loadConfigFile();
       const hasEnvConfig = process.env.AI_TEAMS_AUTH_TOKEN || process.env.SERVER_URL;
       if (!fileConfig2 && !hasEnvConfig) {
@@ -1319,6 +1352,8 @@ if (isCli) {
           run: async () => {
             console.log("  \u2713 \u6B63\u5728\u8FDE\u63A5\u670D\u52A1\u5668...");
             connect2();
+            return new Promise(() => {
+            });
           }
         });
       })();
@@ -1355,6 +1390,7 @@ if (isCli) {
           await runSetup(null);
         }
         applyCliArgsToEnv();
+        reinitializeConfig();
         console.log("  \u2713 \u6B63\u5728\u8FDE\u63A5\u670D\u52A1\u5668...");
         connect2();
       })();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@csdwd/ai-teams-agent",
-  "version": "0.3.4",
+  "version": "0.3.6",
   "description": "AI Teams agent — connects to server via WebSocket, spawns Claude CLI to execute tasks",
   "type": "module",
   "bin": {