From 4d6ba3a0dacded4bb08a694100249bc2648f1324 Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Fri, 19 Jul 2024 14:49:31 +0800
Subject: [PATCH] fix: update default test values
---
tests/unit/llms/test_llm_openai.py | 2 +-
tests/unit/test_hybrid_layer.py | 6 +++---
tests/unit/test_layer.py | 10 +++++-----
3 files changed, 9 insertions(+), 9 deletions(-)
diff --git a/tests/unit/llms/test_llm_openai.py b/tests/unit/llms/test_llm_openai.py
index c5f98710..4287fc88 100644
--- a/tests/unit/llms/test_llm_openai.py
+++ b/tests/unit/llms/test_llm_openai.py
@@ -43,7 +43,7 @@ example_function_schema = {
class TestOpenAILLM:
def test_openai_llm_init_with_api_key(self, openai_llm):
assert openai_llm.client is not None, "Client should be initialized"
- assert openai_llm.name == "gpt-3.5-turbo", "Default name not set correctly"
+ assert openai_llm.name == "gpt-4o", "Default name not set correctly"
def test_openai_llm_init_success(self, mocker):
mocker.patch("os.getenv", return_value="fake-api-key")
diff --git a/tests/unit/test_hybrid_layer.py b/tests/unit/test_hybrid_layer.py
index 416eb934..8beded53 100644
--- a/tests/unit/test_hybrid_layer.py
+++ b/tests/unit/test_hybrid_layer.py
@@ -88,8 +88,8 @@ class TestHybridRouteLayer:
alpha=0.8,
)
assert route_layer.index is not None and route_layer.categories is not None
- assert openai_encoder.score_threshold == 0.82
- assert route_layer.score_threshold == 0.82
+ assert openai_encoder.score_threshold == 0.3
+ assert route_layer.score_threshold == 0.3
assert route_layer.top_k == 10
assert route_layer.alpha == 0.8
assert len(route_layer.index) == 5
@@ -104,7 +104,7 @@ class TestHybridRouteLayer:
route_layer_openai = HybridRouteLayer(
encoder=openai_encoder, sparse_encoder=sparse_encoder
)
- assert route_layer_openai.score_threshold == 0.82
+ assert route_layer_openai.score_threshold == 0.3
def test_add_route(self, openai_encoder):
route_layer = HybridRouteLayer(
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 4566401d..e20aae57 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -155,8 +155,8 @@ class TestRouteLayer:
route_layer = RouteLayer(
encoder=openai_encoder, routes=routes, top_k=10, index=index_cls()
)
- assert openai_encoder.score_threshold == 0.82
- assert route_layer.score_threshold == 0.82
+ assert openai_encoder.score_threshold == 0.3
+ assert route_layer.score_threshold == 0.3
assert route_layer.top_k == 10
assert len(route_layer.index) if route_layer.index is not None else 0 == 5
assert (
@@ -172,7 +172,7 @@ class TestRouteLayer:
assert cohere_encoder.score_threshold == 0.3
assert route_layer_cohere.score_threshold == 0.3
route_layer_openai = RouteLayer(encoder=openai_encoder, index=index_cls())
- assert route_layer_openai.score_threshold == 0.82
+ assert route_layer_openai.score_threshold == 0.3
def test_initialization_no_encoder(self, openai_encoder, index_cls):
os.environ["OPENAI_API_KEY"] = "test_api_key"
@@ -189,8 +189,8 @@ class TestRouteLayer:
route_layer_openai = RouteLayer(
encoder=openai_encoder, routes=dynamic_routes, index=index_cls()
)
- assert openai_encoder.score_threshold == 0.82
- assert route_layer_openai.score_threshold == 0.82
+ assert openai_encoder.score_threshold == 0.3
+ assert route_layer_openai.score_threshold == 0.3
def test_add_route(self, openai_encoder, index_cls):
route_layer = RouteLayer(encoder=openai_encoder, index=index_cls())
--
GitLab