@odla-ai/harness 0.5.0 → 0.5.2

package/dist/code-runtime-cli.cjs

@@ -168,6 +168,7 @@ function parseSnapshot(value) {
   return { host, bindings, commands };
 }
 async function parseSource(value) {
+  const repositoryLimits = { maximumFiles: 1e5, maximumBytes: 80 * 1024 * 1024 };
   const snapshot = record(record(value)?.snapshot);
   if (!snapshot || typeof snapshot.repository !== "string" || typeof snapshot.commitSha !== "string" || typeof snapshot.treeDigest !== "string" || !Array.isArray(snapshot.files)) throw invalid("source");
   const files = snapshot.files.map((value2) => {
@@ -189,12 +190,12 @@ async function parseSource(value) {
       return { path: file.path, content: file.content };
     });
     const source2 = { repository: reference.repository, commitSha: reference.commitSha, files: referenceFiles };
-    const referenceDigest = await (0, import_code.digestCodeRepositorySnapshot)(source2, { maximumFiles: 1e4, maximumBytes: 16 * 1024 * 1024 });
+    const referenceDigest = await (0, import_code.digestCodeRepositorySnapshot)(source2, repositoryLimits);
     if (referenceDigest !== reference.treeDigest) throw invalid("reference source digest");
     references.push({ alias: reference.alias, ...source2, treeDigest: referenceDigest });
   }
   const source = { repository: snapshot.repository, commitSha: snapshot.commitSha, files };
-  const digest = await (0, import_code.digestCodeRepositorySnapshot)(source, { maximumFiles: 1e4, maximumBytes: 16 * 1024 * 1024 });
+  const digest = await (0, import_code.digestCodeRepositorySnapshot)(source, repositoryLimits);
   if (digest !== snapshot.treeDigest) throw invalid("source digest");
   return { ...source, treeDigest: digest, ...references.length ? { references } : {} };
 }
@@ -1266,8 +1267,11 @@ var import_node_os2 = require("os");
 var import_node_path7 = require("path");
 var RESERVED2 = /* @__PURE__ */ new Set([".git", ".odla", ".wrangler", "node_modules", "dist", "coverage"]);
 var SECRET2 = /^(?:\.env(?:\..+)?|\.dev\.vars|credentials(?:\..+)?\.json|dev-token(?:\..+)?\.json)$/i;
