diff --git a/tests/unit/encoders/test_azure.py b/tests/unit/encoders/test_azure.py index 93dffb89194606c2d410c1cf749cc57faf6d1327..3969543e8fb28572923db185ef49f78cb724c080 100644 --- a/tests/unit/encoders/test_azure.py +++ b/tests/unit/encoders/test_azure.py @@ -83,7 +83,7 @@ class TestAzureOpenAIEncoder: ) with pytest.raises(ValueError) as e: openai_encoder(["test document"]) - assert "No embeddings returned. Error" in str(e.value) + assert "No embeddings returned." in str(e.value) def test_openai_encoder_call_failure_non_openai_error(self, openai_encoder, mocker): mocker.patch("os.getenv", return_value="fake-api-key") diff --git a/tests/unit/encoders/test_openai.py b/tests/unit/encoders/test_openai.py index 508e9e9e197a8a0b87fb179554d43c18686c1b44..56ef2b38e5082848279624bedfbc9417ec610d6b 100644 --- a/tests/unit/encoders/test_openai.py +++ b/tests/unit/encoders/test_openai.py @@ -77,7 +77,7 @@ class TestOpenAIEncoder: ) with pytest.raises(ValueError) as e: openai_encoder(["test document"]) - assert "No embeddings returned. Error" in str(e.value) + assert "No embeddings returned." in str(e.value) def test_openai_encoder_call_failure_non_openai_error(self, openai_encoder, mocker): mocker.patch("os.getenv", return_value="fake-api-key")