diff --git a/tests/unit/test_sync.py b/tests/unit/test_sync.py index 3cb425bbd18549a155853c5144b5f97b43c62594..b7acebe924595593512c6dcc41328bda0136e514 100644 --- a/tests/unit/test_sync.py +++ b/tests/unit/test_sync.py @@ -12,7 +12,7 @@ from semantic_router.route import Route from platform import python_version -PINECONE_SLEEP = 12 +PINECONE_SLEEP = 6 def mock_encoder_call(utterances): @@ -199,7 +199,7 @@ class TestRouteLayer: def test_second_initialization_sync(self, openai_encoder, routes, index_cls): index = init_index(index_cls) route_layer = RouteLayer( - encoder=openai_encoder, routes=routes, top_k=10, index=index + encoder=openai_encoder, routes=routes, index=index, auto_sync="local" ) if index_cls is PineconeIndex: time.sleep(PINECONE_SLEEP) # allow for index to be populated @@ -212,7 +212,10 @@ class TestRouteLayer: self, openai_encoder, routes, routes_2, index_cls ): index = init_index(index_cls, sync=None) - _ = RouteLayer(encoder=openai_encoder, routes=routes, index=index) + _ = RouteLayer( + encoder=openai_encoder, routes=routes, index=index, + auto_sync="local" + ) route_layer = RouteLayer(encoder=openai_encoder, routes=routes_2, index=index) if index_cls is PineconeIndex: time.sleep(PINECONE_SLEEP) # allow for index to be populated @@ -223,21 +226,23 @@ class TestRouteLayer: ) def test_utterance_diff(self, openai_encoder, routes, routes_2, index_cls): index = init_index(index_cls) - _ = RouteLayer(encoder=openai_encoder, routes=routes, top_k=10, index=index) - if index_cls is PineconeIndex: - time.sleep(PINECONE_SLEEP) # allow for index to be populated + _ = RouteLayer( + encoder=openai_encoder, routes=routes, index=index, + auto_sync="local" + ) route_layer_2 = RouteLayer( - encoder=openai_encoder, routes=routes_2, top_k=10, index=index + encoder=openai_encoder, routes=routes_2, index=index ) if index_cls is PineconeIndex: time.sleep(PINECONE_SLEEP) # allow for index to be populated - diff = route_layer_2.get_utterance_diff() - assert " Route 1: Hello" in diff - assert "+ Route 1: Hi" in diff - assert "+ Route 2: Au revoir" in diff - assert "+ Route 2: Bye" in diff - assert "+ Route 2: Goodbye" in diff - assert " Route 2: Hi" in diff + diff = route_layer_2.get_utterance_diff(include_metadata=True) + assert "+ Route 1: Hello | None | {'type': 'default'}" in diff + assert "+ Route 1: Hi | None | {'type': 'default'}" in diff + assert "- Route 1: Hello | None | {}" in diff + assert "+ Route 2: Au revoir | None | {}" in diff + assert "- Route 2: Hi | None | {}" in diff + assert "+ Route 2: Bye | None | {}" in diff + assert "+ Route 2: Goodbye | None | {}" in diff @pytest.mark.skipif( os.environ.get("PINECONE_API_KEY") is None, reason="Pinecone API key required"