From 979232257be64231236a52a4e86ae4bacfde3f1e Mon Sep 17 00:00:00 2001 From: ccurme Date: Tue, 10 Sep 2024 14:57:16 -0400 Subject: [PATCH] huggingface[patch]: add integration tests for embeddings (#26272) --- .../test_embeddings_standard.py | 30 +++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 libs/partners/huggingface/tests/integration_tests/test_embeddings_standard.py diff --git a/libs/partners/huggingface/tests/integration_tests/test_embeddings_standard.py b/libs/partners/huggingface/tests/integration_tests/test_embeddings_standard.py new file mode 100644 index 00000000000..4f5cefcbdff --- /dev/null +++ b/libs/partners/huggingface/tests/integration_tests/test_embeddings_standard.py @@ -0,0 +1,30 @@ +"""Test HuggingFace embeddings.""" + +from typing import Type + +from langchain_standard_tests.integration_tests import EmbeddingsIntegrationTests + +from langchain_huggingface.embeddings import ( + HuggingFaceEmbeddings, + HuggingFaceEndpointEmbeddings, +) + + +class TestHuggingFaceEmbeddings(EmbeddingsIntegrationTests): + @property + def embeddings_class(self) -> Type[HuggingFaceEmbeddings]: + return HuggingFaceEmbeddings + + @property + def embedding_model_params(self) -> dict: + return {"model_name": "sentence-transformers/all-mpnet-base-v2"} + + +class TestHuggingFaceEndpointEmbeddings(EmbeddingsIntegrationTests): + @property + def embeddings_class(self) -> Type[HuggingFaceEndpointEmbeddings]: + return HuggingFaceEndpointEmbeddings + + @property + def embedding_model_params(self) -> dict: + return {"model": "sentence-transformers/all-mpnet-base-v2"}