@polos/sdk 0.2.1 → 0.2.3

package/dist/index.cjs

@@ -748,6 +748,11 @@ var OrchestratorClient = class {
     if (request.waitForSubworkflow !== void 0)
       body["wait_for_subworkflow"] = request.waitForSubworkflow;
     if (request.otelTraceparent !== void 0) body["otel_traceparent"] = request.otelTraceparent;
+    if (request.channelContext !== void 0)
+      body["channel_context"] = {
+        channel_id: request.channelContext.channelId,
+        source: request.channelContext.source
+      };
     return this.request("POST", "/api/v1/workflows/batch_run", {
       body
     });
@@ -3381,6 +3386,187 @@ function defineAgent(config) {
   });
   return agentWorkflow;
 }
+
+// src/channels/slack.ts
+var SlackChannel = class {
+  id = "slack";
+  outputMode = "per_step";
+  config;
+  constructor(config) {
+    if (!config.botToken.startsWith("xoxb-")) {
+      throw new Error(
+        `Invalid Slack bot token: must start with "xoxb-". Use the Bot User OAuth Token from your Slack app's OAuth & Permissions page.`
+      );
+    }
+    this.config = config;
+  }
+  async notify(notification) {
+    const overrides = notification.channelOverrides;
+    const channel = overrides?.["channel"] ?? this.config.defaultChannel;
+    const threadTs = overrides?.["thread_ts"] ?? overrides?.["threadTs"];
+    const blocks = this.buildBlocks(notification);
+    const text = notification.title ?? "Agent needs your input";
+    const messageTs = await this.postMessage(channel, threadTs, text, blocks);
+    return {
+      slack_channel: channel,
+      slack_message_ts: messageTs,
+      slack_blocks: blocks
+    };
+  }
+  async sendOutput(context2, event) {
+    const channel = context2.source["channel"];
+    const threadTs = context2.source["threadTs"];
+    if (!channel) return;
+    const text = this.formatOutputEvent(event);
+    if (!text) return;
+    await this.postMessage(channel, threadTs, text);
+  }
+  async postMessage(channel, threadTs, text, blocks) {
+    const body = { channel, text };
+    if (threadTs) body["thread_ts"] = threadTs;
+    if (blocks) body["blocks"] = blocks;
+    const response = await fetch("https://slack.com/api/chat.postMessage", {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${this.config.botToken}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify(body)
+    });
+    const data = await response.json();
+    if (!data.ok) {
+      throw new Error(`Slack API error: ${data.error ?? "unknown"}`);
+    }
+    return data.ts ?? "";
+  }
+  formatOutputEvent(event) {
+    const eventType = event.eventType;
+    if (eventType === "workflow_finish" || eventType === "agent_finish") {
+      const metadata = event.data["_metadata"];
+      const rawResult = event.data["result"];
+      const error = event.data["error"];
+      const workflowId = metadata?.["workflow_id"];
+      if (error) {
+        return `\u274C *${workflowId ?? "Workflow"} failed:* ${error}`;
+      }
+      const result = typeof rawResult === "object" && rawResult !== null && "result" in rawResult ? rawResult["result"] : rawResult;
+      const resultStr = typeof result === "string" ? result : JSON.stringify(result, null, 2);
+      if (resultStr) {
+        return resultStr;
+      }
+      return `\u2705 *${workflowId ?? "Workflow"} finished*`;
+    }
+    if (eventType === "tool_call") {
+      const toolCall = event.data["tool_call"];
+      if (toolCall) {
+        const fn = toolCall["function"];
+        const name = fn?.["name"];
+        if (name) {
+          return `\u{1F527} Calling tool: \`${name}\``;
+        }
+      }
+      return null;
+    }
+    if (eventType === "step_finish") {
+      const stepKey = event.data["step_key"];
+      const error = event.data["error"];
+      if (error) {
+        return `\u26A0\uFE0F Step \`${stepKey ?? "unknown"}\` failed: ${error}`;
+      }
+      return null;
+    }
+    return null;
+  }
+  buildBlocks(n) {
+    const blocks = [];
+    blocks.push({
+      type: "header",
+      text: { type: "plain_text", text: n.title ?? "Agent needs your input" }
+    });
+    if (n.description) {
+      blocks.push({
+        type: "section",
+        text: { type: "mrkdwn", text: n.description }
+      });
+    }
+    if (n.source || n.tool) {
+      const parts = [];
+      if (n.source) parts.push(`*Source:* ${n.source}`);
+      if (n.tool) parts.push(`*Tool:* \`${n.tool}\``);
+      blocks.push({
+        type: "context",
+        elements: [{ type: "mrkdwn", text: parts.join("  |  ") }]
+      });
+    }
+    if (n.context && Object.keys(n.context).length > 0) {
+      const contextText = JSON.stringify(n.context, null, 2);
+      blocks.push({
+        type: "section",
+        text: { type: "mrkdwn", text: "```" + contextText + "```" }
+      });
+    }
+    if (n.expiresAt) {
+      blocks.push({
+        type: "context",
+        elements: [{ type: "mrkdwn", text: `Expires: ${n.expiresAt}` }]
+      });
+    }
+    if (this.isSimpleApproval(n)) {
+      const approveValue = JSON.stringify({
+        executionId: n.executionId,
+        stepKey: n.stepKey,
+        approved: true
+      });
+      const rejectValue = JSON.stringify({
+        executionId: n.executionId,
+        stepKey: n.stepKey,
+        approved: false
+      });
+      blocks.push({
+        type: "actions",
+        elements: [
+          {
+            type: "button",
+            action_id: "polos_approve",
+            text: { type: "plain_text", text: "Approve" },
+            style: "primary",
+            value: approveValue
+          },
+          {
+            type: "button",
+            action_id: "polos_reject",
+            text: { type: "plain_text", text: "Reject" },
+            style: "danger",
+            value: rejectValue
+          },
+          {
+            type: "button",
+            text: { type: "plain_text", text: "View Details" },
+            url: n.approvalUrl
+          }
+        ]
+      });
+    } else {
+      blocks.push({
+        type: "actions",
+        elements: [
+          {
+            type: "button",
+            text: { type: "plain_text", text: "Respond" },
+            url: n.approvalUrl,
+            style: "primary"
+          }
+        ]
+      });
+    }
+    return blocks;
+  }
+  isSimpleApproval(n) {
+    const fields = n.formFields;
+    if (!fields || fields.length === 0) return false;
+    return fields.some((f) => f["key"] === "approved" && f["type"] === "boolean");
+  }
+};
 var logger4 = createLogger({ name: "worker-server" });
 var WorkerServer = class {
   app;
@@ -4234,7 +4420,8 @@ function createOrchestratorStepHelper(orchestratorClient, cachedSteps, execCtx,
       otelTraceparent: getCurrentTraceparent(),
       waitForSubworkflow: options?.waitForSubworkflow ?? false,
       initialState: options?.initialState,
-      runTimeoutSeconds: options?.runTimeoutSeconds
+      runTimeoutSeconds: options?.runTimeoutSeconds,
+      channelContext
     });
     if (options?.waitForSubworkflow) {
       return [void 0, false];
@@ -4422,7 +4609,8 @@ function createOrchestratorStepHelper(orchestratorClient, cachedSteps, execCtx,
         sessionId: execCtx.sessionId,
         userId: execCtx.userId,
         waitForSubworkflow: false,
-        otelTraceparent: getCurrentTraceparent()
+        otelTraceparent: getCurrentTraceparent(),
+        channelContext
       });
       const handleData = response.executions.map((exec, i) => {
         const item = items[i];
@@ -4498,7 +4686,8 @@ function createOrchestratorStepHelper(orchestratorClient, cachedSteps, execCtx,
         sessionId: execCtx.sessionId,
         userId: execCtx.userId,
         waitForSubworkflow: true,
-        otelTraceparent: getCurrentTraceparent()
+        otelTraceparent: getCurrentTraceparent(),
+        channelContext
       });
       const workflowIds = items.map((item) => typeof item.workflow === "string" ? item.workflow : item.workflow.id).join(", ");
       throw new WaitError(`Waiting for sub-workflows [${workflowIds}] to complete`, {
@@ -4694,7 +4883,7 @@ function createOrchestratorStepHelper(orchestratorClient, cachedSteps, execCtx,
         if (channelContext !== void 0) {
           notification.channelContext = channelContext;
         }
-        await Promise.allSettled(
+        const notifyResults = await Promise.allSettled(
           channels.map(async (ch) => {
             try {
               let overrides = notifyConfig?.[ch.id];
@@ -4702,12 +4891,35 @@ function createOrchestratorStepHelper(orchestratorClient, cachedSteps, execCtx,
                 overrides = { ...channelContext.source, ...overrides };
               }
               const n = overrides !== void 0 ? { ...notification, channelOverrides: overrides } : notification;
-              await ch.notify(n);
+              const meta = await ch.notify(n);
+              return meta ? { channelId: ch.id, ...meta } : void 0;
             } catch (err) {
               logger6.warn(`Channel ${ch.id} notification failed`, { error: String(err) });
+              return void 0;
             }
           })
         );
+        const metaEntries = [];
+        for (const r of notifyResults) {
+          if (r.status === "fulfilled" && r.value != null) {
+            metaEntries.push(r.value);
+          }
+        }
+        if (metaEntries.length > 0) {
+          orchestratorClient.publishEvent({
+            topic,
+            events: [
+              {
+                eventType: `notification_meta_${key}`,
+                data: { channels: metaEntries }
+              }
+            ],
+            executionId: execCtx.executionId,
+            rootExecutionId: execCtx.rootExecutionId
+          }).catch((err) => {
+            logger6.warn("Failed to publish notification metadata", { error: String(err) });
+          });
+        }
       }
       throw new WaitError(`Waiting for resume event: ${topic}`, { topic });
     },
@@ -5348,7 +5560,27 @@ function spawnCommand(command, args, options) {
       killed = true;
       proc.kill("SIGKILL");
     }, timeoutMs);
+    let exitGraceTimer = null;
+    let exitCode = null;
+    proc.on("exit", (code) => {
+      exitCode = code;
+      exitGraceTimer = setTimeout(() => {
+        clearTimeout(timer);
+        settle(() => {
+          if (killed) {
+            resolve10({
+              exitCode: 137,
+              stdout,
+              stderr: stderr + "\n[Process killed: timeout exceeded]"
+            });
+          } else {
+            resolve10({ exitCode: exitCode ?? 1, stdout, stderr });
+          }
+        });
+      }, 2e3);
+    });
     proc.on("close", (code) => {
+      if (exitGraceTimer) clearTimeout(exitGraceTimer);
       clearTimeout(timer);
       settle(() => {
         if (killed) {
@@ -5358,11 +5590,12 @@ function spawnCommand(command, args, options) {
             stderr: stderr + "\n[Process killed: timeout exceeded]"
           });
         } else {
-          resolve10({ exitCode: code ?? 1, stdout, stderr });
+          resolve10({ exitCode: code ?? exitCode ?? 1, stdout, stderr });
         }
       });
     });
     proc.on("error", (err) => {
+      if (exitGraceTimer) clearTimeout(exitGraceTimer);
       clearTimeout(timer);
       settle(() => {
         reject(err);
@@ -5608,7 +5841,27 @@ function spawnLocal(command, options) {
       killed = true;
       proc.kill("SIGKILL");
     }, timeoutMs);
+    let exitGraceTimer = null;
+    let exitCode = null;
+    proc.on("exit", (code) => {
+      exitCode = code;
+      exitGraceTimer = setTimeout(() => {
+        clearTimeout(timer);
+        settle(() => {
+          if (killed) {
+            resolve10({
+              exitCode: 137,
+              stdout,
+              stderr: stderr + "\n[Process killed: timeout exceeded]"
+            });
+          } else {
+            resolve10({ exitCode: exitCode ?? 1, stdout, stderr });
+          }
+        });
+      }, 2e3);
+    });
     proc.on("close", (code) => {
+      if (exitGraceTimer) clearTimeout(exitGraceTimer);
       clearTimeout(timer);
       settle(() => {
         if (killed) {
@@ -5618,11 +5871,12 @@ function spawnLocal(command, options) {
             stderr: stderr + "\n[Process killed: timeout exceeded]"
           });
         } else {
-          resolve10({ exitCode: code ?? 1, stdout, stderr });
+          resolve10({ exitCode: code ?? exitCode ?? 1, stdout, stderr });
         }
       });
     });
     proc.on("error", (err) => {
+      if (exitGraceTimer) clearTimeout(exitGraceTimer);
       clearTimeout(timer);
       settle(() => {
         reject(err);
@@ -6253,10 +6507,24 @@ var Worker = class {
   state = "stopped";
   heartbeatInterval = null;
   activeExecutions = /* @__PURE__ */ new Map();
+  /** Tracks running channel bridges so we don't start duplicates on re-dispatch. */
+  activeChannelBridges = /* @__PURE__ */ new Set();
   signalHandler = null;
   constructor(config) {
     this.config = config;
     this.channels = config.channels ?? [];
+    if (!this.channels.some((ch) => ch.id === "slack")) {
+      const slackBotToken = process.env["SLACK_BOT_TOKEN"];
+      if (slackBotToken) {
+        this.channels.push(
+          new SlackChannel({
+            botToken: slackBotToken,
+            defaultChannel: process.env["SLACK_CHANNEL"] ?? "#general"
+          })
+        );
+        logger9.info("Auto-registered SlackChannel from SLACK_BOT_TOKEN env var");
+      }
+    }
     this.maxConcurrentWorkflows = config.maxConcurrentWorkflows ?? 100;
     this.workerServerUrl = config.workerServerUrl ?? `http://localhost:${String(config.port ?? 8e3)}`;
     this.port = config.port ?? 8e3;
@@ -6790,6 +7058,11 @@ var Worker = class {
       };
       const hasWorkflowChannels = workflow.config.channels !== void 0;
       const resolvedChannels = workflow.config.channels ?? this.channels;
+      if (data.channelContext && !data.parentExecutionId) {
+        const rootExecId = data.rootExecutionId ?? executionId;
+        const rootWfId = data.rootWorkflowId ?? workflowId;
+        this.startChannelBridge(rootExecId, rootWfId, data.channelContext, workflow);
+      }
       result = await executeWorkflow({
         workflow,
         payload: data.payload,
@@ -6801,9 +7074,6 @@ var Worker = class {
         sandboxManager: this.sandboxManager,
         channelContext: hasWorkflowChannels ? void 0 : data.channelContext
       });
-      if (data.channelContext) {
-        this.startChannelBridge(executionId, workflowId, data.channelContext, workflow);
-      }
       await this.handleExecutionResult(executionId, workflowId, context2, result);
       retryableFailure = !result.success && !result.waiting && (result.retryable ?? true);
     } catch (error) {
@@ -6946,15 +7216,20 @@ var Worker = class {
   /**
    * Start a channel bridge that streams execution events back to the originating channel.
    */
-  startChannelBridge(executionId, workflowId, channelCtx, workflow) {
+  startChannelBridge(rootExecutionId, rootWorkflowId, channelCtx, workflow) {
+    if (this.activeChannelBridges.has(rootExecutionId)) return;
     const channel = this.channels.find((ch) => ch.id === channelCtx.channelId);
-    if (!channel?.sendOutput) return;
+    if (!channel) {
+      logger9.error(
+        `No channel registered for "${channelCtx.channelId}". ` + (channelCtx.channelId === "slack" ? "Set the SLACK_BOT_TOKEN environment variable so the worker can respond to Slack." : `Register a channel with id "${channelCtx.channelId}" on the worker.`),
+        { executionId: rootExecutionId }
+      );
+      return;
+    }
+    if (!channel.sendOutput) return;
     const outputMode = workflow.config["channelOutputMode"] ?? channel.outputMode ?? "per_step";
     if (outputMode === "none") return;
-    const execution = this.activeExecutions.get(executionId);
-    if (!execution) return;
-    const rootExecutionId = execution.abortController.signal.aborted ? executionId : executionId;
-    const rootWorkflowId = workflowId;
+    this.activeChannelBridges.add(rootExecutionId);
     void (async () => {
       try {
         const stream = this.orchestratorClient.streamEvents({
@@ -6962,13 +7237,14 @@ var Worker = class {
           workflowRunId: rootExecutionId
         });
         for await (const event of stream) {
-          if (execution.abortController.signal.aborted) break;
           if (shouldForwardEvent(event, outputMode)) {
             await channel.sendOutput?.(channelCtx, event);
           }
         }
       } catch (err) {
-        logger9.warn("Channel bridge error", { error: String(err), executionId });
+        logger9.warn("Channel bridge error", { error: String(err), executionId: rootExecutionId });
+      } finally {
+        this.activeChannelBridges.delete(rootExecutionId);
       }
     })();
   }
@@ -7673,181 +7949,6 @@ function sandboxTools(config) {
   return tools;
 }
 
-// src/channels/slack.ts
-var SlackChannel = class {
-  id = "slack";
-  outputMode = "per_step";
-  config;
-  constructor(config) {
-    if (!config.botToken.startsWith("xoxb-")) {
-      throw new Error(
-        `Invalid Slack bot token: must start with "xoxb-". Use the Bot User OAuth Token from your Slack app's OAuth & Permissions page.`
-      );
-    }
-    this.config = config;
-  }
-  async notify(notification) {
-    const overrides = notification.channelOverrides;
-    const channel = overrides?.["channel"] ?? this.config.defaultChannel;
-    const threadTs = overrides?.["thread_ts"];
-    const blocks = this.buildBlocks(notification);
-    const text = notification.title ?? "Agent needs your input";
-    await this.postMessage(channel, threadTs, text, blocks);
-  }
-  async sendOutput(context2, event) {
-    const channel = context2.source["channel"];
-    const threadTs = context2.source["threadTs"];
-    if (!channel) return;
-    const text = this.formatOutputEvent(event);
-    if (!text) return;
-    await this.postMessage(channel, threadTs, text);
-  }
-  async postMessage(channel, threadTs, text, blocks) {
-    const body = { channel, text };
-    if (threadTs) body["thread_ts"] = threadTs;
-    if (blocks) body["blocks"] = blocks;
-    const response = await fetch("https://slack.com/api/chat.postMessage", {
-      method: "POST",
-      headers: {
-        Authorization: `Bearer ${this.config.botToken}`,
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(body)
-    });
-    const data = await response.json();
-    if (!data.ok) {
-      throw new Error(`Slack API error: ${data.error ?? "unknown"}`);
-    }
-  }
-  formatOutputEvent(event) {
-    const eventType = event.eventType;
-    if (eventType === "workflow_finish" || eventType === "agent_finish") {
-      const metadata = event.data["_metadata"];
-      const result = event.data["result"];
-      const error = event.data["error"];
-      const workflowId = metadata?.["workflow_id"];
-      if (error) {
-        return `\u274C *${workflowId ?? "Workflow"} failed:* ${error}`;
-      }
-      const resultStr = typeof result === "string" ? result : JSON.stringify(result, null, 2);
-      if (resultStr) {
-        return `\u2705 *${workflowId ?? "Workflow"} finished:*
-${resultStr}`;
-      }
-      return `\u2705 *${workflowId ?? "Workflow"} finished*`;
-    }
-    if (eventType === "tool_call") {
-      const toolCall = event.data["tool_call"];
-      if (toolCall) {
-        const fn = toolCall["function"];
-        const name = fn?.["name"];
-        if (name) {
-          return `\u{1F527} Calling tool: \`${name}\``;
-        }
-      }
-      return null;
-    }
-    if (eventType === "step_finish") {
-      const stepKey = event.data["step_key"];
-      const error = event.data["error"];
-      if (error) {
-        return `\u26A0\uFE0F Step \`${stepKey ?? "unknown"}\` failed: ${error}`;
-      }
-      return null;
-    }
-    return null;
-  }
-  buildBlocks(n) {
-    const blocks = [];
-    blocks.push({
-      type: "header",
-      text: { type: "plain_text", text: n.title ?? "Agent needs your input" }
-    });
-    if (n.description) {
-      blocks.push({
-        type: "section",
-        text: { type: "mrkdwn", text: n.description }
-      });
-    }
-    if (n.source || n.tool) {
-      const parts = [];
-      if (n.source) parts.push(`*Source:* ${n.source}`);
-      if (n.tool) parts.push(`*Tool:* \`${n.tool}\``);
-      blocks.push({
-        type: "context",
-        elements: [{ type: "mrkdwn", text: parts.join("  |  ") }]
-      });
-    }
-    if (n.context && Object.keys(n.context).length > 0) {
-      const contextText = JSON.stringify(n.context, null, 2);
-      blocks.push({
-        type: "section",
-        text: { type: "mrkdwn", text: "```" + contextText + "```" }
-      });
-    }
-    if (n.expiresAt) {
-      blocks.push({
-        type: "context",
-        elements: [{ type: "mrkdwn", text: `Expires: ${n.expiresAt}` }]
-      });
-    }
-    if (this.isSimpleApproval(n)) {
-      const approveValue = JSON.stringify({
-        executionId: n.executionId,
-        stepKey: n.stepKey,
-        approved: true
-      });
-      const rejectValue = JSON.stringify({
-        executionId: n.executionId,
-        stepKey: n.stepKey,
-        approved: false
-      });
-      blocks.push({
-        type: "actions",
-        elements: [
-          {
-            type: "button",
-            action_id: "polos_approve",
-            text: { type: "plain_text", text: "Approve" },
-            style: "primary",
-            value: approveValue
-          },
-          {
-            type: "button",
-            action_id: "polos_reject",
-            text: { type: "plain_text", text: "Reject" },
-            style: "danger",
-            value: rejectValue
-          },
-          {
-            type: "button",
-            text: { type: "plain_text", text: "View Details" },
-            url: n.approvalUrl
-          }
-        ]
-      });
-    } else {
-      blocks.push({
-        type: "actions",
-        elements: [
-          {
-            type: "button",
-            text: { type: "plain_text", text: "Respond" },
-            url: n.approvalUrl,
-            style: "primary"
-          }
-        ]
-      });
-    }
-    return blocks;
-  }
-  isSimpleApproval(n) {
-    const fields = n.formFields;
-    if (!fields || fields.length === 0) return false;
-    return fields.some((f) => f["key"] === "approved" && f["type"] === "boolean");
-  }
-};
-
 exports.AgentRunConfig = AgentRunConfig;
 exports.DockerEnvironment = DockerEnvironment;
 exports.DuplicateWorkflowError = DuplicateWorkflowError;