agent-relay-server 0.22.0 → 0.24.0

package/src/routes.ts

@@ -139,6 +139,7 @@ import { createToken, deleteTokenProfile, getToken, getTokenProfile, listTokenPr
 import {
   getConnector,
   listConnectors,
+  markConnectorUnreachable,
   readConnectorConfig,
   registerConnectorManifest,
   runConnectorAction,
@@ -2197,7 +2198,7 @@ function restartSpawnParamsForAgent(
   const policy = policyName ? getSpawnPolicy(policyName) : null;
   const requestedBy = opts.resumeId ? "dashboard-resume" : "dashboard-restart";
   if (policy) {
-    const params = { ...managedSpawnParams(policy.value, requestId), agentId: agent.id, requestedBy };
+    const params = { ...buildManagedSpawnParams(policy.value, requestId, { createdBy: "managed-agent" }), agentId: agent.id, requestedBy };
     return opts.resumeId ? withClaudeResumeParams(params, opts.resumeId, agent.id) : params;
   }
 
@@ -4811,11 +4812,6 @@ function policyStatusPayload(policy: SpawnPolicy) {
   };
 }
 
-function managedSpawnParams(policy: SpawnPolicy, requestId: string): Record<string, unknown> {
-  return buildManagedSpawnParams(policy, requestId, { createdBy: "managed-agent" });
-}
-
-
 function requirePolicyAndOrchestrator(name: string): { policy: SpawnPolicy; orch: NonNullable<ReturnType<typeof getOrchestrator>> } | Response {
   const entry = getSpawnPolicy(name);
   if (!entry) return error("spawn policy not found", 404);
@@ -4861,7 +4857,7 @@ function enqueuePolicyStart(policy: SpawnPolicy, reason: string): Command | Resp
     target: orch.agentId,
     correlationId: requestId,
     params: {
-      ...managedSpawnParams(policy, requestId),
+      ...buildManagedSpawnParams(policy, requestId, { createdBy: "managed-agent" }),
       reason,
       orchestratorId: orch.id,
     },
@@ -4875,7 +4871,7 @@ function enqueuePolicyStop(policy: SpawnPolicy, action: "shutdown" | "restart",
   if (!orch) return error("orchestrator not found", 404);
   const state = getManagedAgentState(policy.name);
   const restartRequestId = action === "restart" ? spawnRequestId() : undefined;
-  const restartSpawn = restartRequestId ? managedSpawnParams(policy, restartRequestId) : undefined;
+  const restartSpawn = restartRequestId ? buildManagedSpawnParams(policy, restartRequestId, { createdBy: "managed-agent" }) : undefined;
   const nextState = upsertManagedAgentState({
     policyName: policy.name,
     status: "stopping",
@@ -5784,6 +5780,13 @@ const postConnectorCall: Handler = async (req, params) => {
       headers: { "content-type": res.headers.get("content-type") || "application/json" },
     });
   } catch (e) {
+    // A connection error (refused/reset/DNS) means the advertised endpoint is
+    // genuinely down — reconcile the stale running:true so the dashboard stops
+    // claiming the connector is healthy. A timeout (AbortError) may just be a
+    // slow daemon, so leave its state alone.
+    if ((e as Error).name !== "AbortError") {
+      markConnectorUnreachable(params.id!, `endpoint unreachable: ${(e as Error).message}`);
+    }
     return error(`failed to reach connector: ${(e as Error).message}`, 502);
   }
 };