From b8e861a63b85167a9c47fabf3452e8baba538a7c Mon Sep 17 00:00:00 2001 From: ccurme Date: Thu, 5 Dec 2024 12:00:27 -0500 Subject: [PATCH] openai[patch]: add standard tests for embeddings (#28540) --- .../embeddings/test_base_standard.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 libs/partners/openai/tests/integration_tests/embeddings/test_base_standard.py diff --git a/libs/partners/openai/tests/integration_tests/embeddings/test_base_standard.py b/libs/partners/openai/tests/integration_tests/embeddings/test_base_standard.py new file mode 100644 index 00000000000..66f74f1687f --- /dev/null +++ b/libs/partners/openai/tests/integration_tests/embeddings/test_base_standard.py @@ -0,0 +1,18 @@ +"""Standard LangChain interface tests""" + +from typing import Type + +from langchain_core.embeddings import Embeddings +from langchain_tests.integration_tests.embeddings import EmbeddingsIntegrationTests + +from langchain_openai import OpenAIEmbeddings + + +class TestOpenAIStandard(EmbeddingsIntegrationTests): + @property + def embeddings_class(self) -> Type[Embeddings]: + return OpenAIEmbeddings + + @property + def embedding_model_params(self) -> dict: + return {"model": "text-embedding-3-small", "dimensions": 128}