diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index a54891f3cc8e92444cc49efc9fae0208479440ff..fa5482ab5aec2a21f361f109a6a6c4f3ec83552e 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -8,7 +8,6 @@ from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder from semantic_router.layer import LayerConfig, RouteLayer from semantic_router.llms.base import BaseLLM from semantic_router.route import Route -from semantic_router.index.pinecone import PineconeIndex def mock_encoder_call(utterances): @@ -215,12 +214,12 @@ class TestRouteLayer: route_layer = RouteLayer(encoder=openai_encoder, routes=routes) query_result = route_layer(text="Hello").name assert query_result in ["Route 1", "Route 2"] - + def test_query_filter(self, openai_encoder, routes): route_layer = RouteLayer(encoder=openai_encoder, routes=routes) query_result = route_layer(text="Hello", route_filter=["Route 1"]).name assert query_result in ["Route 1"] - + # def test_query_filter_openai_index(self, openai_encoder, routes): # route_layer = RouteLayer(encoder=openai_encoder, routes=routes, index=PineconeIndex) # query_result = route_layer(text="Hello", route_filter=["Route 1"]).name