diff --git a/redisvl/utils/vectorize/text/voyageai.py b/redisvl/utils/vectorize/text/voyageai.py index 8b181689..1936bd97 100644 --- a/redisvl/utils/vectorize/text/voyageai.py +++ b/redisvl/utils/vectorize/text/voyageai.py @@ -181,9 +181,9 @@ def _get_batch_size(self) -> int: """ if self.model in ["voyage-2", "voyage-02"]: return 72 - elif self.model == "voyage-3-lite": + elif self.model in ["voyage-3-lite", "voyage-3.5-lite"]: return 30 - elif self.model == "voyage-3": + elif self.model in ["voyage-3", "voyage-3.5"]: return 10 else: return 7 # Default for other models