From 7eafd8f8d6d7e7a4770f76fac0042f4c0ee87b72 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 4 Jan 2025 12:39:46 +0400
Subject: [PATCH] chore: lint
---
semantic_router/index/pinecone.py | 4 +++-
tests/unit/test_router.py | 33 +++++++++++++++++++++++--------
2 files changed, 28 insertions(+), 9 deletions(-)
diff --git a/semantic_router/index/pinecone.py b/semantic_router/index/pinecone.py
index 9e8b950e..17eabddd 100644
--- a/semantic_router/index/pinecone.py
+++ b/semantic_router/index/pinecone.py
@@ -145,7 +145,9 @@ class PineconeIndex(BaseIndex):
self.api_key = api_key or os.getenv("PINECONE_API_KEY")
self.base_url = base_url
- logger.warning("Default region changed from us-west-2 to us-east-1 in v0.1.0.dev6")
+ logger.warning(
+ "Default region changed from us-west-2 to us-east-1 in v0.1.0.dev6"
+ )
if self.api_key is None:
raise ValueError("Pinecone API key is required.")
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index c460f121..107c9a7f 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -125,22 +125,27 @@ def cohere_encoder(mocker):
@pytest.fixture
def openai_encoder(mocker):
# Mock the OpenAI client creation and API calls
- mocker.patch('openai.OpenAI')
- mocker.patch('semantic_router.encoders.openai.OpenAI')
+ mocker.patch("openai.OpenAI")
+ mocker.patch("semantic_router.encoders.openai.OpenAI")
# Mock the __call__ method
mocker.patch.object(OpenAIEncoder, "__call__", side_effect=mock_encoder_call)
+
# Mock async call
async def async_mock_encoder_call(docs=None, utterances=None):
# Handle either docs or utterances parameter
texts = docs if docs is not None else utterances
return mock_encoder_call(texts)
+
mocker.patch.object(OpenAIEncoder, "acall", side_effect=async_mock_encoder_call)
# Create and return the mocked encoder
- encoder = OpenAIEncoder(name="text-embedding-3-small", openai_api_key="test_api_key")
+ encoder = OpenAIEncoder(
+ name="text-embedding-3-small", openai_api_key="test_api_key"
+ )
# Mock the initialization/validation step
- mocker.patch.object(encoder, '_validate_api_key')
+ mocker.patch.object(encoder, "_validate_api_key")
return encoder
+
@pytest.fixture
def mock_openai_llm(mocker):
# Mock the OpenAI LLM
@@ -270,7 +275,10 @@ class TestIndexEncoders:
time.sleep(PINECONE_SLEEP) # allow for index to be populated
if isinstance(route_layer, HybridRouter):
- assert route_layer.score_threshold == encoder.score_threshold * route_layer.alpha
+ assert (
+ route_layer.score_threshold
+ == encoder.score_threshold * route_layer.alpha
+ )
else:
assert route_layer.score_threshold == encoder.score_threshold
assert route_layer.top_k == 10
@@ -532,7 +540,10 @@ class TestSemanticRouter:
auto_sync="local",
)
if isinstance(route_layer, HybridRouter):
- assert route_layer.score_threshold == encoder.score_threshold * route_layer.alpha
+ assert (
+ route_layer.score_threshold
+ == encoder.score_threshold * route_layer.alpha
+ )
else:
assert route_layer.score_threshold == encoder.score_threshold
@@ -549,7 +560,10 @@ class TestSemanticRouter:
)
route_layer.add(routes=route_single_utterance)
if isinstance(route_layer, HybridRouter):
- assert route_layer.score_threshold == encoder.score_threshold * route_layer.alpha
+ assert (
+ route_layer.score_threshold
+ == encoder.score_threshold * route_layer.alpha
+ )
else:
assert route_layer.score_threshold == encoder.score_threshold
if index_cls is PineconeIndex:
@@ -571,7 +585,10 @@ class TestSemanticRouter:
time.sleep(PINECONE_SLEEP) # allow for index to be updated
route_layer.add(routes=route_single_utterance)
if isinstance(route_layer, HybridRouter):
- assert route_layer.score_threshold == encoder.score_threshold * route_layer.alpha
+ assert (
+ route_layer.score_threshold
+ == encoder.score_threshold * route_layer.alpha
+ )
else:
assert route_layer.score_threshold == encoder.score_threshold
_ = route_layer("Hello")
--
GitLab