@odla-ai/harness 0.5.0 → 0.5.1

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,7 +1374,10 @@ 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();