Skip to content

Commit 7f3a0fe

Browse files
reset default batch_size to 10 for all vectorizers
1 parent 9a3eb08 commit 7f3a0fe

File tree

6 files changed

+8
-8
lines changed

6 files changed

+8
-8
lines changed

redisvl/utils/vectorize/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def embed_many(
7474
self,
7575
texts: List[str],
7676
preprocess: Optional[Callable] = None,
77-
batch_size: int = 1000,
77+
batch_size: int = 10,
7878
as_buffer: bool = False,
7979
**kwargs,
8080
) -> Union[List[List[float]], List[bytes]]:
@@ -96,7 +96,7 @@ async def aembed_many(
9696
self,
9797
texts: List[str],
9898
preprocess: Optional[Callable] = None,
99-
batch_size: int = 1000,
99+
batch_size: int = 10,
100100
as_buffer: bool = False,
101101
**kwargs,
102102
) -> Union[List[List[float]], List[bytes]]:

redisvl/utils/vectorize/text/azureopenai.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -267,7 +267,7 @@ async def aembed_many(
267267
self,
268268
texts: List[str],
269269
preprocess: Optional[Callable] = None,
270-
batch_size: int = 1000,
270+
batch_size: int = 10,
271271
as_buffer: bool = False,
272272
**kwargs,
273273
) -> Union[List[List[float]], List[bytes]]:

redisvl/utils/vectorize/text/custom.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,7 @@ async def aembed_many(
288288
self,
289289
texts: List[str],
290290
preprocess: Optional[Callable] = None,
291-
batch_size: int = 1000,
291+
batch_size: int = 10,
292292
as_buffer: bool = False,
293293
**kwargs,
294294
) -> Union[List[List[float]], List[bytes]]:

redisvl/utils/vectorize/text/huggingface.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def embed_many(
122122
self,
123123
texts: List[str],
124124
preprocess: Optional[Callable] = None,
125-
batch_size: int = 1000,
125+
batch_size: int = 10,
126126
as_buffer: bool = False,
127127
**kwargs,
128128
) -> Union[List[List[float]], List[bytes]]:
@@ -134,7 +134,7 @@ def embed_many(
134134
preprocess (Optional[Callable], optional): Optional preprocessing
135135
callable to perform before vectorization. Defaults to None.
136136
batch_size (int, optional): Batch size of texts to use when creating
137-
embeddings. Defaults to 1000.
137+
embeddings. Defaults to 10.
138138
as_buffer (bool, optional): Whether to convert the raw embedding
139139
to a byte string. Defaults to False.
140140

redisvl/utils/vectorize/text/mistral.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ async def aembed_many(
217217
self,
218218
texts: List[str],
219219
preprocess: Optional[Callable] = None,
220-
batch_size: int = 1000,
220+
batch_size: int = 10,
221221
as_buffer: bool = False,
222222
**kwargs,
223223
) -> List[List[float]]:

redisvl/utils/vectorize/text/openai.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ async def aembed_many(
218218
self,
219219
texts: List[str],
220220
preprocess: Optional[Callable] = None,
221-
batch_size: int = 1000,
221+
batch_size: int = 10,
222222
as_buffer: bool = False,
223223
**kwargs,
224224
) -> Union[List[List[float]], List[bytes]]:

0 commit comments

Comments
 (0)