@csdwd/ai-teams-agent 0.3.11 → 0.3.14

package/dist/index.js

@@ -875,6 +875,12 @@ function parseServerToEmployeeMessage(value) {
   if (type === "task.cancel") {
     return { type, taskId: nonEmptyStringField(message, "taskId") };
   }
+  if (type === "queue.resume") {
+    return { type };
+  }
+  if (type === "agent.registered") {
+    return { type, consecutiveQueueFailures: nonNegativeNumberField(message, "consecutiveQueueFailures") };
+  }
   throw new ProtocolError(`Unsupported server-to-employee message type: ${type}`);
 }
 function objectValue(value, label) {
@@ -921,6 +927,13 @@ function positiveNumberField(record, key) {
   }
   return value;
 }
+function nonNegativeNumberField(record, key) {
+  const value = record[key];
+  if (typeof value !== "number" || !Number.isFinite(value) || value < 0) {
+    throw new ProtocolError(`${key} must be a non-negative number.`);
+  }
+  return value;
+}
 function optionalCliConfigField(record, key) {
   const value = record[key];
   if (value === void 0 || value === null) {
@@ -1075,6 +1088,8 @@ function connect(state, getMainTask, getQueueTask, onMessage) {
 var mainTask = null;
 var queueTask = null;
 var agentState = loadState();
+var consecutiveQueueFailures = 0;
+var MAX_CONSECUTIVE_QUEUE_FAILURES = 5;
 var connState = {
   socket: null,
   reconnectTimer: null,
@@ -1128,7 +1143,9 @@ function finishTask(taskId, status, payload) {
   slot.set(null);
   recordTaskFinish(current, status, payload);
   if (status === "completed") {
-    if (current.targetMode !== "queue") {
+    if (current.targetMode === "queue") {
+      consecutiveQueueFailures = 0;
+    } else {
       agentState.sessionReady = true;
       persistState(agentState);
     }
@@ -1152,7 +1169,14 @@ function finishTask(taskId, status, payload) {
     taskId,
     error: typeof payload === "string" ? payload : "\u4EFB\u52A1\u6267\u884C\u5931\u8D25\u3002"
   });
-  if (current.targetMode === "queue") requestTask2();
+  if (current.targetMode === "queue") {
+    consecutiveQueueFailures += 1;
+    if (consecutiveQueueFailures >= MAX_CONSECUTIVE_QUEUE_FAILURES) {
+      console.log(`[agent:${EMPLOYEE_ID}] \u8FDE\u7EED ${consecutiveQueueFailures} \u6B21\u961F\u5217\u4EFB\u52A1\u5931\u8D25\uFF0C\u6682\u505C\u63A5\u53D7\u65B0\u7684\u961F\u5217\u4EFB\u52A1\u3002\u7B49\u5F85\u624B\u52A8\u6062\u590D\u3002`);
+      return;
+    }
+    requestTask2();
+  }
 }
 var runnerDeps = {
   findActiveTask,
@@ -1166,6 +1190,14 @@ var runnerDeps = {
   }
 };
 function startTask(message) {
+  if (message.targetMode === "queue" && consecutiveQueueFailures >= MAX_CONSECUTIVE_QUEUE_FAILURES) {
+    send2({
+      type: "task.failed",
+      taskId: message.taskId,
+      error: `Agent \u8FDE\u7EED ${consecutiveQueueFailures} \u6B21\u961F\u5217\u4EFB\u52A1\u5931\u8D25\uFF0C\u6682\u505C\u63A5\u53D7\u65B0\u961F\u5217\u4EFB\u52A1\uFF0C\u7B49\u5F85\u624B\u52A8\u6062\u590D\u3002`
+    });
+    return;
+  }
   const slot = slotForTargetMode(message.targetMode);
   if (slot.get()) {
     send2({
@@ -1219,6 +1251,22 @@ function handleServerMessage(message) {
     startTask(message);
     return;
   }
+  if (message.type === "agent.registered") {
+    const serverCount = message.consecutiveQueueFailures;
+    if (serverCount > consecutiveQueueFailures) {
+      consecutiveQueueFailures = serverCount;
+      if (consecutiveQueueFailures >= MAX_CONSECUTIVE_QUEUE_FAILURES) {
+        console.log(`[agent:${EMPLOYEE_ID}] \u4ECE\u670D\u52A1\u5668\u540C\u6B65\u8FDE\u7EED\u5931\u8D25\u8BA1\u6570 ${consecutiveQueueFailures}\uFF0C\u6682\u505C\u63A5\u53D7\u961F\u5217\u4EFB\u52A1\u3002`);
+      }
+    }
+    return;
+  }
+  if (message.type === "queue.resume") {
+    consecutiveQueueFailures = 0;
+    console.log(`[agent:${EMPLOYEE_ID}] \u961F\u5217\u4EFB\u52A1\u5DF2\u6062\u590D\uFF0C\u91CD\u65B0\u5F00\u59CB\u63A5\u53D7\u4EFB\u52A1\u3002`);
+    requestTask2();
+    return;
+  }
   cancelTask(message.taskId);
 }
 function connect2() {
@@ -1286,7 +1334,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.11");
+    console.log("0.3.14");
     process.exit(0);
   }
   if (args.includes("--help") || args.includes("-h")) {

package/package.json

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