Skip to content
Snippets Groups Projects
Unverified Commit ea1be71c authored by James Briggs's avatar James Briggs
Browse files

fix: use latest model in test

parent 4d6ba3a0
No related branches found
No related tags found
No related merge requests found
...@@ -40,7 +40,7 @@ def cohere_encoder(mocker): ...@@ -40,7 +40,7 @@ def cohere_encoder(mocker):
@pytest.fixture @pytest.fixture
def openai_encoder(mocker): def openai_encoder(mocker):
mocker.patch.object(OpenAIEncoder, "__call__", side_effect=mock_encoder_call) 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 @pytest.fixture
......
...@@ -87,7 +87,7 @@ def cohere_encoder(mocker): ...@@ -87,7 +87,7 @@ def cohere_encoder(mocker):
@pytest.fixture @pytest.fixture
def openai_encoder(mocker): def openai_encoder(mocker):
mocker.patch.object(OpenAIEncoder, "__call__", side_effect=mock_encoder_call) 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 @pytest.fixture
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment