From ea1be71c36374c0e349924eed494236ada35b219 Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Fri, 19 Jul 2024 14:58:14 +0800
Subject: [PATCH] fix: use latest model in test
---
tests/unit/test_hybrid_layer.py | 2 +-
tests/unit/test_layer.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/unit/test_hybrid_layer.py b/tests/unit/test_hybrid_layer.py
index 8beded53..fbf14566 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 e20aae57..74c069fa 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
--
GitLab