Skip to content

Commit d9da17c

Browse files
committed
Correcting VoyageAI batch size param
1 parent 46c00e1 commit d9da17c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

redisvl/utils/vectorize/text/voyageai.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,7 @@ def _embed_many(
273273
274274
Args:
275275
texts: List of texts to embed
276-
batch_size: Deprecated. Token-aware batching is now always used.
276+
batch_size: Maximum number of texts to process in each API call.
277277
**kwargs: Additional parameters to pass to the VoyageAI API.
278278
279279
Returns:
@@ -290,7 +290,7 @@ def _embed_many(
290290
self._validate_input(texts, input_type, truncation)
291291

292292
# Use token-aware batching
293-
batches = self._build_token_aware_batches(texts)
293+
batches = self._build_token_aware_batches(texts, max_batch_size=batch_size)
294294

295295
try:
296296
embeddings: List = []

0 commit comments

Comments
 (0)