Skip to content

Commit 199d3c0

Browse files
authored
Add support for Voyage 3.5 models (#339)
Adding Voyage 3.5 models - it affects the default batch size only
1 parent fb797ba commit 199d3c0

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
@@ -181,9 +181,9 @@ def _get_batch_size(self) -> int:
181181
"""
182182
if self.model in ["voyage-2", "voyage-02"]:
183183
return 72
184-
elif self.model == "voyage-3-lite":
184+
elif self.model in ["voyage-3-lite", "voyage-3.5-lite"]:
185185
return 30
186-
elif self.model == "voyage-3":
186+
elif self.model in ["voyage-3", "voyage-3.5"]:
187187
return 10
188188
else:
189189
return 7 # Default for other models

0 commit comments

Comments
 (0)