From 414f41bf6524bc0444a10b6c9fd9ccf16b5ba0f7 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Fri, 3 Jan 2025 14:45:27 +0400
Subject: [PATCH] chore: lint
---
semantic_router/index/pinecone.py | 11 ++++++++---
tests/unit/test_sync.py | 4 ++--
2 files changed, 10 insertions(+), 5 deletions(-)
diff --git a/semantic_router/index/pinecone.py b/semantic_router/index/pinecone.py
index e70d8b62..3fa22677 100644
--- a/semantic_router/index/pinecone.py
+++ b/semantic_router/index/pinecone.py
@@ -534,7 +534,9 @@ class PineconeIndex(BaseIndex):
scope=scope,
)
- async def _async_read_config(self, field: str, scope: str | None = None) -> ConfigParameter:
+ async def _async_read_config(
+ self, field: str, scope: str | None = None
+ ) -> ConfigParameter:
"""Read a config parameter from the index asynchronously.
:param field: The field to read.
@@ -566,7 +568,9 @@ class PineconeIndex(BaseIndex):
scope=scope,
)
except KeyError:
- raise ValueError(f"Found invalid config record during sync: {config_record}")
+ raise ValueError(
+ f"Found invalid config record during sync: {config_record}"
+ )
else:
logger.warning(f"Configuration for {field} parameter not found in index.")
return ConfigParameter(
@@ -752,7 +756,8 @@ class PineconeIndex(BaseIndex):
"namespace": namespace,
}
async with self.async_client.post(
- f"https://{self.host}/vectors/delete", json=params,
+ f"https://{self.host}/vectors/delete",
+ json=params,
) as response:
return await response.json(content_type=None)
diff --git a/tests/unit/test_sync.py b/tests/unit/test_sync.py
index 390cab49..148e62ee 100644
--- a/tests/unit/test_sync.py
+++ b/tests/unit/test_sync.py
@@ -154,7 +154,7 @@ def cohere_encoder(mocker):
# Handle either docs or utterances parameter
texts = docs if docs is not None else utterances
return mock_encoder_call(texts)
-
+
mocker.patch.object(CohereEncoder, "acall", side_effect=async_mock_encoder_call)
return CohereEncoder(name="test-cohere-encoder", cohere_api_key="test_api_key")
@@ -168,7 +168,7 @@ def openai_encoder(mocker):
# 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)
return OpenAIEncoder(name="text-embedding-3-small", openai_api_key="test_api_key")
--
GitLab