poe-code 3.0.216 → 3.0.217

package/dist/providers/poe-agent.js

@@ -29756,7 +29756,7 @@ function formatError(err) {
   }
 }
 
-// packages/braintrust/src/redact.ts
+// packages/acp-telemetry/src/redact.ts
 var MAX_STRING_BYTES = 65536;
 var MAX_JSON_BYTES = 262144;
 var BINARY_SCAN_BYTES = 1024;
@@ -29795,6 +29795,220 @@ function redactLeaf(value) {
   return value;
 }
 
+// packages/acp-telemetry/src/trace.ts
+function acpToTrace(ctx) {
+  const spawnCtx = ctx;
+  return {
+    root: {
+      name: `agent:${ctx.agent}:${ctx.model ?? "?"}`,
+      kind: "agent",
+      input: redact({
+        prompt: ctx.prompt,
+        mode: ctx.mode,
+        cwd: ctx.cwd
+      }),
+      output: redact(accumulateAgentOutput(ctx.events)),
+      metadata: {
+        sessionId: ctx.sessionId,
+        threadId: ctx.threadId,
+        ...spawnCtx.metadata
+      },
+      metrics: buildMetrics(ctx),
+      children: logToolSpans(ctx.events)
+    }
+  };
+}
+function logToolSpans(events) {
+  const spans = [];
+  for (const [index, event] of events.entries()) {
+    const toolCall = asToolCall(event);
+    if (toolCall === void 0) {
+      continue;
+    }
+    const metadata = collectToolMeta(events, index, readString6(toolCall.toolCallId));
+    spans.push({
+      name: `tool_call:${readString6(toolCall.kind) ?? "unknown"}`,
+      kind: "tool",
+      input: redact(readToolInput(toolCall)),
+      output: redact(assembleToolOutput(events, index, readString6(toolCall.toolCallId))),
+      ...metadata ? { metadata } : {},
+      ...readSpanTimestamps(metadata),
+      children: []
+    });
+  }
+  return spans;
+}
+function collectToolMeta(events, toolCallIndex, toolCallId) {
+  const merged = {};
+  const startMeta = asRecord2(asRecord2(events[toolCallIndex])?._meta);
+  if (startMeta) {
+    for (const [key2, value] of Object.entries(startMeta)) {
+      merged[key2 === "ts" ? "startTs" : key2] = value;
+    }
+  }
+  for (const event of events.slice(toolCallIndex + 1)) {
+    const update = asToolCallUpdate(event);
+    if (update === void 0) continue;
+    if (toolCallId !== void 0 && update.toolCallId !== toolCallId) continue;
+    const updateMeta = asRecord2(update._meta);
+    if (!updateMeta) continue;
+    for (const [key2, value] of Object.entries(updateMeta)) {
+      merged[key2 === "ts" ? "endTs" : key2] = value;
+    }
+  }
+  return Object.keys(merged).length > 0 ? merged : void 0;
+}
+function accumulateAgentOutput(events) {
+  let output = "";
+  for (const event of events) {
+    const record = asRecord2(event);
+    if (record === void 0) {
+      continue;
+    }
+    if (record.event === "agent_message") {
+      output += readString6(record.text) ?? "";
+      continue;
+    }
+    if (record.sessionUpdate === "agent_message_chunk") {
+      output += readContentText(record.content);
+    }
+  }
+  return output;
+}
+function assembleToolOutput(events, toolCallIndex, toolCallId) {
+  const outputs = [];
+  let text5 = "";
+  for (const event of events.slice(toolCallIndex + 1)) {
+    const update = asToolCallUpdate(event);
+    if (update === void 0) {
+      continue;
+    }
+    if (toolCallId !== void 0 && update.toolCallId !== toolCallId) {
+      continue;
+    }
+    if (Object.hasOwn(update, "rawOutput")) {
+      outputs.push(update.rawOutput);
+    }
+    const contentText = readContentText(update.content);
+    if (contentText.length > 0) {
+      text5 += contentText;
+    }
+  }
+  if (outputs.length === 0) {
+    return text5;
+  }
+  if (text5.length > 0) {
+    outputs.push(text5);
+  }
+  return outputs.length === 1 ? outputs[0] : outputs;
+}
+function buildMetrics(ctx) {
+  const usage = ctx.usage;
+  const metrics = {};
+  const promptTokens = readNumber(usage.prompt_tokens) ?? readNumber(usage.inputTokens);
+  const completionTokens = readNumber(usage.completion_tokens) ?? readNumber(usage.outputTokens);
+  addMetric(metrics, "prompt_tokens", promptTokens);
+  addMetric(metrics, "completion_tokens", completionTokens);
+  addMetric(
+    metrics,
+    "tokens",
+    readNumber(usage.tokens) ?? sumIfPresent(promptTokens, completionTokens)
+  );
+  addMetric(
+    metrics,
+    "prompt_cached_tokens",
+    readNumber(usage.prompt_cached_tokens) ?? readNumber(usage.cachedTokens)
+  );
+  addMetric(
+    metrics,
+    "prompt_cache_creation_tokens",
+    readNumber(usage.prompt_cache_creation_tokens)
+  );
+  addMetric(metrics, "durationMs", readNumber(usage.durationMs));
+  return metrics;
+}
+function asToolCall(event) {
+  const record = asRecord2(event);
+  return record?.sessionUpdate === "tool_call" ? record : void 0;
+}
+function asToolCallUpdate(event) {
+  const record = asRecord2(event);
+  return record?.sessionUpdate === "tool_call_update" ? record : void 0;
+}
+function readToolInput(toolCall) {
+  if (Object.hasOwn(toolCall, "input")) {
+    return toolCall.input;
+  }
+  return toolCall.rawInput;
+}
+function readContentText(value) {
+  if (Array.isArray(value)) {
+    return value.map(readContentText).join("");
+  }
+  const record = asRecord2(value);
+  if (record === void 0 || record.type !== "text") {
+    return "";
+  }
+  return readString6(record.text) ?? "";
+}
+function readSpanTimestamps(metadata) {
+  if (metadata === void 0) {
+    return {};
+  }
+  const startTs = readNumber(metadata.startTs);
+  const endTs = readNumber(metadata.endTs);
+  return {
+    ...startTs !== void 0 ? { startTs } : {},
+    ...endTs !== void 0 ? { endTs } : {}
+  };
+}
+function asRecord2(value) {
+  return typeof value === "object" && value !== null ? value : void 0;
+}
+function readString6(value) {
+  return typeof value === "string" ? value : void 0;
+}
+function readNumber(value) {
+  return typeof value === "number" && Number.isFinite(value) ? value : void 0;
+}
+function addMetric(metrics, key2, value) {
+  if (value !== void 0) {
+    metrics[key2] = value;
+  }
+}
+function sumIfPresent(left, right) {
+  return left !== void 0 && right !== void 0 ? left + right : void 0;
+}
+
+// packages/acp-telemetry/src/emit-braintrust.ts
+function emitToBraintrust(trace, parent) {
+  const root = parent.startSpan({ name: trace.root.name, type: "task" });
+  try {
+    emitSpan(root, trace.root);
+  } finally {
+    root.end();
+  }
+}
+function emitSpan(span, traceSpan) {
+  span.log(toBraintrustLogEvent(traceSpan));
+  for (const child of traceSpan.children) {
+    const childSpan = span.startSpan({ name: child.name, type: "tool" });
+    try {
+      emitSpan(childSpan, child);
+    } finally {
+      childSpan.end();
+    }
+  }
+}
+function toBraintrustLogEvent(span) {
+  return {
+    ...Object.hasOwn(span, "input") ? { input: span.input } : {},
+    ...Object.hasOwn(span, "output") ? { output: span.output } : {},
+    ...Object.hasOwn(span, "metadata") ? { metadata: span.metadata } : {},
+    ...Object.hasOwn(span, "metrics") ? { metrics: span.metrics } : {}
+  };
+}
+
 // packages/braintrust/src/row-builder.ts
 function makePipelineRowState(client) {
   const rows = /* @__PURE__ */ new Map();
@@ -29950,8 +30164,8 @@ async function openCurrentChildSpan(client, args, ctx) {
   }
 }
 function buildPipelineCompletionLog(started, completed) {
-  const startRecord = asRecord2(started) ?? {};
-  const completeRecord = asRecord2(completed) ?? {};
+  const startRecord = asRecord3(started) ?? {};
+  const completeRecord = asRecord3(completed) ?? {};
   return {
     input: redact({
       step_name: readPipelineStep(started),
@@ -29974,7 +30188,7 @@ function buildPipelineCompletionLog(started, completed) {
   };
 }
 function buildSuperintendentLog(role, result) {
-  const record = asRecord2(result);
+  const record = asRecord3(result);
   const event = {
     input: redact(record?.input),
     output: redact(record?.output ?? result)
@@ -29988,7 +30202,7 @@ function buildSuperintendentLog(role, result) {
   return event;
 }
 function buildExperimentLog(row, entry) {
-  const entryRecord = asRecord2(entry) ?? {};
+  const entryRecord = asRecord3(entry) ?? {};
   const scores = buildExperimentScores(row.baseline, entry.scores);
   const metrics = { ...row.metrics };
   if (Number.isFinite(entry.durationMs)) {
@@ -30017,12 +30231,12 @@ function buildPipelineMetrics(progress) {
   const metrics = {};
   const usage = progress.usage;
   if (usage !== void 0) {
-    addMetric(metrics, "prompt_tokens", usage.inputTokens);
-    addMetric(metrics, "completion_tokens", usage.outputTokens);
-    addMetric(metrics, "tokens", usage.inputTokens + usage.outputTokens);
-    addMetric(metrics, "prompt_cached_tokens", usage.cachedTokens);
+    addMetric2(metrics, "prompt_tokens", usage.inputTokens);
+    addMetric2(metrics, "completion_tokens", usage.outputTokens);
+    addMetric2(metrics, "tokens", usage.inputTokens + usage.outputTokens);
+    addMetric2(metrics, "prompt_cached_tokens", usage.cachedTokens);
   }
-  addMetric(metrics, "durationMs", progress.durationMs);
+  addMetric2(metrics, "durationMs", progress.durationMs);
   return metrics;
 }
 function buildExperimentScores(baseline, scores) {
@@ -30099,7 +30313,7 @@ function asSpanParent(value) {
   }
   return span;
 }
-function asRecord2(value) {
+function asRecord3(value) {
   return typeof value === "object" && value !== null ? value : void 0;
 }
 function readFirstString(record, keys) {
@@ -30114,7 +30328,7 @@ function readFirstValue(record, keys) {
   }
   return void 0;
 }
-function addMetric(metrics, key2, value) {
+function addMetric2(metrics, key2, value) {
   if (value !== void 0 && Number.isFinite(value)) {
     metrics[key2] = value;
   }
@@ -30166,197 +30380,6 @@ function createPipelineCallbacks(client) {
   };
 }
 
-// packages/braintrust/src/span-builder.ts
-async function logSpawnSession(client, ctx) {
-  try {
-    const { currentSpan } = await import("braintrust");
-    const agentSpan = asSpanParent2(currentSpan()).startSpan({
-      name: `agent:${ctx.agent}:${ctx.model ?? "?"}`,
-      type: "task"
-    });
-    try {
-      logToolSpans(agentSpan, ctx.events);
-      agentSpan.log({
-        input: redact({
-          prompt: ctx.prompt,
-          mode: ctx.mode,
-          cwd: ctx.cwd
-        }),
-        output: redact(accumulateAgentOutput(ctx.events)),
-        metadata: {
-          sessionId: ctx.sessionId,
-          threadId: ctx.threadId,
-          ...ctx.metadata
-        },
-        metrics: buildMetrics(ctx)
-      });
-    } finally {
-      agentSpan.end();
-    }
-  } catch (err) {
-    client.recordError(err, "log spawn session");
-  }
-}
-function logToolSpans(agentSpan, events) {
-  for (const [index, event] of events.entries()) {
-    const toolCall = asToolCall(event);
-    if (toolCall === void 0) {
-      continue;
-    }
-    const toolSpan = agentSpan.startSpan({
-      name: `tool_call:${readString6(toolCall.kind) ?? "unknown"}`,
-      type: "tool"
-    });
-    try {
-      const metadata = collectToolMeta(events, index, readString6(toolCall.toolCallId));
-      toolSpan.log({
-        input: redact(readToolInput(toolCall)),
-        output: redact(assembleToolOutput(events, index, readString6(toolCall.toolCallId))),
-        ...metadata ? { metadata } : {}
-      });
-    } finally {
-      toolSpan.end();
-    }
-  }
-}
-function collectToolMeta(events, toolCallIndex, toolCallId) {
-  const merged = {};
-  const startMeta = asRecord3(asRecord3(events[toolCallIndex])?._meta);
-  if (startMeta) {
-    for (const [key2, value] of Object.entries(startMeta)) {
-      merged[key2 === "ts" ? "startTs" : key2] = value;
-    }
-  }
-  for (const event of events.slice(toolCallIndex + 1)) {
-    const update = asToolCallUpdate(event);
-    if (update === void 0) continue;
-    if (toolCallId !== void 0 && update.toolCallId !== toolCallId) continue;
-    const updateMeta = asRecord3(update._meta);
-    if (!updateMeta) continue;
-    for (const [key2, value] of Object.entries(updateMeta)) {
-      merged[key2 === "ts" ? "endTs" : key2] = value;
-    }
-  }
-  return Object.keys(merged).length > 0 ? merged : void 0;
-}
-function accumulateAgentOutput(events) {
-  let output = "";
-  for (const event of events) {
-    const record = asRecord3(event);
-    if (record === void 0) {
-      continue;
-    }
-    if (record.event === "agent_message") {
-      output += readString6(record.text) ?? "";
-      continue;
-    }
-    if (record.sessionUpdate === "agent_message_chunk") {
-      output += readContentText(record.content);
-    }
-  }
-  return output;
-}
-function assembleToolOutput(events, toolCallIndex, toolCallId) {
-  const outputs = [];
-  let text5 = "";
-  for (const event of events.slice(toolCallIndex + 1)) {
-    const update = asToolCallUpdate(event);
-    if (update === void 0) {
-      continue;
-    }
-    if (toolCallId !== void 0 && update.toolCallId !== toolCallId) {
-      continue;
-    }
-    if (Object.hasOwn(update, "rawOutput")) {
-      outputs.push(update.rawOutput);
-    }
-    const contentText = readContentText(update.content);
-    if (contentText.length > 0) {
-      text5 += contentText;
-    }
-  }
-  if (outputs.length === 0) {
-    return text5;
-  }
-  if (text5.length > 0) {
-    outputs.push(text5);
-  }
-  return outputs.length === 1 ? outputs[0] : outputs;
-}
-function buildMetrics(ctx) {
-  const usage = ctx.usage;
-  const metrics = {};
-  const promptTokens = readNumber(usage.prompt_tokens) ?? readNumber(usage.inputTokens);
-  const completionTokens = readNumber(usage.completion_tokens) ?? readNumber(usage.outputTokens);
-  addMetric2(metrics, "prompt_tokens", promptTokens);
-  addMetric2(metrics, "completion_tokens", completionTokens);
-  addMetric2(
-    metrics,
-    "tokens",
-    readNumber(usage.tokens) ?? sumIfPresent(promptTokens, completionTokens)
-  );
-  addMetric2(
-    metrics,
-    "prompt_cached_tokens",
-    readNumber(usage.prompt_cached_tokens) ?? readNumber(usage.cachedTokens)
-  );
-  addMetric2(
-    metrics,
-    "prompt_cache_creation_tokens",
-    readNumber(usage.prompt_cache_creation_tokens)
-  );
-  addMetric2(metrics, "durationMs", readNumber(usage.durationMs));
-  return metrics;
-}
-function asSpanParent2(value) {
-  const span = value;
-  if (span === void 0 || typeof span.startSpan !== "function") {
-    throw new Error("Braintrust current span unavailable");
-  }
-  return span;
-}
-function asToolCall(event) {
-  const record = asRecord3(event);
-  return record?.sessionUpdate === "tool_call" ? record : void 0;
-}
-function asToolCallUpdate(event) {
-  const record = asRecord3(event);
-  return record?.sessionUpdate === "tool_call_update" ? record : void 0;
-}
-function readToolInput(toolCall) {
-  if (Object.hasOwn(toolCall, "input")) {
-    return toolCall.input;
-  }
-  return toolCall.rawInput;
-}
-function readContentText(value) {
-  if (Array.isArray(value)) {
-    return value.map(readContentText).join("");
-  }
-  const record = asRecord3(value);
-  if (record === void 0 || record.type !== "text") {
-    return "";
-  }
-  return readString6(record.text) ?? "";
-}
-function asRecord3(value) {
-  return typeof value === "object" && value !== null ? value : void 0;
-}
-function readString6(value) {
-  return typeof value === "string" ? value : void 0;
-}
-function readNumber(value) {
-  return typeof value === "number" && Number.isFinite(value) ? value : void 0;
-}
-function addMetric2(metrics, key2, value) {
-  if (value !== void 0) {
-    metrics[key2] = value;
-  }
-}
-function sumIfPresent(left, right) {
-  return left !== void 0 && right !== void 0 ? left + right : void 0;
-}
-
 // packages/braintrust/src/adapters/spawn.ts
 function createSpawnMiddleware(client) {
   return async (ctx, next) => {
@@ -30370,7 +30393,12 @@ function createSpawnMiddleware(client) {
       };
       throw err;
     } finally {
-      await logSpawnSession(client, ctx);
+      try {
+        const { currentSpan } = await import("braintrust");
+        emitToBraintrust(acpToTrace(ctx), currentSpan());
+      } catch (err) {
+        client.recordError(err, "log spawn session");
+      }
     }
   };
 }
@@ -30401,7 +30429,7 @@ function createSuperintendentCallbacks(client) {
 async function logFailedRole(client, role, error2, name) {
   try {
     const { currentSpan } = await import("braintrust");
-    const span = asSpanParent3(currentSpan()).startSpan({
+    const span = asSpanParent2(currentSpan()).startSpan({
       name: name === void 0 ? `role:${role}:failed` : `role:${role}:${name}:failed`,
       type: "task"
     });
@@ -30423,7 +30451,7 @@ async function logFailedRole(client, role, error2, name) {
     client.recordError(err, `superintendent ${role} failed`);
   }
 }
-function asSpanParent3(value) {
+function asSpanParent2(value) {
   const span = value;
   if (span === void 0 || typeof span.startSpan !== "function") {
     throw new Error("Braintrust current span unavailable");