robotrock 0.8.4 → 0.9.0

package/dist/trigger/index.js

@@ -1,6 +1,200 @@
 // src/trigger/index.ts
 import { task, wait } from "@trigger.dev/sdk";
 
+// src/record-handled-to-otel.ts
+import {
+  context,
+  trace,
+  SpanStatusCode
+} from "@opentelemetry/api";
+var TRACER_NAME = "robotrock";
+var WAIT_SPAN_NAME = "robotrock.wait_for_human";
+var HANDLED_EVENT_NAME = "robotrock.task_handled";
+var INVALID_TRACE_ID = "00000000000000000000000000000000";
+function isValidTraceId(traceId) {
+  return typeof traceId === "string" && traceId.length > 0 && traceId !== INVALID_TRACE_ID;
+}
+function resolveSpanContext(span) {
+  const active = span ?? trace.getActiveSpan();
+  const ctx = active?.spanContext();
+  if (!ctx || !isValidTraceId(ctx.traceId)) {
+    return void 0;
+  }
+  return { traceId: ctx.traceId, spanId: ctx.spanId };
+}
+function parseTaskCreatedAt(submittedAt) {
+  if (submittedAt) {
+    const parsed = Date.parse(submittedAt);
+    if (!Number.isNaN(parsed)) {
+      return parsed;
+    }
+  }
+  return Date.now();
+}
+function parseHandledAtMs(handledAt) {
+  if (handledAt instanceof Date) {
+    return handledAt.getTime();
+  }
+  if (typeof handledAt === "number") {
+    return handledAt;
+  }
+  const parsed = Date.parse(handledAt);
+  return Number.isNaN(parsed) ? Date.now() : parsed;
+}
+function toRobotRockHandledOtelInput(handled) {
+  if ("actionId" in handled) {
+    return {
+      taskId: handled.taskId,
+      action: {
+        id: handled.actionId,
+        data: handled.data
+      },
+      handledBy: handled.handledBy,
+      handledAt: handled.handledAt
+    };
+  }
+  return {
+    taskId: handled.taskId,
+    action: {
+      id: handled.action.id,
+      title: handled.action.title,
+      data: handled.action.data
+    },
+    handledBy: handled.handledBy,
+    handledAt: handled.handledAt
+  };
+}
+function captureRobotRockOtelHandle(task2, options) {
+  const spanCtx = resolveSpanContext(options?.span);
+  return {
+    traceId: spanCtx?.traceId ?? "",
+    spanId: spanCtx?.spanId ?? "",
+    taskId: task2.taskId,
+    threadId: task2.threadId,
+    taskCreatedAt: parseTaskCreatedAt(task2.submittedAt)
+  };
+}
+function startRobotRockHumanWaitSpan(handle, options) {
+  const parent = options?.span ?? trace.getActiveSpan();
+  if (!parent) {
+    return void 0;
+  }
+  const tracer = trace.getTracer(options?.tracerName ?? TRACER_NAME);
+  const span = tracer.startSpan(
+    WAIT_SPAN_NAME,
+    {},
+    trace.setSpan(context.active(), parent)
+  );
+  span.setAttribute("robotrock.task_id", handle.taskId);
+  if (handle.threadId) {
+    span.setAttribute("robotrock.thread_id", handle.threadId);
+  }
+  return span;
+}
+function recordRobotRockHandledToOtel(handled, options = {}) {
+  const span = options.span ?? trace.getActiveSpan();
+  if (!span) {
+    return;
+  }
+  const input = toRobotRockHandledOtelInput(handled);
+  const handledAtMs = parseHandledAtMs(input.handledAt);
+  const handle = options.handle;
+  const humanWaitMs = handle != null ? Math.max(0, handledAtMs - handle.taskCreatedAt) : void 0;
+  span.setAttribute("robotrock.task_id", input.taskId);
+  span.setAttribute("robotrock.action.id", input.action.id);
+  if (input.action.title) {
+    span.setAttribute("robotrock.action.title", input.action.title);
+  }
+  if (input.handledBy) {
+    span.setAttribute("robotrock.handled_by", input.handledBy);
+  }
+  span.setAttribute("robotrock.handled_at", new Date(handledAtMs).toISOString());
+  if (humanWaitMs != null) {
+    span.setAttribute("robotrock.human_wait_ms", humanWaitMs);
+  }
+  if (input.threadId) {
+    span.setAttribute("robotrock.thread_id", input.threadId);
+  }
+  const eventAttributes = {
+    "robotrock.task_id": input.taskId,
+    "robotrock.action.id": input.action.id
+  };
+  if (input.handledBy) {
+    eventAttributes["robotrock.handled_by"] = input.handledBy;
+  }
+  if (humanWaitMs != null) {
+    eventAttributes["robotrock.human_wait_ms"] = humanWaitMs;
+  }
+  if (options.includeActionData && input.action.data != null) {
+    const serialized = JSON.stringify(input.action.data);
+    const truncated = serialized.length > 512 ? `${serialized.slice(0, 512)}\u2026` : serialized;
+    span.setAttribute("robotrock.action.data", truncated);
+    eventAttributes["robotrock.action.data"] = truncated;
+  }
+  span.addEvent(HANDLED_EVENT_NAME, eventAttributes);
+}
+function endRobotRockHumanWaitSpan(span, handled, options = {}) {
+  if (!span) {
+    return;
+  }
+  const outcome = options.outcome ?? (handled ? "handled" : "timeout");
+  if (handled && outcome === "handled") {
+    recordRobotRockHandledToOtel(handled, {
+      span,
+      handle: options.handle,
+      includeActionData: options.includeActionData
+    });
+    span.setStatus({ code: SpanStatusCode.OK });
+  } else {
+    span.setAttribute("robotrock.outcome", outcome);
+    span.setStatus({
+      code: SpanStatusCode.ERROR,
+      message: outcome === "timeout" ? "Human response timed out before validUntil" : "Human wait failed"
+    });
+  }
+  span.end();
+}
+
+// src/otel-platform.ts
+function shouldRecordRobotRockOtel(recordOtel) {
+  if (recordOtel === true) {
+    return true;
+  }
+  if (recordOtel === false) {
+    return false;
+  }
+  const env = process.env.ROBOTROCK_OTEL_RECORD_HANDLED;
+  return env === "true" || env === "1";
+}
+function stripPlatformOtelFields(payload) {
+  const { recordOtel: _recordOtel, otelIncludeActionData: _otelIncludeActionData, ...rest } = payload;
+  return rest;
+}
+function beginRobotRockHumanWaitOtel(task2, options = {}) {
+  const recordOtel = shouldRecordRobotRockOtel(options.recordOtel);
+  if (!recordOtel) {
+    return { recordOtel: false };
+  }
+  const handle = captureRobotRockOtelHandle(task2);
+  const waitSpan = startRobotRockHumanWaitSpan(handle);
+  return {
+    recordOtel: true,
+    handle,
+    waitSpan,
+    includeActionData: options.otelIncludeActionData
+  };
+}
+function finishRobotRockHumanWaitOtel(session, handled, outcome) {
+  if (!session.recordOtel) {
+    return;
+  }
+  endRobotRockHumanWaitSpan(session.waitSpan, handled, {
+    handle: session.handle,
+    outcome,
+    includeActionData: session.includeActionData
+  });
+}
+
 // src/schemas/index.ts
 import { z as z2 } from "zod";
 
