diff --git a/tests/unit/test_hybrid_layer.py b/tests/unit/test_hybrid_layer.py index 8beded53fd00209bdf1c172bbce07622cd72b901..fbf14566e3c9051331a3825fe7d952213ec38985 100644 --- a/tests/unit/test_hybrid_layer.py +++ b/tests/unit/test_hybrid_layer.py @@ -40,7 +40,7 @@ def cohere_encoder(mocker): @pytest.fixture def openai_encoder(mocker): mocker.patch.object(OpenAIEncoder, "__call__", side_effect=mock_encoder_call) - return OpenAIEncoder(name="text-embedding-ada-002", openai_api_key="test_api_key") + return OpenAIEncoder(name="text-embedding-3-small", openai_api_key="test_api_key") @pytest.fixture diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index e20aae5751b8c5ce68de3401bd2c8230224fbd7c..74c069fa0ba86481d0142b0b8477c5b3758f837e 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -87,7 +87,7 @@ def cohere_encoder(mocker): @pytest.fixture def openai_encoder(mocker): mocker.patch.object(OpenAIEncoder, "__call__", side_effect=mock_encoder_call) - return OpenAIEncoder(name="text-embedding-ada-002", openai_api_key="test_api_key") + return OpenAIEncoder(name="text-embedding-3-small", openai_api_key="test_api_key") @pytest.fixture