@liy/agent-runner 0.3.2 → 0.3.3

package/dist/index.js

@@ -14126,40 +14126,7 @@ var visualComposerInputSchema = external_exports.object({
   assetGraph: assetGraphSchema.optional()
 }).strict();
 
-// ../mote-core/dist/debug/log-event.js
-var defaultMoteLogRedactPaths = [
-  "attributes.auth.token",
-  "attributes.http.request.header.authorization",
-  "attributes.http.request.header.cookie",
-  "body.headers.authorization",
-  "body.headers.cookie",
-  "body.apiKey",
-  "body.context",
-  "body.password",
-  "body.token",
-  "body.error.stack"
-];
-var moteLogPolicyPresets = {
-  localDebug: {
-    bodyMode: "full",
-    redactionMode: "censor",
-    maxBodyBytes: 256 * 1024,
-    redactPaths: defaultMoteLogRedactPaths
-  },
-  production: {
-    bodyMode: "summary",
-    redactionMode: "remove",
-    maxBodyBytes: 4 * 1024,
-    redactPaths: defaultMoteLogRedactPaths
-  },
-  thirdParty: {
-    bodyMode: "drop",
-    redactionMode: "remove",
-    redactPaths: defaultMoteLogRedactPaths
-  }
-};
-
-// ../mote-core/dist/debug/planner-artifact-pool-snapshot.js
+// ../mote-core/dist/planner-artifact-pool-observability.js
 var DEFAULT_PLANNER_ARTIFACT_POOL_SNAPSHOT_MAX_BYTES = 64 * 1024;
 
 // ../mote-core/dist/contracts/data-result.js