@@ -167,12 +361,12 @@ function validateUrlValue(value, widget) {
   }
   return null;
 }
-function validateContextPublicUrls(context) {
-  if (!context?.data) {
+function validateContextPublicUrls(context2) {
+  if (!context2?.data) {
     return null;
   }
-  for (const [field, value] of Object.entries(context.data)) {
-    const widget = widgetType(context.ui, field);
+  for (const [field, value] of Object.entries(context2.data)) {
+    const widget = widgetType(context2.ui, field);
     if (!widget) {
       continue;
     }
@@ -227,7 +421,8 @@ var contextDataSchema = z.object({
   data: z.record(z.string(), z.unknown()),
   ui: contextUiSchema
 }).optional();
-var taskContextObjectSchema = z.object({
+var TASK_CONTEXT_FORMAT_VERSION = 2;
+var taskContextObjectBaseSchema = z.object({
   /** Source application id; omitted tasks are grouped in the default inbox. */
   app: z.string().min(1).optional(),
   type: z.string().min(1),
@@ -235,9 +430,22 @@ var taskContextObjectSchema = z.object({
   description: z.string().optional(),
   validUntil: z.string().optional(),
   context: contextDataSchema,
+  /** Task context wire format version. @default 2 */
+  contextVersion: z.literal(2).optional(),
+  /** @deprecated Use `contextVersion`. Accepted on ingest only. */
   version: z.literal(2).optional(),
   actions: z.array(taskActionSchema).min(1, "At least one action is required")
 });
+function normalizeTaskContextVersion(data) {
+  const { version: legacyVersion, contextVersion, ...rest } = data;
+  return {
+    ...rest,
+    contextVersion: contextVersion ?? legacyVersion ?? TASK_CONTEXT_FORMAT_VERSION
+  };
+}
+var taskContextObjectSchema = taskContextObjectBaseSchema.transform(
+  normalizeTaskContextVersion
+);
 function refineContextPublicUrls(data, ctx) {
   const error = validateContextPublicUrls(data.context);
   if (error) {
@@ -281,61 +489,11 @@ var threadUpdateInputSchema = z.object({
 });
 var taskPriorities = ["low", "normal", "high", "urgent"];
 var taskPrioritySchema = z.enum(taskPriorities);
-var nonNegativeInt = z.number().int().nonnegative();
-var agentCostTokensSchema = z.object({
-  input: nonNegativeInt.optional(),
-  output: nonNegativeInt.optional(),
-  total: nonNegativeInt.optional()
-});
-var agentCostSchema = z.object({
-  tokens: agentCostTokensSchema.optional(),
-  eur: z.number().nonnegative().optional(),
-  usd: z.number().nonnegative().optional()
-});
-var AGENT_INFO_MAX_KEYS = 32;
-var AGENT_TOOL_CALLS_MAX_KEYS = 64;
-var AGENT_OTEL_SPANS_MAX = 32;
-var agentToolCallsSchema = z.record(z.string().min(1), nonNegativeInt);
-var agentOtelSpanStatusSchema = z.enum(["ok", "error"]);
-var agentOtelSpanSummarySchema = z.object({
-  name: z.string().min(1),
-  durationMs: z.number().nonnegative(),
-  status: agentOtelSpanStatusSchema
-});
-var agentOtelSchema = z.object({
-  traceId: z.string().min(1),
-  spanId: z.string().min(1).optional(),
-  rootDurationMs: z.number().nonnegative().optional(),
-  spans: z.array(agentOtelSpanSummarySchema).max(AGENT_OTEL_SPANS_MAX).optional()
-});
 var agentTelemetrySchema = z.object({
   /** Agent or app version (semver, git SHA, deploy tag — caller-defined). */
-  version: z.string().min(1).optional(),
-  /** Total tool invocations. When omitted but `toolCalls` is set, the sum of counts is used. */
-  toolCallCount: nonNegativeInt.optional(),
-  /** Per-tool invocation counts keyed by tool name. */
-  toolCalls: agentToolCallsSchema.optional(),
-  cost: agentCostSchema.optional(),
-  /** Free-form metadata for experiments (model, prompt variant, etc.). */
-  info: z.record(z.string(), z.unknown()).optional(),
-  /** Structured OpenTelemetry span snapshot for feedback analysis. */
-  otel: agentOtelSchema.optional()
-}).refine(
-  (data) => {
-    const keys = data.info ? Object.keys(data.info) : [];
-    return keys.length <= AGENT_INFO_MAX_KEYS;
-  },
-  { message: `agent.info may contain at most ${AGENT_INFO_MAX_KEYS} keys` }
-).refine(
-  (data) => {
-    const keys = data.toolCalls ? Object.keys(data.toolCalls) : [];
-    return keys.length <= AGENT_TOOL_CALLS_MAX_KEYS;
-  },
-  {
-    message: `agent.toolCalls may contain at most ${AGENT_TOOL_CALLS_MAX_KEYS} tool names`
-  }
-);
-var createTaskBodySchema = taskContextObjectSchema.extend({
+  version: z.string().min(1)
+});
+var createTaskBodySchema = taskContextObjectBaseSchema.extend({
   assignTo: assignToSchema.optional(),
   /**
    * Groups related tasks together. When omitted, the server generates one and
@@ -352,9 +510,9 @@ var createTaskBodySchema = taskContextObjectSchema.extend({
    * the inbox status bar and the thread update log.
    */
   update: threadUpdateInputSchema.optional(),
-  /** Agent telemetry (version, cost, tool calls) — not shown in inbox UI. */
+  /** Agent release version — not shown in inbox UI. */
   agent: agentTelemetrySchema.optional()
-}).superRefine(refineContextPublicUrls);
+}).transform(normalizeTaskContextVersion).superRefine(refineContextPublicUrls);
 
 // src/schemas/index.ts
 var safeUrlSchema2 = z2.string().refine((url) => isPublicHttpUrl(url), {
@@ -398,16 +556,29 @@ var contextDataSchema2 = z2.object({
   data: z2.record(z2.string(), z2.unknown()),
   ui: contextUiSchema2
 }).optional();
-var taskContextObjectSchema2 = z2.object({
+var TASK_CONTEXT_FORMAT_VERSION2 = 2;
+var taskContextObjectBaseSchema2 = z2.object({
   app: z2.string().min(1).optional(),
   type: z2.string().min(1),
   name: z2.string().min(1),
   description: z2.string().optional(),
   validUntil: z2.string().optional(),
   context: contextDataSchema2,
+  contextVersion: z2.literal(2).optional(),
+  /** @deprecated Use `contextVersion`. Accepted on ingest only. */
   version: z2.literal(2).optional(),
   actions: z2.array(taskActionSchema2).min(1, "At least one action is required")
 });
+function normalizeTaskContextVersion2(data) {
+  const { version: legacyVersion, contextVersion, ...rest } = data;
+  return {
+    ...rest,
+    contextVersion: contextVersion ?? legacyVersion ?? TASK_CONTEXT_FORMAT_VERSION2
+  };
+}
+var taskContextObjectSchema2 = taskContextObjectBaseSchema2.transform(
+  normalizeTaskContextVersion2
+);
 function refineContextPublicUrls2(data, ctx) {
   const error = validateContextPublicUrls(data.context);
   if (error) {
@@ -451,56 +622,10 @@ var threadUpdateInputSchema2 = z2.object({
 });
 var taskPriorities2 = ["low", "normal", "high", "urgent"];
 var taskPrioritySchema2 = z2.enum(taskPriorities2);
-var nonNegativeInt2 = z2.number().int().nonnegative();
-var agentCostTokensSchema2 = z2.object({
-  input: nonNegativeInt2.optional(),
-  output: nonNegativeInt2.optional(),
-  total: nonNegativeInt2.optional()
-});
-var agentCostSchema2 = z2.object({
-  tokens: agentCostTokensSchema2.optional(),
-  eur: z2.number().nonnegative().optional(),
-  usd: z2.number().nonnegative().optional()
-});
-var AGENT_INFO_MAX_KEYS2 = 32;
-var AGENT_TOOL_CALLS_MAX_KEYS2 = 64;
-var AGENT_OTEL_SPANS_MAX2 = 32;
-var agentToolCallsSchema2 = z2.record(z2.string().min(1), nonNegativeInt2);
-var agentOtelSpanStatusSchema2 = z2.enum(["ok", "error"]);
-var agentOtelSpanSummarySchema2 = z2.object({
-  name: z2.string().min(1),
-  durationMs: z2.number().nonnegative(),
-  status: agentOtelSpanStatusSchema2
-});
-var agentOtelSchema2 = z2.object({
-  traceId: z2.string().min(1),
-  spanId: z2.string().min(1).optional(),
-  rootDurationMs: z2.number().nonnegative().optional(),
-  spans: z2.array(agentOtelSpanSummarySchema2).max(AGENT_OTEL_SPANS_MAX2).optional()
-});
 var agentTelemetrySchema2 = z2.object({
-  version: z2.string().min(1).optional(),
-  toolCallCount: nonNegativeInt2.optional(),
-  toolCalls: agentToolCallsSchema2.optional(),
-  cost: agentCostSchema2.optional(),
-  info: z2.record(z2.string(), z2.unknown()).optional(),
-  otel: agentOtelSchema2.optional()
-}).refine(
-  (data) => {
-    const keys = data.info ? Object.keys(data.info) : [];
-    return keys.length <= AGENT_INFO_MAX_KEYS2;
-  },
-  { message: `agent.info may contain at most ${AGENT_INFO_MAX_KEYS2} keys` }
-).refine(
-  (data) => {
-    const keys = data.toolCalls ? Object.keys(data.toolCalls) : [];
-    return keys.length <= AGENT_TOOL_CALLS_MAX_KEYS2;
-  },
-  {
-    message: `agent.toolCalls may contain at most ${AGENT_TOOL_CALLS_MAX_KEYS2} tool names`
-  }
-);
-var createTaskBodySchema2 = taskContextObjectSchema2.extend({
+  version: z2.string().min(1)
+});
+var createTaskBodySchema2 = taskContextObjectBaseSchema2.extend({
   assignTo: assignToSchema2.optional(),
   /**
    * Groups related tasks together. When omitted, the server generates one and
@@ -518,7 +643,7 @@ var createTaskBodySchema2 = taskContextObjectSchema2.extend({
    */
   update: threadUpdateInputSchema2.optional(),
   agent: agentTelemetrySchema2.optional()
-}).superRefine(refineContextPublicUrls2);
+}).transform(normalizeTaskContextVersion2).superRefine(refineContextPublicUrls2);
 var threadUpdateBodySchema = threadUpdateInputSchema2;
 
 // src/approval-result.ts
@@ -554,6 +679,10 @@ var DEFAULT_TIMEOUT_MS = 24 * 60 * 60 * 1e3;
 function sleep(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
+function resolveAgentVersionFromEnv() {
+  const fromEnv = process.env.AGENT_VERSION?.trim() || process.env.ROBOTROCK_AGENT_VERSION?.trim();
+  return fromEnv || void 0;
+}
 function parseValidUntilMs(value) {
   if (value === void 0) {
     return void 0;
@@ -593,7 +722,8 @@ var RobotRock = class {
   apiKey;
   baseUrl;
   app;
-  version;
+  agentVersion;
+  contextVersion;
   webhook;
   polling;
   constructor(config) {
@@ -612,7 +742,8 @@ var RobotRock = class {
     const rawBase = config.baseUrl ?? "https://api.robotrock.io/v1";
     this.baseUrl = rawBase.replace(/\/+$/, "");
     this.app = config.app;
-    this.version = config.version ?? 2;
+    this.agentVersion = config.version ?? resolveAgentVersionFromEnv();
+    this.contextVersion = config.advanced?.contextVersion ?? TASK_CONTEXT_FORMAT_VERSION2;
     this.webhook = config.webhook;
     this.polling = config.polling ?? {};
   }
@@ -623,15 +754,17 @@ var RobotRock = class {
     const normalizedTask = normalizeSendToHumanInput(task2, {
       webhook: this.webhook,
       app: this.app,
-      version: this.version
+      contextVersion: this.contextVersion,
+      agentVersion: this.agentVersion
     });
+    const agentVersion = task2.version ?? this.agentVersion;
     const bodyPayload = {
       ...normalizedTask,
       ...task2.assignTo !== void 0 ? { assignTo: task2.assignTo } : {},
       ...task2.threadId !== void 0 ? { threadId: task2.threadId } : {},
       ...task2.priority !== void 0 ? { priority: task2.priority } : {},
       ...task2.update !== void 0 ? { update: task2.update } : {},
-      ...task2.agent !== void 0 ? { agent: task2.agent } : {}
+      ...agentVersion !== void 0 ? { agent: { version: agentVersion } } : {}
     };
     const validation = createTaskBodySchema2.safeParse(bodyPayload);
     if (!validation.success) {
@@ -667,7 +800,8 @@ var RobotRock = class {
     const normalizedTask = normalizeSendToHumanInput(task2, {
       webhook: this.webhook,
       app: this.app,
-      version: this.version
+      contextVersion: this.contextVersion,
+      agentVersion: this.agentVersion
     });
     const createdTaskTask = await this.createTask(task2);
     const hasHandlers = normalizedTask.actions.some(
@@ -811,6 +945,7 @@ function normalizeSendToHumanInput(task2, clientDefaults) {
     threadId: _threadId,
     priority: _priority,
     update: _update,
+    version: _version,
     validUntil,
     app: taskApp,
     ...rest
@@ -820,7 +955,7 @@ function normalizeSendToHumanInput(task2, clientDefaults) {
   const app = taskApp ?? clientDefaults.app;
   return {
     ...rest,
-    version: clientDefaults.version,
+    contextVersion: clientDefaults.contextVersion,
     ...app ? { app } : {},
     ...validUntil !== void 0 ? { validUntil: serializeValidUntil(validUntil) } : {},
     actions: normalizedActions
@@ -940,7 +1075,7 @@ var APPROVE_BY_HUMAN_ACTIONS = [
   { id: "decline", title: "Decline" }
 ];
 async function runSendToHuman(payload) {
-  const { validUntil: validUntilInput, ...taskInput } = payload;
+  const { validUntil: validUntilInput, app, recordOtel, otelIncludeActionData, ...taskFields } = payload;
   const { validUntil, timeout } = resolveWaitTiming(validUntilInput);
   const token = await wait.createToken({ timeout });
   const baseConfig = resolveRobotRockConfig();
@@ -949,40 +1084,68 @@ async function runSendToHuman(payload) {
     baseUrl: baseConfig.baseUrl,
     ...baseConfig.app ? { app: baseConfig.app } : {},
     ...baseConfig.version ? { version: baseConfig.version } : {},
+    ...baseConfig.advanced ? { advanced: baseConfig.advanced } : {},
     webhook: { url: token.url }
   });
+  const taskInput = stripPlatformOtelFields(taskFields);
   const sendResult = await client.sendToHuman({
     ...taskInput,
-    validUntil
+    validUntil,
+    ...app !== void 0 ? { app } : {}
   });
-  const outcome = await wait.forToken(token.id);
-  if (!outcome.ok) {
-    throw new Error(`Human response timed out before validUntil (${timeout})`);
-  }
-  const output = outcome.output;
-  if (!isRobotRockHandlerWebhookPayload(output)) {
-    throw new Error(
-      "Wait token completed with unexpected payload; expected RobotRock handler body (taskId, action.id, action.data)."
-    );
-  }
-  return toDiscriminatedApprovalResult(
-    payload.actions,
-    {
-      id: output.taskId,
-      createdAt: /* @__PURE__ */ new Date(),
-      status: "handled",
-      context: sendResult.task.context,
-      validUntil: Date.now(),
-      handledAt: new Date(output.handledAt).getTime(),
-      handled: {
-        action: {
-          id: output.action.id,
-          data: output.action.data
-        },
-        handledBy: output.handledBy
+  const otelSession = beginRobotRockHumanWaitOtel(sendResult.task, {
+    recordOtel,
+    otelIncludeActionData
+  });
+  let handledPayload = null;
+  let waitOutcome = "pending";
+  try {
+    const outcome = await wait.forToken(token.id);
+    if (!outcome.ok) {
+      waitOutcome = "timeout";
+      throw new Error(`Human response timed out before validUntil (${timeout})`);
+    }
+    const output = outcome.output;
+    if (!isRobotRockHandlerWebhookPayload(output)) {
+      waitOutcome = "error";
+      throw new Error(
+        "Wait token completed with unexpected payload; expected RobotRock handler body (taskId, action.id, action.data)."
+      );
+    }
+    handledPayload = output;
+    waitOutcome = "handled";
+    return toDiscriminatedApprovalResult(
+      payload.actions,
+      {
+        id: output.taskId,
+        createdAt: /* @__PURE__ */ new Date(),
+        status: "handled",
+        context: sendResult.task.context,
+        validUntil: Date.now(),
+        handledAt: new Date(output.handledAt).getTime(),
+        handled: {
+          action: {
+            id: output.action.id,
+            data: output.action.data
+          },
+          handledBy: output.handledBy
+        }
       }
+    );
+  } catch (error) {
+    if (waitOutcome === "pending") {
+      waitOutcome = "error";
     }
-  );
+    throw error;
+  } finally {
+    if (waitOutcome !== "pending") {
+      finishRobotRockHumanWaitOtel(
+        otelSession,
+        handledPayload,
+        waitOutcome === "handled" ? "handled" : waitOutcome
+      );
+    }
+  }
 }
 var sendToHumanTask = task({
   id: "robotrock/send-to-human",