diff --git a/libs/community/langchain_community/embeddings/voyageai.py b/libs/community/langchain_community/embeddings/voyageai.py index 93109d45c65b6..e0bc53852c17b 100644 --- a/libs/community/langchain_community/embeddings/voyageai.py +++ b/libs/community/langchain_community/embeddings/voyageai.py @@ -69,12 +69,12 @@ class VoyageEmbeddings(BaseModel, Embeddings): from langchain_community.embeddings import VoyageEmbeddings - voyage = VoyageEmbeddings(voyage_api_key="your-api-key") + voyage = VoyageEmbeddings(voyage_api_key="your-api-key", model="voyage-2") text = "This is a test query." query_result = voyage.embed_query(text) """ - model: str = "voyage-01" + model: str voyage_api_base: str = "https://api.voyageai.com/v1/embeddings" voyage_api_key: Optional[SecretStr] = None batch_size: int = 8 @@ -121,7 +121,7 @@ def _get_embeddings( embeddings: List[List[float]] = [] if batch_size is None: - batch_size = self.batch_size + batch_size = 72 if self.model in ["voyage-2", "voyage-02"] else 7 if self.show_progress_bar: try: diff --git a/libs/community/tests/integration_tests/embeddings/test_voyageai.py b/libs/community/tests/integration_tests/embeddings/test_voyageai.py index b23dbd7f538b3..abdf08e94d5b0 100644 --- a/libs/community/tests/integration_tests/embeddings/test_voyageai.py +++ b/libs/community/tests/integration_tests/embeddings/test_voyageai.py @@ -2,7 +2,7 @@ from langchain_community.embeddings.voyageai import VoyageEmbeddings # Please set VOYAGE_API_KEY in the environment variables -MODEL = "voyage-01" +MODEL = "voyage-2" def test_voyagi_embedding_documents() -> None: