langchain/libs/community/tests/unit_tests/embeddings/test_ovhcloud.py
Mathis Joffre 60103fc4a5
community: Fix OVHcloud 401 Unauthorized on embedding. (#23260)
They are now rejecting with code 401 calls from users with expired or
invalid tokens (while before they were being considered anonymous).
Thus, the authorization header has to be removed when there is no token.

Related to: #23178

---------

Signed-off-by: Joffref <mariusjoffre@gmail.com>
2024-06-24 12:58:32 -04:00

32 lines
1.0 KiB
Python

import pytest
from langchain_community.embeddings.ovhcloud import OVHCloudEmbeddings
def test_ovhcloud_correct_instantiation() -> None:
llm = OVHCloudEmbeddings(model_name="multilingual-e5-base", access_token="token")
assert isinstance(llm, OVHCloudEmbeddings)
llm = OVHCloudEmbeddings(
model_name="multilingual-e5-base", region="kepler", access_token="token"
)
assert isinstance(llm, OVHCloudEmbeddings)
def test_ovhcloud_empty_model_name_should_raise_error() -> None:
with pytest.raises(ValueError):
OVHCloudEmbeddings(model_name="", region="kepler", access_token="token")
def test_ovhcloud_empty_region_should_raise_error() -> None:
with pytest.raises(ValueError):
OVHCloudEmbeddings(
model_name="multilingual-e5-base", region="", access_token="token"
)
def test_ovhcloud_empty_access_token_should_raise_error() -> None:
with pytest.raises(ValueError):
OVHCloudEmbeddings(
model_name="multilingual-e5-base", region="kepler", access_token=""
)