diff --git a/libs/community/tests/integration_tests/test_ovhcloud.py b/libs/community/tests/integration_tests/test_ovhcloud.py new file mode 100644 index 00000000000..bbbaa9d721c --- /dev/null +++ b/libs/community/tests/integration_tests/test_ovhcloud.py @@ -0,0 +1,8 @@ +from langchain_community.embeddings.ovhcloud import OVHCloudEmbeddings + + +def test_ovhcloud_embed_documents() -> None: + llm = OVHCloudEmbeddings(model_name="multilingual-e5-base") + docs = ["Hello", "World"] + output = llm.embed_documents(docs) + assert len(output) == len(docs) diff --git a/libs/community/tests/unit_tests/embeddings/test_ovhcloud.py b/libs/community/tests/unit_tests/embeddings/test_ovhcloud.py index c456238e69a..caec9811186 100644 --- a/libs/community/tests/unit_tests/embeddings/test_ovhcloud.py +++ b/libs/community/tests/unit_tests/embeddings/test_ovhcloud.py @@ -23,10 +23,3 @@ def test_ovhcloud_empty_access_token_should_not_raise_error() -> None: model_name="multilingual-e5-base", region="kepler", access_token="" ) assert isinstance(llm, OVHCloudEmbeddings) - - -def test_ovhcloud_embed_documents() -> None: - llm = OVHCloudEmbeddings(model_name="multilingual-e5-base") - docs = ["Hello", "World"] - output = llm.embed_documents(docs) - assert len(output) == len(docs)