+var SOURCE_MAX_FILES = 1e5;
+var SOURCE_MAX_BYTES = 80 * 1024 * 1024;
+var SOURCE_SET_MAX_BYTES = 480 * 1024 * 1024;
 async function materializeCodeRuntimeSource(snapshot, tempRoot = (0, import_node_os2.tmpdir)()) {
-  if (!snapshot.files.length || snapshot.files.length > 1e4) throw new TypeError("Code source file count is invalid");
+  if (!snapshot.files.length || snapshot.files.length > SOURCE_MAX_FILES) throw new TypeError("Code source file count is invalid");
   const root = await (0, import_promises6.mkdtemp)((0, import_node_path7.join)(tempRoot, "odla-code-source-"));
   const sourceDir = (0, import_node_path7.join)(root, "source");
   await (0, import_promises6.mkdir)(sourceDir);
@@ -1279,7 +1283,7 @@ async function materializeCodeRuntimeSource(snapshot, tempRoot = (0, import_node
       if (seen.has(file.path)) throw new TypeError("Code source repeats a path");
       seen.add(file.path);
       bytes += Buffer.byteLength(file.path) + Buffer.byteLength(file.content);
-      if (bytes > 16 * 1024 * 1024) throw new TypeError("Code source exceeds its byte bound");
+      if (bytes > SOURCE_MAX_BYTES) throw new TypeError("Code source exceeds its byte bound");
       const target = (0, import_node_path7.resolve)(sourceDir, file.path);
       if (!target.startsWith(`${(0, import_node_path7.resolve)(sourceDir)}${import_node_path7.sep}`)) throw new TypeError("Code source path escapes its root");
       await (0, import_promises6.mkdir)((0, import_node_path7.dirname)(target), { recursive: true });
@@ -1287,14 +1291,14 @@ async function materializeCodeRuntimeSource(snapshot, tempRoot = (0, import_node
     }
     for (const reference of snapshot.references ?? []) {
       validateAlias(reference.alias);
-      if (!reference.files.length || reference.files.length > 1e4) throw new TypeError("Code reference file count is invalid");
+      if (!reference.files.length || reference.files.length > SOURCE_MAX_FILES) throw new TypeError("Code reference file count is invalid");
       for (const file of reference.files) {
         validatePath(file.path);
         const path = `.odla-references/${reference.alias}/${file.path}`;
         if (seen.has(path)) throw new TypeError("Code reference repeats a path");
         seen.add(path);
         bytes += Buffer.byteLength(path) + Buffer.byteLength(file.content);
-        if (bytes > 80 * 1024 * 1024) throw new TypeError("Code source set exceeds its byte bound");
+        if (bytes > SOURCE_SET_MAX_BYTES) throw new TypeError("Code source set exceeds its byte bound");
         const target = (0, import_node_path7.resolve)(sourceDir, path);
         if (!target.startsWith(`${(0, import_node_path7.resolve)(sourceDir)}${import_node_path7.sep}`)) throw new TypeError("Code reference path escapes its root");
         await (0, import_promises6.mkdir)((0, import_node_path7.dirname)(target), { recursive: true });
@@ -1320,7 +1324,7 @@ async function attachCodeRuntimeReferences(workspace, references) {
       validatePath(file.path);
       const path = `.odla-references/${reference.alias}/${file.path}`;
       bytes += Buffer.byteLength(path) + Buffer.byteLength(file.content);
-      if (bytes > 64 * 1024 * 1024) throw new TypeError("Code reference set exceeds its byte bound");
+      if (bytes > SOURCE_SET_MAX_BYTES - SOURCE_MAX_BYTES) throw new TypeError("Code reference set exceeds its byte bound");
       for (const root of [workspace.baselineDir, workspace.workspaceDir]) {
         const target = (0, import_node_path7.resolve)(root, path);
         if (!target.startsWith(`${(0, import_node_path7.resolve)(root)}${import_node_path7.sep}`)) throw new TypeError("Code reference path escapes its root");
@@ -1370,13 +1374,19 @@ async function materializeCommandWorkspace(input) {
       trustedBaseDir: materialized.sourceDir,
       trustedBaseCommitSha: source.commitSha,
       checkpoint: codeCheckpointPayload(command.payload)
-    })).workspace : await stageWorkspace(materialized.sourceDir);
+    })).workspace : await stageWorkspace(materialized.sourceDir, {
+      maxFiles: SOURCE_MAX_FILES,
+      maxBytes: SOURCE_SET_MAX_BYTES
+    });
     return { workspace, sourceDigest: source.treeDigest, requestedLocal: null };
   } finally {
     await materialized.cleanup();
   }
 }
 
+// src/code-runtime-attempt.ts
+var import_ai2 = require("@odla-ai/ai");
+
 // src/code-agent.ts
 var import_ai = require("@odla-ai/ai");
 
@@ -1577,6 +1587,7 @@ async function runCodeAgent(options) {
 // src/code-runtime-attempt.ts
 async function runCodeAgentAttempt(options) {
   try {
+    const surface = options.surface ?? "v2";
     const { run } = await runCodeAgent({
       inference: options.inference,
       broker: options.broker,
@@ -1586,17 +1597,31 @@ async function runCodeAgentAttempt(options) {
       // The brokered route resolves the real model from platform policy; this
       // id only labels the request the control plane is about to rewrite.
       model: "brokered",
-      surface: options.surface ?? "v2",
+      surface,
       ...options.maxSteps === void 0 ? {} : { maxSteps: options.maxSteps },
       ...options.budget ? { budget: options.budget } : {},
       ...options.signal ? { signal: options.signal } : {},
       ...options.onToolCall ? { onToolCall: options.onToolCall } : {}
     });
+    let finalText = run.finalText.trim();
+    if (!finalText && run.stoppedReason !== "refusal") {
+      const closing = await options.inference.chat({
+        model: "brokered",
+        system: `${SYSTEM_PROMPT_FOR[surface]}
+
+Finish with a concise, non-empty answer to the owner. Do not call tools or promise future work.`,
+        messages: [...run.messages, { role: "user", content: "Give the owner the closing answer now, grounded in the repository evidence and tool results above." }],
+        maxTokens: 16384,
+        ...options.signal ? { signal: options.signal } : {}
+      });
+      finalText = (0, import_ai2.extractText)(closing.content).trim();
+    }
+    const missingClosing = !finalText && run.stoppedReason !== "refusal";
     return {
-      status: run.stoppedReason === "refusal" ? "failed" : "completed",
-      finalText: run.finalText,
+      status: run.stoppedReason === "refusal" || missingClosing ? "failed" : "completed",
+      finalText,
       stoppedReason: run.stoppedReason,
-      ...run.stoppedReason === "refusal" ? { error: run.finalText || "the agent refused the task" } : {}
+      ...run.stoppedReason === "refusal" ? { error: finalText || "the agent refused the task" } : missingClosing ? { error: "the Code agent did not produce a closing answer" } : {}
     };
   } catch (cause) {
     const error = (cause instanceof Error ? cause.message : String(cause)).slice(0, 2e3);
@@ -1606,31 +1631,7 @@ async function runCodeAgentAttempt(options) {
 
 // src/code-runtime-inference.ts
 async function handleCodeRuntimeInference(input) {
-  const { command, metadata, request, state } = input;
-  if (state.tokens >= metadata.maxTokensPerInteraction) {
-    if (!state.noticeEmitted) {
-      state.noticeEmitted = true;
-      await input.event({
-        type: "message",
-        actor: "system",
-        body: `The agent paused at the ${metadata.maxTokensPerInteraction.toLocaleString("en-US")}-token per-interaction limit. Send a new instruction to continue.`
-      }).catch(() => void 0);
-    }
-    return {
-      protocolVersion: HARNESS_PROTOCOL_VERSION,
-      type: "inference.response",
-      requestId: request.requestId,
-      response: {
-        id: `budget:${command.commandId}`,
-        provider: "openai",
-        model: "interaction-budget",
-        role: "assistant",
-        content: [{ type: "text", text: "Pause now. The owner-set token limit for this interaction has been reached." }],
-        stopReason: "end_turn",
-        usage: { inputTokens: 0, outputTokens: 0 }
-      }
-    };
-  }
+  const { command, request, state } = input;
   const startedAt = Date.now();
   const response2 = await input.control.infer(command.sessionId, {
     requestId: request.requestId,
@@ -1650,7 +1651,6 @@ async function handleCodeRuntimeInference(input) {
     durationMs: Date.now() - startedAt,
     interactionId: command.commandId,
     interactionTokens: state.tokens,
-    interactionMaxTokens: metadata.maxTokensPerInteraction,
     ...costUsd === void 0 ? {} : { costUsd },
     ...state.costKnown ? { interactionCostUsd: state.costUsd } : {}
   }).catch(() => void 0);
@@ -2758,7 +2758,7 @@ var CodePiRuntimeEngine = class {
       recipeAuthorization: this.options.recipeAuthorization
     }, lease, metadata.role));
     const startedAt = Date.now();
-    const interaction = { tokens: 0, noticeEmitted: false, costUsd: 0, costKnown: true };
+    const interaction = { tokens: 0, costUsd: 0, costKnown: true };
     const inference = createCodeRuntimeInference({
       command,
       metadata,
@@ -2773,31 +2773,30 @@ var CodePiRuntimeEngine = class {
       lease,
       workspaceDir: active.workspace.workspaceDir,
       prompt: metadata.prompt,
-      signal: active.abort.signal,
-      // The owner's per-interaction allowance, enforced by runAgent against
-      // INCREMENTAL usage. The control plane still reserves against the same
-      // ceiling, but this is what stops the loop cleanly at the boundary rather
-      // than letting it discover the limit through a synthesized pause reply.
-      budget: { maxTotalTokens: metadata.maxTokensPerInteraction }
+      signal: active.abort.signal
     });
-    const body = result.finalText.trim() || (result.status === "completed" ? "The agent finished without a closing message." : result.error ?? "The agent failed.");
+    const closing = result.finalText.trim();
+    const completed = result.status === "completed" && Boolean(closing);
+    const detail = result.error?.trim() || (closing ? "the Code agent failed" : "the Code agent did not produce a closing answer");
+    const body = closing || detail;
     await this.#event(command, {
       type: "message",
-      actor: result.status === "completed" ? "agent" : "system",
+      actor: completed ? "agent" : "system",
       body
     }, active.conversationRefs).catch(() => void 0);
     await this.#event(command, {
       type: "status",
-      status: result.status === "completed" ? "idle" : "failed",
+      status: completed ? "idle" : "failed",
       durationMs: Date.now() - startedAt
     }, active.conversationRefs).catch(() => void 0);
-    if (result.status === "failed") {
-      const detail = (result.error ?? "").trim() || "the Code agent failed";
+    if (!completed) {
       await this.#diagnostic(command, active, detail);
       await this.#failure(command, active, detail);
     }
     return {
       ...result,
+      status: completed ? "completed" : "failed",
+      ...!completed ? { error: detail } : {},
       tokens: interaction.tokens,
       ...interaction.costKnown ? { costUsd: interaction.costUsd } : {}
     };