From 8250e32db0e374f71ddd993bc17cbd943a30fdfc Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 4 Jan 2025 16:06:22 +0400
Subject: [PATCH] fix: pinecone delays
---
tests/unit/test_router.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index 3d476b1b..aa0c83fa 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -16,7 +16,7 @@ from semantic_router.route import Route
from platform import python_version
-PINECONE_SLEEP = 12
+PINECONE_SLEEP = 10
def mock_encoder_call(utterances):
@@ -137,7 +137,9 @@ def openai_encoder(mocker):
mocker.patch.object(OpenAIEncoder, "acall", side_effect=async_mock_encoder_call)
# Create and return the mocked encoder
- encoder = OpenAIEncoder(name="text-embedding-3-small")
+ encoder = OpenAIEncoder(
+ name="text-embedding-3-small"
+ )
return encoder
@@ -267,7 +269,7 @@ class TestIndexEncoders:
top_k=10,
)
if index_cls is PineconeIndex:
- time.sleep(PINECONE_SLEEP) # allow for index to be populated
+ time.sleep(PINECONE_SLEEP*2) # allow for index to be populated
if isinstance(route_layer, HybridRouter):
assert (
--
GitLab