Covert MB to bytes for the API

cloudflare · Jan 8, 2025 · 65b67ce · 65b67ce
1 parent def4db0
commit 65b67ce
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/packages/wrangler/src/pipelines/client.ts b/packages/wrangler/src/pipelines/client.ts
@@ -39,7 +39,7 @@ export type PipelineUserConfig = {
 		};
 		batch: {
 			max_duration_s?: number;
-			max_mb?: number;
+			max_bytes?: number;
 			max_rows?: number;
 		};
 		path: {

diff --git a/packages/wrangler/src/pipelines/index.ts b/packages/wrangler/src/pipelines/index.ts
@@ -196,7 +196,9 @@ export function pipelines(pipelineYargs: CommonYargsArgv) {
 					args.compression === undefined ? "gzip" : args.compression;
 
 				const batch = {
-					max_mb: args["batch-max-mb"],
+					max_bytes: args["batch-max-mb"]
+						? args["batch-max-mb"] * 1000 * 1000 // convert to bytes for the API
+						: undefined,
 					max_duration_s: args["batch-max-seconds"],
 					max_rows: args["batch-max-rows"],
 				};
@@ -386,7 +388,8 @@ export function pipelines(pipelineYargs: CommonYargsArgv) {
 					pipelineConfig.destination.compression.type = args.compression;
 				}
 				if (args["batch-max-mb"]) {
-					pipelineConfig.destination.batch.max_mb = args["batch-max-mb"];
+					pipelineConfig.destination.batch.max_bytes =
+						args["batch-max-mb"] * 1000 * 1000; // convert to bytes for the API
 				}
 				if (args["batch-max-seconds"]) {
 					pipelineConfig.destination.batch.max_duration_s =