mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
26 lines
799 B
Python
26 lines
799 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from langchain_community.embeddings.ovhcloud import OVHCloudEmbeddings
|
||
|
|
|
||
|
|
|
||
|
|
def test_ovhcloud_correct_instantiation() -> None:
|
||
|
|
llm = OVHCloudEmbeddings(model_name="multilingual-e5-base")
|
||
|
|
assert isinstance(llm, OVHCloudEmbeddings)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ovhcloud_empty_model_name_should_raise_error() -> None:
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
OVHCloudEmbeddings(model_name="")
|
||
|
|
|
||
|
|
|
||
|
|
def test_ovhcloud_empty_region_should_raise_error() -> None:
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
OVHCloudEmbeddings(model_name="multilingual-e5-base", region="")
|
||
|
|
|
||
|
|
|
||
|
|
def test_ovhcloud_empty_access_token_should_not_raise_error() -> None:
|
||
|
|
llm = OVHCloudEmbeddings(
|
||
|
|
model_name="multilingual-e5-base", region="kepler", access_token=""
|
||
|
|
)
|
||
|
|
assert isinstance(llm, OVHCloudEmbeddings)
|