@@ -15079,7 +15046,7 @@ var QuerySchema = external_exports.object({
   limit: external_exports.number().int().positive().max(1e4).default(1e3),
   accessLevel: AccessLevelSchema.default("serving")
 }).superRefine((query, ctx) => {
-  const surfaces = [query.from.surface, ...query.joins.map((join3) => join3.surface)];
+  const surfaces = [query.from.surface, ...query.joins.map((join2) => join2.surface)];
   const knownSurfaces = /* @__PURE__ */ new Set();
   for (const [index, surface] of surfaces.entries()) {
     if (knownSurfaces.has(surface)) {
@@ -15114,8 +15081,8 @@ var QuerySchema = external_exports.object({
       path: ["time", "bucket", "fn"]
     });
   }
-  for (const [index, join3] of query.joins.entries()) {
-    visitConditionFields(join3.on, (field) => {
+  for (const [index, join2] of query.joins.entries()) {
+    visitConditionFields(join2.on, (field) => {
       validateFieldRef(field, ["joins", index, "on"]);
     });
   }
@@ -15632,92 +15599,6 @@ async function forwardHarnessLines(stream, handler) {
 
 // src/harness-drivers/pi-rpc.ts
 import { spawn as nodeSpawn2 } from "node:child_process";
-
-// src/debug/pi-rpc-trace.ts
-import { once } from "node:events";
-import { createWriteStream, mkdirSync } from "node:fs";
-import { dirname as dirname2, join as join2 } from "node:path";
-var MAX_TRACE_LINE_BYTES = 16384;
-var MAX_TRACE_PREVIEW_CHARS = 8e3;
-function createPiRpcTrace(input) {
-  const filePath = join2(input.workspaceDir, "debug", "pi-rpc.jsonl");
-  try {
-    mkdirSync(dirname2(filePath), { recursive: true });
-    const stream = createWriteStream(filePath, {
-      flags: "a",
-      mode: 384
-    });
-    return createStreamTrace({
-      filePath,
-      stream,
-      taskId: input.taskId,
-      now: input.now ?? (() => /* @__PURE__ */ new Date())
-    });
-  } catch {
-    return createNoopTrace();
-  }
-}
-function createStreamTrace(input) {
-  let closed = false;
-  let failed = false;
-  input.stream.on("error", () => {
-    failed = true;
-  });
-  return {
-    filePath: input.filePath,
-    write(event) {
-      if (closed || failed) {
-        return;
-      }
-      input.stream.write(serializeTraceRecord({
-        ts: input.now().toISOString(),
-        taskId: input.taskId,
-        ...event
-      }));
-    },
-    async close() {
-      if (closed) {
-        return;
-      }
-      closed = true;
-      if (failed) {
-        return;
-      }
-      input.stream.end();
-      await once(input.stream, "finish").then(() => void 0, () => void 0);
-    }
-  };
-}
-function createNoopTrace() {
-  return {
-    write() {
-    },
-    async close() {
-    }
-  };
-}
-function serializeTraceRecord(record2) {
-  const line = `${JSON.stringify(record2)}
-`;
-  if (Buffer.byteLength(line, "utf8") <= MAX_TRACE_LINE_BYTES) {
-    return line;
-  }
-  return `${JSON.stringify({
-    ...record2,
-    truncated: true,
-    payload: previewPayload(record2.payload)
-  })}
-`;
-}
-function previewPayload(payload) {
-  const text = typeof payload === "string" ? payload : JSON.stringify(payload) ?? "undefined";
-  if (text.length <= MAX_TRACE_PREVIEW_CHARS) {
-    return text;
-  }
-  return `${text.slice(0, MAX_TRACE_PREVIEW_CHARS)}...[truncated]`;
-}
-
-// src/harness-drivers/pi-rpc.ts
 var driverOwnedPiFlags = /* @__PURE__ */ new Set([
   "--mode",
   "--provider",
@@ -15775,8 +15656,7 @@ function buildPiRpcArgs(harness) {
 }
 function startPiRpcHarness(input, spawnProcess) {
   const harness = input.harness;
-  const args = buildPiRpcArgs(harness);
-  const child = spawnProcess(harness.command, args, {
+  const child = spawnProcess(harness.command, buildPiRpcArgs(harness), {
     cwd: input.spec.workspaceDir,
     env: input.env,
     stdio: ["pipe", "pipe", "pipe"]
@@ -15784,19 +15664,6 @@ function startPiRpcHarness(input, spawnProcess) {
   const sanitizer = createHarnessOutputSanitizer(input.env, [
     ...input.harness.secretEnv ?? []
   ]);
-  const trace = createPiRpcTrace({
-    taskId: input.spec.taskId,
-    workspaceDir: input.spec.workspaceDir
-  });
-  trace.write({
-    kind: "process.started",
-    stream: "process",
-    payload: {
-      command: harness.command,
-      args,
-      cwd: input.spec.workspaceDir
-    }
-  });
   let cleanupTermination = () => void 0;
   let agentEndReached = false;
   let resolveAgentEnd = () => void 0;
@@ -15813,7 +15680,6 @@ function startPiRpcHarness(input, spawnProcess) {
       input,
       line,
       sanitizer,
-      trace,
       onAgentEnd() {
         agentEndReached = true;
         resolveAgentEnd("agent-end");
@@ -15824,36 +15690,20 @@ function startPiRpcHarness(input, spawnProcess) {
     });
   });
   const stderrDone = forwardJsonlByteStream(child.stderr, (line) => {
-    forwardPiStderrLine(input, line, sanitizer, trace);
+    forwardPiStderrLine(input, line, sanitizer);
   });
-  const promptCommand = {
+  writePiRpcCommand(child, {
     id: `mote-${input.spec.taskId}`,
     type: "prompt",
     message: input.prompt
-  };
-  writePiRpcCommand(child, promptCommand);
-  trace.write({
-    kind: "prompt.sent",
-    stream: "stdin",
-    payload: {
-      id: promptCommand.id,
-      type: promptCommand.type,
-      messageLength: input.prompt.length
-    }
   });
   const result = (async () => {
     let terminal;
     try {
       terminal = await Promise.race([closePromise, agentEndPromise]);
     } catch (error48) {
-      trace.write({
-        kind: "process.error",
-        stream: "process",
-        payload: formatTraceError(error48)
-      });
       cleanupTermination();
       cleanupTermination = terminateChildProcess(child);
-      await trace.close();
       throw error48;
     }
     if (terminal === "agent-end") {
@@ -15862,12 +15712,6 @@ function startPiRpcHarness(input, spawnProcess) {
       const closeResult = await closePromise;
       cleanupTermination();
       await Promise.all([stdoutDone, stderrDone]);
-      trace.write({
-        kind: "process.closed",
-        stream: "process",
-        payload: closeResult
-      });
-      await trace.close();
       return {
         exitCode: closeResult.exitCode,
         signal: closeResult.signal,
@@ -15876,12 +15720,6 @@ function startPiRpcHarness(input, spawnProcess) {
     }
     cleanupTermination();
     await Promise.all([stdoutDone, stderrDone]);
-    trace.write({
-      kind: "process.closed",
-      stream: "process",
-      payload: terminal
-    });
-    await trace.close();
     if (!agentEndReached && terminal.exitCode !== 0) {
       forwardPiHarnessStreamLine(
         input,
@@ -15909,44 +15747,20 @@ function handlePiRpcStdoutLine(input) {
   try {
     parsed = JSON.parse(trimmed);
   } catch {
-    const message = input.sanitizer.redactText(trimmed);
-    input.trace.write({
-      kind: "stdout.text",
-      stream: "stdout",
-      payload: {
-        message
-      }
-    });
-    forwardPiHarnessStreamLine(input.input, "stdout", message);
+    forwardPiHarnessStreamLine(input.input, "stdout", input.sanitizer.redactText(trimmed));
     return;
   }
-  const redacted = input.sanitizer.redactJson(parsed);
   if (isPiRpcResponse(parsed)) {
-    input.trace.write({
-      kind: "stdout.response",
-      stream: "stdout",
-      payload: redacted
-    });
     if (parsed.command === "prompt" && parsed.success === false) {
       input.onPromptRejected(new Error(`Pi RPC prompt was rejected: ${formatPiRpcError(parsed, input.sanitizer.redactText)}`));
     }
     return;
   }
   if (isPiExtensionUiRequest(parsed)) {
-    input.trace.write({
-      kind: "stdout.extension_ui_request",
-      stream: "stdout",
-      payload: redacted
-    });
-    forwardPiExtensionUiRequest(input.input, redacted);
+    forwardPiExtensionUiRequest(input.input, input.sanitizer.redactJson(parsed));
     return;
   }
-  input.trace.write({
-    kind: "stdout.event",
-    stream: "stdout",
-    payload: redacted
-  });
-  forwardPiEvent(input.input, redacted);
+  forwardPiEvent(input.input, input.sanitizer.redactJson(parsed));
   if (parsed.type === "agent_end") {
     input.onAgentEnd();
   }
@@ -15981,16 +15795,8 @@ function forwardPiHarnessStreamLine(input, stream, message) {
     message: trimmed
   }));
 }
-function forwardPiStderrLine(input, line, sanitizer, trace) {
-  const message = sanitizer.redactText(line);
-  trace.write({
-    kind: "stderr.line",
-    stream: "stderr",
-    payload: {
-      message
-    }
-  });
-  forwardPiHarnessStreamLine(input, "stderr", message);
+function forwardPiStderrLine(input, line, sanitizer) {
+  forwardPiHarnessStreamLine(input, "stderr", sanitizer.redactText(line));
 }
 function writePiRpcCommand(child, command) {
   child.stdin.write(`${JSON.stringify(command)}
@@ -16029,17 +15835,6 @@ function formatPiRpcError(parsed, redactText) {
   }
   return "unknown error";
 }
-function formatTraceError(error48) {
-  if (error48 instanceof Error) {
-    return {
-      name: error48.name,
-      message: error48.message
-    };
-  }
-  return {
-    message: String(error48)
-  };
-}
 
 // src/prompt.ts
 import { existsSync, readFileSync } from "node:fs";