@polos/sdk 0.2.1 → 0.2.3

package/dist/index.js

@@ -722,6 +722,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
     });
@@ -3355,6 +3360,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;
@@ -4208,7 +4394,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];
@@ -4396,7 +4583,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];
@@ -4472,7 +4660,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`, {
@@ -4668,7 +4857,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];
@@ -4676,12 +4865,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 });
     },
@@ -5322,7 +5534,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) {
@@ -5332,11 +5564,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);
@@ -5582,7 +5815,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) {
@@ -5592,11 +5845,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);
@@ -6227,10 +6481,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;
@@ -6764,6 +7032,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,
@@ -6775,9 +7048,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) {
@@ -6920,15 +7190,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({
@@ -6936,13 +7211,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);
       }
     })();
   }
@@ -7647,181 +7923,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");
-  }
-};
-
 export { AgentRunConfig, DockerEnvironment, DuplicateWorkflowError, ExecutionHandle, GuardrailError, GuardrailResult2 as GuardrailResult, HookExecutionError, HookResult, LLM, ManagedSandbox, GuardrailResult as MiddlewareGuardrailResult, OrchestratorApiError, OrchestratorClient, Polos, PolosClient, Queue, SandboxManager, SlackChannel, StateSizeError, StateValidationError, StepExecutionError, StreamResult, WaitError, Worker, WorkerServer, WorkflowNotFoundError, agentStreamFunction, assertSafePath, batchAgentInvoke, batchInvoke, buildSummaryMessages, compactIfNeeded, composeGuardrails, composeHooks, conditionalHook, configureLogging, convertFinishReason, convertMiddlewareToolCallToPython, convertPythonToolCallToMiddleware, convertToolResultsToMessages, convertToolsToVercel, convertVercelToolCallToPython, convertVercelUsageToPython, createAskUserTool, createEditTool, createExecTool, createGlobTool, createGrepTool, createLogger, createReadTool, createStepHelper, createStepStore, createWebSearchTool, createWorkflowRegistry, createWriteTool, defineAgent, defineGuardrail, defineHook, defineTool, defineWorkflow, estimateMessageTokens, estimateMessagesTokens, estimateTokens, evaluateAllowlist, executeGuardrailChain, executeGuardrailsOrThrow, executeHookChain, executeHooksOrThrow, executeWorkflow, executedTool, extractTraceparent, generateTraceIdFromExecutionId, getTracer, globalRegistry, hasText, initializeOtel, initializeState, isAgentWorkflow, isBinary, isGuardrail, isHook, isOtelAvailable, isToolWorkflow, isWaitError, llmGenerate, llmStream, maxSteps, maxTokens, normalizeGuardrail, normalizeGuardrails, normalizeHook, normalizeHooks, parseGrepOutput, retry, sandboxTools, serializeFinalState, sleep, stopCondition, stripAnsi, truncateOutput, validateState };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map