Skip to content

Commit 0d4a4ca

Browse files
authored
Lower Vectorize default batch size (#4047)
1 parent 6b1c327 commit 0d4a4ca

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

packages/wrangler/src/vectorize/insert.ts

+8-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@ import type {
1010
StrictYargsOptionsToInterface,
1111
} from "../yargs-types";
1212

13-
const VECTORIZE_UPSERT_BATCH_SIZE = 5_000;
13+
const VECTORIZE_MAX_BATCH_SIZE = 1_000;
14+
const VECTORIZE_UPSERT_BATCH_SIZE = VECTORIZE_MAX_BATCH_SIZE;
1415
const VECTORIZE_MAX_UPSERT_VECTOR_RECORDS = 100_000;
1516

1617
export function options(yargs: CommonYargsArgv) {
@@ -48,6 +49,12 @@ export async function handler(
4849
const config = readConfig(args.config, args);
4950
const rl = createInterface({ input: createReadStream(args.file) });
5051

52+
if (Number(args.batchSize) > VECTORIZE_MAX_BATCH_SIZE) {
53+
logger.error(
54+
`🚨 Vectorize currently limits upload batches to ${VECTORIZE_MAX_BATCH_SIZE} records at a time.`
55+
);
56+
}
57+
5158
let vectorInsertCount = 0;
5259
for await (const batch of getBatchFromFile(rl, args.batchSize)) {
5360
const formData = new FormData();

0 commit comments

Comments
 (0)